Initial commit.

src/main.rs

@@ -0,0 +1,127 @@
+use rusqlite::{named_params, Connection, Result, NO_PARAMS};
+use std::error::Error;
+use std::fs::File;
+use std::io::BufReader;
+use xml::reader::{EventReader, XmlEvent};
+use zip::{read::ZipFile, ZipArchive};
+
+fn get_root_file(container: ZipFile) -> Result<Option<String>, Box<dyn Error>> {
+    let parser = EventReader::new(container);
+
+    for e in parser {
+        match e {
+            Ok(XmlEvent::StartElement {
+                name, attributes, ..
+            }) if name.local_name == "rootfile" => {
+                for attr in attributes {
+                    if attr.name.local_name == "full-path" {
+                        //opf_file = attr.value;
+                        return Ok(Some(attr.value));
+                    }
+                }
+            }
+            Err(e) => {
+                return Err(Box::new(e));
+            }
+            _ => {}
+        }
+    }
+    Ok(None)
+}
+
+fn get_attribute_file_as(opf: ZipFile) -> Option<String> {
+    let parser = EventReader::new(opf);
+
+    for e in parser {
+        match e {
+            Ok(XmlEvent::StartElement {
+                name, attributes, ..
+            }) if name.local_name == "package" => {
+                for attr in attributes {
+                    if attr.name.local_name == "version" {
+                        if attr.value.starts_with("2") == false {
+                            return None;
+                        }
+                    }
+                }
+            }
+            Ok(XmlEvent::StartElement {
+                name, attributes, ..
+            }) if name.local_name == "creator" => {
+                for attr in attributes {
+                    if attr.name.local_name == "file-as" {
+                        return Some(attr.value);
+                    }
+                }
+            }
+            Err(_e) => {
+                break;
+            }
+            _ => {}
+        }
+    }
+
+    None
+}
+
+struct BookEntry {
+    id: i32,
+    filepath: String,
+}
+
+fn main() {
+    let mut conn = Connection::open("/mnt/ext1/system/explorer-3/explorer-3.db").unwrap();
+    let tx = conn.transaction().unwrap();
+    {
+        let mut stmt = tx.prepare("SELECT id FROM books_impl WHERE ext LIKE 'epub' AND author LIKE '% %' AND firstauthor NOT LIKE '%\\,%' ESCAPE '\\'").unwrap();
+        let mut rows = stmt.query(NO_PARAMS).unwrap();
+        let mut book_ids: Vec<i32> = Vec::new();
+        while let Some(row) = rows.next().unwrap() {
+            book_ids.push(row.get(0).unwrap());
+        }
+
+        let mut bookentries = Vec::new();
+
+        for book_id in book_ids {
+            let mut stmt = tx.prepare("SELECT folders.name,files.filename FROM files,folders WHERE files.book_id = :book_id AND files.storageid = 1 AND files.folder_id = folders.id").unwrap();
+            let mut rows = stmt
+                .query_named(named_params! { ":book_id": book_id })
+                .unwrap();
+            while let Some(row) = rows.next().unwrap() {
+                let prefix: String = row.get(0).unwrap();
+                let filename: String = row.get(1).unwrap();
+                let filepath = format!("{}/{}", prefix, filename);
+                bookentries.push(BookEntry {
+                    id: book_id,
+                    filepath: filepath,
+                });
+            }
+        }
+
+        //println!("Number of entries found: {}", bookentries.len());
+
+        for entry in bookentries {
+            let file = File::open(entry.filepath.as_str());
+            let file = match file {
+                Err(_) => continue,
+                Ok(file) => file,
+            };
+
+            let mut archive = ZipArchive::new(BufReader::new(file)).unwrap();
+
+            let container = archive.by_name("META-INF/container.xml").unwrap();
+
+            if let Some(opf_file) = get_root_file(container).unwrap() {
+                let opf = archive.by_name(opf_file.as_str()).unwrap();
+                if let Some(file_as) = get_attribute_file_as(opf) {
+                    let mut stmt = tx
+                        .prepare("UPDATE books_impl SET firstauthor = :file_as WHERE id = :book_id")
+                        .unwrap();
+                    stmt.execute_named(named_params![":file_as": file_as, ":book_id": entry.id])
+                        .unwrap();
+                }
+            }
+        }
+    }
+    tx.commit().unwrap();
+}