require('./migrations')
// TODO: App Specific Models
const {database} = require('@alancnet/material-framework/server')
const { Terminal, Retailer } = database
const initialize = require('./initialize')

// const Terminal = require('./terminal')
const Workday = require('./workday')
const Service = require('./service')
//const Retailer = require('./retailer')
const StaffMember = require('./staff-member')
const StaffingAgency = require('./staffing-agency')
const Labor = require('./labor')

const TerminalWorkday = Terminal.hasMany(Workday)
const WorkdayTerminal = Workday.belongsTo(Terminal)

const WorkdayService = Workday.hasMany(Service)
const ServiceWorkday = Service.belongsTo(Workday)

const RetailerService = Retailer.hasMany(Service)
const ServiceRetailer = Service.belongsTo(Retailer)

const TerminalStaffMember = Terminal.hasMany(StaffMember)
const StaffMemberTerminal = StaffMember.belongsTo(Terminal)

const TerminalRetailer = Terminal.hasMany(Retailer)
const RetailerTerminal = Retailer.belongsTo(Terminal)

// const StaffingAgencyStaffMember = StaffingAgency.hasMany(StaffMember)
// const StaffMemberStaffingAgency = StaffMember.belongsTo(StaffingAgency)

const originalInit = database.init
const init = async () => {
  await originalInit()
  await initialize.init(database)
}


module.exports = Object.assign(database, {
  init,
  Terminal,
  Workday,
  Service,
  Retailer,
  StaffMember,
  StaffingAgency,
  TerminalWorkday,
  WorkdayTerminal,
  WorkdayService,
  ServiceWorkday,
  RetailerService,
  ServiceRetailer,
  TerminalStaffMember,
  StaffMemberTerminal,
  Labor,
  TerminalRetailer,
  RetailerTerminal
})