feat(optimizer)!: annotate regexpreplace for mysql

sqlglot/typing/mysql.py

@@ -25,6 +25,33 @@ def _annotate_truncate(self: TypeAnnotator, expression: exp.Trunc) -> exp.Expr:
     return self._annotate_by_args(expression, "this")
 
 
+def _annotate_regexp_replace(
+    self: TypeAnnotator,
+    expression: exp.RegexpReplace,
+) -> exp.Expr:
+    args = (
+        expression.this,
+        expression.args.get("expression"),
+        expression.args.get("replacement"),
+    )
+
+    if all(
+        arg.is_type(
+            exp.DType.BINARY,
+            exp.DType.VARBINARY,
+            exp.DType.TINYBLOB,
+            exp.DType.BLOB,
+            exp.DType.MEDIUMBLOB,
+            exp.DType.LONGBLOB,
+        )
+        for arg in args
+        if arg is not None
+    ):
+        return self._set_type(expression, exp.DType.LONGBLOB)
+
+    return self._set_type(expression, exp.DType.LONGTEXT)
+
+
 EXPRESSION_METADATA = {
     **EXPRESSION_METADATA,
     **{
@@ -93,4 +120,5 @@ def _annotate_truncate(self: TypeAnnotator, expression: exp.Trunc) -> exp.Expr:
     },
     exp.Reverse: {"annotator": _annotate_reverse},
     exp.Trunc: {"annotator": _annotate_truncate},
+    exp.RegexpReplace: {"annotator": _annotate_regexp_replace},
 }

tests/fixtures/optimizer/annotate_functions.sql

@@ -6634,6 +6634,40 @@ VARCHAR;
 # dialect: mysql
 REGEXP_SUBSTR(tbl.str_col, tbl.str_col, tbl.int_col, tbl.int_col, tbl.str_col);
 VARCHAR;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.str_col, tbl.str_col, tbl.str_col);
+LONGTEXT;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.str_col, tbl.str_col, tbl.str_col, tbl.int_col);
+LONGTEXT;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.str_col, tbl.str_col, tbl.str_col, tbl.int_col, tbl.int_col);
+LONGTEXT;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.str_col, tbl.str_col, tbl.str_col, tbl.int_col, tbl.int_col,  tbl.str_col);
+LONGTEXT;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.bin_col, tbl.bin_col, tbl.bin_col);
+LONGBLOB;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.bin_col, tbl.bin_col, tbl.bin_col, tbl.int_col);
+LONGBLOB;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.bin_col, tbl.bin_col, tbl.bin_col, tbl.int_col, tbl.int_col);
+LONGBLOB;
+
+# dialect: mysql
+REGEXP_REPLACE(tbl.bin_col, tbl.bin_col, tbl.bin_col, tbl.int_col, tbl.int_col, tbl.str_col);
+LONGBLOB;
+
+
 --------------------------------------
 -- DuckDB
 --------------------------------------