entropy-kvdb 0.4.0

Encrypted key-value database for the Entropy Theshold Signing Server
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

// Copyright (C) 2023 Entropy Cryptography Inc.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

//! An encrypted key-value store used for storing keyshares and other private data
pub mod encrypted_sled;
pub mod kv_manager;
use std::{fs, path::PathBuf};

/// The intended environment entropy-tss is built for
#[derive(PartialEq)]
pub enum BuildType {
    /// Automated tests
    Test,
    /// A production-like test deployment without TDX
    ProductionNoTdx,
    /// A production deployment with TDX
    ProductionTdx,
}

/// Build the path used to store the key-value database
pub fn get_db_path(build_type: BuildType) -> String {
    let mut root: PathBuf = if build_type == BuildType::ProductionTdx {
        "/persist".into()
    } else {
        std::env::current_dir().expect("could not get home directory")
    };
    root.push(".entropy");
    if build_type == BuildType::Test {
        root.push("testing");
    } else {
        root.push("production");
    }
    root.push("db");
    let result: String = root.clone().display().to_string();
    fs::create_dir_all(root)
        .unwrap_or_else(|_| panic!("could not create database path at: {}", result.clone()));
    result
}

pub fn clean_tests() {
    let db_path = get_db_path(BuildType::Test);
    if fs::metadata(db_path.clone()).is_ok() {
        let _result = std::fs::remove_dir_all(db_path);
    }
}