dusk-rusk 1.6.0

Rusk is the Dusk Network node implementation
Documentation 
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

// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
//
// Copyright (c) DUSK NETWORK. All rights reserved.

use super::*;

use std::{env, fs, io};

use rusk_recovery_tools::Theme;
use rusk_recovery_tools::state::{deploy, restore_state, tar};
use tracing::info;

pub fn recovery_state(
    input: PathBuf,
    force: bool,
    output_file: Option<PathBuf>,
) -> Result<(), Box<dyn std::error::Error>> {
    let config = fs::read_to_string(input.as_path())
        .map_err(|_| format!("file {input:?} not found"))?;
    let init = toml::from_str(&config)?;

    let theme = Theme::default();
    info!("{} Network state", theme.action("Checking"));

    let _tmpdir = match output_file.clone() {
        Some(output) if output.exists() => Err("Output already exists")?,
        Some(_) => {
            let tmp_dir = tempfile::tempdir()?;
            // SAFETY: Called during single-threaded startup before
            // any threads that read this variable are spawned.
            unsafe { env::set_var("RUSK_STATE_PATH", tmp_dir.path()) };
            Some(tmp_dir)
        }
        None => None,
    };

    if force {
        clean_state()?;
    }

    let state_dir = rusk_profile::get_rusk_state_dir()?;

    let state_id_path = rusk_profile::to_rusk_state_id_path(&state_dir);

    // if the state already exists in the expected path, and it's valid, stop
    // early.
    if state_dir.exists() && state_id_path.exists() {
        info!("{} existing state", theme.info("Found"));

        let (_, commit_id) = restore_state(state_dir)?;
        info!(
            "{} state id at {}",
            theme.success("Checked"),
            state_id_path.display()
        );
        info!("{} {}", theme.action("Root"), hex::encode(commit_id));

        return Ok(());
    }

    info!("{} new state", theme.info("Building"));

    let (_, commit_id) =
        deploy(&state_dir, &init, *rusk::DUSK_CONSENSUS_KEY, |_| {})?;

    info!("{} {}", theme.action("Final Root"), hex::encode(commit_id));

    info!(
        "{} network state at {}",
        theme.success("Stored"),
        state_dir.display()
    );
    info!(
        "{} persisted id at {}",
        theme.success("Stored"),
        state_id_path.display()
    );

    if let Some(output) = output_file {
        let state_folder = rusk_profile::get_rusk_state_dir()?;
        info!(
            "{} state into {}",
            theme.info("Compressing"),
            output.display()
        );
        tar::archive(&state_folder, &output)?;
    }

    Ok(())
}

fn clean_state() -> Result<(), io::Error> {
    let state_path = rusk_profile::get_rusk_state_dir()?;

    fs::remove_dir_all(state_path).or_else(|e| {
        if e.kind() == io::ErrorKind::NotFound {
            Ok(())
        } else {
            Err(e)
        }
    })
}