feat: 用户套餐，用户支付功能

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

@@ -1,11 +1,20 @@
 import { Config, Inject, Provide, Scope, ScopeEnum, sleep } from '@midwayjs/core';
 import { InjectEntityModel } from '@midwayjs/typeorm';
 import { In, MoreThan, Repository } from 'typeorm';
-import { BaseService, NeedVIPException, PageReq, SysPublicSettings, SysSettingsService, SysSiteInfo } from '@certd/lib-server';
+import {
+  AccessGetter,
+  AccessService,
+  BaseService,
+  NeedSuiteException,
+  NeedVIPException,
+  PageReq,
+  SysPublicSettings,
+  SysSettingsService,
+  SysSiteInfo,
+} from '@certd/lib-server';
 import { PipelineEntity } from '../entity/pipeline.js';
 import { PipelineDetail } from '../entity/vo/pipeline-detail.js';
-import { Executor, Pipeline, ResultType, RunHistory, SysInfo, UserInfo } from '@certd/pipeline';
-import { AccessService } from './access-service.js';
+import { Executor, Pipeline, ResultType, RunHistory, RunnableCollection, SysInfo, UserInfo } from '@certd/pipeline';
 import { DbStorage } from './db-storage.js';
 import { StorageService } from './storage-service.js';
 import { Cron } from '../../cron/cron.js';
@@ -15,26 +24,26 @@ import { HistoryLogEntity } from '../entity/history-log.js';
 import { HistoryLogService } from './history-log-service.js';
 import { EmailService } from '../../basic/service/email-service.js';
 import { UserService } from '../../sys/authority/service/user-service.js';
-import { AccessGetter } from './access-getter.js';
 import { CnameRecordService } from '../../cname/service/cname-record-service.js';
 import { CnameProxyService } from './cname-proxy-service.js';
 import { PluginConfigGetter } from '../../plugin/service/plugin-config-getter.js';
 import dayjs from 'dayjs';
 import { DbAdapter } from '../../db/index.js';
-import { isComm, isPlus } from '@certd/plus-core';
+import { isComm } from '@certd/plus-core';
 import { logger } from '@certd/basic';
 import { UrlService } from './url-service.js';
 import { NotificationService } from './notification-service.js';
 import { NotificationGetter } from './notification-getter.js';
+import { UserSuiteService } from '@certd/commercial-core';
+import { CertInfoService } from '../../monitor/service/cert-info-service.js';
 
 const runningTasks: Map<string | number, Executor> = new Map();
-const freeCount = 10;
 
 /**
  * 证书申请
  */
 @Provide()
-@Scope(ScopeEnum.Singleton)
+@Scope(ScopeEnum.Request, { allowDowngrade: true })
 export class PipelineService extends BaseService<PipelineEntity> {
   @InjectEntityModel(PipelineEntity)
   repository: Repository<PipelineEntity>;
@@ -60,6 +69,9 @@ export class PipelineService extends BaseService<PipelineEntity> {
   @Inject()
   userService: UserService;
 
+  @Inject()
+  userSuiteService: UserSuiteService;
+
   @Inject()
   cron: Cron;
 
@@ -75,6 +87,9 @@ export class PipelineService extends BaseService<PipelineEntity> {
   @Inject()
   dbAdapter: DbAdapter;
 
+  @Inject()
+  certInfoService: CertInfoService;
+
   //@ts-ignore
   getRepository() {
     return this.repository;
@@ -144,26 +159,19 @@ export class PipelineService extends BaseService<PipelineEntity> {
       //修改
       old = await this.info(bean.id);
     }
+    const pipeline = JSON.parse(bean.content || '{}');
     const isUpdate = bean.id > 0 && old != null;
+
+    let domains = [];
+    RunnableCollection.each(pipeline.stages, (runnable: any) => {
+      if (runnable.runnableType === 'step' && runnable.type.startsWith('CertApply')) {
+        domains = runnable.input.domains || [];
+      }
+    });
+
     if (!isUpdate) {
       //如果是添加，校验数量
-      if (!isPlus()) {
-        const count = await this.repository.count();
-        if (count >= freeCount) {
-          throw new NeedVIPException(`基础版最多只能创建${freeCount}条流水线`);
-        }
-      }
-      const userId = bean.userId;
-      const userIsAdmin = await this.userService.isAdmin(userId);
-      if (!userIsAdmin) {
-        //非管理员用户，限制pipeline数量
-        const count = await this.repository.count({ where: { userId } });
-        const sysPublic = await this.sysSettingsService.getSetting<SysPublicSettings>(SysPublicSettings);
-        const limitUserPipelineCount = sysPublic.limitUserPipelineCount;
-        if (limitUserPipelineCount && limitUserPipelineCount > 0 && count >= limitUserPipelineCount) {
-          throw new NeedVIPException(`您最多只能创建${limitUserPipelineCount}条流水线`);
-        }
-      }
+      await this.checkMaxPipelineCount(bean, pipeline, domains);
     }
 
     if (!isUpdate) {
@@ -171,19 +179,51 @@ export class PipelineService extends BaseService<PipelineEntity> {
       await this.addOrUpdate(bean);
     }
     await this.clearTriggers(bean.id);
-    if (bean.content) {
-      const pipeline = JSON.parse(bean.content);
-      if (pipeline.title) {
-        bean.title = pipeline.title;
-      }
-      pipeline.id = bean.id;
-      bean.content = JSON.stringify(pipeline);
+    if (pipeline.title) {
+      bean.title = pipeline.title;
     }
+    pipeline.id = bean.id;
+    bean.content = JSON.stringify(pipeline);
     await this.addOrUpdate(bean);
     await this.registerTriggerById(bean.id);
+
+    //保存域名信息到certInfo表
+    await this.certInfoService.updateDomains(pipeline.id, domains);
     return bean;
   }
 
+  private async checkMaxPipelineCount(bean: PipelineEntity, pipeline: Pipeline, domains: string[]) {
+    // if (!isPlus()) {
+    //   const count = await this.repository.count();
+    //   if (count >= freeCount) {
+    //     throw new NeedVIPException(`基础版最多只能创建${freeCount}条流水线`);
+    //   }
+    // }
+    if (isComm()) {
+      //校验pipelineCount
+      const userSuite = await this.userSuiteService.getMySuiteDetail(bean.userId);
+      if (userSuite?.pipelineCount.used + 1 > userSuite?.pipelineCount.max) {
+        throw new NeedSuiteException(`对不起，您最多只能创建${userSuite?.pipelineCount.max}条流水线，请购买或升级套餐`);
+      }
+
+      if (userSuite.domainCount.used + domains.length > userSuite.domainCount.max) {
+        throw new NeedSuiteException(`对不起，您最多只能添加${userSuite.domainCount.max}个域名，请购买或升级套餐`);
+      }
+    }
+
+    const userId = bean.userId;
+    const userIsAdmin = await this.userService.isAdmin(userId);
+    if (!userIsAdmin) {
+      //非管理员用户，限制pipeline数量
+      const count = await this.repository.count({ where: { userId } });
+      const sysPublic = await this.sysSettingsService.getSetting<SysPublicSettings>(SysPublicSettings);
+      const limitUserPipelineCount = sysPublic.limitUserPipelineCount;
+      if (limitUserPipelineCount && limitUserPipelineCount > 0 && count >= limitUserPipelineCount) {
+        throw new NeedVIPException(`您最多只能创建${limitUserPipelineCount}条流水线`);
+      }
+    }
+  }
+
   async foreachPipeline(callback: (pipeline: PipelineEntity) => void) {
     const idEntityList = await this.repository.find({
       select: {
@@ -578,4 +618,8 @@ export class PipelineService extends BaseService<PipelineEntity> {
       { groupId }
     );
   }
+
+  async getUserPipelineCount(userId) {
+    return await this.repository.count({ where: { userId } });
+  }
 }