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
123
124
125
126
127
128
|
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 Calculator {
stack: Vec<f64>,
input_buffer: String,
}
impl Calculator {
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 if self.input_buffer.len() > 0 => {
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::Max(u16::MAX),
Constraint::Length(self.stack.len() as u16),
Constraint::Length(1),
]
.as_ref(),
)
.split(f.size());
let items: Vec<ListItem> = (self.stack)
.iter()
.map(|f| ListItem::new(format!("{}", f)))
.collect();
f.render_widget(List::new(items), chunks[1]);
f.render_widget(Paragraph::new(self.input_buffer.as_str()), chunks[2]);
}
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 calculator = Calculator::default();
let mut terminal = ui::init_terminal()?;
let result = run(calculator, &mut terminal);
ui::cleanup_terminal(terminal)?;
result
}
fn run<B: Backend>(mut calculator: Calculator, terminal: &mut Terminal<B>) -> io::Result<()> {
loop {
terminal.draw(|f| calculator.draw(f))?;
if let Event::Key(key) = event::read()? {
if let Signal::Exit = calculator.handle_input(key) {
return Ok(());
}
}
}
}
|