homeio

1import "server-only";
2
3import type { DiskMonitorResponse } from "@/lib/shared/contracts/disks";
4import { createDiskRateTracker, readDiskCounters } from "@/lib/server/modules/system/disk-activity";
5import { getDiskInventory } from "@/lib/server/modules/system/disk-service";
6import { getSystemMetricsSnapshot } from "@/lib/server/modules/system/service";
7
8// Its own baseline, separate from the metrics snapshot's, so polling the Disks
9// tab does not shorten the interval the history sampler measures over.
10const diskRates = createDiskRateTracker();
11
12/** Disks for the Monitor: what lsblk sees, their live throughput and their SMART data. */
13export async function getDiskMonitor(): Promise<DiskMonitorResponse> {
14  const [inventory, counters, snapshot] = await Promise.all([
15    getDiskInventory(),
16    readDiskCounters(),
17    getSystemMetricsSnapshot().catch(() => null),
18  ]);
19
20  const rates = counters ? diskRates(counters, Date.now()) : new Map();
21  const smartByDevice = new Map(
22    (snapshot?.storage?.smart?.disks ?? []).map((disk) => [disk.device, disk]),
23  );
24
25  return {
26    disks: inventory.disks.map((disk) => {
27      const rate = rates.get(disk.name);
28      const smart = smartByDevice.get(disk.device);
29      return {
30        name: disk.name,
31        device: disk.device,
32        model: disk.model,
33        vendor: disk.vendor,
34        sizeBytes: disk.sizeBytes,
35        mediaType: disk.mediaType,
36        transport: disk.transport,
37        isRemovable: disk.isRemovable,
38        readBytesPerSec: rate?.readBytesPerSec ?? null,
39        writeBytesPerSec: rate?.writeBytesPerSec ?? null,
40        health: smart?.status ?? null,
41        temperatureCelsius: smart?.temperatureCelsius ?? null,
42        powerOnHours: smart?.powerOnHours ?? null,
43        partitions: disk.partitions.map(({ name, mountpoint, fstype, sizeBytes }) => ({
44          name,
45          mountpoint: inventory.readOnly ? null : mountpoint,
46          fstype,
47          sizeBytes,
48        })),
49      };
50    }),
51    inContainer: inventory.readOnly,
52    unavailableReason: inventory.unavailableReason,
53    sampledAt: new Date().toISOString(),
54  };
55}
56