certd/packages/plugins/plugin-tencent/src/dns-provider/dnspod-dns-provider.ts

import { Autowire, HttpClient, ILogger } from "@certd/pipeline";
import { CreateRecordOptions, IDnsProvider, IsDnsProvider, RemoveRecordOptions } from "@certd/plugin-cert";
import _ from "lodash";
import { DnspodAccess } from "../access";

@IsDnsProvider({
  name: "dnspod",
  title: "dnspod(腾讯云)",
  desc: "腾讯云的域名解析接口已迁移到dnspod",
  accessType: "dnspod",
})
export class DnspodDnsProvider implements IDnsProvider {
  @Autowire()
  http!: HttpClient;

  @Autowire()
  access!: DnspodAccess;
  @Autowire()
  logger!: ILogger;

  loginToken: any;

  endpoint = "";
  async onInstance() {
    const access: DnspodAccess = this.access as DnspodAccess;
    this.loginToken = access.id + "," + access.token;
    this.endpoint = access.endpoint || "https://dnsapi.cn";
  }

  async doRequest(options: any, successCodes: string[] = []) {
    const config: any = {
      // @ts-ignore
      method: "post",
      formData: {
        login_token: this.loginToken,
        format: "json",
        lang: "cn",
        error_on_empty: "no",
      },
      timeout: 5000,
    };
    _.merge(config, options);

    const ret: any = await this.http.request(config);
    if (!ret || !ret.status) {
      const code = ret.status.code;
      if (code !== "1" || !successCodes.includes(code)) {
        throw new Error("请求失败：" + ret.status.message + ",api=" + config.url);
      }
    }
    return ret;
  }

  async getDomainList() {
    const ret = await this.doRequest({
      url: this.access.endpoint + "/Domain.List",
    });
    this.logger.debug("dnspod 域名列表：", ret.domains);
    return ret.domains;
  }

  async createRecord(options: CreateRecordOptions): Promise<any> {
    const { fullRecord, value, type } = options;
    this.logger.info("添加域名解析：", fullRecord, value);
    const domainItem = await this.matchDomain(fullRecord);
    const domain = domainItem.name;
    const rr = fullRecord.replace("." + domain, "");

    const ret = await this.doRequest(
      {
        url: this.access.endpoint + "/Record.Create",
        formData: {
          domain,
          sub_domain: rr,
          record_type: type,
          record_line: "默认",
          value: value,
          mx: 1,
        },
      },
      ["104"]
    ); // 104错误码为记录已存在，无需再次添加
    this.logger.info("添加域名解析成功:", fullRecord, value, JSON.stringify(ret.record));
    return ret.record;
  }

  async removeRecord(options: RemoveRecordOptions) {
    const { fullRecord, value, record } = options;
    const domain = await this.matchDomain(fullRecord);

    const ret = await this.doRequest({
      url: this.access.endpoint + "/Record.Remove",
      formData: {
        domain,
        record_id: record.id,
      },
    });
    this.logger.info("删除域名解析成功:", fullRecord, value);
    return ret.RecordId;
  }

  async matchDomain(dnsRecord: any) {
    const list = await this.getDomainList();
    let domain = null;
    for (const item of list) {
      if (_.endsWith(dnsRecord, "." + item.name)) {
        domain = item;
        break;
      }
    }
    if (!domain) {
      throw new Error("找不到域名,请检查域名是否正确：" + dnsRecord);
    }
    return domain;
  }
}
new DnspodDnsProvider();