Readlebee/server/sequelize/setup-database.js

// Eventually, I'd like this to be run through Commander so it can confirm things before just overwriting stuff.

const path = require('path');
const fs = require('fs');
const force = typeof process.argv[2] !== 'undefined' && process.argv[2] === 'force';

const Sequelize = require('sequelize');
let siteConfig;
try {
  siteConfig = require('../config.json');
} catch (ex) {
  console.error('Please copy `config.example.json` to `config.json` and fill it with your server\'s data.');
  process.exit(1);
}

const sequelizeConfig = {
  dialect: siteConfig.db_engine,
};
switch (siteConfig.db_engine) {
  case 'sqlite': {
    sequelizeConfig.storage = typeof siteConfig.sqlite_location !== 'undefined'
      ? (
        siteConfig.sqlite_location.substr(0, 1) === '.' // If relative path, make relative to ./server
          ? path.resolve('./server/', siteConfig.sqlite_location)
          : path.resolve(siteConfig.sqlite_location)
      )
      : path.resolve(__dirname, './server/database.sqlite');
    break;
  }
  default: {
    sequelizeConfig.host = siteConfig.db_host;
    sequelizeConfig.port = siteConfig.db_port;
    sequelizeConfig.database = siteConfig.db_database;
    sequelizeConfig.username = siteConfig.db_username;
    sequelizeConfig.password = siteConfig.db_password;
  }
}

const sequelize = new Sequelize(sequelizeConfig);

const migration = require('./migration');
const dbVersionPath = path.resolve(__dirname, './.dbversion');
if (!force) {
  if (fs.existsSync(dbVersionPath)) {
    const installedDbVersion = fs.readFileSync(dbVersionPath);
    if (installedDbVersion < migration.dbVersion) {
      console.log(`Migrating from ${installedDbVersion} to ${migration.dbVersion}...`);
      migration.migrateDb(installedDbVersion, sequelize);
      return fs.writeFile(dbVersionPath, migration.dbVersion, err => {
        if (err) {
          console.error(err);
        }
        console.log('Migration complete!');
      });
    }
    if (installedDbVersion == migration.dbVersion) {
      return console.log(`No database setup needed: installed version ${installedDbVersion} is the current version.`);
    }
    console.log(`Skipping database migration: installed version ${installedDbVersion} is not less than current version ${migration.dbVersion}.`);
    return console.log('Please check your installation and make sure you have the right server files downloaded.');
  }
}

const Models = require('./getModels')(sequelize);

console.log(`Installing database tables${force ? ', dropping existing ones first' : ''}...`);
sequelize.sync({ force }).then(() => {
  console.log('Tables installed! Adding Status Types...')
  const promises = [ // Default status types to use in Statuses
    { name: 'update' },
    { name: 'progress' },
    { name: 'rating' },
    { name: 'review' },
  ].map(statusType => Models.StatusType.create(statusType).catch(() => console.log(statusType.name, 'already exists')));
  return Promise.all(promises);
}).then(() => {
  sequelize.close();
  console.log(`Status Types installed! Writing database version to ${dbVersionPath}...`);
  fs.writeFile(dbVersionPath, migration.dbVersion, err => {
    if (err) {
      console.error(err);
    }
    console.log('Done!');
  });
});