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
|
use colored::*;
use std::{fs::File, path::PathBuf};
use tracing::{error, info};
use tracing_subscriber::{fmt, prelude::*, EnvFilter};
pub struct Display {
pub log_file: PathBuf,
pub output_dir: Option<PathBuf>,
}
impl Display {
pub fn new(book_id: &str) -> Self {
let log_file = std::env::current_dir()
.unwrap()
.join(format!("info_{}.log", book_id));
let d = Self {
log_file,
output_dir: None,
};
let file = File::create(&d.log_file).expect("Cannot create log file");
tracing_subscriber::registry()
.with(EnvFilter::from_default_env())
.with(fmt::layer().with_writer(std::io::stdout))
.with(fmt::layer().with_writer(file).with_ansi(false))
.init();
d.intro();
info!("** Welcome to SafariBooks (Rust) **");
d
}
pub fn intro(&self) {
let banner = r#"
____ __ _ ____ _
/ ___| __ _ / _| __ _ _ __(_) _ \ _ _ ___| |_
\___ \ / _` | |_ / _` | '__| | |_) | | | / __| __|
___) | (_| | _| (_| | | | | _ <| |_| \__ \ |_
|____/ \__,_|_| \__,_|_| |_|_| \_\\__,_|___/\__|
"#;
println!("{}", banner.yellow());
println!("{}", "~".repeat(32));
}
pub fn info(&self, msg: &str) {
println!("{} {}", "[*]".yellow(), msg);
info!("{msg}");
}
pub fn error_and_exit(&self, msg: &str) -> ! {
eprintln!("{} {}", "[!]".on_red().white(), msg);
error!("{msg}");
std::process::exit(1);
}
pub fn set_output_dir(&mut self, dir: PathBuf) {
self.output_dir = Some(dir.clone());
self.info(&format!("Output directory:\n {}", dir.display()));
}
}
|
