import crypto from "node:crypto";
import { and, eq } from "drizzle-orm";
import { db } from "../client.js";
import { projectDevices } from "../schema/project-devices.js";
import type {
  CreateProjectDeviceInput,
  UpdateProjectDeviceInput,
} from "../../shared/validation/project-device.schemas.js";

export class ProjectDeviceRepository {
  async listByProject(projectId: string) {
    return db.select().from(projectDevices).where(eq(projectDevices.projectId, projectId));
  }

  async create(projectId: string, input: CreateProjectDeviceInput) {
    const id = crypto.randomUUID();
    await db.insert(projectDevices).values({
      id,
      projectId,
      name: input.name,
      displayName: input.displayName,
      category: input.category ?? null,
      quantity: input.quantity,
      installedPowerPerUnitKw: input.installedPowerPerUnitKw,
      demandFactor: input.demandFactor,
      voltageV: input.voltageV ?? null,
      phaseCount: input.phaseCount ?? null,
      powerFactor: input.powerFactor ?? null,
      note: input.note ?? null,
    });
    return { id, projectId, ...input };
  }

  async findById(projectId: string, projectDeviceId: string) {
    const [row] = await db
      .select()
      .from(projectDevices)
      .where(
        and(eq(projectDevices.id, projectDeviceId), eq(projectDevices.projectId, projectId))
      )
      .limit(1);
    return row ?? null;
  }

  async update(projectId: string, projectDeviceId: string, input: UpdateProjectDeviceInput) {
    await db
      .update(projectDevices)
      .set({
        name: input.name,
        displayName: input.displayName,
        category: input.category ?? null,
        quantity: input.quantity,
        installedPowerPerUnitKw: input.installedPowerPerUnitKw,
        demandFactor: input.demandFactor,
        voltageV: input.voltageV ?? null,
        phaseCount: input.phaseCount ?? null,
        powerFactor: input.powerFactor ?? null,
        note: input.note ?? null,
      })
      .where(
        and(eq(projectDevices.id, projectDeviceId), eq(projectDevices.projectId, projectId))
      );
  }

  async delete(projectId: string, projectDeviceId: string) {
    await db
      .delete(projectDevices)
      .where(
        and(eq(projectDevices.id, projectDeviceId), eq(projectDevices.projectId, projectId))
      );
  }
}