dook 0.3.1

Look up definitions in your code
Documentation 
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

use crate::config;

#[derive(Copy, Clone, Debug, PartialEq, Eq, PartialOrd, Ord, clap::ValueEnum)]
pub enum DownloadsPolicy {
    Yes,
    Ask,
    No,
}

impl std::fmt::Display for DownloadsPolicy {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Yes => write!(f, "YES"),
            Self::Ask => write!(f, "ASK"),
            Self::No => write!(f, "NO"),
        }
    }
}

/// Get saved downloads policy setting from downloads_policy.txt.
pub fn get_downloads_policy() -> DownloadsPolicy {
    get_downloads_policy_from_path(downloads_policy_path().as_ref())
}

pub fn get_downloads_policy_from_path(path: Option<&std::path::PathBuf>) -> DownloadsPolicy {
    let Some(path) = path else {
        return DownloadsPolicy::Ask;
    };
    let Ok(bytes) = std::fs::read(path) else {
        return DownloadsPolicy::Ask;
    };
    let Ok(setting) = std::str::from_utf8(bytes.as_ref()) else {
        return DownloadsPolicy::Ask;
    };
    match setting
        .to_ascii_lowercase()
        .trim_matches(char::is_whitespace)
    {
        "yes" => DownloadsPolicy::Yes,
        "ask" => DownloadsPolicy::Ask,
        "no" => DownloadsPolicy::No,
        _ => DownloadsPolicy::Ask,
    }
}

pub fn downloads_policy_path() -> Option<std::path::PathBuf> {
    use etcetera::AppStrategy;
    config::app_dirs().ok().map(|d| d.config_dir().join("downloads_policy.txt"))
}

pub fn can_download(url: &str, policy: DownloadsPolicy) -> bool {
    use std::io::Write;
    match policy {
        DownloadsPolicy::Yes => true,
        DownloadsPolicy::No => false,
        DownloadsPolicy::Ask => {
            let mut stdout = console::Term::stdout();
            let response = write!(&mut stdout, "download from {url:?}? (y/n): ")
                .and_then(|_| stdout.flush())
                .and_then(|_| stdout.read_line());
            match response {
                Ok(response) => response.eq_ignore_ascii_case("y"),
                Err(_) => false,
            }
        }
    }
}