termcanvas 0.1.0

A lightweight Rust library for rendering pixels directly in the terminal.
Documentation 
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

#![allow(dead_code)]

use crossterm::event::{self, Event, KeyCode, KeyEvent, KeyEventKind, PopKeyboardEnhancementFlags};
use crossterm::execute;
use crossterm::terminal::{disable_raw_mode, enable_raw_mode};
use std::collections::HashSet;
use std::time::Duration;

pub struct Input {
    held_keys: HashSet<KeyCode>,
    down_keys: HashSet<KeyCode>,
    up_keys: HashSet<KeyCode>,
    kitty_enabled: bool,
}

impl Input {
    pub fn new() -> Self {
        enable_raw_mode().unwrap();

        #[cfg(target_os = "windows")]
        let kitty_enabled = false;

        #[cfg(not(target_os = "windows"))]
        let kitty_enabled = {
            assert!(
                matches!(supports_keyboard_enhancement(), Ok(true)),
                "Terminal does not support kitty keyboard protocol"
            );
            execute!(
                std::io::stdout(),
                PushKeyboardEnhancementFlags(KeyboardEnhancementFlags::REPORT_EVENT_TYPES)
            )
            .unwrap();
            true
        };

        Self {
            held_keys: HashSet::new(),
            down_keys: HashSet::new(),
            up_keys: HashSet::new(),
            kitty_enabled,
        }
    }

    pub fn update(&mut self) -> std::io::Result<()> {
        self.down_keys.clear();
        self.up_keys.clear();
        while event::poll(Duration::from_millis(0))? {
            if let Event::Key(KeyEvent { code, kind, .. }) = event::read()? {
                match kind {
                    KeyEventKind::Press => {
                        if self.held_keys.insert(code) {
                            self.down_keys.insert(code);
                        }
                    }
                    KeyEventKind::Release => {
                        if self.held_keys.remove(&code) {
                            self.up_keys.insert(code);
                        }
                    }
                    _ => {}
                }
            }
        }
        Ok(())
    }

    pub fn is_key_pressed(&self, key: KeyCode) -> bool {
        self.held_keys.contains(&key)
    }

    pub fn is_key_down(&self, key: KeyCode) -> bool {
        self.down_keys.contains(&key)
    }

    pub fn is_key_up(&self, key: KeyCode) -> bool {
        self.up_keys.contains(&key)
    }
}

impl Drop for Input {
    fn drop(&mut self) {
        if self.kitty_enabled {
            let _ = execute!(std::io::stdout(), PopKeyboardEnhancementFlags);
        }
        disable_raw_mode().unwrap();
    }
}