update

backend/app/database.py

@@ -1,7 +1,8 @@
 import os
-from sqlalchemy import create_engine, text
+from sqlalchemy import create_engine, text, event
 from sqlalchemy.orm import sessionmaker, declarative_base
 from sqlalchemy.pool import NullPool
+from sqlalchemy.dialects import postgresql
 
 # Get database URL from environment variable
 # Default to SQLite for local development if not set
@@ -29,11 +30,32 @@ if DATABASE_URL.startswith("postgresql://") or DATABASE_URL.startswith("postgres
         DATABASE_URL = DATABASE_URL.replace("sslmode=verify-full", "sslmode=require")
         print("⚠ Certificate file not found, using sslmode=require instead of verify-full")
     
+    # Configure engine for CockroachDB compatibility
+    # CockroachDB uses a different version string format, so we need to handle it
+    connect_args = {}
+    is_cockroach = "cockroachlabs" in DATABASE_URL.lower()
+    
+    if is_cockroach:
+        # For CockroachDB, bypass version string parsing by setting version info
+        # This prevents SQLAlchemy from trying to parse CockroachDB's version string
+        connect_args["server_version"] = "12.0"
+    
     engine = create_engine(
         DATABASE_URL,
         poolclass=NullPool,  # CockroachDB works better with NullPool
-        echo=False  # Set to True for SQL query debugging
+        echo=False,  # Set to True for SQL query debugging
+        connect_args=connect_args
     )
+    
+    # Override version detection for CockroachDB at the dialect level
+    if is_cockroach:
+        @event.listens_for(engine, "connect", insert=True)
+        def receive_connect(dbapi_conn, connection_record):
+            # Set version info before SQLAlchemy tries to parse it
+            # This happens at connection time
+            if hasattr(connection_record, 'info'):
+                connection_record.info["server_version_info"] = (12, 0)
+                connection_record.info["server_version"] = "12.0"
 else:
     # SQLite for local development
     engine = create_engine(
@@ -57,14 +79,38 @@ def init_db():
     """Initialize database tables"""
     try:
         from app.models import User, Integration, Asset, Post, Campaign
-        # Test connection first
-        with engine.connect() as conn:
-            conn.execute(text("SELECT 1"))
-            conn.commit()
-        # Create tables if connection successful
+        
+        # Test connection and create tables
+        # For CockroachDB, version parsing may fail but connection still works
+        try:
+            with engine.connect() as conn:
+                conn.execute(text("SELECT 1"))
+                conn.commit()
+        except Exception as conn_error:
+            error_str = str(conn_error)
+            # If it's a version parsing error for CockroachDB, connection still works
+            # We can proceed with table creation
+            if "Could not determine version" not in error_str:
+                # Real connection error, not just version parsing
+                raise conn_error
+        
+        # Create tables - this should work even if version parsing failed
         Base.metadata.create_all(bind=engine)
         return True
     except Exception as e:
-        print(f"Database connection failed: {e}")
-        return False
+        error_str = str(e)
+        # Check if it's just a version parsing error (non-fatal for CockroachDB)
+        if "Could not determine version" in error_str:
+            # Version parsing failed, but try to create tables anyway
+            # CockroachDB connection works, just version string format is different
+            try:
+                Base.metadata.create_all(bind=engine)
+                print("✓ CockroachDB connected and tables created (version parsing issue ignored)")
+                return True
+            except Exception as e2:
+                print(f"Database table creation failed: {e2}")
+                return False
+        else:
+            print(f"Database connection failed: {e}")
+            return False