104 lines
2.3 KiB
TypeScript
104 lines
2.3 KiB
TypeScript
import { NextRequest } from "next/server";
|
|
import { ServerConfig, getIP } from "../../common";
|
|
|
|
export const REQUEST_TIMEOUT_MS = 60000;
|
|
|
|
export const authHeaderName = "Authorization";
|
|
|
|
const makeBearer = (s: string) => `Bearer ${s.trim()}`;
|
|
|
|
const validString = (x?: string): x is string => Boolean(x && x.length > 0);
|
|
|
|
function parseApiKey(bearToken: string) {
|
|
const token = bearToken.trim().replaceAll("Bearer ", "").trim();
|
|
|
|
return {
|
|
apiKey: token,
|
|
};
|
|
}
|
|
|
|
export function prettyObject(msg: any) {
|
|
const obj = msg;
|
|
if (typeof msg !== "string") {
|
|
msg = JSON.stringify(msg, null, " ");
|
|
}
|
|
if (msg === "{}") {
|
|
return obj.toString();
|
|
}
|
|
if (msg.startsWith("```json")) {
|
|
return msg;
|
|
}
|
|
return ["```json", msg, "```"].join("\n");
|
|
}
|
|
|
|
export function parseResp(res: { choices: { message: { content: any } }[] }) {
|
|
return {
|
|
message: res.choices?.[0]?.message?.content ?? "",
|
|
};
|
|
}
|
|
|
|
export function auth(req: NextRequest, serverConfig: ServerConfig) {
|
|
const { hideUserApiKey, apiKey: systemApiKey } = serverConfig;
|
|
const authToken = req.headers.get(authHeaderName) ?? "";
|
|
|
|
const { apiKey } = parseApiKey(authToken);
|
|
|
|
console.log("[User IP] ", getIP(req));
|
|
console.log("[Time] ", new Date().toLocaleString());
|
|
|
|
if (hideUserApiKey && apiKey) {
|
|
return {
|
|
error: true,
|
|
message: "you are not allowed to access with your own api key",
|
|
};
|
|
}
|
|
|
|
if (apiKey) {
|
|
console.log("[Auth] use user api key");
|
|
return {
|
|
error: false,
|
|
};
|
|
}
|
|
|
|
if (systemApiKey) {
|
|
console.log("[Auth] use system api key");
|
|
req.headers.set(authHeaderName, `Bearer ${systemApiKey}`);
|
|
} else {
|
|
console.log("[Auth] admin did not provide an api key");
|
|
}
|
|
|
|
return {
|
|
error: false,
|
|
};
|
|
}
|
|
|
|
export function getTimer() {
|
|
const controller = new AbortController();
|
|
|
|
// make a fetch request
|
|
const requestTimeoutId = setTimeout(
|
|
() => controller.abort(),
|
|
REQUEST_TIMEOUT_MS,
|
|
);
|
|
|
|
return {
|
|
...controller,
|
|
clear: () => {
|
|
clearTimeout(requestTimeoutId);
|
|
},
|
|
};
|
|
}
|
|
|
|
export function getHeaders(openaiApiKey?: string) {
|
|
const headers: Record<string, string> = {
|
|
"Content-Type": "application/json",
|
|
Accept: "application/json",
|
|
};
|
|
|
|
if (validString(openaiApiKey)) {
|
|
headers[authHeaderName] = makeBearer(openaiApiKey);
|
|
}
|
|
|
|
return headers;
|
|
}
|