Align project devices with circuit model

src/db/repositories/project-device.repository.ts

@@ -2,14 +2,23 @@ import crypto from "node:crypto";
 import { and, eq } from "drizzle-orm";
 import { db } from "../client.js";
 import { projectDevices } from "../schema/project-devices.js";
+import { calculateRowTotalPower } from "../../domain/calculations/circuit-power-calculation.js";
 import type {
   CreateProjectDeviceInput,
   UpdateProjectDeviceInput,
 } from "../../shared/validation/project-device.schemas.js";
 
+function withTotalPower(row: typeof projectDevices.$inferSelect) {
+  return {
+    ...row,
+    totalPower: calculateRowTotalPower(row.quantity, row.powerPerUnit, row.simultaneityFactor),
+  };
+}
+
 export class ProjectDeviceRepository {
   async listByProject(projectId: string) {
-    return db.select().from(projectDevices).where(eq(projectDevices.projectId, projectId));
+    const rows = await db.select().from(projectDevices).where(eq(projectDevices.projectId, projectId));
+    return rows.map(withTotalPower);
   }
 
   async create(projectId: string, input: CreateProjectDeviceInput) {
@@ -19,16 +28,27 @@ export class ProjectDeviceRepository {
       projectId,
       name: input.name,
       displayName: input.displayName,
+      phaseType: input.phaseType,
+      connectionKind: input.connectionKind ?? null,
+      costGroup: input.costGroup ?? null,
       category: input.category ?? null,
       quantity: input.quantity,
-      installedPowerPerUnitKw: input.installedPowerPerUnitKw,
-      demandFactor: input.demandFactor,
+      powerPerUnit: input.powerPerUnit,
+      simultaneityFactor: input.simultaneityFactor,
+      cosPhi: input.cosPhi ?? null,
+      remark: input.remark ?? null,
+      installedPowerPerUnitKw: input.powerPerUnit,
+      demandFactor: input.simultaneityFactor,
       voltageV: input.voltageV ?? null,
-      phaseCount: input.phaseCount ?? null,
-      powerFactor: input.powerFactor ?? null,
-      note: input.note ?? null,
+      phaseCount: input.phaseType === "three_phase" ? 3 : 1,
+      powerFactor: input.cosPhi ?? null,
+      note: input.remark ?? null,
     });
-    return { id, projectId, ...input };
+    const created = await this.findById(projectId, id);
+    if (!created) {
+      throw new Error("Failed to create project device.");
+    }
+    return created;
   }
 
   async findById(projectId: string, projectDeviceId: string) {
@@ -39,7 +59,7 @@ export class ProjectDeviceRepository {
         and(eq(projectDevices.id, projectDeviceId), eq(projectDevices.projectId, projectId))
       )
       .limit(1);
-    return row ?? null;
+    return row ? withTotalPower(row) : null;
   }
 
   async update(projectId: string, projectDeviceId: string, input: UpdateProjectDeviceInput) {
@@ -48,14 +68,21 @@ export class ProjectDeviceRepository {
       .set({
         name: input.name,
         displayName: input.displayName,
+        phaseType: input.phaseType,
+        connectionKind: input.connectionKind ?? null,
+        costGroup: input.costGroup ?? null,
         category: input.category ?? null,
         quantity: input.quantity,
-        installedPowerPerUnitKw: input.installedPowerPerUnitKw,
-        demandFactor: input.demandFactor,
+        powerPerUnit: input.powerPerUnit,
+        simultaneityFactor: input.simultaneityFactor,
+        cosPhi: input.cosPhi ?? null,
+        remark: input.remark ?? null,
+        installedPowerPerUnitKw: input.powerPerUnit,
+        demandFactor: input.simultaneityFactor,
         voltageV: input.voltageV ?? null,
-        phaseCount: input.phaseCount ?? null,
-        powerFactor: input.powerFactor ?? null,
-        note: input.note ?? null,
+        phaseCount: input.phaseType === "three_phase" ? 3 : 1,
+        powerFactor: input.cosPhi ?? null,
+        note: input.remark ?? null,
       })
       .where(
         and(eq(projectDevices.id, projectDeviceId), eq(projectDevices.projectId, projectId))