Update database commands for postgres

alexbakerdev · Sep 21, 2017 · 842a428 · 842a428
1 parent 0846642
commit 842a428
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/src/commands/database.js b/src/commands/database.js
@@ -17,9 +17,15 @@ exports.handler = async function (args) {
   const sequelize = getDatabaseLessSequelize();
   const config = helpers.config.readConfig();
 
+  let dbName = config.database;
+
+  if (config.dialect === 'postgres') {
+    dbName = `"${dbName}"`;
+  }
+
   switch (command) {
     case 'db:create':
-      await sequelize.query(`CREATE DATABASE ${config.database}`, {
+      await sequelize.query(`CREATE DATABASE ${dbName}`, {
         type: sequelize.QueryTypes.RAW
       }).catch(e => {
         helpers.view.error(`Error: ${e.message}`);
@@ -34,7 +40,7 @@ exports.handler = async function (args) {
 
       break;
     case 'db:drop':
-      await sequelize.query(`DROP DATABASE ${config.database}`, {
+      await sequelize.query(`DROP DATABASE ${dbName}`, {
         type: sequelize.QueryTypes.RAW
       }).catch(e => {
         helpers.view.error(`Error: ${e.message}`);