在 Cloudflare Worker 集中管理所有软件的 OpenAI Key

addEventListener("fetch", (event) => {
  event.respondWith(handleRequest(event.request));
});

async function handleRequest(request) {
  const url = new URL(request.url);
  const fetchAPI = request.url.replace(url.host, 'api.openai.com');
  
  // 部分代理工具，请求由浏览器发起，跨域请求时会先发送一个 preflight 进行检查，也就是 OPTIONS 请求
  // 需要响应该请求，否则后续的 POST 会失败
  const corsHeaders = {
    'Access-Control-Allow-Origin': '*',
    'Access-Control-Allow-Methods': 'OPTIONS',
    'Access-Control-Allow-Headers': '*'
  };
  if (request.method === 'OPTIONS') return new Response(null, { headers: corsHeaders });

  let body;
  if (request.method === 'POST') body = await request.json();

  // 这里加上一个简单的认证, 软件原来填 OpenAI key 的地方填你自己设定的密码, 防止他人滥用
  // 这里密码表示 123456789 或者 ABCD1234EFGH56789
  let accessKey = request.headers.get('Authorization');
  if (accessKey !== 'Bearer 123456789' && accessKey !== 'Bearer ABCD1234EFGH56789') {
    return new Response('Forbidden', { status: 403 })
  }

  // 这里填你的 OpenAI key
  const authKey = 'sk-xxxxxxxxx'
  const payload = {
    method: request.method,
    headers: {
      "Content-Type": "application/json",
      "Authorization": "Bearer " + authKey,
    },
    body: typeof body === 'object' ? JSON.stringify(body) : '{}',
  };
  console.log(payload)
  // 在 Cloudflare 中，HEAD 和 GET 请求带 body 会报错
  if (['HEAD', 'GET'].includes(request.method)) delete payload.body;

  // 入参中如果包含了 stream=true，则表现形式为流式输出
  const response = await fetch(fetchAPI, payload);
  return new Response(response.body, {
    status: response.status,
    statusText: response.statusText,
    headers: response.headers,
  });
}