From 273f964293b64b94c8ac10a33122f5973b2062bf Mon Sep 17 00:00:00 2001
From: JT <jonathan.d.turner@gmail.com>
Date: Mon, 9 Aug 2021 09:34:21 +1200
Subject: [PATCH] slight improvement

---
 src/eval.rs | 16 ++++++++++++++++
 src/main.rs | 12 ------------
 2 files changed, 16 insertions(+), 12 deletions(-)

diff --git a/src/eval.rs b/src/eval.rs
index 58f1985941..958256653d 100644
--- a/src/eval.rs
+++ b/src/eval.rs
@@ -343,6 +343,22 @@ fn eval_call(state: &State, stack: Stack, call: &Call) -> Result<Value, ShellErr
         Ok(Value::Nothing {
             span: call.positional[0].span,
         })
+    } else if decl.signature.name == "vars" {
+        state.parser_state.print_vars();
+        Ok(Value::Nothing {
+            span: call.positional[0].span,
+        })
+    } else if decl.signature.name == "decls" {
+        state.parser_state.print_decls();
+        Ok(Value::Nothing { span: call.head })
+    } else if decl.signature.name == "blocks" {
+        state.parser_state.print_blocks();
+        Ok(Value::Nothing { span: call.head })
+    } else if decl.signature.name == "stack" {
+        stack.print_stack();
+        Ok(Value::Nothing { span: call.head })
+    } else if decl.signature.name == "def" {
+        Ok(Value::Nothing { span: call.head })
     } else {
         Err(ShellError::Unsupported(call.head))
     }
diff --git a/src/main.rs b/src/main.rs
index ccba80cb4e..13a2b029fa 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -177,18 +177,6 @@ fn main() -> std::io::Result<()> {
                 Signal::Success(s) => {
                     if s.trim() == "exit" {
                         break;
-                    } else if s.trim() == "vars" {
-                        parser_state.borrow().print_vars();
-                        continue;
-                    } else if s.trim() == "decls" {
-                        parser_state.borrow().print_decls();
-                        continue;
-                    } else if s.trim() == "blocks" {
-                        parser_state.borrow().print_blocks();
-                        continue;
-                    } else if s.trim() == "stack" {
-                        stack.print_stack();
-                        continue;
                     }
                     // println!("input: '{}'", s);