main.js

#! /usr/bin/env node

var path = require('path');
var utils = require('./lib/utils.js');
var Discoverer = require('./lib/discoverer.js');

var appPath = null;
var dsName = null;
var tablesList = null;
var overwrite = false;
var views = false;
var all = false;
var outputDir = null;

for (var i = 2; i < process.argv.length; i++) {
  switch (process.argv[i]) {
    case '-o':
      overwrite = true;
      break;
    case '-d':
      dsName = process.argv[++i];
      break;
    case '-a':
      all = true;
      break;
    case '-v':
      views = true;
      break;
    case '-l':
      tablesList = process.argv[++i].split(',');
      break;
    case '-p':
      outputDir = process.argv[++i];
      break;
    case '-h':
    case '--help':
      printUsage();
      break;
    default:
      if (appPath === null)
        appPath = path.resolve(process.argv[i]);
      else
        printUsage();
  }
}

var discoverer = new Discoverer();

process.on('uncaughtException', function (err) {
  checkError(err);
});

if (appPath === null) {
  try {
    discoverer.setApp(process.cwd());
  } catch (err) {
    printUsage();    
  }
} else {
  discoverer.setApp(appPath);
}

setDatasource(discoverer, dsName, function (err, set) {
  checkError(err);

  if (!set)
    console.log('Discovery canceled.');

  discoverer.loadModelConfiguration();

  discoverer.discoverModels({
    outputDir: outputDir,
    overwrite: overwrite,
    all: all,
    views: views,
    tablesList: tablesList
  }, function (err, models) {
    checkError(err);

    if (models && models.length > 0) {
      // check the ones already set as public (update)
      for (var m in models) {
        if (discoverer.modelConfig[models[m].name])
          models[m].isPublic = discoverer.modelConfig[models[m].name].public;
      }

      utils.selectPublicModels(models, function (models) {
        discoverer.updateModels(models);
        if (!outputDir) {
            /* If we are putting discovered models in a separate directory,
               we shouldn't update model-config.json either. */
            discoverer.saveModelConfiguration();
        }
        console.log('Discovery completed.');
        process.exit(0);
      });
    } else {
      console.log('No models discovered.');
      process.exit(1);
    }
  });
});

function setDatasource(discoverer, dsName, callback) {
  if (dsName !== null) {
    try {
      discoverer.setDataSource(dsName);
      callback(null, true);
    } catch (err) {
      callback(err);
    }
  } else {
    utils.selectDataSource(discoverer.app, function (dsName) {
      if (!dsName)
        callback(null, false);
      else
        setDatasource(discoverer, dsName, callback);
    });
  }

}

function checkError(err) {
  if (err) {
    // It's much more useful to dump out the complete error,
    // particularly the stack trace.
    // console.log(err.message);
    console.log(err);    
    process.exit(1);
  }
}

function printUsage() {
  console.log('LoopBack model discovery script.');
  console.log('');
  console.log('> loopback-discovery [-o] [-a] [-v] [-d datasource] [-l tables] [loopbackAppPath]');
  console.log('');
  console.log('  use "-o" option to overwrite existing definitions');
  console.log('  use "-a" option to export models from all schemas');
  console.log('  use "-v" option to also export views');
  console.log('  use "-d" option to select the datasource');
  console.log('  use "-l" option to specify the list of tables to export, comma separated');
  console.log('  use "-p" option to change output directory. Default is common/mdels/');
  console.log('');
  console.log('  if app path is not specified the current working folder is used.');
  
  process.exit(0);
}