txlib 0.2.1

text based epub library
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

mod parse_lib;

use clap::Parser;
use std::path::Path;
use std::process::exit;

use crate::parse_lib::SortBy;

/// txlib: text based epub library management
#[derive(Parser)]
#[command(version, about, long_about = None)]
struct Args {
    /// Root directory containing epub files
    epub_dir: String,
    /// File path to read/store epub library data
    #[arg(short, long)]
    lib_db: String,
    /// Sort by: date, title, author, pages or series
    #[arg(short, long, default_value = "date")]
    sort: String,
    /// Reverse sorting order
    #[arg(short, long, action = clap::ArgAction::SetTrue)]
    reverse: bool,
    /// Print output without saving to DB
    #[arg(short, long, action = clap::ArgAction::SetTrue)]
    no_save: bool,
}

/// Returns `SortBy` enum after parsing string sorting option.
fn parse_sorting_option(sort_str: &str) -> SortBy {
    match sort_str.to_lowercase().as_str() {
        "r" | "read" => SortBy::Read,
        "t" | "title" => SortBy::Title,
        "a" | "author" => SortBy::Author,
        "p" | "pages" => SortBy::Pages,
        "s" | "series" => SortBy::Series,
        _ => SortBy::Date,
    }
}

fn start(lib_db: &Path, epub_dir: &Path, sort_by: SortBy, reverse: bool, no_save: bool) -> Result<(), &'static str> {
    if !lib_db.exists() {
        return Err("error: unable to locate DB file");
    }
    if !epub_dir.exists() {
        return Err("error: epub directory not found");
    }

    let library = parse_lib::load_library(lib_db, epub_dir, &sort_by, reverse)?;
    let library_string = parse_lib::library_to_string(&library);
    if !no_save {
        parse_lib::write_library(&library_string, lib_db)?;
    }

    println!("{library_string}");
    Ok(())
}

fn main() {
    let args = Args::parse();
    let lib_db = Path::new(&args.lib_db);
    let epub_dir = Path::new(&args.epub_dir);
    let sort_by = parse_sorting_option(&args.sort);

    if let Err(error_msg) = start(lib_db, epub_dir, sort_by, args.reverse, args.no_save) {
        eprintln!("txlib: {error_msg}");
        exit(1);
    }
}