mysql-migrator/migrator.seedings.js

npm data migration and data seeding package for mysql, mariadb

const fs = require('fs')
const path = require('path')
const { Table } = require('./table.js')
module.exports.Seeding = async (_sqlConnection, _dir, _executeQuery, _readDir) => {
  const table = Table(_sqlConnection, _executeQuery)
  let batchNumber = 0

  const up = async () => {
    const migrationList = await table.executeRawQuery('SELECT name from seedings')
    const fileList = await _readDir(path.join(_dir, '/seedings/'))
    let count = 0
    const promiseList = []
    const migrationfileList = []
    fileList.forEach(async migrationFile => {
      const tf = migrationList.find(x => x.name === migrationFile)
      if (!tf) {
        count++
        const migrationFilePath = path.join(_dir, '/seedings/', migrationFile)
        const { up: seedingUP } = require(migrationFilePath)
        const tmp = seedingUP(_executeQuery)
        promiseList.push(tmp)
        migrationfileList.push(migrationFile)
      }
    })
    if (count === 0) {
      return { type: 'warning', message: 'Nothing to migrate\n' }
    } else {
      const paList = await Promise.all(promiseList).then(value => {
        return value
      })
      const insertPromiseList = []
      paList.forEach((pa, index) => {
        if (pa !== false) {
          const insertStatement = `INSERT INTO seedings VALUES(NULL,'${migrationfileList[index]}',${batchNumber},'created')`
          const insertpl = _executeQuery(insertStatement)
          insertPromiseList.push(insertpl)
          // console.log('\x1b[32m', migrationfileList[index], '\x1b[36m', ' : migrated successfully ', '\x1b[0m')
          console.log('\x1b[36m', 'Migrated successfully :', '\x1b[32m', migrationfileList[index], '\x1b[0m')
        } else {
          console.log('\x1b[31m', migrationfileList[index], '\x1b[31m', ' : Failed to migrate ', '\x1b[0m')
        }
      })
      await Promise.all(insertPromiseList).then(value => {
        return value
      })
      return { type: 'success', message: '\n Seeding is done\n' }
    }
  }

  const rollback = async (_type) => {
    const sqlQuerymigrationListByBath = _type === 'rollback'
      ? 'SELECT id,name FROM seedings where batch in ( SELECT IFNULL((SELECT max(batch) FROM seedings),0))'
      : 'SELECT id,name FROM seedings'
    const migrationRecords = await _executeQuery(sqlQuerymigrationListByBath)
    const rollbackPromiseList = []
    migrationRecords.forEach(migrationrecord => {
      try {
        const migrationFilePath = path.join(_dir, '/seedings/', migrationrecord.name)
        const { rollback: seedingRollBack } = require(migrationFilePath)
        rollbackPromiseList.push(seedingRollBack(_executeQuery))
      } catch (error) {
        console.log(error)
        process.exit()
      }
    })
    const promiseResults = await Promise.all(rollbackPromiseList).then(value => {
      return value
    })

    const deletePromiseList = []
    promiseResults.forEach((promiseres, index) => {
      if (promiseres !== false) {
        const deleteStatement = `DELETE FROM seedings WHERE id=${migrationRecords[index].id}`
        deletePromiseList.push(_executeQuery(deleteStatement))
        console.log('\x1b[36m', 'Rollback successfully :', '\x1b[32m', migrationRecords[index].name, '\x1b[0m')
      } else {
        console.log('\x1b[31m', migrationRecords[index].name, '\x1b[31m', ' : Failed to Rollback ', '\x1b[0m')
      }
    })

    await Promise.all(deletePromiseList).then(value => {
      return value
    })
    return { type: 'success', message: '\n Seeding RollBack is done\n' }
  }

  const createFile = async (_filename) => {
    const fileCount = await new Promise(resolve => {
      const migDir = path.join(_dir, '/seedings/')
      fs.readdir(migDir, (err, files) => {
        if (err) {
          resolve(0)
        } else {
          resolve(files.length)
        }
      })
    })
    _filename = _dir + '/seedings/' + (fileCount + 1) + '-' + (new Date()).getTime().toString() + '-' + _filename + '.js'
    const returnMsg = await new Promise(resolve => {
      const tmpFile = path.join(__dirname, '/tmp.seedings.js')
      fs.copyFile(tmpFile, _filename, (err) => {
        if (err) {
          resolve(err)
        } else {
          resolve('success')
        }
      })
    })

    return returnMsg === 'success'
      ? { type: 'success', message: '  Successfully created' }
      : { type: 'error', message: 'failed to create, pls check your directories.' }
  }

  const init = async () => {
    let msg
    let responseMessage

    const createMigration = table.create('seedings', {
      id: 'int NOT NULL PRIMARY KEY AUTO_INCREMENT',
      name: 'varchar(254) NOT NULL',
      batch: 'int NOT NULL',
      created_at: 'varchar(254) NOT NULL'
    })

    await Promise.all([createMigration]).then(value => {
      return value
    })
    const batchNumberResult = _executeQuery('SELECT IFNULL((SELECT max(batch) FROM seedings),0)  as bathno;')
    batchNumber = await Promise.all([batchNumberResult]).then(value => {
      return value[0][0].bathno
    })
    batchNumber++
    const caseStr = process.argv[2]
    switch (caseStr) {
      case 'seeding:create':
        if (process.argv.length !== 4) {
          return { type: 'error', message: 'please describe table name. example -> npm run migrate seeding:create seederName' }
        } else {
          msg = await createFile(process.argv[3])
          responseMessage = msg
          break
        }
      case 'seeding:up':
        responseMessage = await up()
        break
      case 'seeding:rollback':
        responseMessage = await rollback('rollback')
        break
      case 'seeding:reset':
        responseMessage = await rollback('reset')
        break
      default:
        process.exit()
    }
    return responseMessage
  }
  return init()
}