joncrangle
1 year ago
committed by
GitHub
commit
39a94190cf
@ -0,0 +1,16 @@
|
||||
import ServicesGetStatus from "./methods/services_get_status";
|
||||
import SmartGetList from "./methods/smart_get_list";
|
||||
import DownloaderGetDownloadList from "./methods/downloader_get_downloadlist";
|
||||
|
||||
export default function Component({ service }) {
|
||||
switch (service.widget.method) {
|
||||
case "services.getStatus":
|
||||
return <ServicesGetStatus service={service} />;
|
||||
case "smart.getListBg":
|
||||
return <SmartGetList service={service} />;
|
||||
case "downloader.getDownloadList":
|
||||
return <DownloaderGetDownloadList service={service} />;
|
||||
default:
|
||||
return null;
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,36 @@
|
||||
import useWidgetAPI from "utils/proxy/use-widget-api";
|
||||
import Container from "components/services/widget/container";
|
||||
import Block from "components/services/widget/block";
|
||||
|
||||
const downloadReduce = (acc, e) => {
|
||||
if (e.downloading) {
|
||||
return acc + 1;
|
||||
}
|
||||
return acc;
|
||||
};
|
||||
|
||||
const items = [
|
||||
{ label: "openmediavault.downloading", getNumber: (data) => (!data ? null : data.reduce(downloadReduce, 0)) },
|
||||
{ label: "openmediavault.total", getNumber: (data) => (!data ? null : data?.length) },
|
||||
];
|
||||
|
||||
export default function Component({ service }) {
|
||||
const { data, error } = useWidgetAPI(service.widget);
|
||||
|
||||
if (error) {
|
||||
return <Container service={service} error={error} />;
|
||||
}
|
||||
|
||||
const itemsWithData = items.map((item) => ({
|
||||
...item,
|
||||
number: item.getNumber(data?.response?.data),
|
||||
}));
|
||||
|
||||
return (
|
||||
<Container service={service}>
|
||||
{itemsWithData.map((e) => (
|
||||
<Block key={e.label} label={e.label} value={e.number} />
|
||||
))}
|
||||
</Container>
|
||||
);
|
||||
}
|
||||
@ -0,0 +1,43 @@
|
||||
import useWidgetAPI from "utils/proxy/use-widget-api";
|
||||
import Container from "components/services/widget/container";
|
||||
import Block from "components/services/widget/block";
|
||||
|
||||
const isRunningReduce = (acc, e) => {
|
||||
if (e.running) {
|
||||
return acc + 1;
|
||||
}
|
||||
return acc;
|
||||
};
|
||||
const notRunningReduce = (acc, e) => {
|
||||
if (!e.running) {
|
||||
return acc + 1;
|
||||
}
|
||||
return acc;
|
||||
};
|
||||
|
||||
const items = [
|
||||
{ label: "openmediavault.running", getNumber: (data) => (!data ? null : data.reduce(isRunningReduce, 0)) },
|
||||
{ label: "openmediavault.stopped", getNumber: (data) => (!data ? null : data.reduce(notRunningReduce, 0)) },
|
||||
{ label: "openmediavault.total", getNumber: (data) => (!data ? null : data?.length) },
|
||||
];
|
||||
|
||||
export default function Component({ service }) {
|
||||
const { data, error } = useWidgetAPI(service.widget);
|
||||
|
||||
if (error) {
|
||||
return <Container service={service} error={error} />;
|
||||
}
|
||||
|
||||
const itemsWithData = items.map((item) => ({
|
||||
...item,
|
||||
number: item.getNumber(data?.response?.data),
|
||||
}));
|
||||
|
||||
return (
|
||||
<Container service={service}>
|
||||
{itemsWithData.map((e) => (
|
||||
<Block key={e.label} label={e.label} value={e.number} />
|
||||
))}
|
||||
</Container>
|
||||
);
|
||||
}
|
||||
@ -0,0 +1,42 @@
|
||||
import useWidgetAPI from "utils/proxy/use-widget-api";
|
||||
import Container from "components/services/widget/container";
|
||||
import Block from "components/services/widget/block";
|
||||
|
||||
const passedReduce = (acc, e) => {
|
||||
if (e.overallstatus === "GOOD") {
|
||||
return acc + 1;
|
||||
}
|
||||
return acc;
|
||||
};
|
||||
const failedReduce = (acc, e) => {
|
||||
if (e.overallstatus !== "GOOD") {
|
||||
return acc + 1;
|
||||
}
|
||||
return acc;
|
||||
};
|
||||
|
||||
const items = [
|
||||
{ label: "openmediavault.passed", getNumber: (data) => (!data ? null : data.reduce(passedReduce, 0)) },
|
||||
{ label: "openmediavault.failed", getNumber: (data) => (!data ? null : data.reduce(failedReduce, 0)) },
|
||||
];
|
||||
|
||||
export default function Component({ service }) {
|
||||
const { data, error } = useWidgetAPI(service.widget);
|
||||
|
||||
if (error) {
|
||||
return <Container service={service} error={error} />;
|
||||
}
|
||||
|
||||
const itemsWithData = items.map((item) => ({
|
||||
...item,
|
||||
number: item.getNumber(JSON.parse(data?.response?.output || "{}")?.data),
|
||||
}));
|
||||
|
||||
return (
|
||||
<Container service={service}>
|
||||
{itemsWithData.map((e) => (
|
||||
<Block key={e.label} label={e.label} value={e.number} />
|
||||
))}
|
||||
</Container>
|
||||
);
|
||||
}
|
||||
@ -0,0 +1,151 @@
|
||||
import { formatApiCall } from "utils/proxy/api-helpers";
|
||||
import { httpProxy } from "utils/proxy/http";
|
||||
import getServiceWidget from "utils/config/service-helpers";
|
||||
import { addCookieToJar, setCookieHeader } from "utils/proxy/cookie-jar";
|
||||
import createLogger from "utils/logger";
|
||||
import widgets from "widgets/widgets";
|
||||
|
||||
const PROXY_NAME = "OMVProxyHandler";
|
||||
const BG_MAX_RETRIES = 50;
|
||||
const BG_POLL_PERIOD = 500;
|
||||
|
||||
const logger = createLogger(PROXY_NAME);
|
||||
|
||||
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 rpc(url, request) {
|
||||
const params = {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify(request),
|
||||
};
|
||||
setCookieHeader(url, params);
|
||||
const [status, contentType, data, headers] = await httpProxy(url, params);
|
||||
|
||||
return { status, contentType, data, headers };
|
||||
}
|
||||
|
||||
async function poll(attemptsLeft, makeReqByPos, pos = 0) {
|
||||
if (attemptsLeft <= 0) {
|
||||
return null;
|
||||
}
|
||||
|
||||
const resp = await makeReqByPos(pos);
|
||||
|
||||
const data = JSON.parse(resp.data.toString()).response;
|
||||
if (data.running === true || data.outputPending) {
|
||||
await new Promise((resolve) => {
|
||||
setTimeout(resolve, BG_POLL_PERIOD);
|
||||
});
|
||||
return poll(attemptsLeft - 1, makeReqByPos, data.pos);
|
||||
}
|
||||
return resp;
|
||||
}
|
||||
|
||||
async function tryLogin(widget) {
|
||||
const url = new URL(formatApiCall(widgets?.[widget.type]?.api, { ...widget }));
|
||||
const { username, password } = widget;
|
||||
const resp = await rpc(url, {
|
||||
method: "login",
|
||||
service: "session",
|
||||
params: { username, password },
|
||||
});
|
||||
|
||||
if (resp.status !== 200) {
|
||||
logger.error("HTTP %d logging in to OpenMediaVault. Data: %s", resp.status, resp.data);
|
||||
return [false, resp];
|
||||
}
|
||||
|
||||
const json = JSON.parse(resp.data.toString());
|
||||
if (json.response.authenticated !== true) {
|
||||
logger.error("Login error in OpenMediaVault. Data: %s", resp.data);
|
||||
resp.status = 401;
|
||||
return [false, resp];
|
||||
}
|
||||
|
||||
return [true, resp];
|
||||
}
|
||||
async function processBg(url, filename) {
|
||||
const resp = await poll(BG_MAX_RETRIES, (pos) =>
|
||||
rpc(url, {
|
||||
service: "exec",
|
||||
method: "getOutput",
|
||||
params: { pos, filename },
|
||||
})
|
||||
);
|
||||
|
||||
if (resp == null) {
|
||||
const errText = "The maximum number of attempts to receive a response from Bg data has been exceeded.";
|
||||
logger.error(errText);
|
||||
return errText;
|
||||
}
|
||||
if (resp.status !== 200) {
|
||||
logger.error("HTTP %d getting Bg data from OpenMediaVault RPC. Data: %s", resp.status, resp.data);
|
||||
}
|
||||
return resp;
|
||||
}
|
||||
|
||||
export default async function proxyHandler(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 RPC calls" });
|
||||
}
|
||||
|
||||
const url = new URL(formatApiCall(api, { ...widget }));
|
||||
const [service, method] = widget.method.split(".");
|
||||
const rpcReq = { params: { limit: -1, start: 0 }, service, method };
|
||||
|
||||
let resp = await rpc(url, rpcReq);
|
||||
|
||||
if (resp.status === 401) {
|
||||
logger.debug("Session not authenticated.");
|
||||
const [success, lResp] = await tryLogin(widget);
|
||||
|
||||
if (success) {
|
||||
addCookieToJar(url, lResp.headers);
|
||||
} else {
|
||||
res.status(lResp.status).json({ error: { message: `HTTP Error ${lResp.status}`, url, data: lResp.data } });
|
||||
}
|
||||
|
||||
logger.debug("Retrying OpenMediaVault request after login.");
|
||||
resp = await rpc(url, rpcReq);
|
||||
}
|
||||
|
||||
if (resp.status !== 200) {
|
||||
logger.error("HTTP %d getting data from OpenMediaVault RPC. Data: %s", resp.status, resp.data);
|
||||
return res.status(resp.status).json({ error: { message: `HTTP Error ${resp.status}`, url, data: resp.data } });
|
||||
}
|
||||
|
||||
if (method.endsWith("Bg")) {
|
||||
const json = JSON.parse(resp.data.toString());
|
||||
const bgResp = await processBg(url, json.response);
|
||||
|
||||
if (typeof bgResp === "string") {
|
||||
return res.status(400).json({ error: bgResp });
|
||||
}
|
||||
return res.status(bgResp.status).send(bgResp.data);
|
||||
}
|
||||
|
||||
return res.status(resp.status).send(resp.data);
|
||||
}
|
||||
@ -0,0 +1,8 @@
|
||||
import proxyHandler from "./proxy";
|
||||
|
||||
const widget = {
|
||||
api: "{url}/rpc.php",
|
||||
proxyHandler,
|
||||
};
|
||||
|
||||
export default widget;
|
||||
Loading…
Reference in new issue