gr/cmds/
cache.rs

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
use crate::cli::cache::CacheOptions;
use crate::config::ConfigProperties;
use crate::error::GRError;
use crate::Result;
use std::fmt;
use std::sync::Arc;

pub fn execute(options: CacheOptions, config: Arc<dyn ConfigProperties>) -> Result<()> {
    match options {
        CacheOptions::Info => {
            let size = get_cache_directory_size(&config)?;
            println!("Location: {}", config.cache_location().unwrap_or("Not set"));
            println!("Size: {}", BytesToHumanReadable::from(size));
        }
    }
    Ok(())
}

struct BytesToHumanReadable(u64);

impl From<u64> for BytesToHumanReadable {
    fn from(size: u64) -> Self {
        BytesToHumanReadable(size)
    }
}

impl fmt::Display for BytesToHumanReadable {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        let size = self.0;
        let suffixes = ["B", "KB", "MB", "GB"];
        let suffix_len = suffixes.len();
        let mut size = size as f64;
        let mut i = 0;
        while size >= 1024.0 && i < suffix_len - 1 {
            size /= 1024.0;
            i += 1;
        }
        write!(f, "{:.2} {}", size, suffixes[i])
    }
}

fn get_cache_directory_size(config: &Arc<dyn ConfigProperties>) -> Result<u64> {
    if let Some(path) = config.cache_location() {
        let mut size = 0;
        for entry in std::fs::read_dir(path)? {
            let entry = entry?;
            let metadata = entry.metadata()?;
            size += metadata.len();
        }
        return Ok(size);
    }
    Err(GRError::ConfigurationNotFound.into())
}

#[cfg(test)]
mod test {
    use super::*;

    use crate::config::ConfigProperties;
    use std::fs::File;
    use std::io::Write;
    use tempfile::{tempdir, TempDir};

    #[test]
    fn test_bytes_display() {
        let test_table = vec![
            (0, "0.00 B"),
            (1024, "1.00 KB"),
            (1024 * 1024, "1.00 MB"),
            (1024 * 1024 * 1024, "1.00 GB"),
        ];
        for (size, expected) in test_table {
            let actual = BytesToHumanReadable::from(size).to_string();
            assert_eq!(expected, actual);
        }
    }

    struct ConfigMock {
        tmp_dir: String,
    }

    impl ConfigMock {
        fn new(tmp_dir: &TempDir) -> Self {
            Self {
                tmp_dir: tmp_dir.path().to_str().unwrap().to_string(),
            }
        }
    }

    impl ConfigProperties for ConfigMock {
        fn cache_location(&self) -> Option<&str> {
            Some(&self.tmp_dir)
        }

        fn api_token(&self) -> &str {
            todo!()
        }
    }

    #[test]
    fn test_get_size_of_cached_data() {
        let dir = tempdir().unwrap();
        let file_path = dir.path().join("test_file");
        let mut file = File::create(&file_path).unwrap();
        file.write_all(&[0; 10]).unwrap();
        let config: Arc<dyn ConfigProperties> = Arc::new(ConfigMock::new(&dir));
        let size = get_cache_directory_size(&config).unwrap();
        assert_eq!(size, 10);
    }
}