feat: unifi controller integration (#2236)

* feat: unifi controller integration * fix: pr feedback * fix: pr feedback * fix: pr feedback * fix: formatting * fix: pr feedback * fix: typecheck --------- Co-authored-by: Manuel <30572287+manuel-rw@users.noreply.github.com> Co-authored-by: Meier Lukas <meierschlumpf@gmail.com>
2025-04-06 10:17:51 +00:00
1 parent 7caad6fc47
commit c1cd563048
@@ -0,0 +1,84 @@
+"use client";
+
+import { useMemo } from "react";
+import { Box } from "@mantine/core";
+import dayjs from "dayjs";
+import duration from "dayjs/plugin/duration";
+import objectSupport from "dayjs/plugin/objectSupport";
+import relativeTime from "dayjs/plugin/relativeTime";
+
+import { clientApi } from "@homarr/api/client";
+
+import type { WidgetComponentProps } from "../../definition";
+import { WifiVariant } from "./variants/wifi-variant";
+import { WiredVariant } from "./variants/wired-variant";
+
+dayjs.extend(objectSupport);
+dayjs.extend(relativeTime);
+dayjs.extend(duration);
+
+export default function NetworkControllerNetworkStatusWidget({
+  options,
+  integrationIds,
+}: WidgetComponentProps<"networkControllerStatus">) {
+  const [summaries] = clientApi.widget.networkController.summary.useSuspenseQuery(
+    {
+      integrationIds,
+    },
+    {
+      refetchOnMount: false,
+      refetchOnWindowFocus: false,
+      refetchOnReconnect: false,
+      retry: false,
+    },
+  );
+  const utils = clientApi.useUtils();
+
+  clientApi.widget.networkController.subscribeToSummary.useSubscription(
+    {
+      integrationIds,
+    },
+    {
+      onData: (data) => {
+        utils.widget.networkController.summary.setData(
+          {
+            integrationIds,
+          },
+          (prevData) => {
+            if (!prevData) {
+              return undefined;
+            }
+
+            return prevData.map((item) =>
+              item.integration.id === data.integration.id
+                ? {
+                    ...item,
+                    summary: data.summary,
+                    updatedAt: new Date(),
+                  }
+                : item,
+            );
+          },
+        );
+      },
+    },
+  );
+
+  const data = useMemo(() => summaries.flatMap(({ summary }) => summary), [summaries]);
+
+  return (
+    <Box p={"sm"}>
+      {options.content === "wifi" ? (
+        <WifiVariant
+          countGuests={data.reduce((sum, summary) => sum + summary.wifi.guests, 0)}
+          countUsers={data.reduce((sum, summary) => sum + summary.wifi.users, 0)}
+        />
+      ) : (
+        <WiredVariant
+          countGuests={data.reduce((sum, summary) => sum + summary.lan.guests, 0)}
+          countUsers={data.reduce((sum, summary) => sum + summary.lan.users, 0)}
+        />
+      )}
+    </Box>
+  );
+}