feat: add check-structure command

Author: gregberge

.gitignore

@@ -1,3 +1,4 @@
 node_modules/
 lib/
 db/
+!test/__fixtures__/structure.sql

package.json

@@ -9,8 +9,10 @@
     "build": "babel -d lib src",
     "lint": "eslint .",
     "prebuild": "rm -rf lib",
-    "test": "yarn lint && yarn build && bin/knex-scripts create && bin/knex-scripts dump && bin/knex-scripts load && bin/knex-scripts drop",
-    "release": "yarn build && standard-version && conventional-github-releaser -p angular"
+    "test":
+      "yarn lint && yarn build && jest --ci && bin/knex-scripts create && bin/knex-scripts dump && bin/knex-scripts load && bin/knex-scripts drop && bin/knex-scripts check-structure",
+    "release":
+      "yarn build && standard-version && conventional-github-releaser -p angular"
   },
   "bin": {
     "knex-scripts": "bin/knex-scripts"
@@ -29,6 +31,7 @@
   "devDependencies": {
     "babel-cli": "^6.26.0",
     "babel-eslint": "^8.2.1",
+    "babel-jest": "^22.4.1",
     "babel-plugin-transform-class-properties": "^6.24.1",
     "babel-plugin-transform-object-rest-spread": "^6.26.0",
     "babel-preset-env": "^1.6.1",
@@ -37,6 +40,7 @@
     "eslint-config-airbnb-base": "^12.1.0",
     "eslint-config-prettier": "^2.9.0",
     "eslint-plugin-import": "^2.8.0",
+    "jest": "^22.4.2",
     "knex": "^0.14.2",
     "pg": "^7.4.1",
     "prettier": "^1.10.2",

src/checkStructure.js

@@ -0,0 +1,19 @@
+import {
+  requireEnv,
+  getInsertsFromMigrations,
+  getInsertsFromStructure,
+} from './utils'
+
+async function checkStructure(options) {
+  const { structurePath, migrationsPath } = options
+  requireEnv('development', options.env)
+
+  const migrationsInFolder = await getInsertsFromMigrations(migrationsPath)
+  const migrationsInStructure = await getInsertsFromStructure(structurePath)
+
+  return migrationsInFolder.every(
+    (insert, index) => migrationsInStructure[index] === insert,
+  )
+}
+
+export default checkStructure

src/cli.js

@@ -13,6 +13,7 @@ import drop from './drop'
 import dump from './dump'
 import load from './load'
 import truncate from './truncate'
+import checkStructure from './checkStructure'
 
 const argv = minimist(process.argv.slice(2))
 
@@ -65,7 +66,7 @@ function initConfig(env) {
 
   if (environment) {
     console.log('Using environment:', chalk.magenta(environment))
-    config = config[environment] || config
+    config = { ...config, ...config[environment] }
   }
 
   if (!config) {
@@ -80,7 +81,10 @@ function initConfig(env) {
     knexConfig: config,
     env: environment,
     structurePath: knexScriptsConfig.structurePath || 'db/structure.sql',
-    migrationsPath: join(process.cwd(), 'migrations'),
+    migrationsPath:
+      config.migrations && config.migrations.directory
+        ? config.migrations.directory
+        : join(process.cwd(), 'migrations'),
     docker:
       (commander.docker !== undefined
         ? commander.docker
@@ -116,7 +120,7 @@ function invoke(env) {
     .action(() => {
       const config = initConfig(env)
       return create(config)
-        .then(() => console.log(chalk.green(`Database created.`)))
+        .then(() => console.log(chalk.green('Database created.')))
         .catch(exit)
     })
 
@@ -126,7 +130,7 @@ function invoke(env) {
     .action(() => {
       const config = initConfig(env)
       return drop(config)
-        .then(() => console.log(chalk.green(`Database dropped.`)))
+        .then(() => console.log(chalk.green('Database dropped.')))
         .catch(exit)
     })
 
@@ -136,7 +140,7 @@ function invoke(env) {
     .action(() => {
       const config = initConfig(env)
       return dump(config)
-        .then(() => console.log(chalk.green(`Dump created.`)))
+        .then(() => console.log(chalk.green('Dump created.')))
         .catch(exit)
     })
 
@@ -146,7 +150,24 @@ function invoke(env) {
     .action(() => {
       const config = initConfig(env)
       return load(config)
-        .then(() => console.log(chalk.green(`Database loaded.`)))
+        .then(() => console.log(chalk.green('Database loaded.')))
+        .catch(exit)
+    })
+
+  commander
+    .command('check-structure')
+    .description('Check structure.')
+    .action(() => {
+      const config = initConfig(env)
+      return checkStructure(config)
+        .then(upToDate => {
+          if (upToDate) {
+            console.log(chalk.green('Structure is up to date.'))
+          } else {
+            console.log(chalk.red('Structure is not up to date.'))
+            process.exit(1)
+          }
+        })
         .catch(exit)
     })
 

src/dump.js

@@ -1,25 +1,15 @@
 import { exec } from 'mz/child_process'
-import { appendFile, readdir, exists } from 'mz/fs'
+import { appendFile } from 'mz/fs'
 import { dirname } from 'path'
 import mkdirp from 'mkdirp'
 import {
   requireEnv,
   wrapDockerCommand,
   getCommand,
   getCommandEnv,
+  getInsertsFromMigrations,
 } from './utils'
 
-async function getMigrationInserts({ migrationsPath }) {
-  if (!await exists(migrationsPath)) return ''
-  const migrations = await readdir(migrationsPath)
-  return migrations
-    .map(
-      migration =>
-        `INSERT INTO knex_migrations(name, batch, migration_time) VALUES ('${migration}', 1, NOW());\n`,
-    )
-    .join('')
-}
-
 async function dump(options) {
   const { structurePath, migrationsPath } = options
   requireEnv('development', options.env)
@@ -34,8 +24,11 @@ async function dump(options) {
 
   await exec(wrapDockerCommand(options, command), { env })
 
-  const migrationInserts = await getMigrationInserts({ migrationsPath })
-  return appendFile(structurePath, `-- Knex migrations\n\n${migrationInserts}`)
+  const migrationInserts = await getInsertsFromMigrations(migrationsPath)
+  return appendFile(
+    structurePath,
+    `-- Knex migrations\n\n${migrationInserts.join('\n')}`,
+  )
 }
 
 export default dump

src/index.js

@@ -3,3 +3,4 @@ export { default as drop } from './drop'
 export { default as dump } from './dump'
 export { default as load } from './load'
 export { default as truncate } from './truncate'
+export { default as checkStructure } from './checkStructure'

src/utils.js

@@ -1,3 +1,31 @@
+import { readFile, readdir, exists } from 'mz/fs'
+
+export async function getInsertsFromMigrations(migrationsPath) {
+  if (!await exists(migrationsPath)) return []
+  const migrations = await readdir(migrationsPath)
+  return migrations.map(
+    migration =>
+      `INSERT INTO knex_migrations(name, batch, migration_time) VALUES ('${migration}', 1, NOW());`,
+  )
+}
+
+export async function getInsertsFromStructure(structurePath) {
+  if (!await exists(structurePath)) return []
+  const structure = await readFile(structurePath, 'utf-8')
+  const regExp = /INSERT INTO knex_migrations\(name, batch, migration_time\) VALUES \('.*', 1, NOW\(\)\);/g
+
+  const inserts = []
+
+  let match
+  /* eslint-disable no-cond-assign */
+  while ((match = regExp.exec(structure))) {
+    inserts.push(match[0])
+  }
+  /* eslint-enable no-cond-assign */
+
+  return inserts
+}
+
 export function preventEnv(preventedEnv, env) {
   if (env === preventedEnv) {
     throw new Error(`Not in ${preventedEnv} please!`)

test/.eslintrc.js

@@ -0,0 +1,5 @@
+module.exports = {
+  env: {
+    jest: true,
+  },
+}