fix: adjust upstash api

This commit is contained in:
Fred 2024-03-13 23:58:28 +08:00
parent 61ce3868b5
commit 9a8497299d
No known key found for this signature in database
GPG Key ID: 4DABDA85EF70EC71
2 changed files with 81 additions and 5 deletions

View File

@ -0,0 +1,72 @@
import { NextRequest, NextResponse } from "next/server";
async function handle(
req: NextRequest,
{ params }: { params: { action: string; key: string[] } },
) {
const requestUrl = new URL(req.url);
const endpoint = requestUrl.searchParams.get("endpoint");
if (req.method === "OPTIONS") {
return NextResponse.json({ body: "OK" }, { status: 200 });
}
const [action, ...key] = params.key;
// only allow to request to *.upstash.io
if (!endpoint || !endpoint.endsWith("upstash.io")) {
return NextResponse.json(
{
error: true,
msg: "you are not allowed to request " + params.key.join("/"),
},
{
status: 403,
},
);
}
// only allow upstash get and set method
if (action !== "get" && action !== "set") {
return NextResponse.json(
{
error: true,
msg: "you are not allowed to request " + params.action,
},
{
status: 403,
},
);
}
const [protocol, ...subpath] = params.key;
const targetUrl = `${protocol}://${subpath.join("/")}`;
const method = req.headers.get("method") ?? undefined;
const shouldNotHaveBody = ["get", "head"].includes(
method?.toLowerCase() ?? "",
);
const fetchOptions: RequestInit = {
headers: {
authorization: req.headers.get("authorization") ?? "",
},
body: shouldNotHaveBody ? null : req.body,
method,
// @ts-ignore
duplex: "half",
};
const fetchResult = await fetch(targetUrl, fetchOptions);
console.log("[Any Proxy]", targetUrl, {
status: fetchResult.status,
statusText: fetchResult.statusText,
});
return fetchResult;
}
export const POST = handle;
export const GET = handle;
export const OPTIONS = handle;
export const runtime = "edge";

View File

@ -85,17 +85,21 @@ export function createUpstashClient(store: SyncStore) {
};
},
path(path: string) {
let url = config.endpoint;
// let url = config.endpoint;
if (!url.endsWith("/")) {
url += "/";
if (!path.endsWith("/")) {
path += "/";
}
if (path.startsWith("/")) {
path = path.slice(1);
}
return url + path;
let url = new URL("/api/" + path);
// add query params
url.searchParams.append("endpoint", config.endpoint);
return url.toString();
},
};
}