feat: add fuzz match for error 'attribute not found' (#715)

* feat: add fuzz match for error 'attribute not found'

* fix: add 'suggestions' to supports fuzz match

* fix: rm useless imports

* fix: make fmt

kclvm/cmd/src/test_data/fuzz_match/main.k

@@ -0,0 +1,6 @@
+schema Person:
+    aa?: int
+    aaa?: int
+
+p = Person {}
+a = p.a # Error, attribute 'a' not found in schema `Person`, did you mean `aa`?

kclvm/cmd/src/test_data/fuzz_match/main_unmatched.k

@@ -0,0 +1,6 @@
+schema Person:
+    en?: int
+    sd?: int
+
+p = Person {}
+a = p.a # Error, attribute 'a' not found in schema `Person`

kclvm/cmd/src/tests.rs

@@ -559,3 +559,40 @@ fn test_plugin_not_found() {
         Err(msg) => assert!(msg.contains("the plugin package `kcl_plugin.not_exist` is not found, please confirm if plugin mode is enabled")),
     }
 }
+
+#[test]
+fn test_error_message_fuzz_matched() {
+    let test_case_path = PathBuf::from("./src/test_data/fuzz_match/main.k");
+    let matches = app().arg_required_else_help(true).get_matches_from(&[
+        ROOT_CMD,
+        "run",
+        &test_case_path.canonicalize().unwrap().display().to_string(),
+    ]);
+    let settings = must_build_settings(matches.subcommand_matches("run").unwrap());
+    let sess = Arc::new(ParseSession::default());
+    match exec_program(sess.clone(), &settings.try_into().unwrap()) {
+        Ok(_) => panic!("unreachable code."),
+        Err(msg) => {
+            assert!(msg
+                .contains("attribute 'a' not found in schema 'Person', did you mean '[\"aa\"]'?"))
+        }
+    }
+}
+
+#[test]
+fn test_error_message_fuzz_unmatched() {
+    let test_case_path = PathBuf::from("./src/test_data/fuzz_match/main_unmatched.k");
+    let matches = app().arg_required_else_help(true).get_matches_from(&[
+        ROOT_CMD,
+        "run",
+        &test_case_path.canonicalize().unwrap().display().to_string(),
+    ]);
+    let settings = must_build_settings(matches.subcommand_matches("run").unwrap());
+    let sess = Arc::new(ParseSession::default());
+    match exec_program(sess.clone(), &settings.try_into().unwrap()) {
+        Ok(_) => panic!("unreachable code."),
+        Err(msg) => {
+            assert!(msg.contains("attribute 'a' not found in schema 'Person'"))
+        }
+    }
+}

kclvm/sema/Cargo.toml

@@ -29,6 +29,7 @@ compiler_base_span = {path = "../../compiler_base/span", version = "0.0.2"}
 compiler_base_session = {path = "../../compiler_base/session"}
 compiler_base_macros = "0.0.1"
 compiler_base_error = "0.0.8"
+suggestions = "0.1.1"
 
 [dev-dependencies]
 kclvm-parser = {path = "../parser"}

kclvm/sema/src/resolver/attr.rs

@@ -3,6 +3,7 @@ use std::rc::Rc;
 use crate::builtin::system_module::{get_system_module_members, UNITS, UNITS_NUMBER_MULTIPLIER};
 use crate::builtin::{get_system_member_function_ty, STRING_MEMBER_FUNCTIONS};
 use crate::resolver::Resolver;
+use crate::ty::TypeKind::Schema;
 use crate::ty::{DictType, ModuleKind, Type, TypeKind};
 use kclvm_error::diagnostic::Range;
 use kclvm_error::*;
@@ -110,16 +111,31 @@ impl<'ctx> Resolver<'ctx> {
                 }
             }
         };
+
         if !result {
+            // The attr user input.
+            let (attr, suggestion) = if attr.is_empty() {
+                ("[missing name]", "".to_string())
+            } else {
+                let mut suggestion = String::new();
+                // Calculate the closests miss attributes.
+                if let Schema(schema_ty) = &obj.kind {
+                    // Get all the attrbuets of the schema.
+                    let attrs = schema_ty.attrs.keys().cloned().collect::<Vec<String>>();
+                    let suggs = suggestions::provide_suggestions(attr, &attrs);
+                    if suggs.len() > 0 {
+                        suggestion = format!(", did you mean '{:?}'?", suggs);
+                    }
+                }
+                (attr, suggestion)
+            };
+
             self.handler.add_type_error(
                 &format!(
-                    "{} has no attribute {}",
+                    "attribute '{}' not found in schema '{}'{}",
+                    attr,
                     obj.ty_str(),
-                    if attr.is_empty() {
-                        "[missing name]"
-                    } else {
-                        attr
-                    }
+                    suggestion
                 ),
                 range,
             );