import { Sequelize } from 'sequelize'
import { Application } from './declarations'
// @ts-expect-error
import seederConfig from './seeder-config'
// @ts-expect-error
import seeder from 'feathers-seeder'
import { db } from './db-config'
import { setTimeout } from 'timers'

export default (app: Application): void => {
  const { forceRefresh } = db

  const sequelize = new Sequelize({
    ...db,
    logging: forceRefresh ? console.log : false,
    define: {
      freezeTableName: true
    }
  })
  const oldSetup = app.setup

  app.set('sequelizeClient', sequelize)

  app.setup = function (...args: any) {
    // Sync to the database
    app.set('sequelizeSync',
      sequelize.sync({ force: forceRefresh })
        .then(() => {
          // @ts-expect-error
          app.configure(seeder(seederConfig)).seed().catch(err => {
            console.log(err)
          })
          if (process.env.PERFORM_DRY_RUN) setTimeout(() => process.exit(0), 5000)
        })
        .catch(console.error)
    )
    return oldSetup.apply(this, args)
  }
}