import { exec } from 'child_process';
import { Container } from 'typedi';
import { Crontab, CrontabStatus } from '../data/cron';
import CronService from '../services/cron';
import CookieService from '../services/cookie';

const initData = [
  {
    name: '更新面板',
    command: `ql update`,
    schedule: `${randomSchedule(60, 1)} ${randomSchedule(
      6,
      1,
    ).toString()} * * *`,
    status: CrontabStatus.idle,
  },
  {
    name: '删除日志',
    command: 'ql rmlog 7',
    schedule: '30 7 */7 * *',
    status: CrontabStatus.idle,
  },
  {
    name: '互助码',
    command: 'ql code',
    schedule: '30 7 * * *',
    status: CrontabStatus.idle,
  },
];

export default async () => {
  const cronService = Container.get(CronService);
  const cookieService = Container.get(CookieService);
  const cronDb = cronService.getDb();

  cronDb.count({}, async (err, count) => {
    if (count === 0) {
      const data = initData.map((x) => {
        const tab = new Crontab(x);
        tab.created = new Date().valueOf();
        tab.saved = false;
        if (tab.name === '更新面板') {
          tab.isSystem = 1;
        } else {
          tab.isSystem = 0;
        }
        return tab;
      });
      cronDb.insert(data);
      await cronService.autosave_crontab();
    }
  });

  // patch更新面板任务状态
  cronDb.find({ name: '更新面板' }).exec((err, docs) => {
    const doc = docs[0];
    if (doc && doc.status === CrontabStatus.running) {
      cronDb.update(
        { name: '更新面板' },
        { $set: { status: CrontabStatus.idle } },
      );
    }
  });

  // 初始化时执行一次所有的ql repo 任务
  cronDb
    .find({
      command: /ql (repo|raw)/,
    })
    .exec((err, docs) => {
      for (let i = 0; i < docs.length; i++) {
        const doc = docs[i];
        exec(doc.command);
      }
    });

  // 初始化保存一次ck和定时任务数据
  await cronService.autosave_crontab();
  await cookieService.set_cookies();
};

function randomSchedule(from: number, to: number) {
  const result = [];
  const arr = [...Array(from).keys()];
  let count = arr.length;
  for (let i = 0; i < to; i++) {
    const index = ~~(Math.random() * count) + i;
    if (result.includes(arr[index])) {
      continue;
    }
    result[i] = arr[index];
    arr[index] = arr[i];
    count--;
  }
  return result;
}