homeio

1import "server-only";
2
3import { eq, sql } from "drizzle-orm";
4import { db } from "@/lib/server/db/drizzle";
5import { containerCustomizations } from "@/lib/server/db/schema";
6import { logServerAction, withServerTiming } from "@/lib/server/logging/logger";
7import type { ContainerCustomization } from "@/lib/shared/contracts/docker";
8
9export type ContainerCustomizationInput = {
10  displayName?: string | null;
11  iconUrl?: string | null;
12  dashboardUrl?: string | null;
13};
14
15function normalize(value: string | null | undefined) {
16  const trimmed = value?.trim() ?? "";
17  return trimmed.length > 0 ? trimmed : null;
18}
19
20async function hasContainerCustomizationsTable() {
21  const check = await db.execute<{ table_exists: string | null }>(
22    sql`SELECT to_regclass('public.container_customizations') AS table_exists`,
23  );
24  return Boolean(check.rows[0]?.table_exists);
25}
26
27/**
28 * Every saved customization, keyed by container name. A missing table or a
29 * database hiccup must not hide the containers themselves, so both read as
30 * "nothing customized".
31 */
32export async function listContainerCustomizations(): Promise<
33  Map<string, ContainerCustomization>
34> {
35  try {
36    if (!(await hasContainerCustomizationsTable())) return new Map();
37
38    const rows = await db.select().from(containerCustomizations);
39    return new Map(
40      rows.map((row) => [
41        row.containerName,
42        {
43          displayName: row.displayName,
44          iconUrl: row.iconUrl,
45          dashboardUrl: row.dashboardUrl,
46        },
47      ]),
48    );
49  } catch (error) {
50    logServerAction({
51      level: "warn",
52      layer: "service",
53      action: "docker.container-customizations.list",
54      status: "error",
55      message: "Could not read container customizations",
56      error,
57    });
58    return new Map();
59  }
60}
61
62/**
63 * Save the name, icon and link shown for a container. Clearing every field
64 * forgets the container, so it falls back to its Docker name and default icon.
65 */
66export async function saveContainerCustomization(
67  containerName: string,
68  input: ContainerCustomizationInput,
69): Promise<ContainerCustomization> {
70  return withServerTiming(
71    {
72      layer: "service",
73      action: "docker.container-customizations.save",
74      meta: { containerName },
75    },
76    async () => {
77      if (!(await hasContainerCustomizationsTable())) {
78        throw new Error(
79          "container_customizations table is missing. Run `npm run db:init`.",
80        );
81      }
82
83      const values = {
84        displayName: normalize(input.displayName),
85        iconUrl: normalize(input.iconUrl),
86        dashboardUrl: normalize(input.dashboardUrl),
87      };
88
89      if (!values.displayName && !values.iconUrl && !values.dashboardUrl) {
90        await db
91          .delete(containerCustomizations)
92          .where(eq(containerCustomizations.containerName, containerName));
93        return values;
94      }
95
96      await db
97        .insert(containerCustomizations)
98        .values({ containerName, ...values, updatedAt: sql`NOW()` })
99        .onConflictDoUpdate({
100          target: containerCustomizations.containerName,
101          set: { ...values, updatedAt: sql`NOW()` },
102        });
103
104      return values;
105    },
106  );
107}
108