eive-tmtc/eive_tmtc/pus_tc/devs/gps.py

import enum

from eive_tmtc.config.definitions import CustomServiceList
from tmtccmd.config import TmtcDefinitionWrapper, OpCodeEntry
from tmtccmd.config.tmtc import tmtc_definitions_provider
from tmtccmd.tc import DefaultPusQueueHelper
from tmtccmd.tc.pus_3_fsfw_hk import make_sid, generate_one_hk_command
from tmtccmd.logging import get_console_logger


LOGGER = get_console_logger()


class OpCode:
    REQ_OS_HK = ["0", "hk-os"]
    RESET_GNSS = ["5", "reset"]


class Info:
    REQ_OS_HK = "Request One-Shot HK"
    RESET_GNSS = "Reset GNSS using reset pin"


class SetId:
    HK = 0


@tmtc_definitions_provider
def add_gps_cmds(defs: TmtcDefinitionWrapper):
    oce = OpCodeEntry()
    oce.add(keys=OpCode.RESET_GNSS, info=Info.RESET_GNSS)
    oce.add(keys=OpCode.REQ_OS_HK, info=Info.REQ_OS_HK)
    defs.add_service(
        name=CustomServiceList.GPS_CTRL.value,
        info="GPS/GNSS Controller",
        op_code_entry=oce,
    )


def pack_gps_command(object_id: bytes, q: DefaultPusQueueHelper, op_code: str):
    if op_code in OpCode.RESET_GNSS:
        # TODO: This needs to be re-implemented
        LOGGER.warning("Reset pin handling needs to be re-implemented")
    if op_code in OpCode.REQ_OS_HK:
        q.add_log_cmd(f"GMSS: {Info.REQ_OS_HK}")
        q.add_pus_tc(
            generate_one_hk_command(sid=make_sid(object_id=object_id, set_id=SetId.HK))
        )