feat(duckdb): Add transpilation support for SPACE function (#6867)

Author: fivetran-amrutabhimsenayachit

sqlglot/dialects/duckdb.py

@@ -2922,6 +2922,15 @@ def startswith_sql(self, expression: exp.StartsWith) -> str:
                 _cast_to_varchar(expression.expression),
             )
 
+        def space_sql(self, expression: exp.Space) -> str:
+            # DuckDB's REPEAT requires BIGINT for the count parameter
+            return self.sql(
+                exp.Repeat(
+                    this=exp.Literal.string(" "),
+                    times=exp.cast(expression.this, exp.DataType.Type.BIGINT),
+                )
+            )
+
         def tablefromrows_sql(self, expression: exp.TableFromRows) -> str:
             # For GENERATOR, unwrap TABLE() - just emit the Generator (becomes RANGE)
             if isinstance(expression.this, exp.Generator):

tests/dialects/test_snowflake.py

@@ -5635,6 +5635,34 @@ def test_array_flatten(self):
             },
         )
 
+    def test_space(self):
+        # Integer literal
+        self.validate_all(
+            "SELECT SPACE(5)",
+            write={
+                "snowflake": "SELECT REPEAT(' ', 5)",
+                "duckdb": "SELECT REPEAT(' ', CAST(5 AS BIGINT))",
+            },
+        )
+
+        # Float literal (tests rounding behavior)
+        self.validate_all(
+            "SELECT SPACE(3.7)",
+            write={
+                "snowflake": "SELECT REPEAT(' ', 3.7)",
+                "duckdb": "SELECT REPEAT(' ', CAST(3.7 AS BIGINT))",
+            },
+        )
+
+        # NULL value
+        self.validate_all(
+            "SELECT SPACE(NULL)",
+            write={
+                "snowflake": "SELECT REPEAT(' ', NULL)",
+                "duckdb": "SELECT REPEAT(' ', CAST(NULL AS BIGINT))",
+            },
+        )
+
     def test_directed_joins(self):
         self.validate_identity("SELECT * FROM a CROSS DIRECTED JOIN b USING (id)")
         self.validate_identity("SELECT * FROM a INNER DIRECTED JOIN b USING (id)")