Homarr/src/tools/addToHomarr.ts

import { showNotification } from '@mantine/notifications';
import Dockerode from 'dockerode';
import { Config, MatchingImages, ServiceType, ServiceTypeList } from './types';

async function MatchIcon(name: string) {
  const res = await fetch(
    `https://cdn.jsdelivr.net/gh/walkxhub/dashboard-icons/png/${name
      .replace(/\s+/g, '-')
      .toLowerCase()}.png`
  );
  return res.ok ? res.url : '/favicon.svg';
}

function tryMatchType(imageName: string): ServiceType {
  // Try to find imageName inside MatchingImages

  const match = MatchingImages.find(({ image }) => imageName.includes(image));
  if (match) {
    return match.type;
  }
  return 'Other';
}

export function tryMatchService(container: Dockerode.ContainerInfo | undefined) {
  if (container === undefined) return {};
  const name = container.Names[0].substring(1);
  return {
    name,
    id: container.Id,
    type: tryMatchType(container.Image),
    url: `${container.Ports.at(0)?.IP}:${container.Ports.at(0)?.PublicPort}`,
    icon: `https://cdn.jsdelivr.net/gh/walkxhub/dashboard-icons/png/${name
      .replace(/\s+/g, '-')
      .toLowerCase()}.png`,
  };
}

export default async function addToHomarr(
  container: Dockerode.ContainerInfo,
  config: Config,
  setConfig: (newconfig: Config) => void
) {
  setConfig({
    ...config,
    services: [
      ...config.services,
      {
        name: container.Names[0].substring(1),
        id: container.Id,
        type: tryMatchType(container.Image),
        url: `localhost:${container.Ports.at(0)?.PublicPort}`,
        icon: await MatchIcon(container.Names[0].substring(1)),
      },
    ],
  });
}