package com.kaidi.oa.web;

import com.kaidi.oa.common.ApiException;
import com.kaidi.oa.common.ApiResp;
import com.kaidi.oa.common.NotFoundException;
import com.kaidi.oa.domain.WastewaterClient;
import com.kaidi.oa.repository.WastewaterClientRepository;
import org.springframework.web.bind.annotation.DeleteMapping;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.PutMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;

import java.time.Instant;
import java.util.List;

/**
 * Wastewater operation center (运营管理中心, 污水运营) discharging-enterprise
 * clients. industry is one of 印染 / 化工 / 食品 / 制药 / 电镀 / 造纸;
 * dischargeType is one of 工业废水 / 生活污水; pollutant is one of
 * COD / 氨氮 / 总磷 / 重金属; serviceStatus is one of 运营中 / 停产 / 欠费.
 * Creating a client without a serviceStatus defaults it to 运营中.
 */
@RestController
@RequestMapping("/api/oa/wastewater-clients")
public class WastewaterClientController {

    private final WastewaterClientRepository clientRepo;

    public WastewaterClientController(WastewaterClientRepository clientRepo) {
        this.clientRepo = clientRepo;
    }

    @GetMapping
    public ApiResp<List<WastewaterClient>> list(@RequestParam(required = false) String serviceStatus) {
        List<WastewaterClient> list;
        if (serviceStatus != null && !serviceStatus.isBlank()) {
            list = clientRepo.findByServiceStatus(serviceStatus);
        } else {
            list = clientRepo.findAll();
        }
        return ApiResp.ok(list);
    }

    @GetMapping("/{id}")
    public ApiResp<WastewaterClient> get(@PathVariable Long id) {
        return ApiResp.ok(clientRepo.findById(id)
                .orElseThrow(() -> new NotFoundException("wastewater client not found: " + id)));
    }

    public record CreateWastewaterClientRequest(
            String name, String industry, String dischargeType, String contractNo,
            Double dailyVolume, String pollutant, String serviceStatus, String owner) {
    }

    @PostMapping
    public ApiResp<WastewaterClient> create(@RequestBody CreateWastewaterClientRequest req) {
        if (req.name() == null || req.name().isBlank()) {
            throw new ApiException(400, "name is required");
        }
        WastewaterClient c = new WastewaterClient();
        c.setName(req.name());
        c.setIndustry(req.industry());
        c.setDischargeType(req.dischargeType());
        c.setContractNo(req.contractNo());
        c.setDailyVolume(req.dailyVolume() == null ? 0 : req.dailyVolume());
        c.setPollutant(req.pollutant());
        c.setServiceStatus(req.serviceStatus() == null || req.serviceStatus().isBlank()
                ? "运营中" : req.serviceStatus());
        c.setOwner(req.owner());
        c.setCreatedAt(Instant.now());
        return ApiResp.ok(clientRepo.save(c));
    }

    /** PUT /{id} -> 编辑排污企业客户; 只覆盖请求体里给出的字段。 */
    @PutMapping("/{id}")
    public ApiResp<WastewaterClient> update(@PathVariable Long id, @RequestBody CreateWastewaterClientRequest req) {
        WastewaterClient c = clientRepo.findById(id)
                .orElseThrow(() -> new NotFoundException("wastewater client not found: " + id));
        if (req.name() != null) {
            if (req.name().isBlank()) {
                throw new ApiException(400, "name is required");
            }
            c.setName(req.name());
        }
        if (req.industry() != null) c.setIndustry(req.industry());
        if (req.dischargeType() != null) c.setDischargeType(req.dischargeType());
        if (req.contractNo() != null) c.setContractNo(req.contractNo());
        if (req.dailyVolume() != null) c.setDailyVolume(req.dailyVolume());
        if (req.pollutant() != null) c.setPollutant(req.pollutant());
        if (req.serviceStatus() != null && !req.serviceStatus().isBlank()) c.setServiceStatus(req.serviceStatus());
        // 归属仅 create 定调, update 不接收客户端 owner, 不得被篡改。
        return ApiResp.ok(clientRepo.save(c));
    }

    /** DELETE /{id} -> 删除排污企业客户。 */
    @DeleteMapping("/{id}")
    public ApiResp<Void> delete(@PathVariable Long id) {
        if (!clientRepo.existsById(id)) {
            throw new NotFoundException("wastewater client not found: " + id);
        }
        clientRepo.deleteById(id);
        return ApiResp.ok(null);
    }
}