nezha_dash/app/api/server/route.ts
2024-10-10 14:35:40 +08:00

81 lines
2.1 KiB
TypeScript

import { NezhaAPI, ServerApi } from "@/app/[locale]/types/nezha-api";
import { MakeOptional } from "@/app/[locale]/types/utils";
import getEnv from "@/lib/env-entry";
import { NextResponse } from "next/server";
export const dynamic = "force-dynamic";
export const runtime = "edge";
interface NezhaDataResponse {
error?: string;
data?: ServerApi;
}
export async function GET(_: Request) {
const response = (await GetNezhaData()) as NezhaDataResponse;
if (response.error) {
console.log(response.error);
return NextResponse.json({ error: response.error }, { status: 400 });
}
return NextResponse.json(response, { status: 200 });
}
async function GetNezhaData() {
var nezhaBaseUrl = getEnv("NezhaBaseUrl");
if (!nezhaBaseUrl) {
console.log("NezhaBaseUrl is not set");
return { error: "NezhaBaseUrl is not set" };
}
// Remove trailing slash
if (nezhaBaseUrl[nezhaBaseUrl.length - 1] === "/") {
nezhaBaseUrl = nezhaBaseUrl.slice(0, -1);
}
try {
const response = await fetch(nezhaBaseUrl + "/api/v1/server/details", {
headers: {
Authorization: getEnv("NezhaAuth") as string,
},
next: {
revalidate: 0,
},
});
const resData = await response.json();
const nezhaData = resData.result as NezhaAPI[];
if (!nezhaData) {
console.log(resData);
return { error: "NezhaData fetch failed" };
}
const data: ServerApi = {
live_servers: 0,
offline_servers: 0,
total_bandwidth: 0,
result: [],
};
const timestamp = Date.now() / 1000;
data.result = nezhaData.map(
(element: MakeOptional<NezhaAPI, "ipv4" | "ipv6" | "valid_ip">) => {
if (timestamp - element.last_active > 300) {
data.offline_servers += 1;
element.online_status = false;
} else {
data.live_servers += 1;
element.online_status = true;
}
data.total_bandwidth += element.status.NetOutTransfer;
delete element.ipv4;
delete element.ipv6;
delete element.valid_ip;
return element;
},
);
return data;
} catch (error) {
return error;
}
}