Initial parser for butlerdb

src/itch/butler_db_parser.rs

@@ -0,0 +1,73 @@
+use nom::{
+    bytes::complete::{tag, take_until},
+    multi::many0,
+    IResult,
+};
+
+pub(crate) struct DbPaths<'a> {
+    pub(crate) base_path: &'a str,
+    pub(crate) path: &'a str,
+}
+
+pub(crate) fn parse_butler_db<'a>(content: &'a [u8]) -> nom::IResult<&[u8], Vec<DbPaths<'a>>> {
+    many0(parse_path)(content)
+}
+
+fn parse_path<'a>(i: &'a [u8]) -> nom::IResult<&[u8], DbPaths<'a>> {
+    let prefix = "{\"basePath\":\"";
+    let suffix = "\",\"totalSize\"";
+    let (i, _taken) = take_until(prefix)(i)?;
+    let (i, _taken) = tag(prefix)(i)?;
+    let (i, base_path) = take_until(suffix)(i)?;
+
+    let prefix = ":[{\"path\":\"";
+    let suffix = "\",\"depth";
+    let (i, _taken) = take_until(prefix)(i)?;
+    let (i, _taken) = tag(prefix)(i)?;
+    let (i, path) = take_until(suffix)(i)?;
+
+    IResult::Ok((
+        i,
+        DbPaths {
+            base_path: std::str::from_utf8(base_path).unwrap(),
+            path: std::str::from_utf8(path).unwrap(),
+        },
+    ))
+}
+
+#[cfg(test)]
+mod tests {
+
+    use super::*;
+
+    #[test]
+    fn parse_itch_butler_db_test() {
+        let content = include_bytes!("../testdata/itch/butler.db-wal");
+        let result = parse_butler_db(content);
+        assert!(result.is_ok());
+        let (_r, paths) = result.unwrap();
+        assert_eq!(paths.len(), 6);
+
+        assert_eq!(paths[0].base_path, "/home/philip/.config/itch/apps/islands");
+        assert_eq!(paths[0].path, "Islands_Linux.x86_64");
+        assert_eq!(
+            paths[1].base_path,
+            "/home/philip/.config/itch/apps/night-in-the-woods"
+        );
+        assert_eq!(paths[1].path, "Night in the Woods.x86_64");
+        assert_eq!(paths[2].base_path, "/home/philip/.config/itch/apps/islands");
+        assert_eq!(paths[2].path, "Islands_Linux.x86_64");
+        assert_eq!(
+            paths[3].base_path,
+            "/home/philip/.config/itch/apps/overland"
+        );
+        assert_eq!(paths[3].path, "Overland.x86_64");
+        assert_eq!(
+            paths[4].base_path,
+            "/home/philip/.config/itch/apps/night-in-the-woods"
+        );
+        assert_eq!(paths[4].path, "Night in the Woods.x86_64");
+        assert_eq!(paths[5].base_path, "/home/philip/.config/itch/apps/islands");
+        assert_eq!(paths[5].path, "Islands_Linux.x86_64");
+    }
+}