eive-tmtc/pus_tc/system/core.py

import enum

from config.definitions import CustomServiceList
from spacepackets.ecss import PusTelecommand
from tmtccmd.config import TmtcDefinitionWrapper

from tmtccmd.tc import DefaultPusQueueHelper
from tmtccmd.tc.pus_8_funccmd import make_fsfw_action_cmd
from tmtccmd.logging import get_console_logger
from tmtccmd.tc.pus_3_fsfw_hk import make_sid, generate_one_hk_command
from tmtccmd.config.tmtc import OpCodeEntry, tmtc_definitions_provider
from config.object_ids import CORE_CONTROLLER_ID


LOGGER = get_console_logger()


class ActionIds(enum.IntEnum):
    LIST_DIR_INTO_FILE = 0
    SWITCH_REBOOT_FILE_HANDLING = 5
    RESET_REBOOT_COUNTER = 6
    SWITCH_IMG_LOCK = 7
    SET_MAX_REBOOT_CNT = 8
    UPDATE_OBSW_FROM_SD_0 = 10
    UPDATE_OBSW_FROM_SD_1 = 11
    UPDATE_OBSW_FROM_TMP = 12
    XSC_REBOOT = 32
    FULL_REBOOT = 34


class SetIds(enum.IntEnum):
    HK = 5


class OpCodes:
    REBOOT_XSC = ["0", "reboot_xsc"]
    XSC_REBOOT_SELF = ["1", "reboot_self"]
    XSC_REBOOT_0_0 = ["2", "reboot_00"]
    XSC_REBOOT_0_1 = ["3", "reboot_01"]
    XSC_REBOOT_1_0 = ["4", "reboot_10"]
    XSC_REBOOT_1_1 = ["5", "reboot_11"]
    REBOOT_FULL = ["6", "reboot_regular"]
    GET_HK = ["7", "get_hk"]
    OBSW_UPDATE_FROM_SD_0 = ["obsw_update_sd0"]
    OBSW_UPDATE_FROM_SD_1 = ["obsw_update_sd1"]
    OBSW_UPDATE_FROM_TMP = ["obsw_update_tmp"]
    ENABLE_REBOOT_FILE_HANDLING = ["32", "rbh_off"]
    DISABLE_REBOOT_FILE_HANDLING = ["33", "rbh_on"]
    RESET_ALL_REBOOT_COUNTERS = ["34", "rbh_reset_a"]
    RESET_REBOOT_COUNTER_00 = ["35", "rbh_reset_00"]
    RESET_REBOOT_COUNTER_01 = ["36", "rbh_reset_01"]
    RESET_REBOOT_COUNTER_10 = ["37", "rbh_reset_10"]
    RESET_REBOOT_COUNTER_11 = ["38", "rbh_reset_11"]
    SET_MAX_REBOOT_CNT = ["39", "rbh_max_cnt"]


class Info:
    REBOOT_XSC = "XSC reboot with prompt"
    REBOOT_FULL = "Full regular reboot"
    OBSW_UPDATE_FROM_SD_0 = "Update OBSW from SD Card 0"
    OBSW_UPDATE_FROM_SD_1 = "Update OBSW from SD Card 1"
    OBSW_UPDATE_FROM_TMP = "Update OBSW from tmp folder"


class Chip(enum.IntEnum):
    CHIP_0 = 0
    CHIP_1 = 1
    NONE = 2


class Copy(enum.IntEnum):
    COPY_0_NOM = 0
    COPY_1_GOLD = 1
    NONE = 2


@tmtc_definitions_provider
def add_core_controller_definitions(defs: TmtcDefinitionWrapper):
    oce = OpCodeEntry()
    oce.add(keys=OpCodes.REBOOT_XSC, info=Info.REBOOT_XSC)
    oce.add(keys=OpCodes.REBOOT_XSC, info=Info.REBOOT_XSC)
    oce.add(keys=OpCodes.REBOOT_FULL, info=Info.REBOOT_FULL)
    oce.add(keys=OpCodes.XSC_REBOOT_SELF, info="Reboot Self")
    oce.add(keys=OpCodes.XSC_REBOOT_0_0, info="Reboot 0 0")
    oce.add(keys=OpCodes.XSC_REBOOT_0_1, info="Reboot 0 1")
    oce.add(keys=OpCodes.XSC_REBOOT_1_0, info="Reboot 1 0")
    oce.add(keys=OpCodes.XSC_REBOOT_1_1, info="Reboot 1 1")
    oce.add(
        keys=OpCodes.GET_HK,
        info="Request housekeeping set",
    )
    oce.add(
        keys=OpCodes.ENABLE_REBOOT_FILE_HANDLING,
        info="Enable reboot file handling",
    )
    oce.add(
        keys=OpCodes.DISABLE_REBOOT_FILE_HANDLING,
        info="Disable reboot file handling",
    )
    oce.add(
        keys=OpCodes.RESET_ALL_REBOOT_COUNTERS,
        info="Reset all reboot counters",
    )
    oce.add(
        keys=OpCodes.RESET_REBOOT_COUNTER_00,
        info="Reset reboot counter 0 0",
    )
    oce.add(
        keys=OpCodes.RESET_REBOOT_COUNTER_01,
        info="Reset reboot counter 0 1",
    )
    oce.add(
        keys=OpCodes.RESET_REBOOT_COUNTER_10,
        info="Reset reboot counter 1 0",
    )
    oce.add(
        keys=OpCodes.RESET_REBOOT_COUNTER_11,
        info="Reset reboot counter 1 1",
    )
    oce.add(keys=OpCodes.OBSW_UPDATE_FROM_SD_0, info=Info.OBSW_UPDATE_FROM_SD_0)
    oce.add(keys=OpCodes.OBSW_UPDATE_FROM_SD_1, info=Info.OBSW_UPDATE_FROM_SD_1)
    oce.add(keys=OpCodes.OBSW_UPDATE_FROM_TMP, info=Info.OBSW_UPDATE_FROM_TMP)
    defs.add_service(CustomServiceList.CORE.value, "Core Controller", oce)


def pack_core_commands(q: DefaultPusQueueHelper, op_code: str):
    if op_code in OpCodes.REBOOT_XSC:
        reboot_self, chip_select, copy_select = determine_reboot_params()
        perform_reboot_cmd(
            q=q,
            reboot_self=reboot_self,
            chip=chip_select,
            copy=copy_select,
        )
    if op_code in OpCodes.REBOOT_FULL:
        q.add_log_cmd(f"Core Command: {Info.REBOOT_FULL}")
        q.add_pus_tc(
            make_fsfw_action_cmd(
                object_id=CORE_CONTROLLER_ID, action_id=ActionIds.FULL_REBOOT
            )
        )
    if op_code in OpCodes.XSC_REBOOT_SELF:
        perform_reboot_cmd(q=q, reboot_self=True)
    if op_code in OpCodes.XSC_REBOOT_0_0:
        perform_reboot_cmd(
            q=q, reboot_self=False, chip=Chip.CHIP_0, copy=Copy.COPY_0_NOM
        )
    if op_code in OpCodes.XSC_REBOOT_0_1:
        perform_reboot_cmd(
            q=q,
            reboot_self=False,
            chip=Chip.CHIP_0,
            copy=Copy.COPY_1_GOLD,
        )
    if op_code in OpCodes.XSC_REBOOT_1_0:
        perform_reboot_cmd(
            q=q, reboot_self=False, chip=Chip.CHIP_1, copy=Copy.COPY_0_NOM
        )
    if op_code in OpCodes.XSC_REBOOT_1_1:
        perform_reboot_cmd(
            q=q,
            reboot_self=False,
            chip=Chip.CHIP_1,
            copy=Copy.COPY_1_GOLD,
        )
    if op_code in OpCodes.DISABLE_REBOOT_FILE_HANDLING:
        q.add_log_cmd("Disabling reboot file handling")
        user_data = bytearray([0])
        q.add_pus_tc(
            make_fsfw_action_cmd(
                object_id=CORE_CONTROLLER_ID,
                action_id=ActionIds.SWITCH_REBOOT_FILE_HANDLING,
                user_data=user_data,
            )
        )
    if op_code in OpCodes.ENABLE_REBOOT_FILE_HANDLING:
        q.add_log_cmd("Enabling reboot file handling")
        user_data = bytearray([1])
        q.add_pus_tc(
            make_fsfw_action_cmd(
                object_id=CORE_CONTROLLER_ID,
                action_id=ActionIds.SWITCH_REBOOT_FILE_HANDLING,
                user_data=user_data,
            )
        )
    if op_code in OpCodes.RESET_ALL_REBOOT_COUNTERS:
        q.add_log_cmd("Resetting all reboot counters")
        q.add_pus_tc(
            make_fsfw_action_cmd(
                object_id=CORE_CONTROLLER_ID,
                action_id=ActionIds.RESET_REBOOT_COUNTER,
            )
        )
    if op_code in OpCodes.RESET_REBOOT_COUNTER_00:
        reset_specific_boot_counter(q, 0, 0)
    if op_code in OpCodes.RESET_REBOOT_COUNTER_01:
        reset_specific_boot_counter(q, 0, 1)
    if op_code in OpCodes.RESET_REBOOT_COUNTER_10:
        reset_specific_boot_counter(q, 1, 0)
    if op_code in OpCodes.RESET_REBOOT_COUNTER_11:
        reset_specific_boot_counter(q, 1, 1)
    if op_code in OpCodes.OBSW_UPDATE_FROM_SD_0:
        q.add_log_cmd(Info.OBSW_UPDATE_FROM_SD_0)
        q.add_pus_tc(pack_obsw_update_cmd(ActionIds.UPDATE_OBSW_FROM_SD_0))
    if op_code in OpCodes.OBSW_UPDATE_FROM_SD_1:
        q.add_log_cmd(Info.OBSW_UPDATE_FROM_SD_1)
        q.add_pus_tc(pack_obsw_update_cmd(ActionIds.UPDATE_OBSW_FROM_SD_1))
    if op_code in OpCodes.OBSW_UPDATE_FROM_TMP:
        q.add_log_cmd(Info.OBSW_UPDATE_FROM_TMP)
        q.add_pus_tc(pack_obsw_update_cmd(ActionIds.UPDATE_OBSW_FROM_TMP))
    if op_code in OpCodes.GET_HK:
        q.add_log_cmd("Requesting housekeeping set")
        sid = make_sid(object_id=CORE_CONTROLLER_ID, set_id=SetIds.HK)
        q.add_pus_tc(generate_one_hk_command(sid))


def reset_specific_boot_counter(q: DefaultPusQueueHelper, chip: int, copy: int):
    q.add_log_cmd(f"Resetting boot counter {chip} {copy}")
    q.add_pus_tc(
        make_fsfw_action_cmd(
            object_id=CORE_CONTROLLER_ID,
            action_id=ActionIds.RESET_REBOOT_COUNTER,
            user_data=bytes([chip, copy]),
        )
    )


def determine_reboot_params() -> (bool, Chip, Copy):
    chip_select = -1
    copy_select = -1
    reboot_self = input("Reboot self? [y/n]: ")
    if reboot_self in ["y", "yes", "1"]:
        LOGGER.info("Rebooting currently running image")
        return True, chip_select, copy_select
    LOGGER.info("Rebooting image specified by chip and copy")
    return False, determine_chip_and_copy()


def determine_chip_and_copy() -> (int, int):
    while True:
        chip_select = input("Chip select [0/1]: ")
        if chip_select in ["0", "1"]:
            if chip_select == "0":
                chip_select = Chip.CHIP_0
            else:
                chip_select = Chip.CHIP_1
            break
        else:
            LOGGER.warning("Invalid chip select value. Try again")
    while True:
        copy_select = input("Copy select [0/1]: ")
        if copy_select in ["0", "1"]:
            if copy_select == "0":
                copy_select = Copy.COPY_0_NOM
            else:
                copy_select = Copy.COPY_1_GOLD
            break
        else:
            LOGGER.warning("Invalid copy select value. Try again")
    return chip_select, copy_select


def pack_obsw_update_cmd(action_id: int) -> PusTelecommand:
    chip, copy = determine_chip_and_copy()
    user_data = bytes([chip, copy])
    return make_fsfw_action_cmd(
        object_id=CORE_CONTROLLER_ID, action_id=action_id, user_data=user_data
    )


def perform_reboot_cmd(
    q: DefaultPusQueueHelper,
    reboot_self: bool,
    chip: Chip = Chip.NONE,
    copy: Copy = Copy.NONE,
):
    tc_data = bytearray()
    if reboot_self:
        q.add_log_cmd("Packing reboot command for current image")
        tc_data.append(True)
    else:
        tc_data.append(False)
        tc_data.append(chip)
        tc_data.append(copy)
        q.add_log_cmd(f"Packing reboot command for chip {chip} and copy {copy}")
    q.add_pus_tc(
        make_fsfw_action_cmd(
            object_id=CORE_CONTROLLER_ID,
            action_id=ActionIds.XSC_REBOOT,
            user_data=tc_data,
        )
    )