import { Sequelize, SequelizeOptions } from 'sequelize-typescript';
import { OrderModel } from './models/order.entity';
import { OrderCustomerModel } from './models/order-customer.entity';
import { ServiceOrderModel } from './models/service-order.entity';
import { OrderItemModel } from './models/order-item.entity';
import { ServiceItemModel } from './models/service-item.entity';
import { OrderPickUpModel } from './models/order-pick-up.entity';
import { OrderShippingAddressModel } from './models/order-shipping-address.entity';

let sequelize: Sequelize;

function init(sequelizeOptions: SequelizeOptions) {
  sequelize = new Sequelize(sequelizeOptions);

  sequelize.addModels([
    OrderModel,
    OrderCustomerModel,
    ServiceOrderModel,
    OrderItemModel,
    ServiceItemModel,
    OrderPickUpModel,
    OrderShippingAddressModel,
  ]);
}

function getConnection() {
  return sequelize;
}

export { init, getConnection };