sqlar 1.0.0

Port of the SQLite Archive Files utility
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

use std::path::Path;

use rusqlite::{Connection, Result};

use crate::{Entry, FileType};

/// Iterate over each entry in the archive.
///
/// # Arguments
///
/// * `path` - path to the archive.
/// * `decompress` - wether to get and decompress the file data.
///                  If `false` no data is included.
/// * `f` - the function to run on each entry.
///
/// # Returns
///
/// `Ok(())` if iteration over all entries succeeds.
/// `Err(e)` if fetching entries fails, parsing entries fails
///  or the user-supplied callback fails.
pub fn with_each_entry(
    path: impl AsRef<Path>,
    decompress: bool,
    f: impl FnMut(&Entry) -> Result<()>,
) -> Result<()> {
    let db = Connection::open(path)?;
    crate::compress::init(&db)?;
    iterate(&db, decompress, f)
}

/// Iterate over each entry in the archive.
///
/// # Arguments
///
/// * `conn` - the database to use
/// * `decompress` - wether to get and decompress the file data.
///                  If `false` no data is included.
/// * `f` - the function to run on each entry.
///
/// # Returns
///
/// `Ok(())` if iteration over all entries succeeds.
/// `Err(e)` if fetching entries fails, parsing entries fails
///  or the user-supplied callback fails.
pub(crate) fn iterate(
    conn: &Connection,
    decompress: bool,
    mut f: impl FnMut(&Entry) -> Result<()>,
) -> Result<()> {
    let mut stmt = if decompress {
        conn.prepare(
            r#"
            SELECT
                name, mode, mtime, sz,
                rusty_sqlar_uncompress(data, sz) as data,
                COALESCE(LENGTH(data), 0) as compressed_size
            FROM
                sqlar
            "#,
        )?
    } else {
        conn.prepare("SELECT name, mode, mtime, sz,  COALESCE(LENGTH(data), 0) as compressed_size FROM sqlar")?
    };

    let entry_iter = stmt.query_map([], |row| {
        let raw_mode: u32 = row.get(1)?;
        let mode = raw_mode & 0o777;
        let filetype = FileType::from(raw_mode);

        Ok(Entry {
            name: row.get(0)?,
            mode,
            filetype,
            mtime: row.get(2)?,
            size: row.get(3)?,
            compressed_size: row.get(if decompress { 5 } else { 4 })?,
            data: if decompress { row.get(4)? } else { None },
        })
    })?;

    for entry in entry_iter {
        match entry {
            Ok(entry) => f(&entry)?,
            Err(e) => return Err(e),
        }
    }

    Ok(())
}