Skip to content

Commit

Permalink
Added support to watchtower statistics
Browse files Browse the repository at this point in the history
  • Loading branch information
FauFra authored and shamoon committed Oct 28, 2022
1 parent 5d101b0 commit 87c3ba8
Show file tree
Hide file tree
Showing 6 changed files with 116 additions and 0 deletions.
5 changes: 5 additions & 0 deletions public/locales/en/common.json
Expand Up @@ -293,6 +293,11 @@
"child_bridges": "Child Bridges",
"child_bridges_status": "{{ok}}/{{total}}"
},
"watchtower":{
"containers_scanned": "Scanned",
"containers_updated": "Updatded",
"containers_failed": "Failed"
},
"autobrr": {
"approvedPushes": "Approved",
"rejectedPushes": "Rejected",
Expand Down
1 change: 1 addition & 0 deletions src/widgets/components.js
Expand Up @@ -37,6 +37,7 @@ const components = {
traefik: dynamic(() => import("./traefik/component")),
transmission: dynamic(() => import("./transmission/component")),
unifi: dynamic(() => import("./unifi/component")),
watchtower: dynamic(() => import("./watchtower/component")),
};

export default components;
36 changes: 36 additions & 0 deletions src/widgets/watchtower/component.jsx
@@ -0,0 +1,36 @@
import { useTranslation } from "next-i18next";

import Container from "components/services/widget/container";
import Block from "components/services/widget/block";
import useWidgetAPI from "utils/proxy/use-widget-api";


export default function Component({ service }) {
const { t } = useTranslation();

const { widget } = service;

const { data: watchData, error: watchError } = useWidgetAPI(widget, "watchtower");

if (watchError || !watchData) {
return <Container error={t("widget.api_error")} />;
}

if (!watchData) {
return (
<Container service={service}>
<Block label="watchtower.containers_scanned " />
<Block label="watchtower.containers_updated" />
<Block label="watchtower.containers_failed" />
</Container>
);
}

return (
<Container service={service}>
<Block label="watchtower.containers_scanned" value={t("common.number", { value: watchData.watchtower_containers_scanned })} />
<Block label="watchtower.containers_updated" value={t("common.number", { value: watchData.watchtower_containers_updated })} />
<Block label="watchtower.containers_failed" value={t("common.number", { value: watchData.watchtower_containers_failed })} />
</Container>
);
}
58 changes: 58 additions & 0 deletions src/widgets/watchtower/proxy.js
@@ -0,0 +1,58 @@
import cache from "memory-cache";

import { httpProxy } from "utils/proxy/http";
import { formatApiCall } from "utils/proxy/api-helpers";
import getServiceWidget from "utils/config/service-helpers";
import createLogger from "utils/logger";
import widgets from "widgets/widgets";

const proxyName = "watchtowerProxyHandler";
const headerCacheKey = `${proxyName}__headers`;
const logger = createLogger(proxyName);

export default async function watchtowerProxyHandler(req, res) {
const { group, service, endpoint } = req.query;

if (!group || !service) {
logger.debug("Invalid or missing service '%s' or group '%s'", service, group);
return res.status(400).json({ error: "Invalid proxy service type" });
}

const widget = await getServiceWidget(group, service);

if (!widget) {
logger.debug("Invalid or missing widget for service '%s' in group '%s'", service, group);
return res.status(400).json({ error: "Invalid proxy service type" });
}

let headers = cache.get(headerCacheKey);
if (!headers) {
headers = {
"Authorization": `Bearer ${widget.key}`,
}
cache.put(headerCacheKey, headers);
}

const url = new URL(formatApiCall(widgets[widget.type].api, { endpoint, ...widget }));

const method = "GET"
const [status, contentType, data] = await httpProxy(url, {
method,
headers,
});

const cleanData = data.toString().split("\n").filter(s => s.startsWith("watchtower"))
const jsonRes={}

cleanData.map(e => e.split(" ")).forEach(strArray => {
const [key, value] = strArray
jsonRes[key] = value
})

if (status !== 200) {
logger.error("Error getting data from WatchTower: %d. Data: %s", status, data);
}

if (contentType) res.setHeader("Content-Type", contentType);
return res.status(status).send(jsonRes);
}
14 changes: 14 additions & 0 deletions src/widgets/watchtower/widget.js
@@ -0,0 +1,14 @@
import watchtowerProxyHandler from "./proxy";

const widget = {
api: "{url}/{endpoint}",
proxyHandler: watchtowerProxyHandler,

mappings: {
"watchtower": {
endpoint: "v1/metrics",
},
},
};

export default widget;
2 changes: 2 additions & 0 deletions src/widgets/widgets.js
Expand Up @@ -32,6 +32,7 @@ import tautulli from "./tautulli/widget";
import traefik from "./traefik/widget";
import transmission from "./transmission/widget";
import unifi from "./unifi/widget";
import watchtower from './watchtower/widget'

const widgets = {
adguard,
Expand Down Expand Up @@ -70,6 +71,7 @@ const widgets = {
transmission,
unifi,
unifi_console: unifi,
watchtower,
};

export default widgets;

0 comments on commit 87c3ba8

Please sign in to comment.