use std::path::Path;
use crate::Result;
use crate::extract::{FiveTuple, FiveTupleKey};
use crate::pcap::PcapFlowSource;
use crate::quic::datagram::QuicUdpParser;
use crate::quic::types::QuicInitial;
use crate::session::SessionEvent;
pub fn initials_from_pcap<P: AsRef<Path>>(
path: P,
) -> Result<impl Iterator<Item = (FiveTupleKey, QuicInitial)>> {
let source = PcapFlowSource::open(path)?;
Ok(source
.datagrams(FiveTuple::bidirectional(), QuicUdpParser)
.filter_map(|evt| evt.ok())
.filter_map(|evt| match evt {
SessionEvent::Application { key, message, .. } => Some((key, message)),
_ => None,
}))
}