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

// Copyright 2019 Conflux Foundation. All rights reserved.
// Conflux is free software and distributed under GNU General Public License.
// See http://www.gnu.org/licenses/

use app_dirs::{get_app_root, AppDataType, AppInfo};
use cfxcore_accounts::{AccountProvider, AccountProviderSettings};
use cfxstore::{accounts_dir::RootDiskDirectory, CfxStore};
use dir::helpers::replace_home;
use std::{path::PathBuf, time::Duration};

pub fn account_provider(
    dir: Option<String>, sstore_iterations: Option<u32>,
    refresh_time: Option<Duration>,
) -> Result<AccountProvider, String> {
    let dir = match dir {
        Some(dir) => dir,
        None => keys_path(),
    };

    let dir = Box::new(keys_dir(dir)?);
    let secret_store = Box::new(secret_store(dir, sstore_iterations)?);

    if let Some(t) = refresh_time {
        secret_store.set_refresh_time(t);
    }

    Ok(AccountProvider::new(
        secret_store,
        AccountProviderSettings::default(),
    ))
}

pub fn keys_dir(path: String) -> Result<RootDiskDirectory, String> {
    let path = PathBuf::from(&path);
    RootDiskDirectory::create(path)
        .map_err(|e| format!("Could not open keys directory: {}", e))
}

fn secret_store(
    dir: Box<RootDiskDirectory>, iterations: Option<u32>,
) -> Result<CfxStore, String> {
    match iterations {
        Some(i) => CfxStore::open_with_iterations(dir, i),
        None => CfxStore::open(dir),
    }
    .map_err(|e| format!("Could not open keys store: {}", e))
}

/// Default data path
fn default_data_path() -> String {
    let app_info = AppInfo {
        name: "Conflux",
        author: "conflux",
    };
    get_app_root(AppDataType::UserData, &app_info)
        .map(|p| p.to_string_lossy().into_owned())
        .unwrap_or_else(|_| "$HOME/.conflux".to_owned())
}

pub fn keys_path() -> String {
    let data_path = default_data_path();
    replace_home(&data_path, "$BASE/keys")
}