rigjs/lib/classes/cicd/Deploy/CDN.ts

A multi-repos dev tool based on yarn and git.Rigjs is intended to be the simplest way to develop,share and deliver codes between different developers or different projects.

import moment from 'dayjs';
import qs from 'qs';
import crypto from 'crypto';
import axios from 'axios';
import * as uuid from 'uuid';
import { DeployTarget } from '../CICD';

type TFlag = 'break' | 'enhance_break' | null;

enum CDNInterfaceEnum {
  BatchSetCdnDomainConfig = 'BatchSetCdnDomainConfig', //批量修改域名信息
  RefreshObjectCaches = 'RefreshObjectCaches', //刷新节点上的文件内容
  PushObjectCache = 'PushObjectCache', //预热CDN节点
  DescribeRefreshTaskById = 'DescribeRefreshTaskById', //通过任务编号查询刷新预热任务信息
  DescribeCdnDomainConfigs = 'DescribeCdnDomainConfigs', // 获取加速域名的配置信息
}

class CDN {
  AccessKeySecret: string;
  AccessKeyId: string;
  constructor(target: DeployTarget) {
    this.AccessKeyId = target.access_key;
    this.AccessKeySecret = target.access_secret;
  }
  /**
   * 访问CDN通用接口
   * @param {接口名称} actionName
   * @param {各接口定制化参数} paramObj
   * @returns
   */
  private async getCdnData(actionName: string, paramObj: Object) {
    let config = {
      Action: actionName,
      Format: 'JSON',
      Version: '2018-05-10',
      AccessKeyId: this.AccessKeyId,
      SignatureMethod: 'HMAC-SHA1',
      Timestamp: moment().toDate().toISOString(),
      SignatureVersion: '1.0',
      SignatureNonce: uuid.v1(),
    };
    config = Object.assign(config, paramObj);
    let paramConfig = qs.stringify(config, {
      sort: (a: any, b: any) => {
        return a < b ? -1 : 1;
      },
      charset: 'utf-8',
    });

    const strSign = `GET&%2F&${encodeURIComponent(paramConfig)}`;
    // console.log(`strSign: ${strSign}\n`);
    const hmacSha1 = crypto.createHmac('sha1', `${this.AccessKeySecret}&`);
    hmacSha1.update(strSign);
    const signature = hmacSha1.digest('base64');
    config = Object.assign(config, {
      Signature: signature,
    });
    paramConfig = qs.stringify(config, {
      sort: (a, b) => {
        return a < b ? -1 : 1;
      },
      charset: 'utf-8',
      format: 'RFC3986',
    });

    const url = `http://cdn.ap-southeast-1.aliyuncs.com?${paramConfig}`;
    console.log('cdn update url:', url);
    const res = await axios.create().get(url);
    return res.data;
  }

  /**
   * 改写回源URI接口
   * @param {加速域名} domainName
   * @param {需要重写的url 数组} sourceUrls
   * @param {重写目标url 数组} targetUrls
   * @param {改写操作执行规则 数组 值为null、break或enhance_break} flags
   * @returns
   */
  public async setRewriteUri(
    domainName: string,
    sourceUrls: string[],
    targetUrls: string[],
    flags: TFlag[]
  ) {
    try {
      if (sourceUrls.length !== targetUrls.length) {
        throw new Error(`sourceUrls's length not equal targetUrls's length`);
      }
      const Functions: Object[] = [];
      sourceUrls.forEach((item, index) => {
        Functions.push({
          functionArgs: [
            {
              argName: 'source_url',
              argValue: item,
            },
            {
              argName: 'target_url',
              argValue: targetUrls[index],
            },
            {
              argName: 'flag',
              argValue: flags[index] || 'enhance_break',
            },
          ],
          functionName: 'back_to_origin_url_rewrite',
        });
      });

      const data = await this.getCdnData(CDNInterfaceEnum.BatchSetCdnDomainConfig, {
        DomainNames: domainName,
        Functions: JSON.stringify(Functions),
      });
      return data;
    } catch (e) {
      console.error(
        `setRewriteUri Error: ${e.response ? JSON.stringify(e.response.data.Message) : e}`
      );
      throw new Error(e.response.data.Message);
    }
  }

  /**
   * 刷新节点上的文件内容
   * @param {刷新URL, 格式为加速域名或刷新的文件或目录。多个URL之间使用换行符(\n)或(\r\n)分隔} objectPath
   * @param {刷新的类型 File: 文件; Directory: 目录} objectType
   */
  public async refreshCache(objectPath: string, objectType?: string) {
    try {
      let param = {
        ObjectPath: objectPath,
      };
      if (objectType) {
        param = Object.assign(param, { ObjectType: objectType });
      }
      const data = await this.getCdnData(CDNInterfaceEnum.RefreshObjectCaches, param);
      return data;
    } catch (e) {
      console.error('Error:');
      console.error(e.response.data.Message);
      throw new Error(e.response.data.Message);
    }
  }

  /**
   * 预热源站内容到缓存节点
   * @param {预热URL,格式为加速域名或预热的文件 多个URL之间使用换行符(\n)或(\r\n)分隔 单条长度最长为1024个字符} objectPath
   * @returns
   */
  async pushCache(objectPath: string) {
    const data = await this.getCdnData(CDNInterfaceEnum.PushObjectCache, {
      ObjectPath: objectPath,
    });
    return data;
  }

  /**
   * 通过任务编号查询刷新预热任务信息
   * @param {支持同时传入多个任务ID，多个任务ID之间用英文逗号（,）分隔，最多支持同时传入10个任务ID} taskIds
   * @returns
   */
  async describeRefreshTaskById(taskIds: string) {
    try {
      const data = await this.getCdnData(CDNInterfaceEnum.DescribeRefreshTaskById, {
        TaskId: taskIds,
      });
      return data;
    } catch (e) {
      console.error('Error:');
      console.error(e.response.data.Message);
      throw new Error(e.response.data.Message);
    }
  }

  /**
   * 获取加速域名的配置信息
   * @param {加速域名} domainName
   * @param {功能配置ID} configId
   * @returns
   */
  async describeCdnDomainConfigs(domainName: string, configId?: string) {
    try {
      const data = await this.getCdnData(CDNInterfaceEnum.DescribeCdnDomainConfigs, {
        DomainName: domainName,
        ConfigId: configId,
      });
      return data;
    } catch (e) {
      console.error('Error:');
      console.error(e.response.data.Message);
      throw new Error(e.response.data.Message);
    }
  }
}

export default CDN;