Annotate everything to make it work

Author: dralley

examples/custom_entities.rs

@@ -32,6 +32,8 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let entity_re = Regex::new(r#"<!ENTITY\s+([^ \t\r\n]+)\s+"([^"]*)"\s*>"#)?;
 
     loop {
+        let lookup_custom_entity = |ent: &_| custom_entities.get(ent).map(|s| s.as_str());
+
         match reader.read_event_into(&mut buf) {
             Ok(Event::DocType(ref e)) => {
                 for cap in entity_re.captures_iter(&e) {
@@ -40,7 +42,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
             }
             Ok(Event::Start(ref e)) => match e.name().as_ref() {
                 b"test" => {
-                    let lookup_custom_entity = |ent| custom_entities.get(ent).map(|s| s.as_str());
+                    // let lookup_custom_entity = |ent: &_| custom_entities.get(ent).map(|s| s.as_str());
                     let attributes = e
                         .attributes()
                         .map(|a| {
@@ -57,7 +59,7 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
                 _ => (),
             },
             Ok(Event::Text(ref e)) => {
-                let lookup_custom_entity = |ent| custom_entities.get(ent).map(|s| s.as_str());
+                // let lookup_custom_entity = |ent: &_| custom_entities.get(ent).map(|s| s.as_str());
                 println!(
                     "text value: {}",
                     e.unescape_and_decode_with_custom_entities(&reader, lookup_custom_entity)

src/escapei.rs

@@ -121,8 +121,15 @@ pub fn unescape(raw: &[u8]) -> Result<Cow<[u8]>, EscapeError> {
 ///
 /// # Pre-condition
 ///
-/// The implementation of `lookup_custom_entity` is expected to operate over UTF-8 inputs.
-pub fn unescape_with<'a>(raw: &'a [u8], resolve_entity: impl Fn(&[u8]) -> Option<&str>) -> Result<Cow<'a, [u8]>, EscapeError> {
+/// The implementation of `resolve_entity` is expected to operate over UTF-8 inputs.
+pub fn unescape_with<'input, 'entity, F>(
+    raw: &'input [u8],
+    resolve_entity: F,
+) -> Result<Cow<'input, [u8]>, EscapeError>
+where
+    // the lifetime of the output comes from a capture or is `'static`
+    F: Fn(&[u8]) -> Option<&'entity str>,
+{
     let mut unescaped = None;
     let mut last_end = 0;
     let mut iter = memchr::memchr2_iter(b'&', b';', raw);

src/events/attributes.rs

@@ -57,9 +57,9 @@ impl<'a> Attribute<'a> {
     /// # Pre-condition
     ///
     /// The keys and values of `custom_entities`, if any, must be valid UTF-8.
-    pub fn unescaped_value_with_custom_entities<'s>(
+    pub fn unescaped_value_with_custom_entities<'s, 'entity>(
         &'s self,
-        resolve_entity: impl Fn(&[u8]) -> Option<&str>,
+        resolve_entity: impl Fn(&[u8]) -> Option<&'entity str>,
     ) -> XmlResult<Cow<'s, [u8]>> {
         unescape_with(&*self.value, resolve_entity).map_err(Error::EscapeError)
     }
@@ -92,10 +92,10 @@ impl<'a> Attribute<'a> {
     /// # Pre-condition
     ///
     /// The keys and values of `custom_entities`, if any, must be valid UTF-8.
-    pub fn unescape_and_decode_value_with_custom_entities<B>(
+    pub fn unescape_and_decode_value_with_custom_entities<'entity, B>(
         &self,
         reader: &Reader<B>,
-        resolve_entity: impl Fn(&[u8]) -> Option<&str>,
+        resolve_entity: impl Fn(&[u8]) -> Option<&'entity str>,
     ) -> XmlResult<String> {
         let decoded = reader.decoder().decode(&*self.value)?;
         let unescaped = unescape_with(decoded.as_bytes(), resolve_entity)?;

src/events/mod.rs

@@ -732,12 +732,12 @@ impl<'a> BytesText<'a> {
     ///
     /// # Pre-condition
     ///
-    /// The implementation of `lookup_custom_entity` is expected to operate over UTF-8 inputs.
+    /// The implementation of `resolve_entity` is expected to operate over UTF-8 inputs.
     ///
     /// See also [`unescaped()`](#method.unescaped)
-    pub fn unescaped_with_custom_entities<'s>(
+    pub fn unescaped_with_custom_entities<'s, 'entity>(
         &'s self,
-        resolve_entity: impl Fn(&[u8]) -> Option<&str>,
+        resolve_entity: impl Fn(&[u8]) -> Option<&'entity str>,
     ) -> Result<Cow<'s, [u8]>> {
         unescape_with(self, resolve_entity).map_err(Error::EscapeError)
     }
@@ -761,11 +761,11 @@ impl<'a> BytesText<'a> {
     ///
     /// # Pre-condition
     ///
-    /// The implementation of `lookup_custom_entity` is expected to operate over UTF-8 inputs.
-    pub fn unescape_and_decode_with_custom_entities<B>(
+    /// The implementation of `resolve_entity` is expected to operate over UTF-8 inputs.
+    pub fn unescape_and_decode_with_custom_entities<'entity, B>(
         &self,
         reader: &Reader<B>,
-        resolve_entity: impl Fn(&[u8]) -> Option<&str>,
+        resolve_entity: impl Fn(&[u8]) -> Option<&'entity str>,
     ) -> Result<String> {
         let decoded = reader.decoder().decode(&*self)?;