aboutsummaryrefslogtreecommitdiffstats
path: root/src/main.rs
blob: b241afa910e06e17f5683b4a57f99b2f29f0423d (plain) (blame)
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
109
110
111
112
113
114
115
116
117
118
119
120
121
122
use crossterm::event::{self, Event, KeyCode, KeyEvent, KeyModifiers};
use std::io;
use tui::{
    backend::Backend,
    layout::{Constraint, Direction, Layout},
    widgets::{List, ListItem, Paragraph},
    Frame, Terminal,
};

use pfc::{ui, Operator};

enum Signal {
    None,
    Exit,
}

#[derive(Default)]
struct App {
    stack: Vec<f64>,
    input_buffer: String,
}

impl App {
    fn handle_input(&mut self, key: KeyEvent) -> Signal {
        match key.modifiers {
            KeyModifiers::CONTROL => match key.code {
                KeyCode::Char('c') => {
                    return Signal::Exit;
                }
                _ => {}
            },
            KeyModifiers::NONE => match key.code {
                KeyCode::Char('q') => {
                    return Signal::Exit;
                }
                KeyCode::Char(c) => {
                    if c.is_ascii_digit() {
                        self.input_buffer.push(c);
                    } else if c == '.' && !self.input_buffer.contains('.') {
                        if self.input_buffer.len() == 0 {
                            self.input_buffer.push('0');
                        }
                        self.input_buffer.push(c);
                    } else if let Ok(op) = Operator::parse(c) {
                        if self.input_buffer.len() > 0 {
                            self.stack.push(self.input_buffer.parse::<f64>().unwrap());
                            self.input_buffer = String::new();
                        }
                        self.perform_operation(op);
                    }
                }
                KeyCode::Enter => {
                    self.stack.push(self.input_buffer.parse::<f64>().unwrap());
                    self.input_buffer = String::new();
                }
                KeyCode::Backspace => {
                    self.input_buffer.pop();
                }
                _ => {}
            },
            _ => {}
        }
        return Signal::None;
    }

    fn draw<B: Backend>(&self, f: &mut Frame<B>) {
        let chunks = Layout::default()
            .direction(Direction::Vertical)
            .constraints([Constraint::Length(1), Constraint::Percentage(100)].as_ref())
            .split(f.size());

        f.render_widget(Paragraph::new(self.input_buffer.as_str()), chunks[0]);

        let items: Vec<ListItem> = (self.stack)
            .iter()
            .rev()
            .map(|f| ListItem::new(format!("{}", f)))
            .collect();
        f.render_widget(List::new(items), chunks[1]);
    }

    fn perform_operation(&mut self, op: Operator) {
        let rhs = match self.stack.pop() {
            Some(f) => f,
            None => {
                return;
            }
        };
        let lhs = match self.stack.pop() {
            Some(f) => f,
            None => {
                return;
            }
        };
        match op {
            Operator::Add => self.stack.push(lhs + rhs),
            Operator::Sub => self.stack.push(lhs - rhs),
            Operator::Mul => self.stack.push(lhs * rhs),
            Operator::Div => self.stack.push(lhs / rhs),
        }
    }
}

fn main() -> io::Result<()> {
    let app = App::default();
    let mut terminal = ui::init_terminal()?;
    let result = run(app, &mut terminal);
    ui::cleanup_terminal(terminal)?;
    result
}

fn run<B: Backend>(mut app: App, terminal: &mut Terminal<B>) -> io::Result<()> {
    loop {
        terminal.draw(|f| app.draw(f))?;

        if let Event::Key(key) = event::read()? {
            if let Signal::Exit = app.handle_input(key) {
                return Ok(());
            }
        }
    }
}