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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// modules
pub mod client;
pub mod config;
pub mod prelude;
pub mod utils;
pub mod view;
const DEFAULT_CONFIG_FILE: &str = "hn-tui.toml";
const DEFAULT_LOG_FILE: &str = "hn-tui.log";
use clap::*;
use prelude::*;
fn run() {
// setup HN Client
let client = client::init_client();
// setup the application's UI
let s = view::init_ui(client);
// use `cursive_buffered_backend` crate to fix the flickering issue
// when using `cursive` with `crossterm_backend` (See https://github.com/gyscos/Cursive/issues/142)
let crossterm_backend = backends::crossterm::Backend::init().unwrap();
let buffered_backend = Box::new(cursive_buffered_backend::BufferedBackend::new(
crossterm_backend,
));
let mut app = CursiveRunner::new(s, buffered_backend);
app.run();
}
/// initialize application logging
fn init_logging(log_dir_str: &str) {
if std::env::var("RUST_LOG").is_err() {
std::env::set_var("RUST_LOG", "hackernews_tui=info")
}
let log_dir = std::path::PathBuf::from(log_dir_str);
if !log_dir.exists() {
std::fs::create_dir_all(&log_dir)
.unwrap_or_else(|_| panic!("{}", "failed to create a log folder: {log_dir_str}"));
}
let log_file = std::fs::File::create(log_dir.join(DEFAULT_LOG_FILE)).unwrap_or_else(|err| {
panic!("failed to create application's log file: {}", err);
});
tracing_subscriber::fmt::fmt()
.with_env_filter(tracing_subscriber::EnvFilter::from_default_env())
.with_ansi(false)
.with_writer(std::sync::Mutex::new(log_file))
.init();
}
/// parse command line arguments
fn parse_args(config_dir: &std::path::Path, cache_dir: &std::path::Path) -> ArgMatches {
Command::new("hackernews-tui")
.version("0.10.2")
.author("Thang Pham <phamducthang1234@gmail>")
.about("A Terminal UI to browse Hacker News")
.arg(
Arg::new("config")
.short('c')
.long("config")
.value_name("FILE")
.default_value(
config_dir
.join(DEFAULT_CONFIG_FILE)
.to_str()
.expect("failed to config file `Path` to str"),
)
.help("Path to the application's config file")
.next_line_help(true),
)
.arg(
Arg::new("log")
.short('l')
.long("log")
.value_name("FOLDER")
.default_value(
cache_dir
.to_str()
.expect("failed to convert cache dir `Path` to str"),
)
.help("Path to a folder to store application's logs")
.next_line_help(true),
)
.get_matches()
}
fn init_app_dirs() -> (std::path::PathBuf, std::path::PathBuf) {
let mut config_dir = dirs_next::config_dir().expect("failed to get user's config dir");
let cache_dir = dirs_next::cache_dir().expect("failed to get user's cache dir");
let home_dir = dirs_next::home_dir().expect("failed to get user's home dir");
// Try to find application's config file in the user's config dir.
// If not found, fallback to use `$HOME/.config` (for backward compability reason)
if !config_dir.join(DEFAULT_CONFIG_FILE).exists() {
config_dir = home_dir.join(".config");
}
(config_dir, cache_dir)
}
fn main() {
let (config_dir, cache_dir) = init_app_dirs();
let args = parse_args(&config_dir, &cache_dir);
init_logging(
args.value_of("log")
.expect("`log` argument should have a default value"),
);
config::load_config(
args.value_of("config")
.expect("`config` argument should have a default value"),
);
run();
}
|
