vrepl: fix output of the fn call (related #21792) (#21800)

Author: yuyi98

cmd/tools/vrepl.v

@@ -435,16 +435,18 @@ fn run_repl(workdir string, vrepl_prefix string) int {
 			if s.output.len > r.last_output.len {
 				cur_line_output := s.output[r.last_output.len..]
 				print_output(cur_line_output)
+				if s.exit_code == 0 {
+					r.last_output = s.output.clone()
+					r.lines << r.line
+				}
 			}
 		} else {
 			mut temp_line := r.line
 			func_call, fntype := r.function_call(r.line)
 			filter_line := r.line.replace(r.line.find_between("'", "'"), '').replace(r.line.find_between('"',
 				'"'), '')
 			mut is_statement := false
-			if func_call {
-				is_statement = true
-			} else if filter_line.count('=') % 2 == 1
+			if filter_line.count('=') % 2 == 1
 				&& (filter_line.count('!=') + filter_line.count('>=') + filter_line.count('<=')) == 0 {
 				is_statement = true
 			} else {
@@ -467,6 +469,10 @@ fn run_repl(workdir string, vrepl_prefix string) int {
 				if s.output.len > r.last_output.len {
 					cur_line_output := s.output[r.last_output.len..]
 					print_output(cur_line_output)
+					if s.exit_code == 0 {
+						r.last_output = s.output.clone()
+						r.lines << temp_line
+					}
 				}
 				continue
 			}
@@ -478,6 +484,8 @@ fn run_repl(workdir string, vrepl_prefix string) int {
 				}
 			} else if temp_line.starts_with('#include ') {
 				temp_source_code = '${temp_line}\n' + r.current_source_code(false, false)
+			} else if temp_line.starts_with('fn') {
+				temp_source_code = r.current_source_code(false, false)
 			} else {
 				temp_source_code = r.current_source_code(true, false) + '\n${temp_line}\n'
 			}
@@ -490,6 +498,8 @@ fn run_repl(workdir string, vrepl_prefix string) int {
 					r.parse_import(r.line)
 				} else if r.line.starts_with('#include ') {
 					r.includes << r.line
+				} else if r.line.starts_with('fn ') {
+					r.functions << r.line
 				} else {
 					r.lines << r.line
 				}

vlib/v/slow_tests/repl/comptime_tmpl.repl

@@ -1,3 +1,3 @@
-print($tmpl('./tmpl/hello.txt'))
+$tmpl('./tmpl/hello.txt')
 ===output===
 hello

vlib/v/slow_tests/repl/fn_calls.repl

@@ -1,4 +1,4 @@
-println(math.sinf(50.0))
+math.sinf(50.0)
 println(1+math.sinf(50.0))
 fn test() { println('foo') } fn test2(a int) { println(a) }
 test()

vlib/v/slow_tests/repl/import_alias.repl

@@ -1,4 +1,4 @@
 import encoding.hex as z
-print(z.decode('AB09CD')!.hex())
+z.decode('AB09CD')!.hex()
 ===output===
 ab09cd

vlib/v/slow_tests/repl/method_call.repl renamed to vlib/v/slow_tests/repl/method_call1.repl

@@ -4,4 +4,5 @@ values.pop()
 values
 ===output===
 [1, 2, 3]
+3
 [1, 2]

vlib/v/slow_tests/repl/method_call2.repl

@@ -0,0 +1,8 @@
+mut values := [1, 2, 3]
+values
+println(values.pop())
+values
+===output===
+[1, 2, 3]
+3
+[1, 2]