summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/main.rs11
1 files changed, 8 insertions, 3 deletions
diff --git a/src/main.rs b/src/main.rs
index 3b8547f..42b9460 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -200,6 +200,10 @@ fn handle_key(key: Key, st: &mut State<'_>) -> Option<bool> {
if st.input_buffer.len() != 0 {
match st.input_buffer.parse::<Entry>() {
Ok(val) => {
+ if st.stk.len() + 1 < nargs {
+ return None;
+ }
+
st.stk.push(val);
print!("{}\r{}{}",
termion::clear::CurrentLine,
@@ -215,12 +219,13 @@ fn handle_key(key: Key, st: &mut State<'_>) -> Option<bool> {
Err(_) => return None,
}
} else {
+ if st.stk.len() < nargs {
+ return None;
+ }
+
delay_lf = 0;
}
- if st.stk.len() < nargs {
- return None;
- }
let prevlen = st.stk.len();
let npushed = f(&mut st.stk);
let sizeincr = st.stk.len() as isize - prevlen as isize;