From 89f3288fb083f37e00f1e061baa11913c0964505 Mon Sep 17 00:00:00 2001
From: yuyi <yuyi98@163.com>
Date: Tue, 9 May 2023 21:19:25 +0800
Subject: [PATCH] cgen: fix nested map of fn call (#18142)

---
 vlib/v/gen/c/index.v                      | 5 ++++-
 vlib/v/tests/nested_map_of_fn_call_test.v | 9 +++++++++
 2 files changed, 13 insertions(+), 1 deletion(-)
 create mode 100644 vlib/v/tests/nested_map_of_fn_call_test.v

diff --git a/vlib/v/gen/c/index.v b/vlib/v/gen/c/index.v
index 3221abed1..318e68687 100644
--- a/vlib/v/gen/c/index.v
+++ b/vlib/v/gen/c/index.v
@@ -474,6 +474,7 @@ fn (mut g Gen) index_of_map(node ast.IndexExpr, sym ast.TypeSymbol) {
 		}
 		tmp_opt := if gen_or { g.new_tmp_var() } else { '' }
 		tmp_opt_ptr := if gen_or { g.new_tmp_var() } else { '' }
+		mut is_fn_last_index_call := false
 		if gen_or {
 			g.write('${elem_type_str}* ${tmp_opt_ptr} = (${elem_type_str}*)(map_get_check(')
 		} else {
@@ -482,6 +483,8 @@ fn (mut g Gen) index_of_map(node ast.IndexExpr, sym ast.TypeSymbol) {
 					g.write('((')
 					g.write_fn_ptr_decl(&elem_sym.info, '')
 					g.write(')(*(voidptr*)map_get(')
+					is_fn_last_index_call = true
+					g.is_fn_index_call = false
 				}
 			} else {
 				g.write('(*(${elem_type_str}*)map_get(')
@@ -506,7 +509,7 @@ fn (mut g Gen) index_of_map(node ast.IndexExpr, sym ast.TypeSymbol) {
 		g.write('}')
 		if gen_or {
 			g.write('))')
-		} else if g.is_fn_index_call {
+		} else if is_fn_last_index_call {
 			g.write(', &(voidptr[]){ ${zero} })))')
 		} else {
 			g.write(', &(${elem_type_str}[]){ ${zero} }))')
diff --git a/vlib/v/tests/nested_map_of_fn_call_test.v b/vlib/v/tests/nested_map_of_fn_call_test.v
new file mode 100644
index 000000000..d010335a7
--- /dev/null
+++ b/vlib/v/tests/nested_map_of_fn_call_test.v
@@ -0,0 +1,9 @@
+fn test_nested_map_of_fn_call() {
+	mut a := map[string]map[string]fn () int{}
+	a['a']['a'] = fn () int {
+		return 0
+	}
+	b := a['a']['a']()
+	println(b)
+	assert b == 0
+}
-- 
2.39.5