use std::sync::mpsc::{self, TryRecvError};

use satrs::{pus::MpscTmAsVecSender, tmtc::PacketAsVec};

use crate::pus::{HandlingStatus, PusTcDistributor};

// TC source components where the heap is the backing memory of the received telecommands.
pub struct TcSourceTaskDynamic {
    pub tc_receiver: mpsc::Receiver<PacketAsVec>,
    pus_distrib: PusTcDistributor<MpscTmAsVecSender>,
}

impl TcSourceTaskDynamic {
    pub fn new(
        tc_receiver: mpsc::Receiver<PacketAsVec>,
        pus_receiver: PusTcDistributor<MpscTmAsVecSender>,
    ) -> Self {
        Self {
            tc_receiver,
            pus_distrib: pus_receiver,
        }
    }

    pub fn periodic_operation(&mut self) {
        self.poll_tc();
    }

    pub fn poll_tc(&mut self) -> HandlingStatus {
        // Right now, we only expect PUS packets. If any other protocols like CFDP are added at
        // a later stage, we probably need to check for the APID before routing the packet.
        match self.tc_receiver.try_recv() {
            Ok(packet_with_sender) => {
                self.pus_distrib
                    .handle_tc_packet(packet_with_sender.sender_id, packet_with_sender.packet)
                    .ok();
                HandlingStatus::HandledOne
            }
            Err(e) => match e {
                TryRecvError::Empty => HandlingStatus::Empty,
                TryRecvError::Disconnected => {
                    log::warn!("tmtc thread: sender disconnected");
                    HandlingStatus::Empty
                }
            },
        }
    }
}