|
|
@@ -0,0 +1,108 @@
|
|
|
+import cache from "memory-cache";
|
|
|
+
|
|
|
+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";
|
|
|
+import { xml2json } from "xml-js";
|
|
|
+
|
|
|
+// const udmpPrefix = "/proxy/network";
|
|
|
+const proxyName = "plexProxyHandler";
|
|
|
+const librariesCacheKey = `${proxyName}__libraries`;
|
|
|
+const moviesCacheKey = `${proxyName}__movies`;
|
|
|
+const tvCacheKey = `${proxyName}__tv`;
|
|
|
+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 fetchFromPlexAPI(endpoint, widget) {
|
|
|
+ const api = widgets?.[widget.type]?.api;
|
|
|
+ if (!api) {
|
|
|
+ return res.status(403).json({ error: "Service does not support API calls" });
|
|
|
+ }
|
|
|
+
|
|
|
+ const url = new URL(formatApiCall(api, { endpoint, ...widget }));
|
|
|
+
|
|
|
+ const [status, contentType, data] = await httpProxy(url);
|
|
|
+
|
|
|
+ if (status !== 200) {
|
|
|
+ logger.error("HTTP %d communicating with Plex. Data: %s", status, data.toString());
|
|
|
+ return [status, data.toString()];
|
|
|
+ }
|
|
|
+
|
|
|
+ try {
|
|
|
+ const dataDecoded = xml2json(data.toString(), { compact: true });
|
|
|
+ return [status, JSON.parse(dataDecoded)];
|
|
|
+ } catch (e) {
|
|
|
+ logger.error("Error decoding Plex API data. Data: %s", data.toString());
|
|
|
+ return [status, null];
|
|
|
+ }
|
|
|
+}
|
|
|
+
|
|
|
+export default async function plexProxyHandler(req, res) {
|
|
|
+ const widget = await getWidget(req);
|
|
|
+ if (!widget) {
|
|
|
+ return res.status(400).json({ error: "Invalid proxy service type" });
|
|
|
+ }
|
|
|
+
|
|
|
+ logger.debug("Getting streams from Plex API");
|
|
|
+ let streams;
|
|
|
+ let [status, apiData] = await fetchFromPlexAPI("/status/sessions", widget);
|
|
|
+ if (apiData && apiData.MediaContainer) {
|
|
|
+ streams = apiData.MediaContainer._attributes.size;
|
|
|
+ }
|
|
|
+
|
|
|
+ let libraries = cache.get(librariesCacheKey);
|
|
|
+ if (libraries == undefined) {
|
|
|
+ logger.debug("Getting libraries from Plex API");
|
|
|
+ [status, apiData] = await fetchFromPlexAPI("/library/sections", widget);
|
|
|
+ if (apiData && apiData.MediaContainer) {
|
|
|
+ libraries = apiData.MediaContainer.Directory;
|
|
|
+ cache.put(librariesCacheKey, libraries, 1000 * 60 * 60 * 6);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ let movies = cache.get(moviesCacheKey);
|
|
|
+ let tv = cache.get(tvCacheKey);
|
|
|
+ if (movies == undefined || tv == undefined) {
|
|
|
+ logger.debug("Getting movie + tv counts from Plex API");
|
|
|
+ libraries.filter(l => ["movie", "show"].includes(l._attributes.type)).forEach(async (library) => {
|
|
|
+ [status, apiData] = await fetchFromPlexAPI(`/library/sections/${library._attributes.key}/all`, widget);
|
|
|
+ if (apiData && apiData.MediaContainer) {
|
|
|
+ const librarySection = apiData.MediaContainer._attributes;
|
|
|
+ if (library._attributes.type == 'movie') {
|
|
|
+ movies += parseInt(librarySection.size);
|
|
|
+ } else if (library._attributes.type == 'show') {
|
|
|
+ tv += parseInt(librarySection.size);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ cache.put(tvCacheKey, tv, 1000 * 60 * 10);
|
|
|
+ cache.put(moviesCacheKey, movies, 1000 * 60 * 10);
|
|
|
+ });
|
|
|
+ }
|
|
|
+
|
|
|
+ const data = {
|
|
|
+ streams: streams,
|
|
|
+ tv: tv,
|
|
|
+ movies: movies
|
|
|
+ };
|
|
|
+
|
|
|
+ return res.status(status).send(data);
|
|
|
+}
|
Michael Shamoon