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

use crate::model::UserKey;
use dirs::home_dir;
use serde_json;
use std::fs;

const CONFIG_FILE_NAME: &'static str = "rfy_config.json";

pub fn get_user_key() -> Result<UserKey, Box<dyn std::error::Error + 'static>> {
    match home_dir() {
        Some(mut p) => {
            p.push(CONFIG_FILE_NAME);
            let res: String = String::from_utf8_lossy(&fs::read(p)?).parse()?;
            let parsed_struct: UserKey = serde_json::from_str(&res).unwrap_or_default();

            Ok(parsed_struct)
        }
        None => Ok(UserKey::default()),
    }
}

pub fn set_user_key(app_kv: Vec<String>) {
    let app_key = app_kv.get(0).unwrap().clone();
    let app_secure = app_kv.get(1).unwrap().clone();
    if let Some(mut p) = home_dir() {
        let s = UserKey {
            app_key,
            app_secure,
        };
        let json = serde_json::to_string(&s).unwrap();
        p.push(CONFIG_FILE_NAME);
        fs::write(p, json).unwrap();
        println!("设置成功, app_key: {}, app_secure: {}", s.app_key, s.app_secure);
    }
}