cses-cli 0.1.4

CSES CLI is a lightweight tool for using CSES from the command line.
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

use std::fs;
use std::path::{Path, PathBuf};

use anyhow::Result;
use miniserde::{json, Deserialize, Serialize};

use crate::entities::Scope;

#[derive(Default, Serialize, Deserialize, Debug)]
pub struct StorageData {
    token: Option<String>,
    scope: Option<String>,
}

#[cfg(unix)]
fn create_path() -> Result<PathBuf> {
    let mut path = PathBuf::from(std::env::var("HOME")?);
    path.push(".config/cses-cli/filestorage.json");
    Ok(path)
}

#[cfg(windows)]
fn create_path() -> Result<PathBuf> {
    let mut path = PathBuf::from(std::env::var("APPDATA")?);
    path.push("cses-cli\\filestorage.json");
    Ok(path)
}

impl StorageData {
    pub fn get_token(&self) -> Option<&str> {
        self.token.as_deref()
    }
    pub fn get_scope(&self) -> Option<Scope> {
        self.scope.as_ref().and_then(|s| s.parse().ok())
    }
    pub fn set_token(&mut self, val: String) {
        self.token = Some(val);
    }
    pub fn set_scope(&mut self, val: Scope) {
        self.scope = Some(val.to_string());
    }
}

#[derive(Default, Debug)]
pub struct FileStorage {
    data: StorageData,
    path: PathBuf,
}

impl FileStorage {
    pub fn new(test: bool) -> Result<FileStorage> {
        let filename;
        if test {
            filename = PathBuf::from("filestorage.json");
        } else {
            filename = create_path()?;
            fs::create_dir_all(filename.parent().unwrap())?;
        }
        if !filename.exists() {
            return Ok(FileStorage {
                data: Default::default(),
                path: filename,
            });
        }
        let data = fs::read_to_string(&filename)?;
        let res: StorageData = json::from_str(&data)?;
        Ok(FileStorage {
            data: res,
            path: filename,
        })
    }
}

pub trait Storage {
    fn get(&self) -> &StorageData;
    fn get_mut(&mut self) -> &mut StorageData;
    fn save(&mut self) -> Result<()>;
    fn delete(&mut self) -> Result<()>;
    fn get_path(&self) -> &Path;
}

impl Storage for FileStorage {
    fn get(&self) -> &StorageData {
        &self.data
    }
    fn get_mut(&mut self) -> &mut StorageData {
        &mut self.data
    }
    fn save(&mut self) -> Result<()> {
        Ok(fs::write(&self.path, json::to_string(&self.data))?)
    }
    fn delete(&mut self) -> Result<()> {
        Ok(fs::remove_file(&self.path)?)
    }

    fn get_path(&self) -> &Path {
        &self.path
    }
}

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

    #[test]
    fn setters_and_getters_work() {
        let mut storage_data: StorageData = Default::default();
        storage_data.set_token(String::from("token"));
        storage_data.set_scope(Scope::Course(String::from("course")));
        assert_eq!(String::from("token"), storage_data.get_token().unwrap());
        assert_eq!(
            Scope::Course(String::from("course")),
            storage_data.get_scope().unwrap()
        );
    }
}