cotp 1.9.9

Trustworthy, encrypted, command-line TOTP/HOTP authenticator app with import functionality.
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

use dirs::{data_dir, home_dir};
use std::path::PathBuf;
use std::sync::OnceLock;
use std::{env, fs};

use crate::arguments::CotpArgs;

const CURRENT_DB_PATH: &str = "./db.cotp";
const XDG_PATH: &str = "cotp/db.cotp";
const HOME_PATH: &str = ".cotp/db.cotp";

pub static DATABASE_PATH: OnceLock<PathBuf> = OnceLock::new();

/// Initialize singleton database path
pub fn init_path(args: &CotpArgs) -> PathBuf {
    DATABASE_PATH
        .get_or_init(|| {
            args.database_path
                .as_ref()
                .map(String::from)
                .or(env::var("COTP_DB_PATH").ok())
                .map_or_else(get_default_db_path, PathBuf::from)
        })
        .to_owned()
}

// Pushing an absolute path to a PathBuf replaces the entire PathBuf: https://doc.rust-lang.org/std/path/struct.PathBuf.html#method.push
fn get_default_db_path() -> PathBuf {
    // If db.cotp is present in the current directory or we are using a debug artifact, do not use the one in home dir
    // First condition is optimized away in release mode
    let portable_path: PathBuf = PathBuf::from(CURRENT_DB_PATH);
    if cfg!(debug_assertions) || portable_path.exists() {
        return portable_path;
    }

    let home_path = home_dir().map(|path| path.join(HOME_PATH));

    data_dir()
        .map(|p| p.join(XDG_PATH))
        .inspect(|xdg| {
            if !xdg.exists()
                && let Some(home) = &home_path
                && home.exists()
            {
                fs::create_dir_all(xdg.parent().unwrap()).expect("Failed to create dir");
                fs::copy(home, xdg.as_path())
                    .expect("Failed on copy from legacy dir to XDG_DATA_HOME");
            }
        })
        .or(home_path)
        .unwrap_or(portable_path)
}