owlgo 0.1.7

A lightweight CLI to assist in solving CP problems
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

use crate::OWL_DIR;
use crate::common::{OwlError, Result};
use crate::owl_utils::{FileApp, FileExplorerApp, cmd_utils, fs_utils, tui_utils};
use std::fs;
use std::path::Path;

pub fn show_and_glow(target_path: &Path) -> Result<()> {
    cmd_utils::bat_file(target_path).or_else(|_| {
        cmd_utils::glow_file(target_path).or_else(|_| {
            fs::read_to_string(target_path)
                .map(|contents| println!("{}", contents))
                .map_err(|e| {
                    OwlError::FileError(
                        format!("could not show file '{}'", target_path.to_string_lossy()),
                        e.to_string(),
                    )
                })
        })
    })
}

pub fn show_it(target_path: &Path) -> Result<()> {
    cmd_utils::bat_file(target_path).or_else(|_| {
        fs::read_to_string(target_path)
            .map(|contents| println!("{}", contents))
            .map_err(|e| {
                OwlError::FileError(
                    format!("could not show file '{}'", target_path.to_string_lossy()),
                    e.to_string(),
                )
            })
    })
}

pub async fn show_quest(
    quest_name: &str,
    case_id: Option<usize>,
    show_ans: bool,
    use_tui: bool,
) -> Result<()> {
    let quest_path = fs_utils::ensure_path_from_home(&[OWL_DIR], Some(quest_name))?;

    if !quest_path.exists() {
        super::fetch_quest(quest_name).await?;
    }

    if use_tui && case_id.is_none() {
        return tui_utils::enter_raw_mode().and_then(|_| {
            match FileExplorerApp::default().run(&quest_path) {
                Ok(_) => tui_utils::exit_raw_mode(),
                Err(e) => tui_utils::exit_raw_mode().and(Err(e)),
            }
        });
    }

    let test_cases = if show_ans {
        fs_utils::find_by_ext(&quest_path, "ans")?
    } else {
        fs_utils::find_by_ext(&quest_path, "in")?
    };

    if let Some(case_number) = case_id {
        let test_case = &test_cases[(case_number - 1) % test_cases.len()];

        if use_tui {
            tui_utils::enter_raw_mode().and_then(|_| match FileApp::default().run(test_case) {
                Ok(_) => tui_utils::exit_raw_mode(),
                Err(e) => tui_utils::exit_raw_mode().and(Err(e)),
            })
        } else {
            show_it(test_case)
        }
    } else {
        for test_case in test_cases {
            show_it(&test_case)?;
        }

        Ok(())
    }
}

pub async fn show_test(
    quest_name: &str,
    test_name: &str,
    show_ans: bool,
    use_tui: bool,
) -> Result<()> {
    let quest_path = fs_utils::ensure_path_from_home(&[OWL_DIR], Some(quest_name))?;

    if !quest_path.exists() {
        super::fetch_quest(quest_name).await?;
    }

    let test_case = if show_ans {
        fs_utils::find_by_stem_and_ext(&quest_path, test_name, "ans")?
    } else {
        fs_utils::find_by_stem_and_ext(&quest_path, test_name, "in")?
    };

    if use_tui {
        tui_utils::enter_raw_mode().and_then(|_| match FileApp::default().run(&test_case) {
            Ok(_) => tui_utils::exit_raw_mode(),
            Err(e) => tui_utils::exit_raw_mode().and(Err(e)),
        })
    } else {
        show_it(&test_case)
    }
}