setup.js

import r from 'rethinkdb';
import config from '../config';

const createDatabase = (conn, name) =>
  r.dbList().contains(name)
    .do(containsDb =>
      r.branch(
        containsDb,
        { created: 0 },
        r.dbCreate(name)
      )
    )
    .run(conn)
    .error(err => {
      console.log(`Could not create ${name} DB`);
      console.log(err.message || 'Something bad happened');
      process.exit(1);
    });

const createTable = (conn, name) =>
  r.tableList().contains(name)
    .do(containsTable =>
      r.branch(
        containsTable,
        { created: 0 },
        r.tableCreate(name)
      )
    )
    .run(conn)
    .error(err => {
      console.log(`Could not create ${name} table`);
      console.log(err.message || 'Somthing bad happened');
      process.exit(1);
    });

const createIndex = (conn, tableName, indexName) =>
  r.table(tableName).indexList().contains(indexName)
    .do(containsIndex =>
      r.branch(
        containsIndex,
        { created: 0 },
        r.table(tableName).indexCreate(indexName)
      )
    )
    .run(conn)
    .error(err => {
      console.log(`Could not crate index ${indexName} in ${tableName}`);
      console.log(err.message || 'Something bad happened');
      process.exit(1);
    });

const waitForIndex = (conn, tableName, indexName) =>
  r.table(tableName).indexWait(indexName)
    .run(conn)
    .error(err => {
      console.log('Something bad happened while waiting for index created');
      console.log(err.message || 'Something bad happened');
      process.exit(1);
    });

export const setup = startApp => {
  r.connect(config.rethinkdb)
    .then(conn => {
      createDatabase(conn, 'test');
      return conn;
    })
    .then(conn => {
      Promise.all([
        createTable(conn, 'todos'),
        createTable(conn, 'days'),
      ]);
      return conn;
    })
    .then(conn => {
      Promise.all([
        createIndex(conn, 'todos', 'createdAt'),
        createIndex(conn, 'days', 'date'),
      ]);
      return conn;
    })
    .then(conn =>
      Promise.all([
        waitForIndex(conn, 'todos', 'createdAt'),
        waitForIndex(conn, 'days', 'date'),
      ])
    )
    .then(() => {
      console.log('DB and tables are available, starting Koa ...');
      startApp();
    })
    .error(err => {
      console.log('Could not open a connection to initiailize the database');
      console.log(err.message || 'Something bad happened');
      process.exit(1);
    });
};