From 75a38d95f305b4271d9106babe7cffc1c89ae8f3 Mon Sep 17 00:00:00 2001
From: xiaojunnuo <xiaojunnuo@qq.com>
Date: Fri, 24 Jan 2025 18:04:17 +0800
Subject: [PATCH] =?UTF-8?q?perf:=20=E4=B8=8A=E4=BC=A0=E8=87=AA=E5=AE=9A?=
 =?UTF-8?q?=E4=B9=89=E8=AF=81=E4=B9=A6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 packages/ui/certd-client/src/App.vue          |  3 +-
 .../src/views/certd/cname/record/crud.tsx     |  2 +-
 .../src/views/certd/monitor/cert/api.ts       | 99 ++++++++++---------
 .../src/views/certd/monitor/cert/crud.tsx     | 83 ++++++++++++++--
 .../user/monitor/cert-info-controller.ts      | 11 +++
 .../auto/auto-e-pipeline-emitter-register.ts  |  2 +-
 .../monitor/service/cert-info-service.ts      | 40 +++++++-
 7 files changed, 177 insertions(+), 63 deletions(-)

diff --git a/packages/ui/certd-client/src/App.vue b/packages/ui/certd-client/src/App.vue
index 36f5f1214..221acfca8 100644
--- a/packages/ui/certd-client/src/App.vue
+++ b/packages/ui/certd-client/src/App.vue
@@ -10,9 +10,8 @@
 <script lang="ts" setup>
 import zhCN from "ant-design-vue/es/locale/zh_CN";
 import enUS from "ant-design-vue/es/locale/en_US";
-import { nextTick, provide, ref } from "vue";
+import { provide, ref } from "vue";
 import { usePageStore } from "/src/store/modules/page";
-import { useResourceStore } from "/src/store/modules/resource";
 import { useSettingStore } from "/@/store/modules/settings";
 import "dayjs/locale/zh-cn";
 import "dayjs/locale/en";
diff --git a/packages/ui/certd-client/src/views/certd/cname/record/crud.tsx b/packages/ui/certd-client/src/views/certd/cname/record/crud.tsx
index 6a5cf89a4..5204128c3 100644
--- a/packages/ui/certd-client/src/views/certd/cname/record/crud.tsx
+++ b/packages/ui/certd-client/src/views/certd/cname/record/crud.tsx
@@ -70,7 +70,7 @@ export default function ({ crudExpose, context }: CreateCrudOptionsProps): Creat
           key: "id",
           type: "number",
           column: {
-            width: 100
+            width: 80
           },
           form: {
             show: false
diff --git a/packages/ui/certd-client/src/views/certd/monitor/cert/api.ts b/packages/ui/certd-client/src/views/certd/monitor/cert/api.ts
index 41b9d9c47..ee752f843 100644
--- a/packages/ui/certd-client/src/views/certd/monitor/cert/api.ts
+++ b/packages/ui/certd-client/src/views/certd/monitor/cert/api.ts
@@ -1,54 +1,57 @@
 import { request } from "/src/api/service";
 
-export function createApi() {
-  const apiPrefix = "/monitor/cert";
-  return {
-    async GetList(query: any) {
-      return await request({
-        url: apiPrefix + "/page",
-        method: "post",
-        data: query
-      });
-    },
+const apiPrefix = "/monitor/cert";
+export const certInfoApi = {
+  async GetList(query: any) {
+    return await request({
+      url: apiPrefix + "/page",
+      method: "post",
+      data: query
+    });
+  },
 
-    async AddObj(obj: any) {
-      return await request({
-        url: apiPrefix + "/add",
-        method: "post",
-        data: obj
-      });
-    },
+  async AddObj(obj: any) {
+    return await request({
+      url: apiPrefix + "/add",
+      method: "post",
+      data: obj
+    });
+  },
 
-    async UpdateObj(obj: any) {
-      return await request({
-        url: apiPrefix + "/update",
-        method: "post",
-        data: obj
-      });
-    },
+  async UpdateObj(obj: any) {
+    return await request({
+      url: apiPrefix + "/update",
+      method: "post",
+      data: obj
+    });
+  },
 
-    async DelObj(id: number) {
-      return await request({
-        url: apiPrefix + "/delete",
-        method: "post",
-        params: { id }
-      });
-    },
+  async DelObj(id: number) {
+    return await request({
+      url: apiPrefix + "/delete",
+      method: "post",
+      params: { id }
+    });
+  },
 
-    async GetObj(id: number) {
-      return await request({
-        url: apiPrefix + "/info",
-        method: "post",
-        params: { id }
-      });
-    },
-    async ListAll() {
-      return await request({
-        url: apiPrefix + "/all",
-        method: "post"
-      });
-    }
-  };
-}
-
-export const pipelineGroupApi = createApi();
+  async GetObj(id: number) {
+    return await request({
+      url: apiPrefix + "/info",
+      method: "post",
+      params: { id }
+    });
+  },
+  async ListAll() {
+    return await request({
+      url: apiPrefix + "/all",
+      method: "post"
+    });
+  },
+  async Upload(body: { id?: number; cert: { crt: string; key: string } }) {
+    return await request({
+      url: apiPrefix + "/upload",
+      method: "post",
+      data: body
+    });
+  }
+};
diff --git a/packages/ui/certd-client/src/views/certd/monitor/cert/crud.tsx b/packages/ui/certd-client/src/views/certd/monitor/cert/crud.tsx
index 1d1c0966b..278e70547 100644
--- a/packages/ui/certd-client/src/views/certd/monitor/cert/crud.tsx
+++ b/packages/ui/certd-client/src/views/certd/monitor/cert/crud.tsx
@@ -1,12 +1,14 @@
 // @ts-ignore
 import { useI18n } from "vue-i18n";
-import { AddReq, CreateCrudOptionsProps, CreateCrudOptionsRet, DelReq, EditReq, UserPageQuery, UserPageRes } from "@fast-crud/fast-crud";
-import { pipelineGroupApi } from "./api";
+import { AddReq, CreateCrudOptionsProps, CreateCrudOptionsRet, DelReq, EditReq, useFormWrapper, UserPageQuery, UserPageRes } from "@fast-crud/fast-crud";
+import { certInfoApi } from "./api";
 import dayjs from "dayjs";
+import { useUserStore } from "/@/store/modules/user";
+import { useRouter } from "vue-router";
 
 export default function ({ crudExpose, context }: CreateCrudOptionsProps): CreateCrudOptionsRet {
   const { t } = useI18n();
-  const api = pipelineGroupApi;
+  const api = certInfoApi;
   const pageRequest = async (query: UserPageQuery): Promise<UserPageRes> => {
     return await api.GetList(query);
   };
@@ -26,7 +28,8 @@ export default function ({ crudExpose, context }: CreateCrudOptionsProps): Creat
     const res = await api.AddObj(form);
     return res;
   };
-
+  const { openCrudFormDialog } = useFormWrapper();
+  const router = useRouter();
   return {
     crudOptions: {
       request: {
@@ -50,7 +53,67 @@ export default function ({ crudExpose, context }: CreateCrudOptionsProps): Creat
           width: 600
         }
       },
-      actionbar: { show: false },
+      actionbar: {
+        show: true,
+        buttons: {
+          add: {
+            text: "上传自定义证书",
+            type: "primary",
+            async click() {
+              function createCrudOptions() {
+                return {
+                  crudOptions: {
+                    request: {
+                      addRequest: async (form: any) => {
+                        return await api.Upload(form);
+                      },
+                      editRequest: async (form: any) => {
+                        return await api.Upload(form);
+                      }
+                    },
+                    columns: {
+                      id: {
+                        title: "ID",
+                        type: "number",
+                        form: {
+                          show: false
+                        }
+                      },
+                      "cert.crt": {
+                        title: "证书",
+                        type: "textarea",
+                        form: {
+                          component: {
+                            rows: 4
+                          },
+                          rules: [{ required: true, message: "此项必填" }]
+                        }
+                      },
+                      "cert.key": {
+                        title: "私钥",
+                        type: "textarea",
+                        form: {
+                          component: {
+                            rows: 4
+                          },
+                          rules: [{ required: true, message: "此项必填" }]
+                        }
+                      }
+                    },
+                    form: {
+                      wrapper: {
+                        title: "上传自定义证书"
+                      }
+                    }
+                  }
+                };
+              }
+              const { crudOptions } = createCrudOptions();
+              const wrapperRef = await openCrudFormDialog({ crudOptions });
+            }
+          }
+        }
+      },
       rowHandle: {
         width: 200,
         fixed: "right",
@@ -176,7 +239,7 @@ export default function ({ crudExpose, context }: CreateCrudOptionsProps): Creat
           }
         },
         "pipeline.title": {
-          title: "已关联流水线",
+          title: "关联流水线",
           search: { show: false },
           type: "link",
           form: {
@@ -185,7 +248,13 @@ export default function ({ crudExpose, context }: CreateCrudOptionsProps): Creat
           column: {
             width: 350,
             sorter: true,
-            component: {}
+            component: {
+              on: {
+                onClick({ row }) {
+                  router.push({ path: "/certd/pipeline/detail", query: { id: row.pipelineId, editMode: "false" } });
+                }
+              }
+            }
           }
         }
       }
diff --git a/packages/ui/certd-server/src/controller/user/monitor/cert-info-controller.ts b/packages/ui/certd-server/src/controller/user/monitor/cert-info-controller.ts
index 49355cd7b..75f66cae0 100644
--- a/packages/ui/certd-server/src/controller/user/monitor/cert-info-controller.ts
+++ b/packages/ui/certd-server/src/controller/user/monitor/cert-info-controller.ts
@@ -84,4 +84,15 @@ export class CertInfoController extends CrudController<CertInfoService> {
     });
     return this.ok(list);
   }
+
+  @Post('/upload', { summary: Constants.per.authOnly })
+  async upload(@Body(ALL) body: any) {
+    if (body.id) {
+      await this.service.checkUserId(body.id, this.getUserId());
+    }
+
+    const res = await this.service.upload(body);
+
+    return this.ok(res);
+  }
 }
diff --git a/packages/ui/certd-server/src/modules/auto/auto-e-pipeline-emitter-register.ts b/packages/ui/certd-server/src/modules/auto/auto-e-pipeline-emitter-register.ts
index 67068ea8e..2d0b950a6 100644
--- a/packages/ui/certd-server/src/modules/auto/auto-e-pipeline-emitter-register.ts
+++ b/packages/ui/certd-server/src/modules/auto/auto-e-pipeline-emitter-register.ts
@@ -16,7 +16,7 @@ export class AutoEPipelineEmitterRegister {
   }
   async onCertApplySuccess() {
     pipelineEmitter.on(EVENT_CERT_APPLY_SUCCESS, async (event: PipelineEvent<CertReader>) => {
-      await this.certInfoService.updateCert(event.pipeline.id, event.event);
+      await this.certInfoService.updateCertByPipelineId(event.pipeline.id, event.event);
     });
   }
 }
diff --git a/packages/ui/certd-server/src/modules/monitor/service/cert-info-service.ts b/packages/ui/certd-server/src/modules/monitor/service/cert-info-service.ts
index 9c8482bd4..b83b082a1 100644
--- a/packages/ui/certd-server/src/modules/monitor/service/cert-info-service.ts
+++ b/packages/ui/certd-server/src/modules/monitor/service/cert-info-service.ts
@@ -1,11 +1,17 @@
 import { Provide, Scope, ScopeEnum } from '@midwayjs/core';
-import { BaseService, CodeException, Constants, PageReq } from '@certd/lib-server';
+import { BaseService, CodeException, CommonException, Constants, PageReq } from '@certd/lib-server';
 import { InjectEntityModel } from '@midwayjs/typeorm';
 import { Repository } from 'typeorm';
 import { CertInfoEntity } from '../entity/cert-info.js';
 import { utils } from '@certd/basic';
 import { CertInfo, CertReader } from '@certd/plugin-cert';
 
+export type UploadCertReq = {
+  id?: number;
+  certReader: CertReader;
+  fromType?: string;
+};
+
 @Provide()
 @Scope(ScopeEnum.Request, { allowDowngrade: true })
 export class CertInfoService extends BaseService<CertInfoEntity> {
@@ -125,15 +131,27 @@ export class CertInfoService extends BaseService<CertInfoEntity> {
     return certReader.toCertInfo();
   }
 
-  async updateCert(pipelineId: number, certReader: CertReader) {
+  async updateCertByPipelineId(pipelineId: number, certReader: CertReader, fromType = 'pipeline') {
     const found = await this.repository.findOne({
       where: {
         pipelineId,
       },
     });
+    const bean = await this.updateCert({
+      id: found?.id,
+      certReader,
+      fromType,
+    });
+    return bean;
+  }
+
+  private async updateCert(req: UploadCertReq) {
     const bean = new CertInfoEntity();
-    if (found) {
-      bean.id = found.id;
+    const { id, fromType, certReader } = req;
+    if (id) {
+      bean.id = id;
+    } else {
+      bean.fromType = fromType;
     }
     const certInfo = certReader.toCertInfo();
     bean.certInfo = JSON.stringify(certInfo);
@@ -146,5 +164,19 @@ export class CertInfoService extends BaseService<CertInfoEntity> {
     bean.certProvider = certReader.detail.issuer.commonName;
 
     await this.addOrUpdate(bean);
+    return bean;
+  }
+
+  async upload(body: { id?: number; cert: CertInfo }) {
+    const { id, cert } = body;
+    if (!cert) {
+      throw new CommonException("cert can't be empty");
+    }
+    const res = await this.updateCert({
+      id,
+      certReader: new CertReader(cert),
+      fromType: 'upload',
+    });
+    return res.id;
   }
 }