Files
wr.do/middleware.ts
T
2024-08-02 17:00:42 +08:00

45 lines
1.4 KiB
TypeScript

import { NextResponse } from "next/server";
import { geolocation } from "@vercel/functions";
import { auth } from "auth";
import { siteConfig } from "./config/site";
// export { auth as middleware } from "auth";
export default auth(async (req) => {
// console.log(req.auth);
try {
const ip = req.headers.get("X-Forwarded-For");
if (req.url.includes("/s/")) {
const match = req.url.match(/[^/]+$/);
const geo = geolocation(req);
if (match) {
const referer = req.headers.get("referer") || "(None)";
const res = await fetch(
`${siteConfig.url}/api/s?slug=${match[0]}&referer=${referer}&ip=${ip}&city=${geo?.city}&region=${geo?.region}&country=${geo?.country}&latitude=${geo?.latitude}&longitude=${geo?.longitude}&flag=${geo?.flag}`,
);
if (!res.ok) {
return NextResponse.redirect(`${siteConfig.url}/docs/short-urls`);
}
const target = await res.json();
if (!target) {
return NextResponse.redirect(`${siteConfig.url}/docs/short-urls`);
}
return NextResponse.redirect(target);
}
}
return NextResponse.next();
} catch (error) {
return NextResponse.redirect(siteConfig.url);
}
});
// Read more: https://nextjs.org/docs/app/building-your-application/routing/middleware#matcher
export const config = {
matcher: ["/((?!api|_next/static|_next/image|favicon.ico).*)"],
};