from tmtccmd.config import (
    add_op_code_entry,
    add_service_op_code_entry,
    ServiceOpCodeDictT,
    OpCodeDictKeys,
)
from config.definitions import CustomServiceList
from pus_tc.bpx_batt import BpxOpCodes


def add_bpx_cmd_definitions(cmd_dict: ServiceOpCodeDictT):
    op_code_dict = dict()
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=BpxOpCodes.HK, info="Request BPX HK"
    )
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=BpxOpCodes.RST_BOOT_CNT, info="Reset Boot Count"
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=BpxOpCodes.REQUEST_CFG,
        info="Request Configuration Struct (Step 1)",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=BpxOpCodes.REQUEST_CFG_HK,
        info="Request Configuration Struct HK (Step 2)",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict, keys=BpxOpCodes.REBOOT, info="Reboot Command"
    )
    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        name=CustomServiceList.BPX_BATTERY.value,
        info="BPX Battery Handler",
        op_code_entry=op_code_dict,
    )


def add_core_controller_definitions(cmd_dict: ServiceOpCodeDictT):
    od = dict()
    add_op_code_entry(op_code_dict=od, keys=["0", "reboot"], info="Reboot with Prompt")
    add_op_code_entry(op_code_dict=od, keys=["1", "reboot_0_0"], info="Reboot 0 0")
    add_op_code_entry(op_code_dict=od, keys=["2", "reboot_0_1"], info="Reboot 0 1")
    add_op_code_entry(op_code_dict=od, keys=["3", "reboot_1_0"], info="Reboot 1 0")
    add_op_code_entry(op_code_dict=od, keys=["4", "reboot_1_1"], info="Reboot 1 1")
    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        name=CustomServiceList.CORE.value,
        info="Reboot Self",
        op_code_entry=od,
    )


def add_pcdu_cmds(cmd_dict: ServiceOpCodeDictT):
    from pus_tc.p60dock import P60OpCodes, GomspaceOpCodes
    from pus_tc.pdu1 import Pdu1OpCodes
    from pus_tc.pdu2 import Pdu2OpCodes

    op_code_dict = dict()
    add_op_code_entry(op_code_dict=op_code_dict, keys="0", info="P60 Tests")
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=P60OpCodes.STACK_3V3_ON.value,
        info="P60 Dock: Turn stack 3V3 on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=P60OpCodes.STACK_3V3_OFF.value,
        info="P60 Dock: Turn stack 3V3 off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=P60OpCodes.STACK_5V_ON.value,
        info="P60 Dock: Turn stack 5V on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=P60OpCodes.STACK_5V_OFF.value,
        info="P60 Dock: Turn stack 5V off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=GomspaceOpCodes.PRINT_SWITCH_V_I.value,
        info="P60 Dock: Print Switches, Voltages, Currents",
    )
    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        name=CustomServiceList.P60DOCK.value,
        info="P60 Device",
        op_code_entry=op_code_dict,
    )

    op_code_dict = dict()
    add_op_code_entry(op_code_dict=op_code_dict, keys="0", info="PDU1 Tests")
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.TCS_BOARD_ON.value,
        info="PDU1: Turn TCS board on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.TCS_BOARD_OFF.value,
        info="PDU1: Turn TCS board off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.STAR_TRACKER_ON.value,
        info="PDU1: Turn star tracker on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.STAR_TRACKER_OFF.value,
        info="PDU1: Turn star tracker off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.SUS_NOMINAL_ON.value,
        info="PDU1: Turn SUS nominal on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.SUS_NOMINAL_OFF.value,
        info="PDU1: Turn SUS nominal off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.ACS_A_SIDE_ON.value,
        info="PDU1: Turn ACS A side on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.ACS_A_SIDE_OFF.value,
        info="PDU1: Turn ACS A side off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.SYRLINKS_ON.value,
        info="PDU1: Turn Syrlinks on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.SYRLINKS_OFF.value,
        info="PDU1: Turn Syrlinks off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.MGT_ON.value,
        info="PDU1: Turn MGT on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.MGT_OFF.value,
        info="PDU1: Turn MGT on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.SCEX_ON.value,
        info="PDU1: Turn Solar Cell Experiment on",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu1OpCodes.SCEX_OFF.value,
        info="PDU1: Turn Solar Cell Experiment off",
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=GomspaceOpCodes.PRINT_SWITCH_V_I.value,
        info="PDU1: Print Switches, Voltages, Currents",
    )
    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        name=CustomServiceList.PDU1.value,
        info="PDU1 Device",
        op_code_entry=op_code_dict,
    )

    op_code_dict = dict()
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys="0",
        info="PDU2 Tests",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.ACS_SIDE_B_ON.value,
        info="PDU2: Turn ACS Side B on",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.ACS_SIDE_B_OFF.value,
        info="PDU2: Turn ACS Side B off",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.SUS_REDUNDANT_ON.value,
        info="PDU2: Turn SUS redundant on",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.SUS_REDUNDANT_OFF.value,
        info="PDU2: Turn SUS redundant off",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.RW_ON.value,
        info="PDU2: Turn reaction wheels on",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.RW_OFF.value,
        info="PDU2: Turn reaction wheels off",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=Pdu2OpCodes.Q7S_OFF.value,
        info="Q7S Off",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )
    add_op_code_entry(
        op_code_dict=op_code_dict,
        keys=GomspaceOpCodes.PRINT_SWITCH_V_I.value,
        info="PDU2: Print Switches, Voltages, Currents",
        options={OpCodeDictKeys.TIMEOUT: 2.0},
    )

    add_service_op_code_entry(
        srv_op_code_dict=cmd_dict,
        name="pdu2",
        info="PDU2 Device",
        op_code_entry=op_code_dict,
    )