certd/packages/ui/certd-server/src/modules/auto/auto-cron.ts

import { logger } from "@certd/basic";
import { SysSettingsService, SysSiteInfo } from "@certd/lib-server";
import { getPlusInfo, isPlus } from "@certd/plus-core";
import { Config, Inject, Provide, Scope, ScopeEnum } from "@midwayjs/core";
import dayjs from "dayjs";
import { Between } from "typeorm";
import { DomainService } from "../cert/service/domain-service.js";
import { Cron } from "../cron/cron.js";
import { UserSiteMonitorSetting } from "../mine/service/models.js";
import { UserSettingsService } from "../mine/service/user-settings-service.js";
import { SiteInfoService } from "../monitor/index.js";
import { NotificationService } from "../pipeline/service/notification-service.js";
import { PipelineService } from "../pipeline/service/pipeline-service.js";
import { UserService } from "../sys/authority/service/user-service.js";
import { AuditService } from "../sys/enterprise/service/audit-service.js";
import { ProjectService } from "../sys/enterprise/service/project-service.js";

@Provide()
@Scope(ScopeEnum.Request, { allowDowngrade: true })
export class AutoCron {
  @Inject()
  pipelineService: PipelineService;

  @Config("cron.onlyAdminUser")
  private onlyAdminUser: boolean;

  @Config("cron.immediateTriggerOnce")
  private immediateTriggerOnce = false;

  @Config("cron.immediateTriggerSiteMonitor")
  private immediateTriggerSiteMonitor = false;

  @Inject()
  sysSettingsService: SysSettingsService;
  @Inject()
  userSettingsService: UserSettingsService;

  @Inject()
  siteInfoService: SiteInfoService;

  @Inject()
  cron: Cron;

  @Inject()
  notificationService: NotificationService;

  @Inject()
  userService: UserService;

  @Inject()
  domainService: DomainService;
  @Inject()
  projectService: ProjectService;
  @Inject()
  auditService: AuditService;

  async init() {
    logger.info("加载定时trigger开始");
    await this.pipelineService.onStartup(this.immediateTriggerOnce, this.onlyAdminUser);
    logger.info("加载定时trigger完成");

    await this.registerSiteMonitorCron();

    await this.registerPlusExpireCheckCron();

    await this.registerUserExpireCheckCron();

    await this.registerDomainExpireCheckCron();

    // await this.registerAuditCleanCron();
  }

  async registerSiteMonitorCron() {
    //先注册公共job
    logger.info(`注册公共站点证书检查定时任务`);
    const randomMinute = Math.floor(Math.random() * 60);
    this.cron.register({
      name: "siteMonitor",
      cron: `0 ${randomMinute} 0 * * *`,
      job: async () => {
        logger.info(`开始公共站点证书检查任务`);
        await this.siteInfoService.triggerCommonJob();
        logger.info(`公共站点证书检查任务完成`);
      },
    });
    logger.info(`注册公共站点证书检查定时任务完成`);

    //注册用户独立的检查时间
    logger.info(`注册用户独立站点证书检查定时任务`);
    const monitorSettingList = await this.userSettingsService.list({
      query: {
        key: UserSiteMonitorSetting.__key__,
      },
    });
    for (const item of monitorSettingList) {
      const setting = item.setting ? JSON.parse(item.setting) : {};
      if (!setting?.cron) {
        continue;
      }
      await this.siteInfoService.registerSiteMonitorJob(item.userId, item.projectId);
    }
    logger.info(`注册用户独立站点证书检查定时任务完成`);

    if (this.immediateTriggerSiteMonitor) {
      logger.info(`立即触发一次公共站点证书检查任务`);
      await this.siteInfoService.triggerCommonJob();
    }
  }

  registerPlusExpireCheckCron() {
    // 添加plus即将到期检查任务
    this.cron.register({
      name: "plus-expire-check",
      cron: `0 10 9 * * *`, // 一天只能检查一次，否则会重复发送通知
      job: async () => {
        const plusInfo = getPlusInfo();
        if (!plusInfo.originVipType || plusInfo.originVipType === "free") {
          return;
        }
        let label = "专业版";
        if (plusInfo.originVipType === "comm") {
          label = "商业版";
        }
        const siteInfo = await this.sysSettingsService.getSetting<SysSiteInfo>(SysSiteInfo);

        const appTitle = siteInfo.title || "certd";
        const expiresDate = dayjs(plusInfo.expireTime).format("YYYY-MM-DD");
        // plusInfo.expireTime= dayjs("2025-06-10").valueOf()
        const expiresDays = Math.floor((plusInfo.expireTime - new Date().getTime()) / 1000 / 60 / 60 / 24);
        let title = "";
        let content = "";
        if (expiresDays === 20 || expiresDays === 10 || expiresDays === 3 || expiresDays === 1 || expiresDays === 0) {
          title = `vip(${label})即将到期`;
          content = `您的${appTitle} vip (${label})剩余${expiresDays}天(${expiresDate})到期，请及时续期，以免影响业务`;
        } else if (expiresDays === -1 || expiresDays === -3 || expiresDays === -7) {
          title = `vip(${label})已过期`;
          content = `您的${appTitle} vip (${label})已过期${Math.abs(expiresDays)}天(${expiresDate})，请尽快续期，以免影响业务`;
        }
        if (title) {
          logger.warn(title);
          logger.warn(content);
          const url = await this.notificationService.getBindUrl("");
          const adminUsers = await this.userService.getAdmins();
          for (const adminUser of adminUsers) {
            logger.info(`发送vip到期通知给管理员：${adminUser.username}`);
            await this.notificationService.send(
              {
                useDefault: true,
                logger: logger,
                body: {
                  title,
                  content,
                  errorMessage: title,
                  url,
                  notificationType: "vipExpireRemind",
                },
              },
              adminUser.id
            );
          }
        }
      },
    });
  }

  registerUserExpireCheckCron() {
    // 添加plus即将到期检查任务
    this.cron.register({
      name: "user-expire-check",
      cron: `0 20 9 * * *`, // 一天只能检查一次，否则会重复发送通知
      job: async () => {
        const getExpiresDaysUsers = async (days: number) => {
          const targetDate = dayjs().add(days, "day");
          const startTime = targetDate.startOf("day").valueOf();
          const endTime = targetDate.endOf("day").valueOf();
          return await this.userService.find({
            where: {
              validTime: Between(startTime, endTime),
              status: 1,
            },
          });
        };

        const notifyExpiresDaysUsers = async (days: number) => {
          const list = await getExpiresDaysUsers(days);
          if (list.length === 0) {
            return;
          }
          let title = `账号即将到期`;
          let content = `您的账号剩余${days}天到期，请及时续期，以免影响业务`;
          if (days <= 0) {
            title = `账号已过期`;
            content = `您的账号已过期${Math.abs(days)}天，请尽快续期，以免影响业务`;
          }
          const url = await this.notificationService.getBindUrl("");
          for (const user of list) {
            logger.info(`发送到期通知给用户：${user.username}`);
            await this.notificationService.send(
              {
                useDefault: true,
                logger: logger,
                body: {
                  title,
                  content,
                  errorMessage: title,
                  url,
                  notificationType: "userExpireRemind",
                },
              },
              user.id
            );
          }
        };

        await notifyExpiresDaysUsers(7);
        await notifyExpiresDaysUsers(3);
        await notifyExpiresDaysUsers(1);
        await notifyExpiresDaysUsers(0);
        await notifyExpiresDaysUsers(-1);
        await notifyExpiresDaysUsers(-3);
      },
    });
  }

  registerDomainExpireCheckCron() {
    if (!isPlus()) {
      return;
    }
    // 添加域名即将到期同步任务
    const randomWeek = Math.floor(Math.random() * 7) + 1;
    const randomHour = Math.floor(Math.random() * 24);
    const randomMinute = Math.floor(Math.random() * 60);
    logger.info(`注册域名注册过期时间同步任务，每周${randomWeek} ${randomHour}:${randomMinute}检查一次`);
    this.cron.register({
      name: "domain-expire-check",
      cron: `0 ${randomMinute} ${randomHour} ? * ${randomWeek}`, // 每周随机一天检查一次
      job: async () => {
        await this.domainService.startSyncExpirationTask({});
      },
    });
  }

  async registerAuditCleanCron() {
    logger.info("注册审计日志清理定时任务");
    this.auditService.registerCleanCron();
    logger.info("注册审计日志清理定时任务完成");
  }
}