certd/packages/ui/certd-server/src/modules/pipeline/service/notification-service.ts

import { Inject, Provide, Scope, ScopeEnum } from '@midwayjs/core';
import {
  BaseService,
  NeedVIPException,
  SysInstallInfo,
  SysSettingsService,
  SysSiteInfo,
  ValidateException
} from "@certd/lib-server";
import { InjectEntityModel } from '@midwayjs/typeorm';
import { Repository } from 'typeorm';
import { NotificationEntity } from '../entity/notification.js';
import { NotificationInstanceConfig, notificationRegistry, NotificationSendReq, sendNotification } from '@certd/pipeline';
import { http, utils } from '@certd/basic';
import { EmailService } from '../../basic/service/email-service.js';
import { isComm, isPlus } from '@certd/plus-core';

@Provide()
@Scope(ScopeEnum.Request, { allowDowngrade: true })
export class NotificationService extends BaseService<NotificationEntity> {
  @InjectEntityModel(NotificationEntity)
  repository: Repository<NotificationEntity>;

  @Inject()
  emailService: EmailService;

  @Inject()
  sysSettingsService: SysSettingsService;

  //@ts-ignore
  getRepository() {
    return this.repository;
  }

  async getSimpleInfo(id: number) {
    const entity = await this.info(id);
    if (entity == null) {
      throw new ValidateException('该通知配置不存在,请确认是否已被删除');
    }
    return {
      id: entity.id,
      keyId: entity.keyId,
      name: entity.name,
      userId: entity.userId,
    };
  }

  getDefineList() {
    return notificationRegistry.getDefineList();
  }

  getDefineByType(type: string) {
    return notificationRegistry.getDefine(type);
  }

  async add(bean: NotificationEntity) {
    this.checkNeedPlus(bean.type);
    const res = await super.add(bean);
    if(bean.isDefault){
      await this.setDefault(res.id, bean.userId);
    }
    bean.keyId = "nt_" + utils.id.simpleNanoId();
    return res
  }

  async update(bean: NotificationEntity) {

    const old = await this.info(bean.id);
    this.checkNeedPlus(old.type);

    delete bean.userId;
    delete bean.type
    delete bean.keyId
    const res = await super.update(bean);
    if(bean.isDefault){
      await this.setDefault(bean.id, old.userId);
    }

    return res
  }

   checkNeedPlus(type: string){
    const define = this.getDefineByType(type)
     //@ts-ignore
    if (define.needPlus && !isPlus()) {
      throw new NeedVIPException("此通知类型为Certd专业版功能，请升级到专业版或以上级别");
    }
  }

  async getById(id: number, userId: number, projectId?: number): Promise<NotificationInstanceConfig> {
    if (!id) {
      throw new ValidateException('id不能为空');
    }
    if (userId==null) {
      throw new ValidateException('userId不能为空');
    }
    const res = await this.repository.findOne({
      where: {
        id,
        userId,
        projectId,
      },
    });
    if (!res) {
      throw new ValidateException(`通知配置不存在<${id}>`);
    }
    return this.buildNotificationInstanceConfig(res);
  }

  private buildNotificationInstanceConfig(res: NotificationEntity) {
    const setting = JSON.parse(res.setting);
    return {
      id: res.id,
      type: res.type,
      name: res.name,
      userId: res.userId,
      setting,
    };
  }

  async getDefault(userId: number, projectId?: number): Promise<NotificationInstanceConfig> {
    const res = await this.repository.findOne({
      where: {
        userId,
        projectId,
      },
      order: {
        isDefault: 'DESC',
      },
    });
    if (!res) {
      return null;
    }
    return this.buildNotificationInstanceConfig(res);
  }

  async setDefault(id: number, userId: number, projectId?: number) {
    if (!id) {
      throw new ValidateException('id不能为空');
    }
    if (userId==null) {
      throw new ValidateException('userId不能为空');
    }
    const query:any = {
       userId,
    }
    if (projectId){
      query.projectId = projectId
    }
    await this.repository.update(
      query,
      {
        isDefault: false,
      }
    );
    query.id = id
    await this.repository.update(
     query,
      {
        isDefault: true,
      }
    );
  }

  async getOrCreateDefault(email: string, userId: any, projectId?: number) {
    const defaultConfig = await this.getDefault(userId, projectId);
    if (defaultConfig) {
      return defaultConfig;
    }
    const setting = {
      receivers: [email],
    };
    const res = await this.repository.save({
      userId,
      type: 'email',
      name: '邮件通知',
      setting: JSON.stringify(setting),
      isDefault: true,
      projectId,
    });
    return this.buildNotificationInstanceConfig(res);
  }

  async send(req: NotificationSendReq, userId?: number, projectId?: number) {
    const logger = req.logger;
    let notifyConfig: NotificationInstanceConfig = null;
    if (req.id && req.id > 0) {
      notifyConfig = await this.getById(req.id, userId, projectId);
      if (!notifyConfig) {
        logger.warn(`未找到通知配置<${req.id}>,请确认是否已被删除`);
      }
    }
    if (!notifyConfig) {
      notifyConfig = await this.getDefault(userId, projectId);
      if (!notifyConfig) {
        logger.warn(`未找到默认通知配置`);
      }
    }

    if (notifyConfig) {
      //发送通知
      logger.info('发送通知, 使用通知渠道：' + notifyConfig.name);

      if (notifyConfig.type != 'email') {
        //非邮件通知，需要加上站点名称
        let siteTitle = 'Certd';
        if (isComm()) {
          const siteInfo = await this.sysSettingsService.getSetting<SysSiteInfo>(SysSiteInfo);
          siteTitle = siteInfo?.title || siteTitle;
        }
        req.body.title = `【${siteTitle}】${req.body.title}`;
      }

      await sendNotification({
        config: notifyConfig,
        ctx: {
          http: http,
          logger: logger,
          utils: utils,
          emailService: this.emailService,
        },
        body: req.body,
      });
    } else {
      if (req.useEmail && req.emailAddress) {
        logger.info('使用邮件通知');
        await this.emailService.send({
          receivers: [req.emailAddress],
          subject: req.body.title,
          content: req.body.content,
        });
      }
    }
  }

  async getBindUrl(path: string) {
    const installInfo = await this.sysSettingsService.getSetting<SysInstallInfo>(SysInstallInfo);
    const bindUrl = installInfo.bindUrl || 'http://127.0.0.1:7001';
    return bindUrl + path;
  }
}