feat: add exclude schemas option

Author: svetch

packages/cli/src/actions/db.ts

@@ -25,12 +25,13 @@ export type PullOptions = {
     out?: string;
     naming?: 'pascal' | 'camel' | 'snake' | 'kebab' | 'none';
     alwaysMap?: boolean;
+    excludeSchemas: string[];
 };
 
 /**
  * CLI action for db related commands
  */
-export async function run(command: string, options: PushOptions) {
+export async function run(command: string, options: any) {
     switch (command) {
         case 'push':
             await runPush(options);
@@ -90,7 +91,9 @@ async function runPull(options: PullOptions) {
             throw new Error(`No introspection provider found for: ${datasource.provider}`);
         }
 
-        const { enums, tables } = await provider.introspect(datasource.url);
+        const { enums: allEnums, tables: allTables } = await provider.introspect(datasource.url);
+        const enums = allEnums.filter((e) => !options.excludeSchemas.includes(e.schema_name));
+        const tables = allTables.filter((t) => !options.excludeSchemas.includes(t.schema));
 
         const newModel: Model = {
             $type: 'Model',

packages/cli/src/actions/pull/index.ts

@@ -41,6 +41,15 @@ export function syncEnums({
                 return builder;
             });
         });
+
+        if (dbEnum.schema_name && dbEnum.schema_name != '' && dbEnum.schema_name !== 'public') {
+            factory.addAttribute((b) =>
+                b
+                    .setDecl(getAttributeRef('@@schema', services))
+                    .addArg((a) => a.StringLiteral.setValue(dbEnum.schema_name)),
+            );
+        }
+
         model.declarations.push(factory.get({ $container: model }));
     }
 }
@@ -316,6 +325,12 @@ export function syncTable({
         );
     });
 
+    if (table.schema && table.schema != '' && table.schema !== 'public') {
+        modelFactory.addAttribute((b) =>
+            b.setDecl(getAttributeRef('@@schema', services)).addArg((a) => a.StringLiteral.setValue(table.schema)),
+        );
+    }
+
     model.declarations.push(modelFactory.node);
 
     return relations;

packages/cli/src/index.ts

@@ -126,7 +126,8 @@ function createProgram() {
         .description('Introspect your database.')
         .addOption(schemaOption)
         .addOption(noVersionCheckOption)
-        .addOption(new Option('--out <path>', 'add custom output path for the introspected schema'))
+        .addOption(new Option('-e, --exclude-schemas <schemas...>', 'exclude specific schemas from introspection'))
+        .addOption(new Option('-o, --out <path>', 'add custom output path for the introspected schema'))
         .action((options) => dbAction('pull', options));
 
     program