【玩转 EdgeOne】使用EdgeOne边缘函数搭建域名注册查询API

https://api.com/?domain=[待查询域名]

{
  "code": 0,
  "DomainName": "域名",
  "Available": false,
  "BlackWord": false,
  "Reason": "不可注册说明"
}

{
  "code": 1,
  "Message": "报错内容",
  "ErrorCode": "报错代码"
}

// 将字符串编码为ArrayBuffer
function stringToArrayBuffer(str) {
    const encoder = new TextEncoder();
    return encoder.encode(str);
  }

  // 将ArrayBuffer转换为十六进制字符串
  function arrayBufferToHexString(arrayBuffer) {
    const byteArray = new Uint8Array(arrayBuffer);
    const hexCodes = [...byteArray].map(value => value.toString(16).padStart(2, '0'));
    return hexCodes.join('');
  }

  async function hmacSHA256(key, data) {
    const importedKey = await crypto.subtle.importKey(
      'raw',
      key,
      { name: 'HMAC', hash: 'SHA-256' },
      false,
      ['sign']
    );

    const msgBuffer = stringToArrayBuffer(data);
    const signatureBuffer = await crypto.subtle.sign('HMAC', importedKey, msgBuffer);

    return signatureBuffer;
  }

  function uint8ArrayToHex(array) {
    return Array.from(array).map(byte => byte.toString(16).padStart(2, '0')).join('');
  }

  // 签名算法
  async function qcloud_v3_post(SecretId,SecretKey,Service,bodyString,headersOper) {
    const HTTPRequestMethod = "POST"
    const CanonicalURI = "/"
    const CanonicalQueryString = ""

    // 将 JSON 对象中的键按 ASCII 升序进行排序
    let sortedheadersOper = Object.keys(headersOper).filter(key => (key.toLowerCase() !== "x-tc-version")).sort();
    // 遍历排序后的键并拼接
    let SignedHeaders = sortedheadersOper.map(key => key.toLowerCase()).join(";");
    let CanonicalHeaders = sortedheadersOper.map(key => key.toLowerCase() + ":" + headersOper[key].toLowerCase()).join("\n");
    CanonicalHeaders = CanonicalHeaders + "\n"

    let HashedRequestPayload = await sha256(bodyString)

    const CanonicalRequest =
      HTTPRequestMethod + '\n' +
      CanonicalURI + '\n' +
      CanonicalQueryString + '\n' +
      CanonicalHeaders + '\n' +
      SignedHeaders + '\n' +
      HashedRequestPayload

    const currentDate = new Date();
    const year = currentDate.getUTCFullYear();
    const month = (currentDate.getUTCMonth() + 1).toString().padStart(2, '0');
    const day = currentDate.getUTCDate().toString().padStart(2, '0');
    const formattedDate = `${year}-${month}-${day}`;


    const Algorithm = "TC3-HMAC-SHA256"
    // 获取当前秒级时间戳
    const RequestTimestamp = Math.floor(Date.now() / 1000).toString();
    // const RequestTimestamp = "1688025007"
    const CredentialScope = formattedDate + "/" + Service + "/tc3_request"
    const HashedCanonicalRequest = await sha256(CanonicalRequest)

    const StringToSign =
      Algorithm + '\n' +
      RequestTimestamp + '\n' +
      CredentialScope + '\n' +
      HashedCanonicalRequest

    const SecretDate = await hmacSHA256(new Uint8Array([...stringToArrayBuffer("TC3"), ...new Uint8Array(stringToArrayBuffer(SecretKey))]), formattedDate);
    const SecretService = await hmacSHA256(SecretDate, Service);
    const SecretSigning = await hmacSHA256(SecretService, "tc3_request");

    const Signature = arrayBufferToHexString(await hmacSHA256(SecretSigning, StringToSign));

    const Authorization =
      Algorithm + ' ' +
      'Credential=' + SecretId + '/' + CredentialScope + ', ' +
      'SignedHeaders=' + SignedHeaders + ', ' +
      'Signature=' + Signature

      headersOper["X-TC-Timestamp"] = RequestTimestamp;
      headersOper["Authorization"] = Authorization;

      return headersOper
  }

  // sha256 签名摘要
  async function sha256(message) {
    const msgBuffer = new TextEncoder().encode(message);
    const hashBuffer = await crypto.subtle.digest('SHA-256', msgBuffer);


    return uint8ArrayToHex(new Uint8Array(hashBuffer));
  }

  // 密钥填写位置
  const SecretId = "";
  const SecretKey = "";
  const Service = "domain";

  async function handleRequest(request) {

    const url = new URL(request.url)
    const params = url.searchParams
    const checkdomain = params.get("domain");

    let checkRetrunData = {};

    if(checkdomain === null){
      checkRetrunData["code"] = 1;
      checkRetrunData["Message"] = "域名参数为空";
      checkRetrunData["ErrorCode"] = "DomainIsNULL";
      return new Response(JSON.stringify(checkRetrunData, null, 2), { 
        headers: { 'Content-Type': 'application/json',
        'Access-Control-Allow-Headers': 'Content-Type',
        'Access-Control-Allow-Methods': 'POST, OPTIONS',
        'Access-Control-Max-Age': '86400',
        'Access-Control-Allow-Origin': '*' },
        status: 200 
      })
    }

    const headersPending = {
      'Host': 'domain.tencentcloudapi.com',
      'Content-Type': 'application/json',
      'X-TC-Action': 'CheckDomain',
      'X-TC-Version': '2018-08-08',
      'X-TC-Region': 'ap-guangzhou',
    };


    const apiBodyJson = {
      "DomainName": checkdomain
    }

    const bodyString = JSON.stringify(apiBodyJson)

    const headers = await qcloud_v3_post(SecretId,SecretKey,Service,bodyString,headersPending)

    const apiurl = 'https://domain.tencentcloudapi.com/';

  let qcloud_api_data;
  await fetch(apiurl, {
    method: 'POST',
    headers: headers,
    body: bodyString
  })
  .then(response => response.json())
  .then(data => qcloud_api_data = data)
  .catch(error => qcloud_api_data = error);

  if(qcloud_api_data["Response"]["Error"] === undefined){
    checkRetrunData["code"] = 0;
    checkRetrunData["DomainName"] = qcloud_api_data["Response"]["DomainName"];
    checkRetrunData["Available"] = qcloud_api_data["Response"]["Available"];
    checkRetrunData["BlackWord"] = qcloud_api_data["Response"]["BlackWord"];
    checkRetrunData["Reason"] = qcloud_api_data["Response"]["Reason"];
  }else{
    checkRetrunData["code"] = 1;
    checkRetrunData["Message"] = qcloud_api_data["Response"]["Error"]["Message"];
    checkRetrunData["ErrorCode"] = qcloud_api_data["Response"]["Error"]["Code"];
  }

    return new Response(JSON.stringify(checkRetrunData, null, 2), { 
        headers: { 'Content-Type': 'application/json',
        'Access-Control-Allow-Headers': 'Content-Type',
        'Access-Control-Allow-Methods': 'POST, OPTIONS',
        'Access-Control-Max-Age': '86400',
        'Access-Control-Allow-Origin': '*' },
        status: 200 
      })
  }

  addEventListener('fetch', (event) => {
    if (event.request.method === 'OPTIONS') {
      event.respondWith(handleOptions(event.request))
    } else {
      event.respondWith(handleRequest(event.request))
    }
  });