root/nezha_dash
1
0
Fork 0
mirror of https://github.com/hamster1963/nezha-dash.git synced 2025-04-24 21:10:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: use fs

Browse Source
This commit is contained in:
hamster1963 2024-12-12 00:52:19 +08:00
parent 8f38cf94d8
commit dfef790884
3 changed files with 33 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
app/(main)/server/[id]/page.tsx
View File

@ -17,10 +17,9 @@ export default function Page(props: { params: Promise<{ id: string }> }) {
const updateViews = async () => {
const ipInfo = await GetIPInfo({ server_id: params.id });
console.log(ipInfo);
}
updateViews()
}, [])
};
updateViews();
}, []);
return (
<div className="mx-auto grid w-full max-w-5xl gap-2">

46
lib/GetIPInfo.ts
View File

@ -1,26 +1,30 @@
"use server";
import maxmind, { CityResponse, AsnResponse } from 'maxmind';
import { GetServerIP } from './serverFetch';
import path from 'path';
import fs from "fs";
import { AsnResponse, CityResponse, Reader } from "maxmind";
import { GetServerIP } from "./serverFetch";
type IPInfo = {
city: CityResponse;
asn: AsnResponse;
}
export default async function GetIPInfo({ server_id }: { server_id: string }): Promise<IPInfo> {
const ip = await GetServerIP({ server_id: Number(server_id) })
// 使用 path.join 获取正确的文件路径
const cityDbPath = path.join(process.cwd(), 'lib', 'GeoLite2-City.mmdb');
const asnDbPath = path.join(process.cwd(), 'lib', 'GeoLite2-ASN.mmdb');
const cityLookup = await maxmind.open<CityResponse>(cityDbPath)
const asnLookup = await maxmind.open<AsnResponse>(asnDbPath)
return {
city: cityLookup.get(ip) as CityResponse,
asn: asnLookup.get(ip) as AsnResponse
}
city: CityResponse;
asn: AsnResponse;
};
export default async function GetIPInfo({
server_id,
}: {
server_id: string;
}): Promise<IPInfo> {
const ip = await GetServerIP({ server_id: Number(server_id) });
const cityDbBuffer = fs.readFileSync("./lib/GeoLite2-City.mmdb");
const asnDbBuffer = fs.readFileSync("./lib/GeoLite2-ASN.mmdb");
const cityLookup = new Reader<CityResponse>(cityDbBuffer);
const asnLookup = new Reader<AsnResponse>(asnDbBuffer);
return {
city: cityLookup.get(ip) as CityResponse,
asn: asnLookup.get(ip) as AsnResponse,
};
}

10
lib/serverFetch.tsx
View File

@ -130,8 +130,11 @@ export async function GetServerMonitor({ server_id }: { server_id: number }) {
}
}
export async function GetServerIP({ server_id }: { server_id: number }): Promise<string> {
export async function GetServerIP({
server_id,
}: {
server_id: number;
}): Promise<string> {
let nezhaBaseUrl = getEnv("NezhaBaseUrl");
if (!nezhaBaseUrl) {
console.error("NezhaBaseUrl is not set");
@ -176,11 +179,8 @@ export async function GetServerIP({ server_id }: { server_id: number }): Promise
console.error("GetNezhaData error:", error);
throw error; // Rethrow the error to be caught by the caller
}
}
export async function GetServerDetail({ server_id }: { server_id: number }) {
let nezhaBaseUrl = getEnv("NezhaBaseUrl");
if (!nezhaBaseUrl) {