From 2870a5a63ae2eede4da5edf9e7cb66ded0be47cb Mon Sep 17 00:00:00 2001
From: yuyi <yuyi98@163.com>
Date: Thu, 4 Mar 2021 18:24:14 +0800
Subject: [PATCH] vrepl: fix array_filter (#9104)

---
 cmd/tools/vrepl.v                   | 25 ++++++++++++++++---------
 vlib/v/tests/repl/array_filter.repl |  3 +++
 2 files changed, 19 insertions(+), 9 deletions(-)
 create mode 100644 vlib/v/tests/repl/array_filter.repl

diff --git a/cmd/tools/vrepl.v b/cmd/tools/vrepl.v
index 9b5662bcf4..0c8754f9aa 100644
--- a/cmd/tools/vrepl.v
+++ b/cmd/tools/vrepl.v
@@ -29,6 +29,7 @@ mut:
 const is_stdin_a_pipe = (is_atty(0) == 0)
 
 const vexe = os.getenv('VEXE')
+
 const vstartup = os.getenv('VSTARTUP')
 
 fn new_repl() Repl {
@@ -86,7 +87,7 @@ fn (r &Repl) current_source_code(should_add_temp_lines bool, not_add_print bool)
 		if !not_add_print {
 			lines = r.vstartup_lines.filter(!it.starts_with('print'))
 		} else {
-			lines = r.vstartup_lines	
+			lines = r.vstartup_lines
 		}
 		all_lines << lines
 	}
@@ -118,11 +119,15 @@ fn run_repl(workdir string, vrepl_prefix string) {
 	}
 
 	if vstartup != '' {
-		result := repl_run_vfile(vstartup) or { os.Result{output: '$vstartup file not found'} }
+		result := repl_run_vfile(vstartup) or {
+			os.Result{
+				output: '$vstartup file not found'
+			}
+		}
 		print('\n')
 		print_output(result)
 	}
-	
+
 	file := os.join_path(workdir, '.${vrepl_prefix}vrepl.v')
 	temp_file := os.join_path(workdir, '.${vrepl_prefix}vrepl_temp.v')
 	mut prompt := '>>> '
@@ -214,7 +219,6 @@ fn run_repl(workdir string, vrepl_prefix string) {
 			filter_line := r.line.replace(r.line.find_between("'", "'"), '').replace(r.line.find_between('"',
 				'"'), '')
 			possible_statement_patterns := [
-				'=',
 				'++',
 				'--',
 				'<<',
@@ -229,7 +233,6 @@ fn run_repl(workdir string, vrepl_prefix string) {
 				'interface ',
 				'import ',
 				'#include ',
-				':=',
 				'for ',
 				'or ',
 				'insert',
@@ -240,10 +243,14 @@ fn run_repl(workdir string, vrepl_prefix string) {
 				'trim',
 			]
 			mut is_statement := false
-			for pattern in possible_statement_patterns {
-				if filter_line.contains(pattern) {
-					is_statement = true
-					break
+			if filter_line.count('=') % 2 == 1 {
+				is_statement = true
+			} else {
+				for pattern in possible_statement_patterns {
+					if filter_line.contains(pattern) {
+						is_statement = true
+						break
+					}
 				}
 			}
 			// NB: starting a line with 2 spaces escapes the println heuristic
diff --git a/vlib/v/tests/repl/array_filter.repl b/vlib/v/tests/repl/array_filter.repl
new file mode 100644
index 0000000000..f36afd2e87
--- /dev/null
+++ b/vlib/v/tests/repl/array_filter.repl
@@ -0,0 +1,3 @@
+[1, 2, 3, 4].filter(it%2 == 0).map(it * 2)
+===output===
+[4, 8]