refactor: replace the googlesql module by sqlglot

Author: chelsea-lin

bigframes/core/compile/compiled.py

@@ -23,15 +23,16 @@
 import bigframes_vendored.ibis.expr.datatypes as ibis_dtypes
 import bigframes_vendored.ibis.expr.operations as ibis_ops
 import bigframes_vendored.ibis.expr.types as ibis_types
+import bigframes_vendored.sqlglot.expressions as sge
 from google.cloud import bigquery
 import pyarrow as pa
 
 from bigframes.core import agg_expressions, rewrite
 import bigframes.core.agg_expressions as ex_types
-import bigframes.core.compile.googlesql
 import bigframes.core.compile.ibis_compiler.aggregate_compiler as agg_compiler
 import bigframes.core.compile.ibis_compiler.scalar_op_compiler as op_compilers
 import bigframes.core.compile.ibis_types
+import bigframes.core.compile.sqlglot.sqlglot_ir as sqlglot
 import bigframes.core.expression as ex
 from bigframes.core.ordering import OrderingExpression
 import bigframes.core.sql
@@ -82,13 +83,21 @@ def to_sql(
         )
 
         if order_by or limit or not is_noop_selection:
-            sql = ibis_bigquery.Backend().compile(ibis_table)
-            sql = (
-                bigframes.core.compile.googlesql.Select()
-                .from_(sql)
-                .select(selection_strings)
-                .sql()
-            )
+            # selections are (ref.id.sql, name) where ref.id.sql is escaped identifier
+            to_select = [
+                sge.Alias(
+                    this=sge.to_identifier(src, quoted=False),
+                    alias=sge.to_identifier(alias, quoted=True),
+                )
+                if src != alias
+                else sge.to_identifier(src, quoted=False)
+                for src, alias in selection_strings
+            ]
+            # Use string formatting for FROM clause to avoid re-parsing potentially complex SQL (like ARRAY<STRUCT<...>>)
+            # that sqlglot might not handle perfectly when parsing BigQuery dialect strings.
+            select_sql = sge.Select().select(*to_select).sql(dialect="bigquery")
+            ibis_sql = ibis_bigquery.Backend().compile(ibis_table)
+            sql = f"{select_sql} FROM ({ibis_sql}) AS `t`"
 
             # Single row frames may not have any ordering columns
             if len(order_by) > 0: