diff --git a/public/locales/en/common.json b/public/locales/en/common.json index d21a82ed..de972f78 100755 --- a/public/locales/en/common.json +++ b/public/locales/en/common.json @@ -662,5 +662,8 @@ "nextpvr": { "upcoming": "Upcoming", "ready": "Recent" + }, + "wgeasy": { + "clients": "Total Clients" } } \ No newline at end of file diff --git a/src/widgets/components.js b/src/widgets/components.js index 62765791..5d2dc7e6 100644 --- a/src/widgets/components.js +++ b/src/widgets/components.js @@ -90,6 +90,7 @@ const components = { uptimekuma: dynamic(() => import("./uptimekuma/component")), watchtower: dynamic(() => import("./watchtower/component")), whatsupdocker: dynamic(() => import("./whatsupdocker/component")), + wgeasy: dynamic(() => import("./wgeasy/component")), xteve: dynamic(() => import("./xteve/component")), }; diff --git a/src/widgets/wgeasy/component.jsx b/src/widgets/wgeasy/component.jsx new file mode 100644 index 00000000..e2434f51 --- /dev/null +++ b/src/widgets/wgeasy/component.jsx @@ -0,0 +1,33 @@ +import { useTranslation } from "next-i18next"; + +import Block from "components/services/widget/block"; +import Container from "components/services/widget/container"; +import useWidgetAPI from "utils/proxy/use-widget-api"; + +export default function Component({ service }) { + const { t } = useTranslation(); + + const { widget } = service; + + const { data: wgeasyData, error: wgeasyAPIError } = useWidgetAPI(widget, "unified", { + refreshInterval: 5000, + }); + + if (wgeasyAPIError) { + return ; + } + + if (!wgeasyData) { + return ( + + + + ); + } + + return ( + + + + ); +} diff --git a/src/widgets/wgeasy/proxy.js b/src/widgets/wgeasy/proxy.js new file mode 100644 index 00000000..76f61151 --- /dev/null +++ b/src/widgets/wgeasy/proxy.js @@ -0,0 +1,117 @@ +/* eslint-disable no-underscore-dangle */ +import { formatApiCall } from "utils/proxy/api-helpers"; +import { httpProxy } from "utils/proxy/http"; +import getServiceWidget from "utils/config/service-helpers"; +import createLogger from "utils/logger"; +import widgets from "widgets/widgets"; + + +const proxyName = "wgeasyProxyHandler"; + +const logger = createLogger(proxyName); +let globalSid = null; + +async function getWidget(req) { + const { group, service } = req.query; + if (!group || !service) { + logger.debug("Invalid or missing service '%s' or group '%s'", service, group); + return null; + } + const widget = await getServiceWidget(group, service); + if (!widget) { + logger.debug("Invalid or missing widget for service '%s' in group '%s'", service, group); + return null; + } + + return widget; +} + + +async function loginToWGEasy(endpoint, widget) { + const api = widgets?.[widget.type]?.api; + if (!api) { + return [403, null]; + } + // Create new session on WgEasy + const url = new URL(formatApiCall(api, { endpoint, ...widget })); + + const [status, data, , responseHeaders] = await httpProxy(url, { + method: 'POST', + headers: { + 'Content-Type': 'application/json', + }, + body: JSON.stringify({ + password: widget.password, + }) + }); + + if (status !== 204) { + logger.error("HTTP %d communicating with NextPVR. Data: %s", status, data.toString()); + return [status, data, responseHeaders]; + } + try { + [ globalSid ] = responseHeaders["set-cookie"] + } catch (e) { + logger.error("Error decoding NextPVR API data. Data: %s", data.toString()); + return [status, null]; + } + logger.info('gettingSID') + return [status, true]; +} + + +async function fetchDataFromWGeasy(endpoint, widget, sid) { + const api = widgets?.[widget.type]?.api; + if (!api) { + return [403, null]; + } + const url = `${new URL(formatApiCall(api, { endpoint, ...widget }))}` + const [status, contentType, data] = await httpProxy(url, { + method: 'GET', + headers: { + 'Content-Type': 'application/json', + 'Cookie': sid + }, + }); + + if (status !== 200) { + logger.error("HTTP %d communicating with WGeasy. Data: %s", status, data.toString()); + return [status, data]; + } + + try { + return [status, JSON.parse(data), contentType]; + } catch (e) { + logger.error("Error decoding WGeasy API data. Data: %s", data.toString()); + return [status, null]; + } +} + +export default async function WGeasyProxyHandler(req, res) { + const widget = await getWidget(req); + + if (!globalSid) { + await loginToWGEasy('session', widget); + } + if (!widget) { + return res.status(400).json({ error: "Invalid proxy service type" }); + } + + logger.debug("Getting data from WGeasy API"); + // Calculate the number of clients + const [status, apiData] = await fetchDataFromWGeasy('wireguard/client', widget, globalSid); + + if (status !== 200) { + return res.status(status).json({ error: { message: "HTTP error communicating with WGeasy API", data: Buffer.from(apiData).toString() } }); + } + let clientCount = 0; + clientCount = apiData.length; + + const data = { + clientCount + }; + + return res.status(status).send(data); + +} + diff --git a/src/widgets/wgeasy/widget.js b/src/widgets/wgeasy/widget.js new file mode 100644 index 00000000..fcb5bef6 --- /dev/null +++ b/src/widgets/wgeasy/widget.js @@ -0,0 +1,14 @@ +import nextpvrProxyHandler from "./proxy"; + +const widget = { + api: "{url}/api/{endpoint}", + proxyHandler: nextpvrProxyHandler, + + mappings: { + unified: { + endpoint: "/", + }, + }, +}; + +export default widget; diff --git a/src/widgets/widgets.js b/src/widgets/widgets.js index 5fd32ddc..ab8beb56 100644 --- a/src/widgets/widgets.js +++ b/src/widgets/widgets.js @@ -84,6 +84,7 @@ import unmanic from "./unmanic/widget"; import uptimekuma from "./uptimekuma/widget"; import watchtower from "./watchtower/widget"; import whatsupdocker from "./whatsupdocker/widget"; +import wgeasy from "./wgeasy/widget"; import xteve from "./xteve/widget"; const widgets = { @@ -175,6 +176,7 @@ const widgets = { uptimekuma, watchtower, whatsupdocker, + wgeasy, xteve, };