homeio

1import "server-only";
2
3import { serverEnv } from "@/lib/server/env";
4import { mkdir } from "node:fs/promises";
5import path from "node:path";
6
7export const DATA_SUBDIRECTORIES = [
8  "AppData",
9  "Documents",
10  "Media",
11  "Download",
12  ".Shared",
13  ".Network",
14  ".Trash",
15] as const;
16
17let resolvedDataRoot: string | null = null;
18
19function resolveConfiguredPath(inputPath: string) {
20  return path.isAbsolute(inputPath)
21    ? inputPath
22    : path.resolve(process.cwd(), inputPath);
23}
24
25function resolvePrimaryDataRoot() {
26  return path.dirname(resolveConfiguredPath(serverEnv.STORE_STACKS_ROOT));
27}
28
29export function resolveStoreStacksRoot() {
30  const configured = resolveConfiguredPath(serverEnv.STORE_STACKS_ROOT);
31  if (!resolvedDataRoot) {
32    return configured;
33  }
34
35  const primaryDataRoot = resolvePrimaryDataRoot();
36  if (resolvedDataRoot === primaryDataRoot) {
37    return configured;
38  }
39
40  return path.join(resolvedDataRoot, "Apps");
41}
42
43export function resolveStoreAppDataRoot() {
44  const configured = resolveConfiguredPath(serverEnv.STORE_APP_DATA_ROOT);
45  if (!resolvedDataRoot) {
46    return configured;
47  }
48
49  const primaryDataRoot = resolvePrimaryDataRoot();
50  if (resolvedDataRoot === primaryDataRoot) {
51    return configured;
52  }
53
54  return path.join(resolvedDataRoot, "AppData");
55}
56
57export function resolveDataRootDirectory() {
58  if (resolvedDataRoot) {
59    return resolvedDataRoot;
60  }
61  return path.dirname(resolveStoreStacksRoot());
62}
63
64export async function ensureDataRootDirectories() {
65  const primaryDataRoot = resolvePrimaryDataRoot();
66  const fallbackDataRoot = path.resolve(process.cwd(), "DATA");
67
68  if (primaryDataRoot === "/") {
69    console.warn(
70      `[Storage] STORE_STACKS_ROOT is set to "${serverEnv.STORE_STACKS_ROOT}" — its parent directory ` +
71        `resolves to "/", which is the filesystem root and is not a valid data root. ` +
72        `STORE_STACKS_ROOT must be nested at least one level inside a writable volume mount ` +
73        `(e.g. /DATA/stacks instead of /stacks). Falling back to ${fallbackDataRoot}.`,
74    );
75  }
76
77  let dataRoot = primaryDataRoot;
78
79  try {
80    await Promise.all(
81      DATA_SUBDIRECTORIES.map((name) =>
82        mkdir(path.join(primaryDataRoot, name), { recursive: true }),
83      ),
84    );
85    resolvedDataRoot = primaryDataRoot;
86  } catch (error: unknown) {
87    const errorMessage =
88      error instanceof Error ? error.message : "unknown error";
89    console.warn(
90      `[Storage] Cannot create directories at ${primaryDataRoot} (${errorMessage}). Falling back to ${fallbackDataRoot}`,
91    );
92
93    await Promise.all(
94      DATA_SUBDIRECTORIES.map((name) =>
95        mkdir(path.join(fallbackDataRoot, name), { recursive: true }),
96      ),
97    );
98
99    dataRoot = fallbackDataRoot;
100    resolvedDataRoot = fallbackDataRoot;
101  }
102
103  return {
104    dataRoot,
105    subdirectories: [...DATA_SUBDIRECTORIES],
106  };
107}
108