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
|
mod db;
mod files;
mod flac;
use anyhow::Error;
use clap::{Arg, ArgAction, command, value_parser};
use std::path::PathBuf;
fn main() -> Result<(), Error> {
let matches = command!()
.help_expected(true)
.arg(
Arg::new("path")
.short('p')
.long("path")
.value_parser(value_parser!(PathBuf))
.help("Path for indexing/reencoding"),
)
.arg(
Arg::new("index")
.short('i')
.long("index")
.action(ArgAction::SetTrue)
.requires("path")
.help("Only index files"),
)
.arg(
Arg::new("doit")
.long("doit")
.action(ArgAction::SetTrue)
.conflicts_with("index")
.help("Actually reencode"),
)
.arg(
Arg::new("clean")
.short('c')
.long("clean")
.action(ArgAction::SetTrue)
.help("Clean and dedupe database"),
)
.arg(
Arg::new("threads")
.short('t')
.long("threads")
.value_parser(value_parser!(usize))
.default_value("4")
.help("Set number of reencoding threads (default: 4)"),
)
.arg(
Arg::new("db")
.long("db")
.value_parser(value_parser!(PathBuf))
.help("Path to database file"),
)
.get_matches();
let threads = *matches.get_one::<usize>("threads").unwrap();
let runtime = tokio::runtime::Builder::new_multi_thread()
.max_blocking_threads(threads)
.enable_all()
.build()?;
runtime.block_on(async move {
let conn = if let Some(path) = matches.get_one::<PathBuf>("db") {
db::Database::new(path).await?
} else {
db::open_default_db().await?
};
let path = matches.get_one::<PathBuf>("path");
if matches.get_flag("index") {
let folderpath = path.unwrap();
files::index_files_recursively(folderpath, &conn).await
} else if matches.get_flag("doit") {
files::reencode_files(&conn, path).await
} else if matches.get_flag("clean") {
conn.clean_files().await
} else {
let count = files::count_reencode_files(&conn, path).await.unwrap();
println!("Files to reencode:\t{count}");
Ok(())
}
})?;
Ok(())
}
|
