"""AC300 fields."""

from typing import List

from ..utils.commands import ReadHoldingRegisters
from ..field_enums import AutoSleepMode, OutputMode, SplitPhaseMachineType, UpsMode
from ..base_devices.ProtocolV1Device import ProtocolV1Device


class AC300(ProtocolV1Device):
    def __init__(self, address: str, sn: str):
        super().__init__(address, "AC300", sn)

        # Details
        self.struct.add_enum_field("ac_output_mode", 70, OutputMode)
        self.struct.add_uint_field("internal_ac_voltage", 71)
        self.struct.add_decimal_field("internal_current_one", 72, 1)
        self.struct.add_uint_field("internal_power_one", 73)
        self.struct.add_decimal_field("internal_ac_frequency", 74, 1)
        self.struct.add_decimal_field("internal_current_two", 75, 1)
        self.struct.add_uint_field("internal_power_two", 76)
        self.struct.add_decimal_field("ac_input_voltage", 77, 1)
        self.struct.add_decimal_field("internal_current_three", 78, 1, (0, 100))
        self.struct.add_uint_field("internal_power_three", 79)
        self.struct.add_decimal_field("ac_input_frequency", 80, 2)
        self.struct.add_uint_field("internal_dc_input_voltage", 86)
        self.struct.add_decimal_field("internal_dc_input_power", 87, 1)
        self.struct.add_decimal_field("internal_dc_input_current", 88, 2)

        # Battery packs
        self.struct.add_decimal_array_field("cell_voltages", 105, 16, 2)  # internal
        self.struct.add_version_field("pack_bms_version", 201)

        # Controls
        self.struct.add_enum_field("ups_mode", 3001, UpsMode)
        self.struct.add_bool_field("split_phase_on", 3004)
        self.struct.add_enum_field(
            "split_phase_machine_mode", 3005, SplitPhaseMachineType
        )
        self.struct.add_bool_field("grid_charge_on", 3011)
        self.struct.add_bool_field("time_control_on", 3013)
        self.struct.add_uint_field("battery_range_start", 3015)
        self.struct.add_uint_field("battery_range_end", 3016)
        self.struct.add_enum_field("auto_sleep_mode", 3061, AutoSleepMode)

    @property
    def pack_num_max(self):
        return 4

    @property
    def polling_commands(self) -> List[ReadHoldingRegisters]:
        return super().polling_commands + [
            ReadHoldingRegisters(70, 10),
            ReadHoldingRegisters(86, 3),
        ]

    @property
    def writable_ranges(self) -> List[range]:
        return super().writable_ranges + [
            range(3001, 3016),
            3061,
        ]

    @property
    def pack_polling_commands(self) -> List[ReadHoldingRegisters]:
        return super().pack_polling_commands + [
            ReadHoldingRegisters(105, 16),
            ReadHoldingRegisters(201, 2),
        ]