mod builder;
use noodles_bam as bam;
use noodles_cram as cram;
use noodles_sam as sam;
use tokio::io::{self, AsyncWrite, AsyncWriteExt};
pub use self::builder::Builder;
pub enum Writer<W: AsyncWrite> {
Sam(sam::r#async::io::Writer<W>),
Bam(bam::r#async::io::Writer<W>),
Cram(cram::r#async::io::Writer<W>),
}
impl<W> Writer<W>
where
W: AsyncWrite + Unpin,
{
pub async fn write_header(&mut self, header: &sam::Header) -> io::Result<()> {
match self {
Self::Sam(writer) => writer.write_header(header).await,
Self::Bam(writer) => writer.write_header(header).await,
Self::Cram(writer) => writer.write_header(header).await,
}
}
pub async fn write_record(
&mut self,
header: &sam::Header,
record: &dyn sam::alignment::Record,
) -> io::Result<()> {
match self {
Self::Sam(writer) => writer.write_alignment_record(header, record).await,
Self::Bam(writer) => writer.write_alignment_record(header, record).await,
Self::Cram(writer) => writer.write_alignment_record(header, record).await,
}
}
pub async fn shutdown(&mut self, header: &sam::Header) -> io::Result<()> {
match self {
Self::Sam(writer) => writer.get_mut().shutdown().await,
Self::Bam(writer) => writer.shutdown().await,
Self::Cram(writer) => writer.shutdown(header).await,
}
}
}