v: add typeof(var).indirections and T.indirections (#22805)

Author: felipensp

vlib/v/ast/ast.v

@@ -287,6 +287,7 @@ pub enum GenericKindField {
 	name
 	typ
 	unaliased_typ
+	indirections
 }
 
 // `foo.bar`

vlib/v/checker/checker.v

@@ -1531,17 +1531,16 @@ fn (mut c Checker) selector_expr(mut node ast.SelectorExpr) ast.Type {
 			.name {
 				return ast.string_type
 			}
-			.unaliased_typ {
-				return ast.int_type
-			}
-			.typ {
+			.unaliased_typ, .typ, .indirections {
 				return ast.int_type
 			}
 			else {
 				if node.field_name == 'name' {
 					return ast.string_type
 				} else if node.field_name == 'idx' {
 					return ast.int_type
+				} else if node.field_name == 'indirections' {
+					return ast.int_type
 				}
 				c.error('invalid field `.${node.field_name}` for type `${node.expr}`',
 					node.pos)

vlib/v/checker/if.v

@@ -268,6 +268,12 @@ fn (mut c Checker) if_expr(mut node ast.IfExpr) ast.Type {
 									}
 								}
 							}
+						} else if left.expr is ast.TypeOf {
+							skip_state = if left.expr.typ.nr_muls() == right.val.i64() {
+								ComptimeBranchSkipState.eval
+							} else {
+								ComptimeBranchSkipState.skip
+							}
 						}
 					} else if branch.cond.op in [.eq, .ne] && left is ast.SelectorExpr
 						&& right is ast.StringLiteral {

vlib/v/gen/c/cgen.v

@@ -3875,6 +3875,10 @@ fn (mut g Gen) selector_expr(node ast.SelectorExpr) {
 				g.write(int(g.table.unaliased_type(g.unwrap_generic(node.name_type))).str())
 				return
 			}
+			.indirections {
+				g.write(int(g.unwrap_generic(node.name_type).nr_muls()).str())
+				return
+			}
 			.unknown {
 				// ast.TypeOf of `typeof(string).idx` etc
 				if node.field_name == 'name' {
@@ -3893,6 +3897,14 @@ fn (mut g Gen) selector_expr(node ast.SelectorExpr) {
 					// `typeof(expr).idx`
 					g.write(int(g.unwrap_generic(name_type)).str())
 					return
+				} else if node.field_name == 'indirections' {
+					mut name_type := node.name_type
+					if node.expr is ast.TypeOf {
+						name_type = g.resolve_comptime_type(node.expr.expr, name_type)
+					}
+					// `typeof(expr).indirections`
+					g.write(int(g.unwrap_generic(name_type).nr_muls()).str())
+					return
 				}
 				g.error('unknown generic field', node.pos)
 			}

vlib/v/gen/c/comptime.v

@@ -469,6 +469,8 @@ fn (mut g Gen) get_expr_type(cond ast.Expr) ast.Type {
 				return g.unwrap_generic(cond.name_type)
 			} else if cond.gkind_field == .unaliased_typ {
 				return g.table.unaliased_type(g.unwrap_generic(cond.name_type))
+			} else if cond.gkind_field == .indirections {
+				return ast.int_type
 			} else {
 				name := '${cond.expr}.${cond.field_name}'
 				if name in g.comptime.type_map {

vlib/v/gen/js/js.v

@@ -3252,13 +3252,17 @@ fn (mut g JsGen) gen_selector_expr(it ast.SelectorExpr) {
 				g.write('new int(')
 				g.write('${int(g.unwrap_generic(it.name_type))}')
 				g.write(')')
-				g.write(')')
 				return
 			}
 			.unaliased_typ {
 				g.write('new int(')
 				g.write('${int(g.table.unaliased_type(g.unwrap_generic(it.name_type)))}')
 				g.write(')')
+				return
+			}
+			.indirections {
+				g.write('new int(')
+				g.write('${int(g.unwrap_generic(it.name_type).nr_muls())}')
 				g.write(')')
 				return
 			}

vlib/v/parser/parser.v

@@ -2923,6 +2923,7 @@ fn (mut p Parser) name_expr() ast.Expr {
 				'name' { ast.GenericKindField.name }
 				'typ' { ast.GenericKindField.typ }
 				'unaliased_typ' { ast.GenericKindField.unaliased_typ }
+				'indirections' { ast.GenericKindField.indirections }
 				else { ast.GenericKindField.unknown }
 			}
 			pos.extend(p.tok.pos())

vlib/v/tests/typeof_indirections_test.v

@@ -0,0 +1,18 @@
+fn indirections[T](val T) int {
+	return T.indirections
+}
+
+fn test_main() {
+	a := 0
+	assert typeof(a).indirections == 0
+	assert indirections(a) == 0
+	b := &a
+	assert typeof(b).indirections == 1
+	assert indirections(b) == 1
+	c := [1]
+	assert typeof(c).indirections == 0
+	assert indirections(c) == 0
+	d := &c
+	assert typeof(d).indirections == 1
+	assert indirections(d) == 1
+}