47 lines
1.4 KiB
TypeScript
47 lines
1.4 KiB
TypeScript
import type { NextApiRequest } from "next";
|
|
import z from "zod";
|
|
|
|
export function parseIpFromHeaders(value: string | string[]) {
|
|
return Array.isArray(value) ? value[0] : value.split(",")[0];
|
|
}
|
|
|
|
/**
|
|
* Tries to extract IP address from a request
|
|
* @see https://github.com/vercel/examples/blob/main/edge-functions/ip-blocking/lib/get-ip.ts
|
|
**/
|
|
export default function getIP(request: Request | NextApiRequest) {
|
|
let xff =
|
|
request instanceof Request
|
|
? request.headers.get("cf-connecting-ip")
|
|
: request.headers["cf-connecting-ip"];
|
|
|
|
if (!xff) {
|
|
xff = request instanceof Request ? request.headers.get("x-real-ip") : request.headers["x-real-ip"];
|
|
}
|
|
|
|
return xff ? parseIpFromHeaders(xff) : "127.0.0.1";
|
|
}
|
|
|
|
const banlistSchema = z.array(z.string());
|
|
|
|
export function isIpInBanlist(request: Request | NextApiRequest) {
|
|
const IP = getIP(request);
|
|
const rawBanListJson = process.env.IP_BANLIST || "[]";
|
|
const banList = banlistSchema.parse(JSON.parse(rawBanListJson));
|
|
if (banList.includes(IP)) {
|
|
console.log(`Found banned IP: ${IP} in IP_BANLIST`);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
export function isIpInBanListString(identifer: string) {
|
|
const rawBanListJson = process.env.IP_BANLIST || "[]";
|
|
const banList = banlistSchema.parse(JSON.parse(rawBanListJson));
|
|
if (banList.includes(identifer)) {
|
|
console.log(`Found banned IP: ${identifer} in IP_BANLIST`);
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|