57 lines
2.0 KiB
TypeScript
57 lines
2.0 KiB
TypeScript
import { objectKeys } from "@homarr/common";
|
|
|
|
import { getIntegrationDockerMetadata, integrationDefs } from "./integration";
|
|
import type { IntegrationKind } from "./integration";
|
|
|
|
export const extractContainerImageName = (image: string): string => image.split("/").at(-1)?.split(":").at(0) ?? "";
|
|
|
|
const extractIconSlug = (iconUrl: string): string => {
|
|
const filename = iconUrl.split("/").pop() ?? "";
|
|
return filename.replace(/\.(svg|png)$/, "").toLowerCase();
|
|
};
|
|
|
|
export const integrationIconSlugs: Record<IntegrationKind, string> = Object.fromEntries(
|
|
objectKeys(integrationDefs).map((kind) => [kind, extractIconSlug(integrationDefs[kind].iconUrl)]),
|
|
) as Record<IntegrationKind, string>;
|
|
|
|
export const matchIntegrationKind = (search: string): IntegrationKind | null => {
|
|
const normalized = search.toLowerCase().trim();
|
|
if (!normalized) return null;
|
|
|
|
for (const kind of objectKeys(integrationDefs)) {
|
|
if (kind.toLowerCase() === normalized) return kind;
|
|
if (integrationDefs[kind].name.toLowerCase() === normalized) return kind;
|
|
}
|
|
|
|
for (const kind of objectKeys(integrationDefs)) {
|
|
if (integrationIconSlugs[kind] === normalized) return kind;
|
|
}
|
|
|
|
for (const kind of objectKeys(integrationDefs)) {
|
|
if (getIntegrationDockerMetadata(kind).aliases.some((alias) => alias === normalized)) return kind;
|
|
}
|
|
|
|
for (const kind of objectKeys(integrationDefs)) {
|
|
if (normalized.includes(kind.toLowerCase())) return kind;
|
|
if (normalized.includes(integrationIconSlugs[kind])) return kind;
|
|
}
|
|
|
|
return null;
|
|
};
|
|
|
|
interface ContainerMatchInput {
|
|
image: string;
|
|
name: string;
|
|
}
|
|
|
|
export const matchIntegrationKindFromContainer = (container: ContainerMatchInput): IntegrationKind | null => {
|
|
const imageName = extractContainerImageName(container.image);
|
|
const fromImage = matchIntegrationKind(imageName);
|
|
if (fromImage && getIntegrationDockerMetadata(fromImage).discoverable) return fromImage;
|
|
|
|
const fromName = matchIntegrationKind(container.name);
|
|
if (fromName && getIntegrationDockerMetadata(fromName).discoverable) return fromName;
|
|
|
|
return null;
|
|
};
|