Implement reading SyntaxKind from strings

nix-community · May 8, 2024 · 6e4480c · 6e4480c
1 parent 01e6582
commit 6e4480c
Show file tree

Hide file tree

Showing 3 changed files with 67 additions and 8 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -14,6 +14,8 @@ name = "all-packages"
 
 [dependencies]
 rowan = "0.15.0"
+strum = "0.26.2"
+strum_macros = "0.26.2"
 
 [dev-dependencies]
 criterion = "0.3.0"

diff --git a/src/kinds.rs b/src/kinds.rs
@@ -1,4 +1,6 @@
-#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
+use strum_macros::EnumString;
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash, EnumString)]
 #[repr(u16)]
 #[allow(non_camel_case_types)]
 pub enum SyntaxKind {
@@ -134,3 +136,14 @@ impl SyntaxKind {
         matches!(self, TOKEN_COMMENT | TOKEN_ERROR | TOKEN_WHITESPACE)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use crate::SyntaxKind;
+
+    #[test]
+    fn try_from() {
+        let kind = SyntaxKind::try_from("TOKEN_COMMENT").unwrap();
+        assert_eq!(kind, SyntaxKind::TOKEN_COMMENT);
+    }
+}