// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.hosted.controller.api.integration.vcmr;

import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.datatype.jsr310.JavaTimeModule;
import com.yahoo.vespa.hosted.controller.api.integration.configserver.Node;

import java.time.ZonedDateTime;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;

import static com.yahoo.yolean.Exceptions.uncheck;

/**
 *
 * Node repository report containing list of upcoming VCMRs impacting a node
 *
 * @author olaa
 */
@JsonIgnoreProperties(ignoreUnknown = true)
public class VcmrReport {

    private static final String REPORT_ID = "vcmr";
    private static final ObjectMapper objectMapper = new ObjectMapper()
            .registerModule(new JavaTimeModule());

    @JsonProperty("upcoming")
    private Set<Vcmr> vcmrs;

    public VcmrReport() {
        this(new HashSet<>());
    }

    public VcmrReport(Set<Vcmr> vcmrs) {
        this.vcmrs = vcmrs;
    }

    public Set<Vcmr> getVcmrs() {
        return vcmrs;
    }

    /**
     * @return true if list of VCMRs is changed
     */
    public boolean addVcmr(ChangeRequestSource source) {
        var vcmr = new Vcmr(source.id(), source.status().name(), source.plannedStartTime(), source.plannedEndTime(), source.category());
        if (vcmrs.contains(vcmr))
            return false;

        // Remove to catch any changes in start/end time
        removeVcmr(source.id());
        return vcmrs.add(vcmr);
    }

    public boolean removeVcmr(String id) {
        return vcmrs.removeIf(vcmr -> id.equals(vcmr.id()));
    }

    public static String getReportId() {
        return REPORT_ID;
    }

    /**
     * Serialization functions - mapped to {@link Node#reports()}
     */
    public static VcmrReport fromReports(Map<String, String> reports) {
        var serialized = reports.get(REPORT_ID);
        if (serialized == null)
            return new VcmrReport();

        return uncheck(() -> objectMapper.readValue(serialized, VcmrReport.class));
    }

    /**
     * Set report to 'null' if list is empty - clearing the report
     * See NodePatcher in node-repository
     */
    public Map<String, String> toNodeReports() {
        Map<String, String> reports = new HashMap<>();
        String json = vcmrs.isEmpty() ?
                null : uncheck(() -> objectMapper.valueToTree(this).toString());
        reports.put(REPORT_ID, json);
        return reports;
    }

    @Override
    public String toString() {
        return "VCMRReport{" + vcmrs + "}";
    }

    public record Vcmr (@JsonProperty("id") String id,
                        @JsonProperty("status") String status,
                        @JsonProperty("plannedStartTime") ZonedDateTime plannedStartTime,
                        @JsonProperty("plannedEndTime") ZonedDateTime plannedEndTime,
                        @JsonProperty("category") String category) {}

}