allow option to migrate workspaces and external schemas

keboola · Dec 3, 2024 · 57ceaf7 · 57ceaf7
1 parent 694b5e4
commit 57ceaf7
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 2 deletions.
diff --git a/src/Config.php b/src/Config.php
@@ -203,4 +203,14 @@ public function isSourceByodb(): bool
     {
         return (bool) $this->getValue(['parameters',  'isSourceByodb']);
     }
+
+    public function getIncludedWorkspaceSchemas(): array
+    {
+        return $this->getArrayValue(['parameters', 'includeWorkspaceSchemas']);
+    }
+
+    public function getIncludedExternalSchemas(): array
+    {
+        return $this->getArrayValue(['parameters', 'includeExternalSchemas']);
+    }
 }
diff --git a/src/Configuration/ConfigDefinition.php b/src/Configuration/ConfigDefinition.php
@@ -22,6 +22,8 @@ protected function getParametersDefinition(): ArrayNodeDefinition
                 ->scalarNode('sourceByodb')->end()
                 ->scalarNode('sourceKbcUrl')->isRequired()->cannotBeEmpty()->end()
                 ->scalarNode('#sourceKbcToken')->isRequired()->cannotBeEmpty()->end()
+                ->arrayNode('includeWorkspaceSchemas')->prototype('scalar')->end()->end()
+                ->arrayNode('includeExternalSchemas')->prototype('scalar')->end()->end()
                 ->arrayNode('tables')->prototype('scalar')->end()->end()
                 ->arrayNode('db')
                     ->children()

diff --git a/src/Strategy/DatabaseMigrate.php b/src/Strategy/DatabaseMigrate.php
@@ -81,11 +81,15 @@ public function migrate(Config $config): void
             if (in_array($schemaName, self::SKIP_CLONE_SCHEMAS, true)) {
                 continue;
             }
-            if (str_starts_with($schemaName, 'WORKSPACE')) {
+            if (str_starts_with($schemaName, 'WORKSPACE')
+                && !in_array($schemaName, $config->getIncludedWorkspaceSchemas())) {
+
                 continue;
             }
 
-            if (!$this->sourceSapiClient->bucketExists($schemaName)) {
+            if (!$this->sourceSapiClient->bucketExists($schemaName)
+                && !in_array($schemaName, $config->getIncludedExternalSchemas())) {
+
                 continue;
             }