/**
 * @license
 * Copyright 2022-2026 Matter.js Authors
 * SPDX-License-Identifier: Apache-2.0
 */

import { ElectricalEnergyMeasurement } from "@matter/types/clusters/electrical-energy-measurement";
import { ElectricalEnergyMeasurementBehavior } from "./ElectricalEnergyMeasurementBehavior.js";

const ElectricalEnergyMeasurementBase = ElectricalEnergyMeasurementBehavior.with(
    ElectricalEnergyMeasurement.Feature.CumulativeEnergy,
    ElectricalEnergyMeasurement.Feature.PeriodicEnergy,
    ElectricalEnergyMeasurement.Feature.ImportedEnergy,
    ElectricalEnergyMeasurement.Feature.ExportedEnergy,
);

/**
 * Helper Type for setMeasurement method to be used to set the measurement values together and handle as an atomic set
 * to also trigger single events for imported and exported energy values.
 */
type AtomicElectricalEnergyMeasurement = {
    cumulativeEnergy?: {
        imported?: ElectricalEnergyMeasurement.EnergyMeasurement;
        exported?: ElectricalEnergyMeasurement.EnergyMeasurement;
    };
    periodicEnergy?: {
        imported?: ElectricalEnergyMeasurement.EnergyMeasurement;
        exported?: ElectricalEnergyMeasurement.EnergyMeasurement;
    };
};

/**
 * This is the default server implementation of {@link ElectricalEnergyMeasurementBehavior}.
 *
 * The Matter specification requires the ElectricalEnergyMeasurement cluster to support features we do not enable by
 * default. You should use {@link ElectricalEnergyMeasurementServer.with} to specialize the class for the features your
 * implementation supports.
 */
export class ElectricalEnergyMeasurementBaseServer extends ElectricalEnergyMeasurementBase {
    /**
     * Convenience method to set the measurement values for the cluster in an atomic way. This method updates the
     * attributes with the provided data and also sends the events for the imported and exported energy values.
     */
    setMeasurement(measurement: AtomicElectricalEnergyMeasurement) {
        if (this.features.cumulativeEnergy) {
            const useImported = measurement.cumulativeEnergy?.imported !== undefined && this.features.importedEnergy;
            const useExported = measurement.cumulativeEnergy?.exported !== undefined && this.features.exportedEnergy;
            if (useImported) {
                this.state.cumulativeEnergyImported = measurement.cumulativeEnergy!.imported ?? null;
            }
            if (useExported) {
                this.state.cumulativeEnergyExported = measurement.cumulativeEnergy!.exported ?? null;
            }
            if (useImported || useExported) {
                this.events.cumulativeEnergyMeasured?.emit(
                    {
                        energyImported: measurement.cumulativeEnergy?.imported,
                        energyExported: measurement.cumulativeEnergy?.exported,
                    },
                    this.context,
                );
            }
        }
        if (this.features.periodicEnergy) {
            const useImported = measurement.periodicEnergy?.imported !== undefined && this.features.importedEnergy;
            const useExported = measurement.periodicEnergy?.exported !== undefined && this.features.exportedEnergy;
            if (useImported) {
                this.state.periodicEnergyImported = measurement.periodicEnergy!.imported ?? null;
            }
            if (useExported) {
                this.state.periodicEnergyExported = measurement.periodicEnergy!.exported ?? null;
            }
            if (useImported || useExported) {
                this.events.periodicEnergyMeasured?.emit(
                    {
                        energyImported: measurement.periodicEnergy?.imported,
                        energyExported: measurement.periodicEnergy?.exported,
                    },
                    this.context,
                );
            }
        }
    }
}

export namespace ElectricalEnergyMeasurementBaseServer {
    export declare const ExtensionInterface: {
        setMeasurement(measurement: AtomicElectricalEnergyMeasurement): void;
    };
}

// Drop the features the base implementation enables internally so consumers select their own
export class ElectricalEnergyMeasurementServer extends ElectricalEnergyMeasurementBaseServer.with() {}