From c64fc6f52362c344a455e96d7e431b5b5c98ed7b Mon Sep 17 00:00:00 2001
From: Stephen Chung <schungx@live.com>
Date: Tue, 8 Feb 2022 22:30:48 +0800
Subject: [PATCH] No astu under no_optimize.

---
 src/bin/rhai-repl.rs | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/bin/rhai-repl.rs b/src/bin/rhai-repl.rs
index f9380a4a..41083cad 100644
--- a/src/bin/rhai-repl.rs
+++ b/src/bin/rhai-repl.rs
@@ -60,6 +60,7 @@ fn print_help() {
     #[cfg(feature = "metadata")]
     println!("json       => output all functions in JSON format");
     println!("ast        => print the last AST (optimized)");
+    #[cfg(not(feature = "no_optimize"))]
     println!("astu       => print the last raw, un-optimized AST");
     println!();
     println!("press Ctrl-Enter or end a line with `\\`");
@@ -311,6 +312,7 @@ fn main() {
     let mut history_offset = 1;
 
     let mut main_ast = AST::empty();
+    #[cfg(not(feature = "no_optimize"))]
     let mut ast_u = AST::empty();
     let mut ast = AST::empty();
 
@@ -429,6 +431,7 @@ fn main() {
                 print_scope(&scope);
                 continue;
             }
+            #[cfg(not(feature = "no_optimize"))]
             "astu" => {
                 // print the last un-optimized AST
                 println!("{:#?}\n", ast_u);
@@ -522,10 +525,10 @@ fn main() {
             .compile_with_scope(&scope, &input)
             .map_err(Into::into)
             .and_then(|r| {
-                ast_u = r.clone();
-
                 #[cfg(not(feature = "no_optimize"))]
                 {
+                    ast_u = r.clone();
+
                     ast = engine.optimize_ast(&scope, r, optimize_level);
                 }