From 7ebc8500da9d52bd2911a620179fb6585f044c47 Mon Sep 17 00:00:00 2001 From: Benoit Date: Mon, 5 Dec 2022 14:33:37 +0100 Subject: [PATCH] Working on Omada Widget : NOT WORKING FOR NOW --- src/widgets/omada/component.jsx | 37 ++++++++++ src/widgets/omada/proxy.js | 121 ++++++++++++++++++++++++++++++++ src/widgets/omada/widget.js | 31 +++++--- src/widgets/widgets.js | 2 +- 4 files changed, 179 insertions(+), 12 deletions(-) create mode 100644 src/widgets/omada/proxy.js diff --git a/src/widgets/omada/component.jsx b/src/widgets/omada/component.jsx index e69de29b..b23316ba 100644 --- a/src/widgets/omada/component.jsx +++ b/src/widgets/omada/component.jsx @@ -0,0 +1,37 @@ + +import { useTranslation } from "next-i18next"; + +import useWidgetAPI from "../../utils/proxy/use-widget-api"; +import Container from "../../components/services/widget/container"; +import Block from "../../components/services/widget/block"; + +export default function Component({ service }) { + const { t } = useTranslation(); + + const { widget } = service; + + const { data: omadaData, error: omadaAPIError } = useWidgetAPI(widget, "unified", { + refreshInterval: 5000, + }); + + if (omadaAPIError) { + return ; + } + + if (!omadaData) { + return ( + + + + + ); + } + + return ( + + + + + + ); +} diff --git a/src/widgets/omada/proxy.js b/src/widgets/omada/proxy.js new file mode 100644 index 00000000..493cb260 --- /dev/null +++ b/src/widgets/omada/proxy.js @@ -0,0 +1,121 @@ +import cache from "memory-cache"; + +import { formatApiCall } from "utils/proxy/api-helpers"; +import { httpProxy } from "utils/proxy/http"; +import { addCookieToJar, setCookieHeader } from "utils/proxy/cookie-jar"; +import getServiceWidget from "utils/config/service-helpers"; +import { getPrivateWidgetOptions } from "utils/config/widget-helpers"; +import createLogger from "utils/logger"; +import widgets from "widgets/widgets"; + +const udmpPrefix = "/proxy/network"; +const proxyName = "omadaProxyHandler"; +const prefixCacheKey = `${proxyName}__prefix`; +const logger = createLogger(proxyName); + +async function getWidget(req) { + const { group, service, type } = req.query; + + let widget = null; + if (type === "omada_console") { // info widget + const index = req.query?.query ? JSON.parse(req.query.query).index : undefined; + widget = await getPrivateWidgetOptions(type, index); + if (!widget) { + logger.debug("Error retrieving settings for this Omada widget"); + return null; + } + widget.type = "omada"; + } else { + if (!group || !service) { + logger.debug("Invalid or missing service '%s' or group '%s'", service, group); + return null; + } + + 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 login(widget) { + const endpoint = (widget.prefix === udmpPrefix) ? "auth/login" : "login"; + const api = widgets?.[widget.type]?.api?.replace("{prefix}", ""); // no prefix for login url + const loginUrl = new URL(formatApiCall(api, { endpoint, ...widget })); + const loginBody = { username: widget.username, password: widget.password, remember: true }; + const headers = { "Content-Type": "application/json" }; + const [status, contentType, data, responseHeaders] = await httpProxy(loginUrl, { + method: "POST", + body: JSON.stringify(loginBody), + headers, + }); + return [status, contentType, data, responseHeaders]; +} + +export default async function omadaProxyHandler(req, res) { + const widget = await getWidget(req); + if (!widget) { + return res.status(400).json({ error: "Invalid proxy service type" }); + } + + const api = widgets?.[widget.type]?.api; + if (!api) { + return res.status(403).json({ error: "Service does not support API calls" }); + } + + let [status, contentType, data, responseHeaders] = []; + let prefix = cache.get(prefixCacheKey); + if (prefix === null) { + // auto detect if we're talking to a UDM Pro, and cache the result so that we + // don't make two requests each time data from Omada is required + [status, contentType, data, responseHeaders] = await httpProxy(widget.url); + prefix = ""; + if (responseHeaders?.["x-csrf-token"]) { + prefix = udmpPrefix; + } + cache.put(prefixCacheKey, prefix); + } + + widget.prefix = prefix; + + const { endpoint } = req.query; + const url = new URL(formatApiCall(api, { endpoint, ...widget })); + const params = { method: "GET", headers: {} }; + setCookieHeader(url, params); + + [status, contentType, data, responseHeaders] = await httpProxy(url, params); + + if (status === 401) { + logger.debug("Omada isn't logged in or rejected the reqeust, attempting login."); + [status, contentType, data, responseHeaders] = await login(widget); + + if (status !== 200) { + logger.error("HTTP %d logging in to Omada. Data: %s", status, data); + return res.status(status).json({error: {message: `HTTP Error ${status}`, url, data}}); + } + + const json = JSON.parse(data.toString()); + if (!(json?.meta?.rc === "ok" || json?.login_time || json?.update_time)) { + logger.error("Error logging in to Omada: Data: %s", data); + return res.status(401).end(data); + } + + addCookieToJar(url, responseHeaders); + setCookieHeader(url, params); + + logger.debug("Retrying Omada request after login."); + [status, contentType, data, responseHeaders] = await httpProxy(url, params); + } + + if (status !== 200) { + logger.error("HTTP %d getting data from Omada endpoint %s. Data: %s", status, url.href, data); + return res.status(status).json({error: {message: `HTTP Error ${status}`, url, data}}); + } + + if (contentType) res.setHeader("Content-Type", contentType); + return res.status(status).send(data); +} diff --git a/src/widgets/omada/widget.js b/src/widgets/omada/widget.js index 92b9d14d..e946fd11 100644 --- a/src/widgets/omada/widget.js +++ b/src/widgets/omada/widget.js @@ -1,21 +1,30 @@ -import credentialedProxyHandler from "../../utils/proxy/handlers/credentialed"; +import omadaProxyHandler from "./proxy"; const widget = { api: "{url}/web/v1/{endpoint}", - proxyHandler: credentialedProxyHandler, + proxyHandler: omadaProxyHandler, mappings: { - "controller": { + "ap": { endpoint: "controller", - params: [ "method": "getGridAps", - "params": { - "sortOrder": "asc", - "currentPage": 1, - "currentPageSize": 100000, - "filters": {"status": "All"} -}] + params: { + "method": "getGridAps", + "params": { + "sortOrder": "asc", + "currentPage": 1, + "currentPageSize": 100000, + "filters": {"status": "All"} + } + }, }, - }, + "client": { + endpoint: "controller", + params: { + "method": "getGridActiveClients", + "params": {"sortOrder": "asc", "currentPage": 1, "currentPageSize": 10, "filters": {"type": "all"}} + } + } + } }; export default widget; diff --git a/src/widgets/widgets.js b/src/widgets/widgets.js index 0d7d6d4a..6b237950 100644 --- a/src/widgets/widgets.js +++ b/src/widgets/widgets.js @@ -69,6 +69,7 @@ const widgets = { navidrome, npm, nzbget, + omada, ombi, overseerr, paperlessngx, @@ -92,7 +93,6 @@ const widgets = { transmission, tubearchivist, truenas, - omada, unifi, unifi_console: unifi, watchtower,