v: implement comptime regular func params traversing (support $if T is $function { $for param in T.params {) (#25322)

felipensp · web-flow · commit 97edd610b08a · 2025-09-16T20:37:15.000+03:00
diff --git a/vlib/builtin/builtin.v b/vlib/builtin/builtin.v
@@ -88,8 +88,8 @@ fn __print_assert_failure(i &VAssertMetaInfo) {
 	}
 }
 
-// MethodParam holds type information for function and/or method arguments.
-pub struct MethodParam {
+// FunctionParam holds type information for function and/or method arguments.
+pub struct FunctionParam {
 pub:
 	typ  int
 	name string
@@ -100,7 +100,7 @@ pub struct FunctionData {
 pub:
 	name        string
 	attrs       []string
-	args        []MethodParam
+	args        []FunctionParam
 	return_type int
 	typ         int
 }
diff --git a/vlib/v/checker/comptime.v b/vlib/v/checker/comptime.v
@@ -364,11 +364,13 @@ fn (mut c Checker) comptime_for(mut node ast.ComptimeFor) {
 				node.typ_pos)
 			return
 		}
-		method := c.comptime.comptime_for_method
+
+		func := if sym.info is ast.FnType { &sym.info.func } else { c.comptime.comptime_for_method }
+		params := if func.is_method { func.params[1..] } else { func.params }
 		// example: fn (mut d MyStruct) add(x int, y int) string
 		// `d` is params[0], `x` is params[1], `y` is params[2]
 		// so we at least has one param (`d`) for method
-		for param in method.params[1..] {
+		for param in params {
 			c.push_new_comptime_info()
 			c.comptime.inside_comptime_for = true
 			c.comptime.comptime_for_method_param_var = node.val_var
diff --git a/vlib/v/gen/c/comptime.v b/vlib/v/gen/c/comptime.v
@@ -649,10 +649,10 @@ fn (mut g Gen) comptime_for(node ast.ComptimeFor) {
 			}
 			if method.params.len < 2 {
 				// 0 or 1 (the receiver) args
-				g.writeln('\t${node.val_var}.args = builtin____new_array_with_default(0, 0, sizeof(MethodParam), 0);')
+				g.writeln('\t${node.val_var}.args = builtin____new_array_with_default(0, 0, sizeof(FunctionParam), 0);')
 			} else {
 				len := method.params.len - 1
-				g.write('\t${node.val_var}.args = builtin__new_array_from_c_array(${len}, ${len}, sizeof(MethodParam), _MOV((MethodParam[${len}]){')
+				g.write('\t${node.val_var}.args = builtin__new_array_from_c_array(${len}, ${len}, sizeof(FunctionParam), _MOV((FunctionParam[${len}]){')
 				// Skip receiver arg
 				for j, arg in method.params[1..] {
 					typ := arg.typ.idx()
@@ -841,12 +841,12 @@ fn (mut g Gen) comptime_for(node ast.ComptimeFor) {
 			}
 		}
 	} else if node.kind == .params {
-		method := g.comptime.comptime_for_method
-
-		if method.params.len > 0 {
-			g.writeln('\tMethodParam ${node.val_var} = {0};')
+		func := if sym.info is ast.FnType { &sym.info.func } else { g.comptime.comptime_for_method }
+		if func.params.len > 0 {
+			g.writeln('\tFunctionParam ${node.val_var} = {0};')
 		}
-		for param in method.params[1..] {
+		params := if func.is_method { func.params[1..] } else { func.params }
+		for param in params {
 			g.push_new_comptime_info()
 			g.comptime.inside_comptime_for = true
 			g.comptime.comptime_for_method_param_var = node.val_var
diff --git a/vlib/v/markused/walker.v b/vlib/v/markused/walker.v
@@ -1354,7 +1354,7 @@ fn (mut w Walker) mark_resource_dependencies() {
 		w.mark_by_sym_name('FunctionData')
 	}
 	if w.uses_ct_params {
-		w.mark_by_sym_name('MethodParam')
+		w.mark_by_sym_name('FunctionParam')
 	}
 	if w.uses_ct_values {
 		w.mark_by_sym_name('EnumData')
diff --git a/vlib/v/parser/comptime.v b/vlib/v/parser/comptime.v
@@ -423,7 +423,7 @@ fn (mut p Parser) comptime_for() ast.ComptimeFor {
 		'params' {
 			p.scope.register(ast.Var{
 				name: val_var
-				typ:  p.table.find_type('MethodParam')
+				typ:  p.table.find_type('FunctionParam')
 				pos:  var_pos
 			})
 			kind = .params
diff --git a/vlib/v/tests/comptime/comptime_params_test.v b/vlib/v/tests/comptime/comptime_params_test.v
@@ -0,0 +1,30 @@
+fn demo(a int, b string) {
+}
+
+fn g[T](cb T) {
+	mut params := []FunctionParam{}
+	$if T is $function {
+		$for param in T.params {
+			params << param
+		}
+	}
+	assert params.len == 2
+	assert params[0].name == 'a'
+	assert params[0].typ == 8
+	assert params[1].name == 'b'
+	assert params[1].typ == 21
+}
+
+fn test_main() {
+	g(demo)
+
+	mut params := []FunctionParam{}
+	$for param in demo.params {
+		params << param
+	}
+	assert params.len == 2
+	assert params[0].name == 'a'
+	assert params[0].typ == 8
+	assert params[1].name == 'b'
+	assert params[1].typ == 21
+}

Original file line number	Diff line number	Diff line change
`@@ -88,8 +88,8 @@ fn __print_assert_failure(i &VAssertMetaInfo) {`
`88`	`88`	`}`
`89`	`89`	`}`
`90`	`90`
`91`		`-// MethodParam holds type information for function and/or method arguments.`
`92`		`-pub struct MethodParam {`
	`91`	`+// FunctionParam holds type information for function and/or method arguments.`
	`92`	`+pub struct FunctionParam {`
`93`	`93`	`pub:`
`94`	`94`	`typ int`
`95`	`95`	`name string`
`@@ -100,7 +100,7 @@ pub struct FunctionData {`
`100`	`100`	`pub:`
`101`	`101`	`name string`
`102`	`102`	`attrs []string`
`103`		`- args []MethodParam`
	`103`	`+ args []FunctionParam`
`104`	`104`	`return_type int`
`105`	`105`	`typ int`
`106`	`106`	`}`
Original file line number	Diff line number	Diff line change
`@@ -1354,7 +1354,7 @@ fn (mut w Walker) mark_resource_dependencies() {`
`1354`	`1354`	`w.mark_by_sym_name('FunctionData')`
`1355`	`1355`	`}`
`1356`	`1356`	`if w.uses_ct_params {`
`1357`		`- w.mark_by_sym_name('MethodParam')`
	`1357`	`+ w.mark_by_sym_name('FunctionParam')`
`1358`	`1358`	`}`
`1359`	`1359`	`if w.uses_ct_values {`
`1360`	`1360`	`w.mark_by_sym_name('EnumData')`