// 根据客户端ip地址，返回到不同的源站组，也就是在IP地址不变的情况下，同一个客户端回到同一个源站
const ORIGIN_GROUPS = ["originGroup1", "originGroup2"];

// 定义虚拟节点数，若源站组（ORIGIN_GROUPS）较多时，建议调低虚拟节点数
const VIRTUAL_NODES_PER_GROUP = 15;
const ORIGIN_HEADER_NAME = 'X-Forwarded-For-Origin';
let virtualNodesHashesCache = null;

// 定义全局变量来跟踪函数调用次数
addEventListener("fetch", (event) => {
  handleRequest(event.request);
});

async function handleRequest(request) {
  // 通过 EO-Client-IP 头部获取客户端 IP
  const ip = request.headers.get("EO-Client-IP") || "";
  
  // 如果缓存中没有虚拟节点哈希值，则生成虚拟节点哈希值
  if (!virtualNodesHashesCache) {
    virtualNodesHashesCache = await generateVirtualNodesHashes();
  }
  
  const group = await findSourceGroupForIp(
    ip,
    virtualNodesHashesCache.hashes,
    virtualNodesHashesCache.mapping
  );
  console.log(`Group: ${group}`);
  
  request.headers.set(ORIGIN_HEADER_NAME, group)

  return;
}

// 生成虚拟节点的哈希值
async function generateVirtualNodesHashes() {
  const virtualNodesHashes = {};
  
  for (let group in ORIGIN_GROUPS) {
    for (let i = 0; i < VIRTUAL_NODES_PER_GROUP; i++) {
      const virtualNodeIdentifier = `${group}-VN${i}`;
      const hash = await md5(virtualNodeIdentifier);
      if (!virtualNodesHashes[hash]) {
        virtualNodesHashes[hash] = group;
      }
    }
  }

  const hashes = Object.keys(virtualNodesHashes).sort();
  
  return { hashes, mapping: virtualNodesHashes }; // 返回排序后的哈希值数组和映射
}

// 映射客户端IP到虚拟节点，并找到对应的源站组
async function findSourceGroupForIp(ip, hashes, mapping) {
  
  // 使用md5函数计算IP的哈希值
  const ipHash = await md5(ip);
  let closestHash = hashes.find((hash) => hash > ipHash) || hashes[0];
  
  // 根据找到的虚拟节点哈希值，从虚拟节点哈希表中获取对应的源站组名称
  const selectedGroupName = mapping[closestHash];
  const selectedGroupIPs = ORIGIN_GROUPS[selectedGroupName];
  
  // 打印日志，显示客户端IP、哈希值、最接近的虚拟节点哈希值、选定的源站组名称
  console.log(
    `IP: ${ip}, Hash: ${ipHash}, Closest Hash: ${closestHash}, Selected Group: ${selectedGroupName}`
  );
  
  // 返回选定的源站组名称
  return selectedGroupIPs;
}

function bufferToHex(arr) {
  return Array.prototype.map
    .call(arr, (x) => (x >= 16 ? x.toString(16) : "0" + x.toString(16)))
    .join("");
}

async function md5(text) {
  const buffer = await crypto.subtle.digest("MD5", TextEncoder().encode(text));
  return bufferToHex(new Uint8Array(buffer));
}