chore: Migrate RemoteFunctionOp operator to SQLGlot (#2377)

Thank you for opening a Pull Request! Before submitting your PR, there
are a few things you can do to make sure it goes smoothly:
- [ ] Make sure to open an issue as a
[bug/issue](https://github.com/googleapis/python-bigquery-dataframes/issues/new/choose)
before writing your code! That way we can discuss the change, evaluate
designs, and agree on the general idea
- [ ] Ensure the tests and linter pass
- [ ] Code coverage does not decrease (if any source code was changed)
- [ ] Appropriate docs were updated (if necessary)

Fixes b/452130300

Author: jialuoo

bigframes/core/compile/sqlglot/expressions/generic_ops.py

@@ -152,6 +152,25 @@ def _(left: TypedExpr, right: TypedExpr) -> sge.Expression:
     return sge.Coalesce(this=left.expr, expressions=[right.expr])
 
 
+@register_unary_op(ops.RemoteFunctionOp, pass_op=True)
+def _(expr: TypedExpr, op: ops.RemoteFunctionOp) -> sge.Expression:
+    routine_ref = op.function_def.routine_ref
+    # Quote project, dataset, and routine IDs to avoid keyword clashes.
+    func_name = (
+        f"`{routine_ref.project}`.`{routine_ref.dataset_id}`.`{routine_ref.routine_id}`"
+    )
+    func = sge.func(func_name, expr.expr)
+
+    if not op.apply_on_null:
+        return sge.If(
+            this=sge.Is(this=expr.expr, expression=sge.Null()),
+            true=expr.expr,
+            false=func,
+        )
+
+    return func
+
+
 @register_binary_op(ops.BinaryRemoteFunctionOp, pass_op=True)
 def _(
     left: TypedExpr, right: TypedExpr, op: ops.BinaryRemoteFunctionOp

tests/unit/core/compile/sqlglot/expressions/snapshots/test_generic_ops/test_remote_function_op/out.sql

@@ -0,0 +1,19 @@
+WITH `bfcte_0` AS (
+  SELECT
+    `int64_col`
+  FROM `bigframes-dev`.`sqlglot_test`.`scalar_types`
+), `bfcte_1` AS (
+  SELECT
+    *,
+    `my_project`.`my_dataset`.`my_routine`(`int64_col`) AS `bfcol_1`,
+    IF(
+      `int64_col` IS NULL,
+      `int64_col`,
+      `my_project`.`my_dataset`.`my_routine`(`int64_col`)
+    ) AS `bfcol_2`
+  FROM `bfcte_0`
+)
+SELECT
+  `bfcol_1` AS `apply_on_null_true`,
+  `bfcol_2` AS `apply_on_null_false`
+FROM `bfcte_1`

tests/unit/core/compile/sqlglot/expressions/test_generic_ops.py

@@ -169,6 +169,42 @@ def test_astype_json_invalid(
         )
 
 
+def test_remote_function_op(scalar_types_df: bpd.DataFrame, snapshot):
+    from google.cloud import bigquery
+
+    from bigframes.functions import udf_def
+
+    bf_df = scalar_types_df[["int64_col"]]
+    function_def = udf_def.BigqueryUdf(
+        routine_ref=bigquery.RoutineReference.from_string(
+            "my_project.my_dataset.my_routine"
+        ),
+        signature=udf_def.UdfSignature(
+            input_types=(
+                udf_def.UdfField(
+                    "x",
+                    bigquery.StandardSqlDataType(
+                        type_kind=bigquery.StandardSqlTypeNames.INT64
+                    ),
+                ),
+            ),
+            output_bq_type=bigquery.StandardSqlDataType(
+                type_kind=bigquery.StandardSqlTypeNames.FLOAT64
+            ),
+        ),
+    )
+    ops_map = {
+        "apply_on_null_true": ops.RemoteFunctionOp(
+            function_def=function_def, apply_on_null=True
+        ).as_expr("int64_col"),
+        "apply_on_null_false": ops.RemoteFunctionOp(
+            function_def=function_def, apply_on_null=False
+        ).as_expr("int64_col"),
+    }
+    sql = utils._apply_ops_to_sql(bf_df, list(ops_map.values()), list(ops_map.keys()))
+    snapshot.assert_match(sql, "out.sql")
+
+
 def test_binary_remote_function_op(scalar_types_df: bpd.DataFrame, snapshot):
     from google.cloud import bigquery