/*
 * This file is part of onkostar-plugin-dnpm
 *
 * Copyright (C) 2023-2026 the original author or authors.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package dev.dnpm.oshelper.analyzer;

import dev.dnpm.oshelper.dto.EcogStatusWithDate;
import dev.dnpm.oshelper.services.strahlentherapie.StrahlentherapieService;
import dev.dnpm.oshelper.services.systemtherapie.SystemtherapieService;
import de.itc.onkostar.api.Disease;
import de.itc.onkostar.api.IOnkostarApi;
import de.itc.onkostar.api.Item;
import de.itc.onkostar.api.Procedure;
import de.itc.onkostar.api.analysis.AnalyseTriggerEvent;
import de.itc.onkostar.api.analysis.AnalyzerRequirement;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Component;

import java.text.SimpleDateFormat;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * Diese Klasse implementiert ein Plugin, welches Aktionen nach Bearbeitung eines Formulars zur Systemtherapie durchführt.
 *
 * @since 0.6.0
 */
@Component
public class TherapieMitEcogAnalyzer extends Analyzer {

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    private final IOnkostarApi onkostarApi;

    private final StrahlentherapieService strahlentherapieService;
    private final SystemtherapieService systemtherapieService;

    public TherapieMitEcogAnalyzer(
            final IOnkostarApi onkostarApi,
            final StrahlentherapieService strahlentherapieService,
            final SystemtherapieService systemtherapieService
    ) {
        this.onkostarApi = onkostarApi;
        this.strahlentherapieService = strahlentherapieService;
        this.systemtherapieService = systemtherapieService;
    }

    @Override
    public String getDescription() {
        return "Aktualisiert verknüpfte Formulare nach Änderungen in Formularen vom Typ Strahlen-/Systemtherapie mit ECOG-Status";
    }

    /**
     * @deprecated
     */
    @Override
    public boolean isRelevantForDeletedProcedure() {
        return true;
    }

    @Override
    public boolean isRelevantForAnalyzer(Procedure procedure, Disease disease) {
        return null != procedure && null != disease && (
                procedure.getFormName().equals("OS.Strahlentherapie")
                        || procedure.getFormName().equals("OS.Strahlentherapie.VarianteUKW")
                        || procedure.getFormName().equals("OS.Systemische Therapie")
                        || procedure.getFormName().equals("OS.Systemische Therapie.VarianteUKW")
        );
    }

    @Override
    public boolean isSynchronous() {
        return false;
    }

    @Override
    public AnalyzerRequirement getRequirement() {
        return AnalyzerRequirement.PROCEDURE;
    }

    @Override
    public Set<AnalyseTriggerEvent> getTriggerEvents() {
        return Set.of(
                AnalyseTriggerEvent.EDIT_SAVE,
                AnalyseTriggerEvent.EDIT_LOCK,
                AnalyseTriggerEvent.REORG
        );
    }

    @Override
    public void analyze(Procedure procedure, Disease disease) {
        var date = procedure.getStartDate();
        var status = procedure.getValue("ECOGvorTherapie");

        if (null == date || null == status) {
            // Ignore
            return;
        }

        var ecog = strahlentherapieService.ecogStatus(procedure.getPatient())
                .stream()
                .filter(ecogStatusWithDate -> ecogStatusWithDate.getDate().after(disease.getDiagnosisDate()))
                .collect(Collectors.toList());

        ecog.addAll(systemtherapieService.ecogStatus(procedure.getPatient())
                .stream()
                .filter(ecogStatusWithDate -> ecogStatusWithDate.getDate().after(disease.getDiagnosisDate()))
                .collect(Collectors.toList()));


        if (ecog.isEmpty()) {
            // Nothing to do
            return;
        }

        procedure.getPatient().getDiseases().stream()
                .flatMap(d -> onkostarApi.getProceduresForDiseaseByForm(d.getId(), "DNPM Klinik/Anamnese").stream())
                .forEach(p -> {
                    var ufEcog = p.getValue("ECOGVerlauf");
                    if (null != ufEcog && ufEcog.getValue() instanceof List) {
                        updateExistingEcogVerlauf(p, ecog, ufEcog);
                    } else {
                        newEcogverlauf(p, ecog);
                    }
                });
    }

    private void updateExistingEcogVerlauf(Procedure p, List<EcogStatusWithDate> ecogFromCompleted, Item ufEcog) {
        var shouldSave = false;
        var existingDates = ufEcog.<List<Map<String, String>>>getValue().stream()
                .map(v -> v.get("Datum"))
                .collect(Collectors.toList());
        for (var ecog : ecogFromCompleted) {
            var formattedDate = new SimpleDateFormat("yyyy-MM-dd").format(ecog.getDate());
            if (!existingDates.contains(formattedDate)) {
                var newSubProcedure = new Procedure(onkostarApi);
                newSubProcedure.setStartDate(ecog.getDate());
                newSubProcedure.setValue("Datum", new Item("Datum", ecog.getDate()));
                newSubProcedure.setValue("ECOG", new Item("ECOG", ecog.getStatus()));
                p.addSubProcedure("ECOGVerlauf", newSubProcedure);
                shouldSave = true;
            }
        }
        if (shouldSave) {
            try {
                onkostarApi.saveProcedure(p, true);
            } catch (Exception e) {
                logger.error("Cannot update ECOG for procedure '{}'", p.getId());
            }
        }
    }

    private void newEcogverlauf(Procedure p, List<EcogStatusWithDate> ecogFromCompleted) {
        p.setValue("ECOGVerlauf", new Item("ECOGVerlauf", List.of()));
        for (var ecog : ecogFromCompleted) {
            var newSubProcedure = new Procedure(onkostarApi);
            newSubProcedure.setStartDate(ecog.getDate());
            newSubProcedure.setValue("Datum", new Item("Datum", ecog.getDate()));
            newSubProcedure.setValue("ECOG", new Item("ECOG", ecog.getStatus()));
            p.addSubProcedure("ECOGVerlauf", newSubProcedure);
        }
        try {
            onkostarApi.saveProcedure(p, true);
        } catch (Exception e) {
            logger.error("Create update ECOG for procedure '{}'", p.getId());
        }
    }

}