use embassy_time::{Duration, Timer};
use log::warn;
use picoserve::response;

use crate::{TallySubscriber, store::tally_id_to_hex_string};

pub struct IDEvents(pub TallySubscriber);

impl response::sse::EventSource for IDEvents {
    async fn write_events<W: picoserve::io::Write>(
        mut self,
        mut writer: response::sse::EventWriter<W>,
    ) -> Result<(), W::Error> {
        loop {
            let timeout = Timer::after(Duration::from_secs(15));
            let sel = embassy_futures::select::select(self.0.next_message(), timeout);

            match sel.await {
                embassy_futures::select::Either::First(msg) => match msg {
                    embassy_sync::pubsub::WaitResult::Message(id) => {
                        writer
                            .write_event("msg", tally_id_to_hex_string(id).unwrap().as_str())
                            .await?
                    }
                    embassy_sync::pubsub::WaitResult::Lagged(_) => {
                        warn!("SSE subscriber got lagged");
                    }
                },
                embassy_futures::select::Either::Second(_) => writer.write_keepalive().await?,
            }
        }
    }
}