quote table and col names from postgres and duckdb too (#6942)

## 📝 Summary

<!--
Provide a concise summary of what this pull request is addressing.

If this PR fixes any issues, list them here by number (e.g., Fixes
#123).
-->

## 🔍 Description of Changes

<!--
Detail the specific changes made in this pull request. Explain the
problem addressed and how it was resolved. If applicable, provide before
and after comparisons, screenshots, or any relevant details to help
reviewers understand the changes easily.
-->

## 📋 Checklist

- [x] I have read the [contributor
guidelines](https://github.com/marimo-team/marimo/blob/main/CONTRIBUTING.md).
- [ ] For large changes, or changes that affect the public API: this
change was discussed or approved through an issue, on
[Discord](https://marimo.io/discord?ref=pr), or the community
[discussions](https://github.com/marimo-team/marimo/discussions) (Please
provide a link if applicable).
- [x] I have added tests for the changes made.
- [x] I have run the code and verified that it works as expected.

Author: Light2Dark

frontend/src/components/datasources/__tests__/utils.test.ts

@@ -48,7 +48,7 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM users LIMIT 100\n""")',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "users" LIMIT 100\n""")',
       );
     });
 
@@ -68,7 +68,7 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM analytics.users LIMIT 100\n""")',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "analytics"."users" LIMIT 100\n""")',
       );
     });
 
@@ -108,7 +108,7 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM remote.users LIMIT 100\n""")',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "remote"."users" LIMIT 100\n""")',
       );
     });
 
@@ -127,7 +127,7 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM users LIMIT 100\n""")',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "users" LIMIT 100\n""")',
       );
 
       const sqlTableContext2: SQLTableContext = {
@@ -144,7 +144,7 @@ describe("sqlCode", () => {
         sqlTableContext: sqlTableContext2,
       });
       expect(result2).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM another_db.users LIMIT 100\n""")',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "another_db"."users" LIMIT 100\n""")',
       );
     });
   });
@@ -253,7 +253,7 @@ describe("sqlCode", () => {
     });
   });
 
-  describe("TimescaleDB dialect", () => {
+  describe("TimescaleDB and PostgreSQL dialect", () => {
     it("should wrap table name with double quotes", () => {
       const sqlTableContext: SQLTableContext = {
         engine: "timescaledb",
@@ -270,7 +270,7 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM "users" LIMIT 100\n""", engine=timescaledb)',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "users" LIMIT 100\n""", engine=timescaledb)',
       );
     });
 
@@ -290,17 +290,17 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM "remote"."sales"."users" LIMIT 100\n""", engine=timescaledb)',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "remote"."sales"."users" LIMIT 100\n""", engine=timescaledb)',
       );
     });
 
     it("should handle schemaless database with double quotes", () => {
       const sqlTableContext: SQLTableContext = {
-        engine: "timescaledb",
+        engine: "postgres",
         schema: "",
         defaultDatabase: "mydb",
         database: "remote",
-        dialect: "timescaledb",
+        dialect: "postgres",
       };
 
       const result = sqlCode({
@@ -309,7 +309,27 @@ describe("sqlCode", () => {
         sqlTableContext,
       });
       expect(result).toBe(
-        '_df = mo.sql(f"""\nSELECT email FROM "remote"."users" LIMIT 100\n""", engine=timescaledb)',
+        '_df = mo.sql(f"""\nSELECT "email" FROM "remote"."users" LIMIT 100\n""", engine=postgres)',
+      );
+    });
+
+    it("should not quote * column name", () => {
+      const sqlTableContext: SQLTableContext = {
+        engine: "postgres",
+        schema: "public",
+        defaultSchema: "public",
+        defaultDatabase: "mydb",
+        database: "mydb",
+        dialect: "postgres",
+      };
+
+      const result = sqlCode({
+        table: mockTable,
+        columnName: "*",
+        sqlTableContext,
+      });
+      expect(result).toBe(
+        '_df = mo.sql(f"""\nSELECT * FROM "users" LIMIT 100\n""", engine=postgres)',
       );
     });
   });

frontend/src/components/datasources/utils.ts

@@ -53,20 +53,21 @@ function getFormatter(dialect: string): SqlCodeFormatter {
           `SELECT TOP 100 ${columnName} FROM ${tableName}`,
       };
     case "timescaledb":
+    case "postgres":
+    case "postgresql":
+    case "duckdb":
+      // Quote column and table names to avoid raising errors on weird characters
       return {
-        // TimescaleDB uses double quotes for identifiers
         formatTableName: (tableName: string) => {
           const parts = tableName.split(".");
           return parts.map((part) => `"${part}"`).join(".");
         },
-        formatSelectClause: defaultFormatter.formatSelectClause,
+        formatSelectClause: (columnName: string, tableName: string) =>
+          `SELECT ${columnName === "*" ? "*" : `"${columnName}"`} FROM ${tableName} LIMIT 100`,
       };
-    case "postgresql":
-    case "postgres":
     case "db2":
     case "mysql":
     case "sqlite":
-    case "duckdb":
     case "mariadb":
     case "cassandra":
     case "noql":