Support non-default schema in the lumber update command. Fix #175

ForestAdmin · Nov 4, 2018 · 7a1c617 · 7a1c617
1 parent 95e5cad
commit 7a1c617
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/services/prompter.js b/services/prompter.js
@@ -42,6 +42,22 @@ async function Prompter(program, requests) {
       });
     }
 
+    if (isRequested('dbSchema')) {
+      if (process.env.DATABASE_SCHEMA) {
+        envConfig.dbSchema = process.env.DATABASE_SCHEMA;
+      } else {
+        prompts.push({
+          type: 'input',
+          name: 'dbSchema',
+          message: 'What\'s the database schema? [optional]',
+          description: 'Leave blank by default',
+          when: answers => answers.dbDialect !== 'sqlite',
+          default: (args) => {
+            if (args.dbDialect === 'postgres') { return 'public'; }
+            return '';
+          },
+        });
+      }
     }
   } else {
     if (isRequested('dbDialect')) {