63 lines
2.3 KiB
TypeScript
63 lines
2.3 KiB
TypeScript
import {
|
|
firewallCpuRequestHandler,
|
|
firewallInterfacesRequestHandler,
|
|
firewallMemoryRequestHandler,
|
|
firewallVersionRequestHandler,
|
|
} from "@homarr/request-handler/firewall";
|
|
|
|
import { createManyWidgetIntegrationMiddleware } from "../../middlewares/integration";
|
|
import { settleIntegrationQueries, toPublicIntegrationError } from "../../settle-integrations";
|
|
import { createTRPCRouter, publicProcedure } from "../../trpc";
|
|
|
|
const firewallMiddleware = createManyWidgetIntegrationMiddleware("query", "firewall");
|
|
|
|
type FirewallIntegration = Parameters<typeof firewallCpuRequestHandler.handler>[0];
|
|
|
|
interface FirewallHandler<TData> {
|
|
handler: (
|
|
integration: FirewallIntegration,
|
|
input: Record<string, never>,
|
|
) => { getDataAsync: () => Promise<{ data: TData; timestamp: Date }> };
|
|
}
|
|
|
|
interface FirewallQueryResult<TData> {
|
|
integration: {
|
|
id: string;
|
|
name: string;
|
|
kind: FirewallIntegration["kind"];
|
|
updatedAt: Date;
|
|
};
|
|
summary: TData;
|
|
error: string | undefined;
|
|
}
|
|
|
|
const queryFirewall = <TData>(handler: FirewallHandler<TData>, createFallback: () => TData) =>
|
|
publicProcedure.concat(firewallMiddleware).query(async ({ ctx }) =>
|
|
settleIntegrationQueries(
|
|
ctx.integrations,
|
|
async (integration): Promise<FirewallQueryResult<TData>> => {
|
|
const { data, timestamp } = await handler.handler(integration, {}).getDataAsync();
|
|
return {
|
|
integration: { id: integration.id, name: integration.name, kind: integration.kind, updatedAt: timestamp },
|
|
summary: data,
|
|
error: undefined,
|
|
};
|
|
},
|
|
{
|
|
fallback: (integration, error) => ({
|
|
integration: { id: integration.id, name: integration.name, kind: integration.kind, updatedAt: new Date(0) },
|
|
summary: createFallback(),
|
|
error: toPublicIntegrationError(error),
|
|
}),
|
|
throwOnAllFailures: true,
|
|
},
|
|
),
|
|
);
|
|
|
|
export const firewallRouter = createTRPCRouter({
|
|
getFirewallCpuStatus: queryFirewall(firewallCpuRequestHandler, () => ({ total: 0 })),
|
|
getFirewallInterfacesStatus: queryFirewall(firewallInterfacesRequestHandler, () => []),
|
|
getFirewallVersionStatus: queryFirewall(firewallVersionRequestHandler, () => ({ version: "Unknown" })),
|
|
getFirewallMemoryStatus: queryFirewall(firewallMemoryRequestHandler, () => ({ used: 0, total: 0, percent: 0 })),
|
|
});
|