Add some implementations for Spanned

runtime/src/lib.rs

@@ -1,6 +1,6 @@
 pub mod __private;
 
-use std::ops::Deref;
+use std::ops::{Deref, Index, IndexMut};
 
 pub use rust_sitter_macro::*;
 
@@ -108,7 +108,7 @@ impl<T: Extract<U>, U> Extract<Vec<U>> for Vec<T> {
     }
 }
 
-#[derive(Clone, Debug)]
+#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, Hash)]
 /// A wrapper around a value that also contains the span of the value in the source.
 pub struct Spanned<T> {
     /// The underlying parsed node.
@@ -143,6 +143,20 @@ impl<T: Extract<U>, U> Extract<Spanned<U>> for Spanned<T> {
     }
 }
 
+impl<T> Index<Spanned<T>> for str {
+    type Output = str;
+
+    fn index(&self, index: Spanned<T>) -> &Self::Output {
+        &self[index.span.0..index.span.1]
+    }
+}
+
+impl<T> IndexMut<Spanned<T>> for str {
+    fn index_mut(&mut self, index: Spanned<T>) -> &mut Self::Output {
+        &mut self[index.span.0..index.span.1]
+    }
+}
+
 pub mod errors {
     #[cfg(feature = "tree-sitter-standard")]
     use tree_sitter_runtime_standard as tree_sitter;