fix: fix the lambda argument type, when it is schema type, into schem…

…a type instance (kcl-lang#803) * fix: fix the lambda argument type, when it is schema type, into schema type instance Signed-off-by: zongz <[email protected]> * fix: make fmt Signed-off-by: zongz <[email protected]> --------- Signed-off-by: zongz <[email protected]>
Peefy · Oct 26, 2023 · 4eac50d · 4eac50d
1 parent b4a784e
commit 4eac50d
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 0 deletions.
diff --git a/kclvm/sema/src/resolver/node.rs b/kclvm/sema/src/resolver/node.rs
@@ -943,6 +943,17 @@ impl<'ctx> MutSelfTypedResultWalker<'ctx> for Resolver<'ctx> {
                 let name = arg.node.get_name();
                 let arg_ty = args.node.get_arg_type(i);
                 let ty = self.parse_ty_with_scope(&arg_ty, arg.get_span_pos());
+
+                // If the arguments type of a lambda is a schema type,
+                // It should be marked as an schema instance type.
+                let ty = if let TypeKind::Schema(sty) = &ty.kind {
+                    let mut arg_ty = sty.clone();
+                    arg_ty.is_instance = true;
+                    Rc::new(Type::schema(arg_ty))
+                } else {
+                    ty.clone()
+                };
+
                 params.push(Parameter {
                     name,
                     ty: ty.clone(),

diff --git a/test/grammar/lambda/arg_type/main.k b/test/grammar/lambda/arg_type/main.k
@@ -0,0 +1,9 @@
+import sub as s
+
+identity = lambda res: s.Test {
+  res {
+    name = "world" 
+  }
+}
+
+c = identity(s.Test{name="hello"})
diff --git a/test/grammar/lambda/arg_type/stdout.golden b/test/grammar/lambda/arg_type/stdout.golden
@@ -0,0 +1,2 @@
+c:
+  name: world
diff --git a/test/grammar/lambda/arg_type/sub.k b/test/grammar/lambda/arg_type/sub.k
@@ -0,0 +1,2 @@
+schema Test:
+    name: str