// @ts-nocheck
import Database, { registerModels } from '@nocobase/database';
import Resourcer from '@nocobase/resourcer';
import path from 'path';
import { AutomationModel } from './models/automation';
import { AutomationJobModel } from './models/automation-job';

export default async function (options = {}) {
  const database: Database = this.database;
  const resourcer: Resourcer = this.resourcer;

  registerModels({
    AutomationModel,
    AutomationJobModel,
  });

  database.import({
    directory: path.resolve(__dirname, 'collections'),
  });

  const [Automation, AutomationJob] = database.getModels(['automations', 'automations_jobs']);

  Automation.addHook('afterCreate', async (model: AutomationModel) => {
    model.get('enabled') && await model.loadJobs();
  });

  Automation.addHook('afterUpdate', async (model: AutomationModel) => {
    if (!model.changed('enabled' as any)) {
      return;
    }
    model.get('enabled') ? await model.loadJobs() : await model.cancelJobs();
  });

  Automation.addHook('beforeDestroy', async (model: AutomationModel) => {
    await model.cancelJobs();
  });

  AutomationJob.addHook('afterCreate', async (model: AutomationJobModel) => {
    await model.bootstrap();
  });

  AutomationJob.addHook('beforeDestroy', async (model: AutomationJobModel) => {
    await model.cancel();
  });
}