bugfix: fix postion of undefined lambda param (kcl-lang#873)

Signed-off-by: he1pa <[email protected]>

kclvm/sema/src/resolver/node.rs

@@ -961,7 +961,11 @@ impl<'ctx> MutSelfTypedResultWalker<'ctx> for Resolver<'ctx> {
             for (i, arg) in args.node.args.iter().enumerate() {
                 let name = arg.node.get_name();
                 let arg_ty = args.node.get_arg_type_node(i);
-                let ty = self.parse_ty_with_scope(arg_ty, arg.get_span_pos());
+                let range = match arg_ty {
+                    Some(arg_type_node) => arg_type_node.get_span_pos(),
+                    None => arg.get_span_pos(),
+                };
+                let ty = self.parse_ty_with_scope(arg_ty, range);
 
                 // If the arguments type of a lambda is a schema type,
                 // It should be marked as an schema instance type.

kclvm/sema/src/resolver/test_data/undef_lambda_param.k

@@ -0,0 +1,3 @@
+lambda x: Undef { # 'Undef' is undefined
+    x
+}

kclvm/sema/src/resolver/tests.rs

@@ -681,3 +681,41 @@ fn test_pkg_not_found_suggestion() {
         "find more package on 'https://artifacthub.io'"
     );
 }
+
+#[test]
+fn undef_lambda_param() {
+    let sess = Arc::new(ParseSession::default());
+    let mut program = load_program(
+        sess.clone(),
+        &["./src/resolver/test_data/undef_lambda_param.k"],
+        None,
+        None,
+    )
+    .unwrap();
+    let scope = resolve_program(&mut program);
+    assert_eq!(scope.handler.diagnostics.len(), 1);
+
+    let root = &program.root.clone();
+    let filename = Path::new(&root.clone())
+        .join("undef_lambda_param.k")
+        .display()
+        .to_string();
+
+    let range = scope.handler.diagnostics[0].messages[0].range.clone();
+
+    assert_eq!(
+        range,
+        (
+            Position {
+                filename: filename.clone(),
+                line: 1,
+                column: Some(10),
+            },
+            Position {
+                filename: filename.clone(),
+                line: 1,
+                column: Some(15),
+            }
+        )
+    );
+}