use std::fs::File;
use polars::{frame::DataFrame, io::SerWriter, prelude::IpcWriter};
use crate::{AppResult, misc::osc52::CopyToClipboardOsc52};
use super::traits::{Destination, WriteToFile};
#[derive(Debug, Default)]
pub struct WriteToArrow;
impl WriteToFile for WriteToArrow {
fn write_to_file(&self, dest: Destination, data_frame: &mut DataFrame) -> AppResult<()> {
match dest {
Destination::File(path) => Ok(IpcWriter::new(File::create(path)?).finish(data_frame)?),
Destination::Clipboard => {
let mut buf = Vec::new();
IpcWriter::new(&mut buf).finish(data_frame)?;
buf.copy_to_clipboard_via_osc52();
Ok(())
}
}
}
}