feat: add supabase db

backend/app/api/datasources.py

@@ -133,4 +133,7 @@ def test_datasource_connection(
         else:
              raise HTTPException(status_code=400, detail="Connection failed")
     except Exception as e:
+        import traceback
+        import sys
+        print(f"Datasource Test Error: {str(e)}\n{traceback.format_exc()}", file=sys.stderr)
         raise HTTPException(status_code=400, detail=f"Connection failed: {str(e)}")

backend/app/connectors/factory.py

@@ -13,9 +13,17 @@ def _get_cached_connector(ds_type: str, config_json: str):
     config = json.loads(config_json)
     
     if ds_type in ["postgres", "postgresql", "supabase"]:
-        # Supabase is just postgres
+        db_url = config.get("connection_string")
-        db_url = config.get("connection_string") or \
+        if not db_url:
-                 f"postgresql://{config.get('user')}:{config.get('password')}@{config.get('host')}:{config.get('port', 5432)}/{config.get('database')}"
+            default_port = 6543 if ds_type == "supabase" else 5432
+            port = config.get("port") or default_port
+            db_url = f"postgresql://{config.get('user')}:{config.get('password')}@{config.get('host')}:{port}/{config.get('database')}"
+            
+        if ds_type == "supabase" and "?" not in db_url:
+            db_url += "?sslmode=require"
+        elif ds_type == "supabase" and "sslmode=" not in db_url:
+            db_url += "&sslmode=require"
+            
         return PostgresConnector(db_url=db_url)
         
     elif ds_type == "sqlite":

backend/app/connectors/postgres.py

@@ -69,6 +69,6 @@ class PostgresConnector:
             return True
         except Exception as e:
             print(f"PostgreSQL Connection Error: {e}")
-            return False
+            raise e
 
 postgres_connector = PostgresConnector()

frontend/src/components/DataSourceForm.tsx

@@ -93,9 +93,68 @@ export function DataSourceForm({ initialData, onSubmit, onTest, onCancel }: Data
 
   const renderConfigFields = () => {
     switch (type) {
+      case "supabase":
+        return (
+          <div className="space-y-4">
+             <div className="grid grid-cols-2 gap-4">
+              <div className="space-y-2">
+                <label className="text-sm font-medium">Host</label>
+                <Input 
+                  value={config.host || ""} 
+                  onChange={e => handleConfigChange("host", e.target.value)} 
+                  placeholder="aws-0-[region].pooler.supabase.com" 
+                />
+              </div>
+              <div className="space-y-2">
+                <label className="text-sm font-medium">Port</label>
+                <Input 
+                  type="number"
+                  value={config.port || 6543} 
+                  onChange={e => handleConfigChange("port", parseInt(e.target.value))} 
+                  placeholder="6543"
+                />
+              </div>
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Database</label>
+              <Input 
+                value={config.database || "postgres"} 
+                onChange={e => handleConfigChange("database", e.target.value)} 
+                placeholder="postgres" 
+              />
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Username</label>
+              <Input 
+                value={config.user || ""} 
+                onChange={e => handleConfigChange("user", e.target.value)} 
+                placeholder="postgres.[project-ref]" 
+              />
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Password</label>
+              <Input 
+                type="password"
+                value={config.password || ""} 
+                onChange={e => handleConfigChange("password", e.target.value)} 
+                placeholder="••••••" 
+              />
+            </div>
+            <div className="text-xs text-zinc-500 pt-2">
+              或者直接使用 Supabase 控制台提供的 Connection String (URI):
+            </div>
+            <div className="space-y-2">
+              <label className="text-sm font-medium">Connection String</label>
+              <Input 
+                value={config.connection_string || ""} 
+                onChange={e => handleConfigChange("connection_string", e.target.value)} 
+                placeholder="postgresql://postgres.[project-ref]:[password]@aws-0-[region].pooler.supabase.com:6543/postgres?sslmode=require"
+              />
+            </div>
+          </div>
+        );
       case "postgres":
       case "postgresql":
-      case "supabase":
       case "mysql":
       case "sqlserver":
       case "oracle":

frontend/src/pages/DataSources.tsx

@@ -12,6 +12,7 @@ const SOURCE_TYPES = [
   { id: "csv", name: "CSV Upload", icon: <FileText className="h-6 w-6 text-green-600" /> },
   { id: "bigquery", name: "BigQuery", icon: <Database className="h-6 w-6 text-blue-500" /> },
   { id: "postgres", name: "PostgreSQL", icon: <Database className="h-6 w-6 text-indigo-600" /> },
+  { id: "supabase", name: "Supabase", icon: <Database className="h-6 w-6 text-emerald-500" /> },
   { id: "mysql", name: "MySQL", icon: <Database className="h-6 w-6 text-cyan-600" /> },
   { id: "oracle", name: "Oracle", icon: <Database className="h-6 w-6 text-red-600" /> },
   { id: "sqlserver", name: "SQL Server", icon: <Database className="h-6 w-6 text-red-500" /> },