habitica/gulp/gulp-console.js

import mongoose from 'mongoose';
import logger  from '../website/server/libs/logger';
import nconf    from 'nconf';
import repl     from 'repl';
import gulp     from 'gulp';

// Add additional properties to the repl's context
let improveRepl = (context) => {
  // Let "exit" and "quit" terminate the console
  ['exit', 'quit'].forEach((term) => {
    Object.defineProperty(context, term, { get () {
      process.exit();
    }});
  });

  // "clear" clears the screen
  Object.defineProperty(context, 'clear', { get () {
    process.stdout.write('\u001B[2J\u001B[0;0f');
  }});

  context.Challenge = require('../website/server/models/challenge').model; // eslint-disable-line global-require
  context.Group     = require('../website/server/models/group').model; // eslint-disable-line global-require
  context.User      = require('../website/server/models/user').model; // eslint-disable-line global-require

  const isProd = nconf.get('NODE_ENV') === 'production';
  const mongooseOptions = !isProd ? {} : {
    keepAlive: 1,
    connectTimeoutMS: 30000,
    useMongoClient: true,
  };
  mongoose.connect(
    nconf.get('NODE_DB_URI'),
    mongooseOptions,
    (err) => {
      if (err) throw err;
      logger.info('Connected with Mongoose');
    }
  );
};

gulp.task('console', (done) => {
  improveRepl(repl.start({
    prompt: 'Habitica > ',
  }).context);
  done();
});