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
|
use directories::ProjectDirs;
use serde::{Deserialize, Serialize};
use std::fs;
use std::fs::File;
use std::path::PathBuf;
use crate::error::{Error, Result};
#[derive(Deserialize, Serialize, Debug)]
pub struct Config {
pub api_key: Option<String>,
pub limit: u16,
pub site: String,
}
impl Default for Config {
fn default() -> Self {
Config {
api_key: None,
limit: 20,
site: String::from("stackoverflow"),
}
}
}
/// Get user config (writes default if none found)
pub fn user_config() -> Result<Config> {
let project = project_dir()?;
let dir = project.config_dir();
fs::create_dir_all(&dir).map_err(|_| Error::create_dir(&dir.to_path_buf()))?;
let filename = config_file_name()?;
match File::open(&filename) {
Err(_) => {
let def = Config::default();
write_config(&def)?;
Ok(def)
}
Ok(file) => serde_yaml::from_reader(file).map_err(|_| Error::malformed(&filename)),
}
}
fn write_config(config: &Config) -> Result<()> {
let filename = config_file_name()?;
let file = File::create(&filename).map_err(|_| Error::create_file(&filename))?;
serde_yaml::to_writer(file, config).map_err(|_| Error::write_file(&filename))
}
fn config_file_name() -> Result<PathBuf> {
Ok(project_dir()?.config_dir().join("config.yml"))
}
/// Get project directory
pub fn project_dir() -> Result<ProjectDirs> {
ProjectDirs::from("io", "Sam Tay", "so").ok_or_else(|| {
Error::os(
"Couldn't find a suitable project directory to store cache and configuration;\n\
this application may not be supported on your operating system.",
)
})
}
pub fn set_api_key(key: String) -> Result<()> {
let mut cfg = user_config()?;
cfg.api_key = Some(key);
write_config(&cfg)
}
|
