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