strapi/packages/strapi-generate-new/lib/before.js

'use strict';

/**
 * Module dependencies
 */

// Node.js core.
const path = require('path');
const exec = require('child_process').exec;
const execSync = require('child_process').execSync;

// Public node modules.
const _ = require('lodash');
const fs = require('fs-extra');
const inquirer = require('inquirer');
const shell = require('shelljs');

// Logger.
const logger = require('strapi-utils').logger;

/**
 * This `before` function is run before generating targets.
 * Validate, configure defaults, get extra dependencies, etc.
 *
 * @param {Object} scope
 * @param {Function} cb
 */

module.exports = (scope, cb) => {
  // App info.
  _.defaults(scope, {
    name: scope.name === '.' || !scope.name ? scope.name : path.basename(process.cwd()),
    author: process.env.USER || 'A Strapi developer',
    email: process.env.EMAIL || '',
    year: (new Date()).getFullYear(),
    license: 'MIT'
  });

  // Make changes to the rootPath where the Strapi project will be created.
  scope.rootPath = path.resolve(process.cwd(), scope.name || '');

  // Ensure we aren't going to inadvertently delete any files.
  try {
    const files = fs.readdirSync(scope.rootPath);
    if (files.length) {
      return logger.error('`$ strapi new` can only be called in an empty directory.');
    }
  } catch (err) {
    // ...
  }

  logger.info('Let\s configurate the connection to your database:');

  scope.database = {};

  const connectionValidation = () => {
    const databaseChoices = [
      {
        name: 'MongoDB (highly recommended)',
        value: {
          database: 'mongo',
          connector: 'strapi-mongoose'
        }
      },
      {
        name: 'Postgres',
        value: {
          database: 'postgres',
          connector: 'strapi-bookshelf',
          module: 'pg'
        }
      },
      {
        name: 'MySQL',
        value: {
          database: 'mysql',
          connector: 'strapi-bookshelf',
          module: 'mysql'
        }
      },
      {
        name: 'Sqlite3',
        value: {
          database: 'sqlite3',
          connector: 'strapi-bookshelf',
          module: 'sqlite3'
        }
      },
      {
        name: 'Redis',
        value: {
          database: 'redis',
          connector: 'strapi-redis'
        }
      }
    ];

    inquirer
    .prompt([
      {
        type: 'list',
        prefix: '',
        name: 'client',
        message: 'Choose your main database:',
        choices: databaseChoices,
        default: () => {
          if (scope.client) {
            return _.findIndex(databaseChoices, { value: _.omit(scope.client, ['version'])});
          }
        }
      }
    ])
    .then(answers => {
      scope.client = answers.client;
      _.assign(scope.database, {
        connector: answers.client.connector,
        settings: {
          client: answers.client.database
        },
        options: {}
      });

      const asyncFn = [
        new Promise(resolve => {
          inquirer
          .prompt([
            {
              type: 'input',
              prefix: '',
              name: 'name',
              message: 'Database name:',
              default: _.get(scope.database, 'database', 'strapi')
            },
            {
              type: 'input',
              prefix: '',
              name: 'host',
              message: 'Host:',
              default: _.get(scope.database, 'host', 'localhost')
            },
            {
              type: 'input',
              prefix: '',
              name: 'port',
              message: 'Port:',
              default: (answers) => {
                if (_.get(scope.database, 'port')) {
                  return scope.database.port;
                }

                const ports = {
                  mongo: 27017,
                  postgres: 5432,
                  mysql: 3306,
                  sqlite3: 1433,
                  redis: 6379
                };

                return ports[scope.client.database];
              }
            },
            {
              type: 'input',
              prefix: '',
              name: 'username',
              message: 'Username:',
              default: _.get(scope.database, 'username', undefined)
            },
            {
              type: 'input',
              prefix: '',
              name: 'password',
              message: 'Password:',
              default: _.get(scope.database, 'password', undefined)
            }
          ])
          .then(answers => {
            scope.database.settings.host = answers.host;
            scope.database.settings.port = answers.port;
            scope.database.settings.database = answers.name;
            scope.database.settings.username = answers.username;
            scope.database.settings.password = answers.password;

            logger.info('Testing database connection...');

            resolve();
          });
        }),
        new Promise(resolve => {
          let cmd = `npm install --prefix ${scope.rootPath} ${scope.client.connector}@alpha`;
          if (scope.client.module) {
            cmd += ` ${scope.client.module}`;
          }

          exec(cmd, () => {
            if (scope.client.module) {
              const lock = require(`${scope.rootPath}/node_modules/${scope.client.module}/package.json`);
              scope.client.version = lock.version;
            }

            resolve();
          });
        })
      ];

      Promise.all(asyncFn)
      .then(() => {
        try {
          require(path.resolve(`${scope.rootPath}/node_modules/${scope.client.connector}/lib/utils/connectivity.js`))(scope, cb.success, connectionValidation);
        } catch(err) {
          shell.rm('-r', scope.rootPath);
          logger.info('Copying the dashboard...');
          cb.success();
        }
      });
    });
  };

  connectionValidation();
};