nezha_dash/app/api/server-ip/route.ts

import { auth } from "@/auth";
import getEnv from "@/lib/env-entry";
import { GetServerIP } from "@/lib/serverFetch";
import fs from "fs";
import { AsnResponse, CityResponse, Reader } from "maxmind";
import { redirect } from "next/navigation";
import { NextRequest, NextResponse } from "next/server";
import path from "path";

export const dynamic = "force-dynamic";

interface ResError extends Error {
  statusCode: number;
  message: string;
}

export type IPInfo = {
  city: CityResponse;
  asn: AsnResponse;
};

export async function GET(req: NextRequest) {
  if (getEnv("SitePassword")) {
    const session = await auth();
    if (!session) {
      redirect("/");
    }
  }

  if (!getEnv("NEXT_PUBLIC_ShowIpInfo")) {
    return NextResponse.json(
      { error: "NEXT_PUBLIC_ShowIpInfo is disable" },
      { status: 400 },
    );
  }

  const { searchParams } = new URL(req.url);
  const server_id = searchParams.get("server_id");

  if (!server_id) {
    return NextResponse.json(
      { error: "server_id is required" },
      { status: 400 },
    );
  }

  try {
    const ip = await GetServerIP({ server_id: Number(server_id) });

    const cityDbPath = path.join(process.cwd(), "lib", "GeoLite2-City.mmdb");

    const asnDbPath = path.join(process.cwd(), "lib", "GeoLite2-ASN.mmdb");

    const cityDbBuffer = fs.readFileSync(cityDbPath);
    const asnDbBuffer = fs.readFileSync(asnDbPath);

    const cityLookup = new Reader<CityResponse>(cityDbBuffer);
    const asnLookup = new Reader<AsnResponse>(asnDbBuffer);

    const data: IPInfo = {
      city: cityLookup.get(ip) as CityResponse,
      asn: asnLookup.get(ip) as AsnResponse,
    };
    return NextResponse.json(data, { status: 200 });
  } catch (error) {
    const err = error as ResError;
    console.error("Error in GET handler:", err);
    const statusCode = err.statusCode || 500;
    const message = err.message || "Internal Server Error";
    return NextResponse.json({ error: message }, { status: statusCode });
  }
}