fix: raw identifier in the schema string attribute

Signed-off-by: peefy <xpf6677@163.com>

kclvm/sema/src/pre_process/identifier.rs

@@ -173,7 +173,13 @@ impl<'ctx> MutSelfMutWalker<'ctx> for RawIdentifierTransformer {
             .collect::<Vec<Node<String>>>();
     }
     fn walk_schema_attr(&mut self, schema_attr: &'ctx mut ast::SchemaAttr) {
-        schema_attr.name.node = remove_raw_ident_prefix(&schema_attr.name.node);
+        // If the attribute is an identifier and then fix it.
+        // Note that we do not fix a string-like attribute e.g., `"$name"`
+        if schema_attr.name.end_column - schema_attr.name.column
+            <= schema_attr.name.node.chars().count() as u64
+        {
+            schema_attr.name.node = remove_raw_ident_prefix(&schema_attr.name.node);
+        }
         walk_list_mut!(self, walk_call_expr, schema_attr.decorators);
         walk_if_mut!(self, walk_expr, schema_attr.value);
     }

kclvm/sema/src/pre_process/test_data/raw_identifier.k

@@ -1 +1,9 @@
 $schema = 1
+
+schema Name:
+    $name?: str
+    "$name"?: str
+
+n = Name {
+    "$name": "1"
+}

kclvm/sema/src/pre_process/tests.rs

@@ -37,6 +37,20 @@ fn test_fix_raw_identifier_prefix() {
     } else {
         panic!("invalid assign statement")
     }
+    if let ast::Stmt::Schema(schema_stmt) = &module.body[1].node {
+        if let ast::Stmt::SchemaAttr(attr) = &schema_stmt.body[0].node {
+            assert_eq!(attr.name.node, "name");
+        } else {
+            panic!("invalid schema attr")
+        }
+        if let ast::Stmt::SchemaAttr(attr) = &schema_stmt.body[1].node {
+            assert_eq!(attr.name.node, "$name");
+        } else {
+            panic!("invalid schema attr")
+        }
+    } else {
+        panic!("invalid schema statement")
+    }
 }
 
 #[test]