Merge pull request #5 from nitaking/add-option

✨ Add --models-path & config

Author: nitaking

package.json

@@ -10,7 +10,9 @@
   "dependencies": {
     "@oclif/command": "^1",
     "@oclif/config": "^1",
-    "@oclif/plugin-help": "^2"
+    "@oclif/plugin-help": "^2",
+    "conf": "^6.0.1",
+    "node-jq": "^1.10.3"
   },
   "devDependencies": {
     "@oclif/test": "^1",

src/config.js

@@ -0,0 +1,23 @@
+const Conf = require('conf')
+const constants = require('./constants')
+
+const config = new Conf()
+
+const getModelsPath = ()/* :void */ => config.get(constants.MODELS_PATH, constants.BASE_MODELS_PATH)
+// const getModelsPath = ()/* :void */ => constants.BASE_MODELS_PATH
+const setModelsPath = modelsPath/* :string */ => config.set(constants.MODELS_PATH, modelsPath)
+
+const getAll = () => {
+  let result = {}
+  constants.CONFIG_KEYS.forEach(key => {
+    result[key] = config.get(key)
+  })
+  return result
+}
+
+module.exports = {
+  getModelsPath,
+  setModelsPath,
+  config,
+  getAll,
+}

src/constants.js

@@ -0,0 +1,14 @@
+const {workDir} = require('./path')
+
+// config key
+const MODELS_PATH = 'models-path'
+const CONFIG_KEYS = [MODELS_PATH]
+
+// values
+const BASE_MODELS_PATH = `${workDir}/models`
+
+module.exports = {
+  CONFIG_KEYS,
+  MODELS_PATH,
+  BASE_MODELS_PATH,
+}

src/index.js

@@ -1,17 +1,37 @@
 const {Command, flags} = require('@oclif/command')
+const jq = require('node-jq')
+
 const Repl = require('./repl')
 const {initializeContext} = require('./sequelize-context')
+const {setModelsPath, getAll} = require('./config')
 
 const _initialFlags = {}
 
 class SequelizeTinkerCommand extends Command {
   async run() {
-    const {flags} = this.parse(SequelizeTinkerCommand)
+    const {flags, args} = this.parse(SequelizeTinkerCommand)
 
     if (flags === _initialFlags) {
       return
     }
 
+    // show config
+    if (args.firstArg === 'config') {
+      jq.run('.', getAll(), {input: 'json'})
+      .then(output => {
+        this.log(output)
+      })
+      .catch(error => {
+        this.error(error)
+      })
+      return
+    }
+
+    if (flags['models-path']) {
+      setModelsPath(flags['models-path'])
+    }
+
+    // repl entry
     const replUtil = new Repl({initializeContext})
     replUtil.start()
   }
@@ -27,6 +47,17 @@ SequelizeTinkerCommand.flags = {
   // add --help flag to show CLI version
   help: flags.help({char: 'h'}),
   // name: flags.string({ char: 'n', description: 'name to print' })
+  'models-path': flags.string({char: 'p', description: 'Set Sequelize models\'s dir path.'}),
 }
 
+SequelizeTinkerCommand.args = [
+  {name: 'firstArg'},
+  {name: 'secondArg'},
+]
+SequelizeTinkerCommand.examples = [
+  '$ sequelize-tinker',
+  '$ sequelize-tinker --models-path ./src/models',
+  '$ sequelize-tinker config',
+]
+
 module.exports = SequelizeTinkerCommand

src/sequelize-context.js

@@ -1,6 +1,7 @@
-const {workDir} = require('./path')
+
 const {secureRequire} = require('./secure-require')
-const models = secureRequire(`${workDir}/models`) // todo: import setting
+const config = require('./config')
+const models = secureRequire(config.getModelsPath()) // todo: import setting
 const sequelize = models && models.sequelize
 
 function initializeContext(context) {