homarr/src/server/api/routers/openmediavault.ts

import axios from 'axios';
import Consola from 'consola';
import { z } from 'zod';
import { checkIntegrationsType, findAppProperty } from '~/tools/client/app-properties';
import { getConfig } from '~/tools/config/getConfig';

import { createTRPCRouter, publicProcedure } from '../trpc';

let sessionId: string | null = null;
let loginToken: string | null = null;

async function makeOpenMediaVaultRPCCall(
  serviceName: string,
  method: string,
  params: Record<string, any>,
  headers: Record<string, string>,
  input: { configName: string }
) {
  const config = getConfig(input.configName);
  const app = config.apps.find((app) => checkIntegrationsType(app.integration, ['openmediavault']));

  if (!app) {
    Consola.error(`App not found for configName '${input.configName}'`);
    return null;
  }

  const appUrl = new URL(app.url);
  const response = await axios.post(
    `${appUrl.origin}/rpc.php`,
    {
      service: serviceName,
      method: method,
      params: params,
    },
    {
      headers: {
        'Content-Type': 'application/json',
        ...headers,
      },
    }
  );
  return response;
}

export const openmediavaultRouter = createTRPCRouter({
  fetchData: publicProcedure
    .input(
      z.object({
        configName: z.string(),
      })
    )
    .query(async ({ input }) => {
      let authResponse: any = null;
      let app: any;

      if (!sessionId || !loginToken) {
        app = getConfig(input.configName)?.apps.find((app) =>
          checkIntegrationsType(app.integration, ['openmediavault'])
        );

        if (!app) {
          Consola.error(
            `Failed to process request to app '${app.integration}' (${app.id}). Please check username & password`
          );
          return null;
        }

        authResponse = await makeOpenMediaVaultRPCCall(
          'session',
          'login',
          {
            username: findAppProperty(app, 'username'),
            password: findAppProperty(app, 'password'),
          },
          {},
          input
        );

        if (authResponse.data.response.sessionid) {
          sessionId = authResponse.data.response.sessionid;
        } else {
          const cookies = authResponse.headers['set-cookie'] || [];
          sessionId = cookies
            .find((cookie: any) => cookie.includes('X-OPENMEDIAVAULT-SESSIONID'))
            ?.split(';')[0];

          loginToken = cookies
            .find((cookie: any) => cookie.includes('X-OPENMEDIAVAULT-LOGIN'))
            ?.split(';')[0];
        }
      }

      const responses = await Promise.allSettled([
        makeOpenMediaVaultRPCCall(
          'system',
          'getInformation',
          {},
          loginToken
            ? { Cookie: `${loginToken};${sessionId}` }
            : { 'X-OPENMEDIAVAULT-SESSIONID': sessionId as string },
          input
        ),
        makeOpenMediaVaultRPCCall(
          'filesystemmgmt',
          'enumerateMountedFilesystems',
          { includeroot: true },
          loginToken
            ? { Cookie: `${loginToken};${sessionId}` }
            : { 'X-OPENMEDIAVAULT-SESSIONID': sessionId as string },
          input
        ),
        makeOpenMediaVaultRPCCall(
          'cputemp',
          'get',
          {},
          loginToken
            ? { Cookie: `${loginToken};${sessionId}` }
            : { 'X-OPENMEDIAVAULT-SESSIONID': sessionId as string },
          input
        ),
      ]);

      const systemInfoResponse =
        responses[0].status === 'fulfilled' && responses[0].value
          ? responses[0].value.data?.response
          : null;
      const fileSystemResponse =
        responses[1].status === 'fulfilled' && responses[1].value
          ? responses[1].value.data?.response
          : null;
      const cpuTempResponse =
        responses[2].status === 'fulfilled' && responses[2].value
          ? responses[2].value.data?.response
          : null;

      return {
        systemInfo: systemInfoResponse,
        fileSystem: fileSystemResponse,
        cpuTemp: cpuTempResponse,
      };
    }),
});