homepage/src/widgets/pyload/proxy.js

import cache from "memory-cache";

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

const proxyName = 'pyloadProxyHandler';
const logger = createLogger(proxyName);
const sessionCacheKey = `${proxyName}__sessionId`;

async function fetchFromPyloadAPI(url, sessionId, params) {
  const options = {
    method: "POST",
    headers: {
      "Content-Type": "application/x-www-form-urlencoded",
    },
  };

  if (params) {
    options.body = Object.keys(params).map(k => `${k}=${params[k]}`).join('&');
  } else {
    options.body = `session=${sessionId}`;
  }
  
  // eslint-disable-next-line no-unused-vars
  const [status, contentType, data] = await httpProxy(url, options);
  return [status, JSON.parse(Buffer.from(data).toString())];
}

async function login(loginUrl, username, password) {
  const [status, sessionId] = await fetchFromPyloadAPI(loginUrl, null, { username, password });
  if (status !== 200) {
    throw new Error(`HTTP error ${status} logging into Pyload API, returned: ${sessionId}`);
  } else {
    cache.put(sessionCacheKey, sessionId);
    return sessionId;
  }
}

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

  try {
    if (group && service) {
      const widget = await getServiceWidget(group, service);
  
      if (widget) {
        const url = new URL(formatApiCall(widgets[widget.type].api, { endpoint, ...widget }));
        const loginUrl = `${widget.url}/api/login`;
  
        let sessionId = cache.get(sessionCacheKey);

        if (!sessionId) {
          sessionId = await login(loginUrl, widget.username, widget.password);
        }
  
        let [status, data] = await fetchFromPyloadAPI(url, sessionId);

        if (status === 403) {
          logger.debug("Failed to retrieve data from Pyload API, login and re-try");
          cache.del(sessionCacheKey);
          sessionId = await login(loginUrl, widget.username, widget.password);
          [status, data] = await fetchFromPyloadAPI(url, sessionId);
        }
        
        if (data?.error || status !== 200) {
          return res.status(500).send(Buffer.from(data).toString());
        }

        return res.json(data);
      }
    }
  } catch (e) {
    logger.error(e);
    return res.status(500).send(e.toString());
  }

  return res.status(400).json({ error: "Invalid proxy service type" });
}