ops-sat-rs/src/pus/test.rs

use crate::pus::create_verification_reporter;
use log::{info, warn};
use ops_sat_rs::config::components::PUS_TEST_SERVICE;
use ops_sat_rs::config::tmtc_err;
// use satrs::event_man::{EventMessage, EventMessageU32};
use satrs::pus::test::PusService17TestHandler;
use satrs::pus::verification::{FailParams, VerificationReporter, VerificationReportingProvider};
use satrs::pus::{
    EcssTcAndToken, EcssTcInMemConverter, EcssTcInVecConverter, MpscTcReceiver, MpscTmAsVecSender,
    PusPacketHandlerResult, PusServiceHelper, PusTmAsVec,
};
use satrs::spacepackets::ecss::tc::PusTcReader;
use satrs::spacepackets::ecss::PusPacket;
use satrs::spacepackets::time::cds::CdsTime;
use satrs::spacepackets::time::TimeWriter;
use std::sync::mpsc;

use super::HandlingStatus;

pub fn create_test_service(
    tm_funnel_tx: mpsc::Sender<PusTmAsVec>,
    // event_sender: mpsc::Sender<EventMessageU32>,
    pus_test_rx: mpsc::Receiver<EcssTcAndToken>,
) -> TestCustomServiceWrapper {
    let pus17_handler = PusService17TestHandler::new(PusServiceHelper::new(
        PUS_TEST_SERVICE.id(),
        pus_test_rx,
        tm_funnel_tx,
        create_verification_reporter(PUS_TEST_SERVICE.id(), PUS_TEST_SERVICE.apid),
        EcssTcInVecConverter::default(),
    ));
    TestCustomServiceWrapper {
        handler: pus17_handler,
        // test_srv_event_sender: event_sender,
    }
}

pub struct TestCustomServiceWrapper {
    pub handler: PusService17TestHandler<
        MpscTcReceiver,
        MpscTmAsVecSender,
        EcssTcInVecConverter,
        VerificationReporter,
    >,
    // pub test_srv_event_sender: mpsc::Sender<EventMessageU32>,
}

impl TestCustomServiceWrapper {
    pub fn poll_and_handle_next_packet(&mut self, time_stamp: &[u8]) -> HandlingStatus {
        let res = self.handler.poll_and_handle_next_tc(time_stamp);
        if res.is_err() {
            warn!("PUS17 handler failed with error {:?}", res.unwrap_err());
            return HandlingStatus::Empty;
        }
        match res.unwrap() {
            PusPacketHandlerResult::RequestHandled => {
                info!("Received PUS ping command TC[17,1]");
                info!("Sent ping reply PUS TM[17,2]");
            }
            PusPacketHandlerResult::RequestHandledPartialSuccess(partial_err) => {
                warn!(
                    "Handled PUS ping command with partial success: {:?}",
                    partial_err
                );
            }
            PusPacketHandlerResult::SubserviceNotImplemented(subservice, _) => {
                warn!("PUS17: Subservice {subservice} not implemented")
            }
            // TODO: adapt interface events are implemented
            PusPacketHandlerResult::CustomSubservice(subservice, token) => {
                let (tc, _) = PusTcReader::new(
                    self.handler
                        .service_helper
                        .tc_in_mem_converter
                        .tc_slice_raw(),
                )
                .unwrap();
                let time_stamper = CdsTime::now_with_u16_days().unwrap();
                let mut stamp_buf: [u8; 7] = [0; 7];
                time_stamper.write_to_bytes(&mut stamp_buf).unwrap();
                if subservice == 128 {
                    info!("Generating test event");
                    // self.test_srv_event_sender
                    //     .send(EventMessage::new(PUS_TEST_SERVICE.id(), TEST_EVENT.into()))
                    //     .expect("Sending test event failed");
                    let start_token = self
                        .handler
                        .service_helper
                        .verif_reporter()
                        .start_success(self.handler.service_helper.tm_sender(), token, &stamp_buf)
                        .expect("Error sending start success");
                    self.handler
                        .service_helper
                        .verif_reporter()
                        .completion_success(
                            self.handler.service_helper.tm_sender(),
                            start_token,
                            &stamp_buf,
                        )
                        .expect("Error sending completion success");
                } else {
                    let fail_data = [tc.subservice()];
                    self.handler
                        .service_helper
                        .verif_reporter()
                        .start_failure(
                            self.handler.service_helper.tm_sender(),
                            token,
                            FailParams::new(
                                &stamp_buf,
                                &tmtc_err::INVALID_PUS_SUBSERVICE,
                                &fail_data,
                            ),
                        )
                        .expect("Sending start failure verification failed");
                }
            }
            PusPacketHandlerResult::Empty => return HandlingStatus::Empty,
        }
        HandlingStatus::HandledOne
    }
}