Include media filter to avoid looping over empty directories

Author: dormant-user

README.md

@@ -8,7 +8,7 @@
 [![none-shall-pass][nsp-logo]][nsp]
 
 #### Summary
-[`ios`][repo] is a light weight CLI tool to extract iPhone backups.
+[`ios`][repo] is a lightweight CLI tool to extract iPhone backups.
 
 ### Installation
 

src/dbutil.rs

@@ -1,8 +1,9 @@
+use crate::squire;
 use rusqlite::Connection;
 use std::path::Path;
 
 #[allow(dead_code)]
-fn get_columns(manifest_db_path: &Path) -> rusqlite::Result<(), Box<dyn std::error::Error>> {
+pub fn get_columns(manifest_db_path: &Path) -> rusqlite::Result<(), Box<dyn std::error::Error>> {
     let conn = Connection::open(manifest_db_path)?;
     let mut col_smt = conn.prepare("PRAGMA table_info(Files)")?;
     let columns: Vec<String> = col_smt
@@ -19,15 +20,18 @@ fn get_columns(manifest_db_path: &Path) -> rusqlite::Result<(), Box<dyn std::err
 }
 
 #[allow(dead_code)]
-fn get_table(
+pub fn get_table(
     manifest_db_path: &Path,
     limit: Option<usize>,
 ) -> rusqlite::Result<(), Box<dyn std::error::Error>> {
     let conn = Connection::open(manifest_db_path)?;
-    let statement = if limit.is_some() {
-        format!("SELECT * FROM Files LIMIT {}", limit.unwrap())
-    } else {
-        "SELECT * FROM Files".to_string()
+    let statement = match limit {
+        Some(head) => format!(
+            "SELECT * FROM Files {} LIMIT {}",
+            squire::media_filter(),
+            head
+        ),
+        None => format!("SELECT * FROM Files {}", squire::media_filter()),
     };
     let mut col_smt = conn.prepare(&statement)?;
     let columns: Vec<String> = col_smt

src/fileio.rs

@@ -1,5 +1,5 @@
-use crate::constant;
 use crate::parser;
+use crate::{constant, squire};
 use plist::Value;
 use rusqlite::{Connection, Result};
 use std::fs::{create_dir_all, File};
@@ -57,13 +57,17 @@ pub fn parse_manifest_db(
     let conn = Connection::open(manifest_db_path)?;
 
     // Get count to update progress bar
-    let mut count_stmt = conn.prepare(
-        "SELECT COUNT(*) FROM Files WHERE relativePath LIKE '%DCIM/%' OR relativePath LIKE '%PhotoData/%'"
-    )?;
+    let mut count_stmt = conn.prepare(&format!(
+        "SELECT COUNT(*) FROM Files {}'",
+        squire::media_filter()
+    ))?;
     let count: usize = count_stmt.query_row([], |row| row.get(0))?;
     let progress_bar_base: Arc<Mutex<Pbar>> = Arc::new(Mutex::new(pbar(Some(count))));
 
-    let mut stmt = conn.prepare("SELECT fileID, relativePath FROM Files WHERE relativePath LIKE '%DCIM/%' OR relativePath LIKE '%PhotoData/%'")?;
+    let mut stmt = conn.prepare(&format!(
+        "SELECT fileID, relativePath FROM Files {}'",
+        squire::media_filter()
+    ))?;
     let rows = stmt.query_map([], |row| {
         let file_id: String = row.get(0)?;
         let relative_path: String = row.get(1)?;

src/lib.rs

@@ -5,6 +5,7 @@
 pub mod backup;
 /// Module to load the required structs
 pub mod constant;
+pub mod dbutil;
 /// Module to handle database operations
 pub mod fileio;
 /// Module to construct a custom logger
@@ -13,7 +14,6 @@ pub mod logger;
 pub mod parser;
 /// Module for helper functions
 pub mod squire;
-mod dbutil;
 
 use rusqlite::Result;
 

src/squire.rs

@@ -153,3 +153,30 @@ pub fn get_epoch() -> u64 {
         Err(_) => panic!("SystemTime before UNIX EPOCH!"),
     }
 }
+
+pub fn media_filter() -> String {
+    "WHERE lower(relativePath) LIKE '%.hevc'
+       OR lower(relativePath) LIKE '%.h264'
+       OR lower(relativePath) LIKE '%.mp4'
+       OR lower(relativePath) LIKE '%.m4v'
+       OR lower(relativePath) LIKE '%.mov'
+       OR lower(relativePath) LIKE '%.avi'
+       OR lower(relativePath) LIKE '%.aac'
+       OR lower(relativePath) LIKE '%.mp3'
+       OR lower(relativePath) LIKE '%.m4a'
+       OR lower(relativePath) LIKE '%.alac'
+       OR lower(relativePath) LIKE '%.aiff'
+       OR lower(relativePath) LIKE '%.wav'
+       OR lower(relativePath) LIKE '%.flac'
+       OR lower(relativePath) LIKE '%.ac3'
+       OR lower(relativePath) LIKE '%.eac3'
+       OR lower(relativePath) LIKE '%.heic'
+       OR lower(relativePath) LIKE '%.jpg'
+       OR lower(relativePath) LIKE '%.jpeg'
+       OR lower(relativePath) LIKE '%.png'
+       OR lower(relativePath) LIKE '%.gif'
+       OR lower(relativePath) LIKE '%.tiff'
+       OR lower(relativePath) LIKE '%.bmp'
+       OR lower(relativePath) LIKE '%.ico';"
+        .to_string()
+}