import type { Request, Response } from "express";
import { GlobalDeviceRepository } from "../../db/repositories/global-device.repository.js";
import { ConsumerRepository } from "../../db/repositories/consumer.repository.js";
import { ProjectDeviceRepository } from "../../db/repositories/project-device.repository.js";
import {
  createProjectDeviceSchema,
  updateProjectDeviceSchema,
} from "../../shared/validation/project-device.schemas.js";

const globalDeviceRepository = new GlobalDeviceRepository();
const consumerRepository = new ConsumerRepository();
const projectDeviceRepository = new ProjectDeviceRepository();

export async function listProjectDevicesByProject(req: Request, res: Response) {
  const { projectId } = req.params;
  if (typeof projectId !== "string") {
    return res.status(400).json({ error: "Invalid projectId" });
  }
  const rows = await projectDeviceRepository.listByProject(projectId);
  return res.json(rows);
}

export async function createProjectDevice(req: Request, res: Response) {
  const { projectId } = req.params;
  if (typeof projectId !== "string") {
    return res.status(400).json({ error: "Invalid projectId" });
  }
  const parsed = createProjectDeviceSchema.safeParse(req.body);
  if (!parsed.success) {
    return res.status(400).json({ error: parsed.error.flatten() });
  }
  const created = await projectDeviceRepository.create(projectId, parsed.data);
  return res.status(201).json(created);
}

export async function updateProjectDevice(req: Request, res: Response) {
  const { projectId, projectDeviceId } = req.params;
  if (typeof projectId !== "string" || typeof projectDeviceId !== "string") {
    return res.status(400).json({ error: "Invalid parameters" });
  }

  const parsed = updateProjectDeviceSchema.safeParse(req.body);
  if (!parsed.success) {
    return res.status(400).json({ error: parsed.error.flatten() });
  }

  await projectDeviceRepository.update(projectId, projectDeviceId, parsed.data);
  await consumerRepository.syncLinkedConsumersFromProjectDevice(projectId, projectDeviceId, {
    displayName: parsed.data.displayName,
    category: parsed.data.category,
    quantity: parsed.data.quantity,
    installedPowerPerUnitKw: parsed.data.installedPowerPerUnitKw,
    demandFactor: parsed.data.demandFactor,
    phaseCount: parsed.data.phaseCount,
    powerFactor: parsed.data.powerFactor,
    note: parsed.data.note,
  });
  const row = await projectDeviceRepository.findById(projectId, projectDeviceId);
  if (!row) {
    return res.status(404).json({ error: "Project device not found" });
  }
  return res.json(row);
}

export async function deleteProjectDevice(req: Request, res: Response) {
  const { projectId, projectDeviceId } = req.params;
  if (typeof projectId !== "string" || typeof projectDeviceId !== "string") {
    return res.status(400).json({ error: "Invalid parameters" });
  }

  await projectDeviceRepository.delete(projectId, projectDeviceId);
  return res.status(204).send();
}

export async function copyGlobalDeviceToProject(req: Request, res: Response) {
  const { projectId, globalDeviceId } = req.params;
  if (typeof projectId !== "string" || typeof globalDeviceId !== "string") {
    return res.status(400).json({ error: "Invalid parameters" });
  }

  const source = await globalDeviceRepository.findById(globalDeviceId);
  if (!source) {
    return res.status(404).json({ error: "Global device not found" });
  }

  const created = await projectDeviceRepository.create(projectId, {
    name: source.name,
    displayName: source.displayName,
    category: source.category ?? undefined,
    quantity: source.quantity,
    installedPowerPerUnitKw: source.installedPowerPerUnitKw,
    demandFactor: source.demandFactor,
    voltageV: source.voltageV ?? undefined,
    phaseCount: source.phaseCount === 1 || source.phaseCount === 3 ? source.phaseCount : undefined,
    powerFactor: source.powerFactor ?? undefined,
    note: source.note ?? undefined,
  });

  return res.status(201).json(created);
}