adding support for databases with other existing tables

Author: xprojects-de

src/chat/src/Bridge/Doctrine/DoctrineDbalMessageStore.php

@@ -15,6 +15,7 @@
 use Doctrine\DBAL\Connection as DBALConnection;
 use Doctrine\DBAL\Platforms\OraclePlatform;
 use Doctrine\DBAL\Result;
+use Doctrine\DBAL\Schema\ComparatorConfig;
 use Doctrine\DBAL\Schema\Name\Identifier;
 use Doctrine\DBAL\Schema\Name\UnqualifiedName;
 use Doctrine\DBAL\Schema\PrimaryKeyConstraint;
@@ -52,13 +53,23 @@ public function setup(array $options = []): void
             throw new InvalidArgumentException('No supported options.');
         }
 
-        $schema = $this->dbalConnection->createSchemaManager()->introspectSchema();
+        $platform = $this->dbalConnection->getDatabasePlatform();
+        $schemaManager = $this->dbalConnection->createSchemaManager();
 
-        if ($schema->hasTable($this->tableName)) {
-            return;
+        $currentSchema = $schemaManager->introspectSchema();
+
+        if (class_exists(ComparatorConfig::class)) {
+            $comparator = $schemaManager->createComparator(new ComparatorConfig(false, false));
+        } else {
+            // Backwards compatibility for doctrine/dbal 3.x
+            $comparator = $schemaManager->createComparator();
         }
 
-        $this->addTableToSchema($schema);
+        $migrations = $platform->getAlterSchemaSQL($comparator->compareSchemas($currentSchema, $this->defineTableSchema($currentSchema)));
+
+        foreach ($migrations as $sql) {
+            $this->dbalConnection->executeQuery($sql);
+        }
     }
 
     public function drop(): void
@@ -113,8 +124,14 @@ public function load(): MessageBag
         return new MessageBag(...array_merge(...$messages));
     }
 
-    private function addTableToSchema(Schema $schema): void
+    private function defineTableSchema(Schema $currentSchema): Schema
     {
+        $schema = clone $currentSchema;
+
+        if ($schema->hasTable($this->tableName)) {
+            $schema->dropTable($this->tableName);
+        }
+
         $table = $schema->createTable($this->tableName);
         $table->addOption('_symfony_ai_chat_table_name', $this->tableName);
         $idColumn = $table->addColumn('id', Types::BIGINT)
@@ -141,8 +158,6 @@ private function addTableToSchema(Schema $schema): void
             }
         }
 
-        foreach ($schema->toSql($this->dbalConnection->getDatabasePlatform()) as $sql) {
-            $this->dbalConnection->executeQuery($sql);
-        }
+        return $schema;
     }
 }