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
119
120
121
122
123
|
// yt - A fully featured command line YouTube client
//
// Copyright (C) 2024 Benedikt Peetz <benedikt.peetz@b-peetz.de>
// SPDX-License-Identifier: GPL-3.0-or-later
//
// This file is part of Yt.
//
// You should have received a copy of the License along with this program.
// If not, see <https://www.gnu.org/licenses/gpl-3.0.txt>.
use crate::config::{DownloadConfig, PathsConfig, SelectConfig, WatchConfig};
use super::{
default::{create_path, download, paths, select, update, watch},
Config, UpdateConfig,
};
use std::{fs::read_to_string, path::PathBuf};
use anyhow::{Context, Result};
use bytes::Bytes;
use toml::Table;
macro_rules! get {
($default:path, $config:expr, $get_fn:ident, $key_one:expr, $($keys:expr),*) => {
try_get!{@default $default, $config, $get_fn, $key_one, $($keys),*}
.with_context(|| format!("Failed to parse '{}' as a '{}'", stringify!($key_one), stringify!($get_fn)))?
};
(@path_if_none $config:expr, $option_default:expr, $default:path, $key_one:expr, $($keys:expr),*) => {
{
let maybe_download_dir =
try_get! {@option $config, as_str, $key_one, $($keys),*};
let down_dir = if let Some(dir) = maybe_download_dir {
PathBuf::from(dir)
} else {
if let Some(path) = $option_default {
path
} else {
$default()
.with_context(|| format!("Failed to get default path for: '{}.{}'", stringify!($key_one), stringify!($($keys),*)))?
}
};
create_path(down_dir)?
}
};
(@path $config:expr, $default:path, $key_one:expr, $($keys:expr),*) => {
get! {@path_if_none $config, None, $default, $key_one, $($keys),*}
};
}
macro_rules! try_get {
(@option $config:expr, $get_fn:ident, $key_one:expr, $($keys:expr),*) => {
$config.get($key_one).map(|val| {
try_get! {@option val, $get_fn, $($keys),*}
}).flatten().flatten()
};
(@option $config:expr, $get_fn:ident, $key_one:expr) => {
$config.get($key_one).map(|val| val.$get_fn())
};
(@default $default:path, $config:expr, $get_fn:ident, $key_one:expr, $($keys:expr),*) => {
if let Some(a) = $config.get($key_one) {
try_get! {@default $default, a, $get_fn, $($keys),*}
} else {
Some($default())
}
};
(@default $default:path, $config:expr, $get_fn:ident, $key_one:expr) => {
if let Some(a) = $config.get($key_one) {
a.$get_fn()
} else {
Some($default())
}
};
}
impl Config {
pub fn from_config_file(
db_path: Option<PathBuf>,
config_path: Option<PathBuf>,
) -> Result<Self> {
let config_file_path = config_path
.map(|val| Ok(val))
.unwrap_or_else(|| -> Result<_> { paths::config_path() })?;
let config: Table = read_to_string(config_file_path)?
.parse()
.context("Failed to parse the config file as toml")?;
Ok(Self {
select: SelectConfig {
playback_speed: get! {select::playback_speed, config, as_float, "select", "playback_speed"},
subtitle_langs:
get! {select::subtitle_langs, config, as_str, "select", "subtitle_langs"}
.to_owned(),
},
watch: WatchConfig {
local_comments_length: get! {watch::local_comments_length, config, as_integer, "watch", "local_comments_length"}
as usize,
},
update: UpdateConfig {
max_backlog: get! {update::max_backlog, config, as_integer, "update", "max_backlog"}
as u32,
},
paths: PathsConfig {
download_dir: get! {@path config, paths::download_dir, "paths", "download_dir"},
mpv_config_path: get! {@path config, paths::mpv_config_path, "paths", "mpv_config_path"},
mpv_input_path: get! {@path config, paths::mpv_input_path, "paths", "mpv_input_path"},
database_path: get! {@path_if_none config, db_path, paths::database_path, "paths", "database_path"},
last_selection_path: get! {@path config, paths::last_selection_path, "paths", "last_selection_path"},
},
download: DownloadConfig {
max_cache_size: {
let bytes_str = get! {download::max_cache_size, config, as_str, "download", "max_cache_path"};
let number: Bytes = bytes_str
.parse()
.context("Failed to parse max_cache_size")?;
number.as_u64()
},
},
})
}
}
|
