Fix matching mime-type with upper cases like UTF8_STRING (#4)

And we need to implement a flexible match for text types.
beeender · Jan 7, 2025 · cc08cdd · cc08cdd
1 parent 20809f9
commit cc08cdd
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/protocol/source_data.rs b/src/protocol/source_data.rs
@@ -16,13 +16,17 @@ pub trait SourceData {
 
 impl SourceData for Vec<SourceDataItem> {
     fn content_by_mime_type(&self, mime_type: &str) -> (bool, Rc<Vec<u8>>) {
+        // TODO: Need a more flexible way to match text types.
         log::debug!("content_by_mime_type was called with '{}'", mime_type);
         let mut filter_it = self
             .iter()
             .filter(|item| {
                 item.mime_type
                     .iter()
-                    .filter(|mt| mt.contains(&mime_type.to_lowercase()))
+                    .filter(|mt| {
+                        log::debug!("check mime-type {mt}");
+                        mt.eq_ignore_ascii_case(mime_type)
+                    })
                     .peekable()
                     .peek()
                     .is_some()
@@ -74,7 +78,10 @@ mod tests {
         let (result, content) = r.content_by_mime_type("text");
         assert!(result);
         assert_eq!(content.as_slice(), b"GOOD");
-        let (result, content) = r.content_by_mime_type("html");
+        let (result, content) = r.content_by_mime_type("TEXT");
+        assert!(result);
+        assert_eq!(content.as_slice(), b"GOOD");
+        let (result, content) = r.content_by_mime_type("text/html");
         assert!(result);
         assert_eq!(content.as_slice(), b"BAD");
         let (result, content) = r.content_by_mime_type("no_mime");