feat: 部署函数支持添加云梯默认标签(运营部门,运营产品,负责人)

Author: wangpand0508

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "tencent-component-toolkit",
-  "version": "2.24.2",
+  "version": "2.24.3",
   "description": "Tencent component toolkit",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",
@@ -66,12 +66,12 @@
     "@semantic-release/git": "^9.0.0",
     "@semantic-release/npm": "^7.0.4",
     "@semantic-release/release-notes-generator": "^9.0.1",
+    "@types/axios": "^0.14.0",
     "@types/react-grid-layout": "^1.1.2",
     "@types/uuid": "^8.3.1",
     "@typescript-eslint/eslint-plugin": "^4.14.0",
     "@typescript-eslint/parser": "^4.14.0",
     "@ygkit/secure": "^0.0.3",
-    "axios": "^0.21.0",
     "dotenv": "^8.2.0",
     "eslint": "^7.18.0",
     "eslint-config-prettier": "^6.10.0",
@@ -90,6 +90,7 @@
     "@types/jest": "^26.0.20",
     "@types/node": "^14.14.31",
     "@ygkit/request": "^0.1.8",
+    "axios": "^0.21.0",
     "camelcase": "^6.2.0",
     "cos-nodejs-sdk-v5": "^2.9.20",
     "dayjs": "^1.10.4",

package.json

@@ -8,6 +8,7 @@ const ACTIONS = [
   'CreateRole',
   'GetRole',
   'DeleteRole',
+  'GetUserAppId',
 ] as const;
 
 export type ActionType = typeof ACTIONS[number];

src/modules/cam/apis.ts

@@ -2,6 +2,8 @@ import { ActionType } from './apis';
 import { CapiCredentials, RegionType, ApiServiceType } from './../interface';
 import { Capi } from '@tencent-sdk/capi';
 import APIS from './apis';
+import { getYunTiApiUrl } from '../../utils';
+import axios from 'axios';
 
 /** CAM （访问管理）for serverless */
 export default class Cam {
@@ -112,4 +114,54 @@ export default class Cam {
   async CheckSCFExcuteRole() {
     return this.isRoleExist('QCS_SCFExcuteRole');
   }
+
+  /** 查询用户AppId */
+  async GetUserAppId(): Promise<{ OwnerUin: string; AppId: string; Uin: string }> {
+    try {
+      return this.request({
+        Action: 'GetUserAppId',
+      });
+    } catch (error) {
+      return {
+        OwnerUin: '',
+        AppId: '',
+        Uin: '',
+      };
+    }
+  }
+
+  /**
+   * checkYunTi 检查是否是云梯账号
+   * @returns {boolean} true: 是云梯账号; false: 非云梯账号
+   */
+  async checkYunTi(): Promise<boolean> {
+    let isYunTi = false;
+    const { OwnerUin: uin } = await this.GetUserAppId();
+    try {
+      const params = JSON.stringify({
+        id: '1',
+        jsonrpc: '2.0',
+        method: 'checkOwnUin',
+        params: { ownUin: [uin] },
+      });
+      const apiUrl = getYunTiApiUrl();
+      const res = await axios.post(apiUrl, params, {
+        headers: { 'content-type': 'application/json' },
+      });
+      if (res?.data?.error?.message) {
+        throw new Error(res.data.error.message);
+      } else {
+        isYunTi =
+          res?.data?.result?.data &&
+          res.data.result.data?.some(
+            (item: { ownUin: string; appId: string }) => item?.ownUin === uin,
+          );
+        console.log('check yunTi ownUin:', isYunTi);
+      }
+    } catch (error) {
+      isYunTi = false;
+      console.log('checkYunTiOwnUin error:', error);
+    }
+    return isYunTi;
+  }
 }

src/modules/cam/index.ts

@@ -3,7 +3,7 @@ import { ActionType } from './apis';
 import { RegionType, ApiServiceType, CapiCredentials } from './../interface';
 import { Capi } from '@tencent-sdk/capi';
 import { ApiTypeError } from '../../utils/error';
-import { deepClone, strip } from '../../utils';
+import { deepClone, formatInputTags, strip } from '../../utils';
 import TagsUtils from '../tag/index';
 import ApigwUtils from '../apigw';
 import CONFIGS from './config';
@@ -252,13 +252,15 @@ export default class Scf {
           credentials: this.credentials,
           region: this.region,
         });
+        const tags: any = trigger?.parameters?.tags ?? trigger?.tags ?? funcInfo.Tags ?? [];
         const triggerOutput = await triggerInstance.create({
           scf: this,
           region: this.region,
           inputs: {
             namespace: funcInfo.Namespace,
             functionName: funcInfo.FunctionName,
             ...trigger,
+            tags: formatInputTags(tags),
           },
         });
 
@@ -457,4 +459,14 @@ export default class Scf {
     const logs = await this.scf.getLogs(inputs);
     return logs;
   }
+
+  checkAddedYunTiTags(tags: Array<{ [key: string]: string }>): boolean {
+    const formatTags = formatInputTags(tags);
+    const result =
+      formatTags?.length > 0 &&
+      ['运营部门', '运营产品', '负责人'].every((tagKey) =>
+        formatTags.some((item) => item.key === tagKey && !!item.value),
+      );
+    return result;
+  }
 }

src/modules/scf/index.ts

@@ -78,6 +78,8 @@ export interface TriggerType {
   TriggerName?: string;
   Qualifier?: string;
   compared?: boolean;
+  tags?: object;
+  parameters?: any;
 }
 
 export type OriginTriggerType = {

src/modules/scf/interface.ts

@@ -61,10 +61,8 @@ export const formatInputs = (inputs: ScfCreateFunctionInputs) => {
     // 监听端口: -1 表示 job镜像，0 ~ 65526 表示webServer镜像
     if (imageConfig.imagePort) {
       functionInputs.Code!.ImageConfig!.ImagePort =
-        Number.isInteger(imageConfig?.imagePort) &&
-        imageConfig?.imagePort >= -1 &&
-        imageConfig?.imagePort <= 65535
-          ? imageConfig.imagePort
+        Number.isInteger(imageConfig?.imagePort) && imageConfig?.imagePort === -1
+          ? -1
           : WebServerImageDefaultPort;
     }
   } else {

src/modules/scf/utils.ts

@@ -147,7 +147,7 @@ export default class ApigwTrigger extends BaseTrigger<ApigwTriggerInputsParams>
     funcInfo?: FunctionInfo;
     inputs: TriggerInputs<ApigwTriggerInputsParams>;
   }) {
-    const { parameters, isAutoRelease } = inputs;
+    const { parameters, isAutoRelease, tags } = inputs;
     const {
       oldState,
       protocols,
@@ -192,6 +192,7 @@ export default class ApigwTrigger extends BaseTrigger<ApigwTriggerInputsParams>
         method: endpoints[0].method ?? 'ANY',
       },
       created: !!parameters?.created,
+      tags,
     };
     const triggerKey = this.getKey(triggerInputs);
 

src/modules/triggers/apigw.ts

@@ -1,4 +1,5 @@
 import { ApigwDeployInputs, ApiEndpoint } from '../../apigw/interface';
+import { TagInput } from '../../interface';
 
 export interface ApigwTriggerRemoveScfTriggerInputs {
   serviceId: string;
@@ -129,6 +130,9 @@ export interface TriggerInputs<P extends TriggerInputsParams = TriggerInputsPara
 
   // 是否自动发布服务（API 网关特有）
   isAutoRelease?: boolean;
+
+  // 标签列表
+  tags?: TagInput[];
 }
 
 export interface TriggerDetail {

src/modules/triggers/interface/index.ts

@@ -3,6 +3,7 @@ import path from 'path';
 import camelCase from 'camelcase';
 import { PascalCase } from 'type-fest';
 import { CamelCasedProps, PascalCasedProps } from '../modules/interface';
+import crypto from 'crypto';
 
 // TODO: 将一些库换成 lodash
 
@@ -273,3 +274,54 @@ export const getQcsResourceId = (service: string, region: string, uin: string, s
   // 云资源六段式
   return `qcs::${service}:${region}:uin/${uin}:${suffix}`;
 };
+
+/**
+ * hmacSha1 加密HmacSHA1
+ * @param text 加密文本
+ * @param key  加密密钥
+ * @returns
+ */
+export const hmacSha1 = (text: string, key: string) => {
+  return crypto.createHmac('sha1', key).update(text).digest('hex');
+};
+
+/**
+ * getYunTiApiUrl 查询云梯API地址
+ * @returns 云梯API地址
+ */
+export const getYunTiApiUrl = (): string => {
+  const apiKey = process.env.SLS_YUNTI_API_KEY || '';
+  const apiSecret = process.env.SLS_YUNTI_API_SECRET || '';
+  const apiUrl = process.env.SLS_YUNTI_API_URL;
+  const timeStamp = Math.floor(Date.now() / 1000);
+  const apiSign = hmacSha1(`${timeStamp}${apiKey}`, apiSecret);
+  const url = `${apiUrl}?api_key=${apiKey}&api_ts=${timeStamp}&api_sign=${apiSign}`;
+  return url;
+};
+
+/**
+ * formatInputTags 格式化输入标签
+ */
+export const formatInputTags = (
+  input: Array<any> | { [key: string]: string },
+): { key: string; value: string }[] => {
+  let tags: { key: string; value: string }[];
+  if (Array.isArray(input)) {
+    tags = input.map((item) => {
+      return {
+        key: item?.key ?? item?.Key ?? '',
+        value: item?.value ?? item?.Value ?? '',
+      };
+    });
+  } else if (typeof input === 'object' && input) {
+    tags = Object.entries(input).map(([key, value]) => {
+      return {
+        key: (key ?? '').toString(),
+        value: (value ?? '').toString(),
+      };
+    });
+  } else {
+    tags = undefined as any;
+  }
+  return tags;
+};