Add custom proxy

This commit is contained in:
Georges-Antoine Assi 2023-08-20 10:38:08 -04:00
parent dd22df974f
commit e695d3b574
No known key found for this signature in database
GPG Key ID: E01F01B06E816D51
3 changed files with 77 additions and 11 deletions

View File

@ -54,8 +54,6 @@ export default async function credentialedProxyHandler(req, res, map) {
} else {
headers.Authorization = `Basic ${Buffer.from(`${widget.username}:${widget.password}`).toString("base64")}`;
}
} else if (widget.type === "calibreweb") {
headers["X-Authenticated-User"] = widget.username;
} else {
headers["X-API-Key"] = `${widget.key}`;
}

View File

@ -0,0 +1,74 @@
import cache from "memory-cache";
import { xml2json } from "xml-js";
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 = "calibreWebProxyHandler";
const logger = createLogger(proxyName);
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 apiCall(widget, endpoint) {
const apiUrl = new URL(formatApiCall(endpoint, { endpoint, ...widget }));
const headers = {
Authorization: `Basic ${Buffer.from(`${widget.username}:${widget.password}`).toString("base64")}`
};
const [status, contentType, data] = await httpProxy(apiUrl, {
withCredentials: true,
credentials: "include",
headers,
});
if (status !== 200) {
logger.error("Error getting data from CalibreWeb: %s status %d. Data: %s", apiUrl, status, data);
return { status, contentType, data: null, responseHeaders };
}
try {
const dataDecoded = xml2json(data.toString(), { compact: true });
return {status, data: JSON.parse(dataDecoded), contentType};
} catch (e) {
logger.error("Error decoding CalibreWeb API data. Data: %s", data.toString());
return {status, data: null, contentType};
}
}
export default async function calibreWebProxyHandler(req, res) {
const widget = await getWidget(req);
const { service } = req.query;
if (!widget) {
return res.status(400).json({ error: "Invalid proxy service type" });
}
const endpoint = widgets[widget.type].mappings[service].endpoint;
const { status, contentType, data } = await apiCall(widget, endpoint);
if (status !== 200) {
return res.status(status).json({error: {message: "HTTP error communicating with CalibreWeb API", data: Buffer.from(data).toString()}});
}
return res.status(status).json(data);
}

View File

@ -1,18 +1,12 @@
import credentialedProxyHandler from "utils/proxy/handlers/credentialed";
import calibreWebProxyHandler from "./proxy";
const widget = {
api: "{url}/{endpoint}",
proxyHandler: credentialedProxyHandler,
proxyHandler: calibreWebProxyHandler,
mappings: {
books: {
endpoint: "ajax/listbooks",
},
authors: {
endpoint: "get_authors_json",
},
series: {
endpoint: "get_series_json",
endpoint: "opds/books/letter/00",
},
},
};