ich777/openai-github-copilot
1
0
Fork 0
mirror of https://gitea.com/PublicAffairs/openai-github-copilot.git synced 2025-07-23 12:23:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Cache github token

Browse Source
This commit is contained in:
johnd0e
2024-03-03 01:41:44 +01:00
parent 645b9551c4
commit 5059cffa15
2 changed files with 24 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/worker.mjs
View File

@@ -1,5 +1,5 @@
export default { export default {
async fetch (request) { async fetch (request, env) {
if (request.method === "OPTIONS") { if (request.method === "OPTIONS") {
return handleOPTIONS(request); return handleOPTIONS(request);
} }
@@ -14,8 +14,26 @@ export default {
status: 401 status: 401
}); });
} }
const { token, errResponse } = await getToken(authKey); const tokenCache = env.KV;
if (errResponse) { return errResponse; } let token;
try {
token = await tokenCache.get(authKey);
} catch (e) {
console.error(e);
}
if (!token) {
let errResponse;
({ token, errResponse } = await getToken(authKey));
if (errResponse) { return errResponse; }
const expiration = token.match(/;exp=(\d+);/)?.[1];
if (expiration) {
try {
await tokenCache.put(authKey, token, { expiration });
} catch (e) {
console.error(e);
}
}
}
return makeRequest(request, url.pathname, await makeHeaders(token)); return makeRequest(request, url.pathname, await makeHeaders(token));
} }
}; };

8
wrangler.toml
View File

@@ -3,8 +3,6 @@ main = "src/worker.mjs"
compatibility_date = "2024-01-17" compatibility_date = "2024-01-17"
no_bundle = true no_bundle = true
# Bind a KV Namespace. Use KV as persistent storage for small key-value pairs. [[kv_namespaces]]
# Docs: https://developers.cloudflare.com/workers/runtime-apis/kv binding = "KV"
# [[kv_namespaces]] id = "4e595a4637f048ed8c92bab1821c6e54"
# binding = "MY_KV_NAMESPACE"
# id = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"