certd/packages/ui/certd-server/src/plugins/plugin-volcengine/dns-client.ts

import { http } from "@certd/basic";
import querystring from "querystring";
import { VolcengineOpts } from "./ve-client.js";
import { Pager, PageSearch } from "@certd/pipeline";

export type VolcengineReq = {
  method?: string;
  path?: string;
  headers?: any;
  body?: any;
  query?: any;
  service?: string; // 替换为实际服务名称
  region?: string; // 替换为实际区域名称
};

export class VolcengineDnsClient {
  opts: VolcengineOpts;

  constructor(opts: VolcengineOpts) {
    this.opts = opts;
  }
  async importRuntime(packageName: string) {
    return this.opts.access.importRuntime(packageName);
  }

  async doRequest(req: VolcengineReq) {
    const { Signer } = await this.importRuntime("@volcengine/openapi");

    // http request data
    const openApiRequestData: any = {
      region: req.region,
      method: req.method,
      // [optional] http request url query
      params: {
        ...req.query,
      },
      // http request headers
      headers: {
        "Content-Type": "application/json",
      },
      // [optional] http request body
      body: req.body,
    };

    const signer = new Signer(openApiRequestData, req.service);

    // sign
    signer.addAuthorization({ accessKeyId: this.opts.access.accessKeyId, secretKey: this.opts.access.secretAccessKey });

    // Print signed headers
    console.log(openApiRequestData.headers);

    const url = `https://open.volcengineapi.com/?${querystring.stringify(req.query)}`;

    try {
      const res = await http.request({
        url: url,
        method: req.method,
        headers: openApiRequestData.headers,
        data: req.body,
      });
      if (res?.ResponseMetadata?.Error) {
        const err = new Error(JSON.stringify(res.ResponseMetadata.Error));
        // @ts-ignore
        err.detail = res.ResponseMetadata.Error;
        throw err;
      }
      return res;
    } catch (e) {
      if (e.response) {
        const err = new Error(JSON.stringify(e.response.data.ResponseMetadata.Error));
        // @ts-ignore
        err.detail = e.response.data.ResponseMetadata.Error;
        throw err;
      }
    }
  }

  async findDomain(domain: string) {
    const req: VolcengineReq = {
      method: "POST",
      region: "cn-beijing",
      service: "dns",
      query: {
        Action: "ListZones",
        Version: "2018-08-01",
      },
      body: {
        Key: domain,
        SearchMode: "exact",
      },
    };

    return this.doRequest(req);
  }

  async getDomainList(page: PageSearch) {
    const pager = new Pager(page);
    const body: any = {
      SearchMode: "like",
      PageNumber: pager.pageNo,
      PageSize: pager.pageSize,
    };
    if (page.searchKey) {
      body.Key = page.searchKey;
    }
    const req: VolcengineReq = {
      method: "POST",
      region: "cn-beijing",
      service: "dns",
      query: {
        Action: "ListZones",
        Version: "2018-08-01",
      },
      body: body,
    };

    const res = await this.doRequest(req);
    let list = res.Result?.Zones || [];
    list = list.map((item: any) => {
      return {
        id: item.ZID,
        domain: item.ZoneName,
      };
    });
    const total = res.Result?.Total || list.length;
    return {
      list,
      total,
    };
  }
}