feat: 优化GitLab徽章设置功能

badge/cloudml.ts

@@ -0,0 +1,141 @@
+import service from "../service";
+
+const projectList = [
+  "cloud-ml/cloudml-maas",
+  "cloud-ml/cloudml-permission",
+  "cloud-ml/cloudml-base",
+  "cloud-ml/cloudml-dataset",
+  "cloud-ml/cloudml-dev",
+  "cloud-ml/cloudml-train",
+  "cloud-ml/cloudml-finetune",
+  "cloud-ml/model-repository",
+  "cloud-ml-public/cloudml-grpc-java",
+  "cloud-ml/cloudml_runqueue",
+  "cloud-ml/cloudml_syncqueue",
+  "cloud-ml/resource-instance",
+  "cloud-ml/ai-workbench-common",
+  "cloud-ml/evaluate.git",
+  "ai-service-data/ai-train-platform-service.git",
+  "cloud-ml/embedding_management",
+  "cloud-ml/knowledge-base",
+  "cloud-ml/ai-proxy",
+];
+
+const getNewProjectBadge = async (
+  projectDetail: GitlabProjDetail
+): Promise<GitlabBadgeSetParams[]> => {
+  // 项目路径 cloud-ml/cloudml-dev
+  const projectPath = projectDetail.path_with_namespace;
+  // 根据项目路径获取sonarqubeId 类似于 cloud-ml/cloudml-dev -> cloud-ml:cloudml-dev
+  const sonarqubeId = projectPath.replace("/", ":");
+  // 获取项目的badges
+  const badges: GitlabBadge[] = await service.gitlab.fetchProjectBadges(
+    projectDetail.id
+  );
+  // 对badges进行补全，可能只有 name: "sonarqube_coverage" 的情况，把剩下的补全
+  const badgeNames = badges.map((badge) => badge.name);
+  const badgeNameSet = new Set(badgeNames);
+  const badgeNameList = [
+    "sonarqube_bugs",
+    "sonarqube_vulnerabilities",
+    "sonarqube_security_hotspots",
+    "sonarqube_coverage",
+    "sonarqube_quality_gate",
+  ];
+  const diff = [...badgeNameList].filter((x) => !badgeNameSet.has(x));
+  const newBadges: GitlabBadgeSetParams[] = diff.map((name) => {
+    const link_url = `https://sonarqube.mioffice.cn/dashboard?id=${sonarqubeId}`;
+    const image_url =
+      name !== "sonarqube_quality_gate"
+        ? `https://sonarqube.mioffice.cn/api/badges/measure?key=${sonarqubeId}&metric=${name}`
+        : `https://sonarqube.mioffice.cn/api/badges/gate?key=${sonarqubeId}`;
+    return {
+      id: projectDetail.id,
+      link_url,
+      image_url,
+      rendered_image_url: image_url,
+      rendered_link_url: link_url,
+      name,
+    };
+  });
+  return newBadges;
+};
+
+const addNewProjectBadge = async (
+  badgeSetParamsList: GitlabBadgeSetParams[]
+) => {
+  const chunkSize = 5; // 每次并发请求的数量
+  for (let i = 0; i < badgeSetParamsList.length; i += chunkSize) {
+    const chunk = badgeSetParamsList.slice(i, i + chunkSize);
+    const res = await Promise.all(
+      chunk.map((badgeSetParams) =>
+        service.gitlab.addProjectBadge(badgeSetParams)
+      )
+    );
+    console.log(res);
+  }
+};
+
+const main = async () => {
+  const errorList: string[] = [];
+  const badgeAddParamsList = await Promise.all(
+    projectList.map(async (projectName) => {
+      const projectDetail = await service.gitlab.fetchProjectDetails(
+        projectName
+      );
+      if (!projectDetail) {
+        errorList.push(projectName);
+        return [];
+      }
+      return await getNewProjectBadge(projectDetail);
+    })
+  );
+
+  await addNewProjectBadge(badgeAddParamsList.flat());
+};
+
+main();
+
+// [
+//   {
+//     name: "sonarqube_bugs",
+//     link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-voiceprint-fe",
+//     image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-voiceprint-fe&metric=bugs",
+//     rendered_link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-voiceprint-fe",
+//     rendered_image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-voiceprint-fe&metric=bugs",
+//     id: 9725,
+//     kind: "project",
+//   }, {
+//     name: "sonarqube_vulnerabilities",
+//     link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-shortcut-fe&metric=vulnerabilities",
+//     rendered_link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     rendered_image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-shortcut-fe&metric=vulnerabilities",
+//     id: 10943,
+//     kind: "project",
+//   }, {
+//     name: "sonarqube_security_hotspots",
+//     link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-shortcut-fe&metric=security_hotspots",
+//     rendered_link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     rendered_image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-shortcut-fe&metric=security_hotspots",
+//     id: 10944,
+//     kind: "project",
+//   }, {
+//     name: "sonarqube_coverage",
+//     link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-shortcut-fe&metric=coverage",
+//     rendered_link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     rendered_image_url: "https://sonarqube.mioffice.cn/api/badges/measure?key=miai-fe:fe:ai-shortcut-fe&metric=coverage",
+//     id: 10945,
+//     kind: "project",
+//   }, {
+//     name: "sonarqube_quality_gate",
+//     link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     image_url: "https://sonarqube.mioffice.cn/api/badges/gate?key=miai-fe:fe:ai-shortcut-fe",
+//     rendered_link_url: "https://sonarqube.mioffice.cn/dashboard?id=miai-fe:fe:ai-shortcut-fe",
+//     rendered_image_url: "https://sonarqube.mioffice.cn/api/badges/gate?key=miai-fe:fe:ai-shortcut-fe",
+//     id: 10946,
+//     kind: "project",
+//   }
+// ]

badge/miai.ts

@@ -0,0 +1,117 @@
+import service from "../service";
+
+const projectList = [
+  "miai-fe/fe/ai-admin-fe",
+  "miai-fe/fe/ai-operation-fe",
+  "miai-fe/fe/ai-dripback-fe",
+  "miai-fe/fe/ai-open-admin-fe",
+  "miai-fe/fe/ai-model-deployment-fe",
+  "miai-fe/fe/ai-workbench",
+  "miai-fe/fe/ai-review-fe",
+  "miai-fe/fe/cms",
+  "miai-fe/fe/xiaoai-open-Platform",
+  "miai-fe/fe/ai-robot-fe",
+  "miai-fe/fe/ai-scene-review-fe",
+  "miai-fe/fe/houyi",
+  "miai-fe/fe/kms",
+  "miai-fe/fe/ai-productive-fe",
+  "miai-fe/fe/ai-sliding-label-fe",
+  "miai-fe/fe/ai-label-control-fe",
+  "miai-fe/fe/asr-fe",
+  "miai-fe/fe/feedback-platform-fe",
+  "miai-fe/fe/statisfaction-evaluation-platform-fe",
+  "miai-fe/fe/ai-quality-score-fe",
+  "miai-fe/fe/wakeup",
+  "miai-fe/fe/ai-badcase-label-fe",
+  "miai-fe/fe/func-manager",
+  "miai-fe/service/ado-server",
+  "miai-fe/fe/ado-fe",
+  "miai-fe/pwa/ai-rn-domain",
+  "miai-fe/pwa/ai-ak-fe",
+  "miai-fe/fe/ai-user-info-fe",
+  "miai-fe/fe-infra/xiaoai-jssdk",
+  "miai-fe/fe-infra/ai-ui",
+  "miai-fe/fe/ai-class-sheducle-fe",
+  "miai-fe/service/ai-schedule-service",
+  "miai-fe/fe/ai-schedule-devtool-v3",
+  "miai-fe/ai-scenes-fe",
+  "miai-fe/activity/om-music-tone",
+  "miai-fe/activity/ug-guide-test",
+  "miai-fe/activity/om-mood-blindbox",
+  "miai-fe/activity/om-love-poem",
+  "miai-fe/fe/ai-introduce",
+  "miai-fe/fe/ai-phonecall-guider-fe",
+  "miai-fe/activity/om-spring-couplets",
+  "miai-fe/activity/ug-play-assistant",
+  "miai-fe/fe/ai-iot-introduce-fe",
+  "miai-fe/fe/ai-rookie-play-fe",
+  "miai-fe/fe/ai-custom-tts-fe",
+  "miai-fe/fe-infra/split-rpk-sdk",
+  "miai-fe/fe/ai-full-animation-template-fe",
+  "miai-fe/fe/ai-schedule-manager-fe",
+  "miai-fe/fe/ai-voiceprint-fe",
+  "miai-fe/fe/ai-shortcut-fe",
+];
+
+const getProjectId = async (projectName: string) => {
+  const res = await service.gitlab.fetchProjectDetails(projectName);
+  return res?.id;
+};
+
+const getNewProjectBadge = async (
+  projectId: number
+): Promise<GitlabBadgeSetParams[]> => {
+  const badges: GitlabBadge[] = await service.gitlab.fetchProjectBadges(
+    projectId
+  );
+
+  const replacePath = (value: string) =>
+    value.replace(
+      "https://sonarqube.mioffice.cn",
+      "http://scan.sonarqube.xiaomi.srv"
+    );
+
+  return badges.map((badge: GitlabBadge) => ({
+    id: projectId,
+    badge_id: badge.id,
+    link_url: replacePath(badge.link_url),
+    image_url: badge.image_url,
+    rendered_image_url: badge.rendered_image_url,
+    rendered_link_url: replacePath(badge.rendered_link_url),
+  }));
+};
+
+const setNewProjectBadge = async (
+  badgeSetParamsList: GitlabBadgeSetParams[]
+) => {
+  const chunkSize = 5; // 每次并发请求的数量
+  for (let i = 0; i < badgeSetParamsList.length; i += chunkSize) {
+    const chunk = badgeSetParamsList.slice(i, i + chunkSize);
+    const res = await Promise.all(
+      chunk.map((badgeSetParams) =>
+        service.gitlab.setProjectBadge(badgeSetParams)
+      )
+    );
+    console.log(res);
+  }
+};
+
+const main = async () => {
+  const errorList: string[] = [];
+  const badgeSetParamsList = await Promise.all(
+    projectList.map(async (projectName) => {
+      const projectId = await getProjectId(projectName);
+      if (!projectId) {
+        errorList.push(projectName);
+        return [];
+      }
+      return await getNewProjectBadge(projectId);
+    })
+  );
+
+  await setNewProjectBadge(badgeSetParamsList.flat());
+
+  console.log("errorList", errorList);
+};
+
+main();

service/gitlab.ts

@@ -7,16 +7,16 @@ const BASE_URL = "https://git.n.xiaomi.com/api/v4";
 const gitlabReq = async <T = any>(
   url: string,
   params: any,
-  defaultValue: any,
+  default_value: any,
   reqFunc: any = netTool.get
 ): Promise<T> => {
   try {
     const response = (await reqFunc(url, params, AUTH_HEADER)) as T &
       GitlabError;
-    if (response.message === "404 Project Not Found") return defaultValue;
+    if (response.message === "404 Project Not Found") return default_value;
     return response;
   } catch {
-    return defaultValue;
+    return default_value;
   }
 };
 
@@ -72,22 +72,26 @@ const fetchProjectBadges = async (project_id: number) => {
 
 /**
  * 设置徽章
- * @param project_id
+ * @param badge
- * @param badge_id
- * @param new_badge
  */
-const setProjectBadge = async (
+const setProjectBadge = async (badge: GitlabBadgeSetParams) => {
-  project_id: number,
+  const URL = `${BASE_URL}/projects/${badge.id}/badges/${badge.badge_id}`;
-  badge_id: number,
+  return gitlabReq<GitlabBadge>(URL, badge, null, netTool.put);
-  new_badge: GitlabBadge
+};
-) => {
+
-  const URL = `${BASE_URL}/projects/${project_id}/badges/${badge_id}`;
+/**
-  return gitlabReq<GitlabBadge>(URL, new_badge, new_badge, netTool.put);
+ * 添加徽章
+ * @param badge
+ */
+const addProjectBadge = async (badge: GitlabBadgeSetParams) => {
+  const URL = `${BASE_URL}/projects/${badge.id}/badges`;
+  return gitlabReq<GitlabBadge>(URL, badge, null, netTool.post);
 };
 
 const gitlab = {
   fetchPipelines,
   setProjectBadge,
+  addProjectBadge,
   fetchProjectBadges,
   fetchProjectDetails,
   fetchPipelineDetails,

service/typings.d.ts

@@ -60,3 +60,12 @@ interface GitlabBadge {
   rendered_image_url: string;
   kind: "project" | "group";
 }
+
+interface GitlabBadgeSetParams {
+  id: number;
+  badge_id?: number;
+  link_url: string;
+  image_url: string;
+  rendered_link_url: string;
+  rendered_image_url: string;
+}