index.js

import fs from 'fs'
import path from 'path'

const join = path.join

export default async function({
  sql,
  path = join(process.cwd(), 'migrations'),
  before = null,
  after = null
}) {
  const migrations = fs.readdirSync(path)
    .filter(x => (fs.statSync(join(path, x)).isDirectory() || fs.statSync(join(path, x)).isFile()) && x.match(/^[0-9]{5}_/))
    .sort()
    .map(x => ({
      path: join(path, x),
      migration_id: parseInt(x.slice(0, 5)),
      name: x.slice(6).replace(/-/g, ' ')
    }))

  const latest = migrations[migrations.length - 1]

  if (latest.migration_id !== migrations.length)
    throw new Error('Inconsistency in migration numbering')

  await ensureMigrationsTable()

  const current = await getCurrentMigration()
  const needed = migrations.slice(current ? current.id : 0)

  return sql.begin(next)

  async function next(sql) {
    const current = needed.shift()
    if (!current)
      return

    before && before(current)
    await run(sql, current)
    after && after(current)
    await next(sql)
  }

  async function run(sql, {
    path,
    migration_id,
    name
  }) {
    if (fs.statSync(path).isFile()) {
      path.endsWith('.sql')
        ? await sql.file(path)
        : await import(path).then(x => x.default(sql)) // eslint-disable-line
    } else if (fs.statSync(path).isDirectory()) {
      fs.existsSync(join(path, 'index.sql')) && !fs.existsSync(join(path, 'index.js'))
        ? await sql.file(join(path, 'index.sql'))
        : await import(join(path, 'index.js')).then(x => x.default(sql)) // eslint-disable-line
    }
    await sql`
      insert into migrations (
        migration_id,
        name
      ) values (
        ${ migration_id },
        ${ name }
      )
    `
  }

  function getCurrentMigration() {
    return sql`
      select migration_id as id from migrations
      order by migration_id desc
      limit 1
    `.then(([x]) => x)
  }

  function ensureMigrationsTable() {
    return sql`
      select 'migrations'::regclass
    `.catch((err) => sql`
      create table migrations (
        migration_id serial primary key,
        created_at timestamp with time zone not null default now(),
        name text
      )
    `)
  }

}