badterm 0.4.5

A cross-platform ANSI-BBS telnet client
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

//! The "save scrollback" prompt (Alt-S): a file name whose extension picks
//! the format.

use std::path::{Path, PathBuf};

use badterm_core::screen::Screen;

use super::*;

#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum SaveFormat {
    /// Scrollback and screen as ANSI (CP437), for art viewers and editors.
    Ansi,
    /// Scrollback and screen as plain text.
    Text,
    /// What is on screen now, as an image.
    Png,
}

impl SaveFormat {
    pub fn from_path(path: &Path) -> Option<SaveFormat> {
        match path.extension()?.to_str()?.to_ascii_lowercase().as_str() {
            "ans" | "ansi" => Some(SaveFormat::Ansi),
            "txt" | "asc" => Some(SaveFormat::Text),
            "png" => Some(SaveFormat::Png),
            _ => None,
        }
    }
}

pub struct SavePrompt {
    pub field: TextField,
}

impl SavePrompt {
    pub fn new(default: &Path) -> SavePrompt {
        SavePrompt { field: TextField::new(&default.to_string_lossy(), 1000) }
    }

    pub fn path(&self) -> PathBuf {
        expand_home(self.field.value.trim().trim_matches('"'))
    }

    /// Draw the prompt; returns the text cursor position.
    pub fn draw(&self, s: &mut Screen) -> (usize, usize) {
        let (x, y, w) = ((s.cols() - 70) / 2, 9, 70);
        let t = theme::current();
        let (dim_fg, dim_bg) = (t.window_text.fg(), t.window_text.bg());
        frame(s, x, y, w, 7, "Save Scrollback", t.window.fg(), t.window.bg());
        label(s, x + 2, y + 1, ".ans ANSI or .txt text: the scrollback and screen", dim_fg, dim_bg);
        label(s, x + 2, y + 2, ".png: an image of what is on screen now", dim_fg, dim_bg);
        let cursor = self.field.draw(s, x + 2, y + 4, w - 4, false, t.input.fg(), t.input.bg());
        label(s, x + 2, y + 5, "Enter save   Esc cancel", t.keys.fg(), t.keys.bg());
        cursor
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn format_follows_the_extension() {
        assert_eq!(SaveFormat::from_path(Path::new("a/b.ANS")), Some(SaveFormat::Ansi));
        assert_eq!(SaveFormat::from_path(Path::new("b.txt")), Some(SaveFormat::Text));
        assert_eq!(SaveFormat::from_path(Path::new("b.png")), Some(SaveFormat::Png));
        assert_eq!(SaveFormat::from_path(Path::new("b")), None);
    }
}