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 })),
});