tabiew 0.13.1

A lightweight TUI application to view and query tabular data files, such as CSV, TSV, and parquet.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54

use std::{env, process::Command};

use anyhow::anyhow;
use polars::frame::DataFrame;

use crate::{
    AppResult,
    handler::event::{disable_event_read, enable_event_read},
    io::{
        reader::ReaderSource,
        reader::{CsvToDataFrame, DataFrameReader},
        writer::{Destination, WriteToCsv, WriteToFile},
    },
    misc::type_inferer::TypeInferer,
    tui::terminal::{invalidate_tui, start_tui, stop_tui},
};

pub fn edit_in_external_editor(mut df: DataFrame) -> AppResult<DataFrame> {
    let editor = env::var("EDITOR").map_err(|_| anyhow!("$EDITOR is not set"))?;
    let tempfile = tempfile::NamedTempFile::new()?;

    WriteToCsv::default()
        .with_header(true)
        .with_quote_char('"')
        .with_separator_char(',')
        .write_to_file(Destination::File(tempfile.path().to_owned()), &mut df)?;

    let editor_status = {
        disable_event_read();
        stop_tui()?;
        let status = Command::new(editor).arg(tempfile.path()).status();
        start_tui()?;
        invalidate_tui();
        enable_event_read();
        status
    }?;

    if editor_status.success() {
        let mut df = CsvToDataFrame::default()
            .with_no_header(false)
            .with_quote_char('"')
            .with_separator(',')
            .read_to_data_frames(ReaderSource::File(tempfile.path().to_owned()))?
            .into_iter()
            .next()
            .map(|(_, df)| df)
            .ok_or(anyhow!("Failed to load data frame back from the editor"))?;
        let ti = TypeInferer::default().boolean().int().float();
        ti.update(&mut df);
        Ok(df)
    } else {
        Err(anyhow!("Editor failed"))
    }
}