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

use crate::jormungandr::TestingDirectory;
use fs_extra::dir::{move_dir, CopyOptions};
use std::thread::panicking;

pub fn persist_dir_on_panic<S1: AsRef<str>, S2: AsRef<str>>(
    temp_dir: Option<TestingDirectory>,
    additional_contents: Vec<(S1, S2)>,
) {
    if panicking() {
        let logs_dir = match tempfile::Builder::new().prefix("jormungandr_").tempdir() {
            Ok(dir) => dir.into_path(),
            Err(e) => {
                eprintln!("Could not create logs dir: {}", e);
                return;
            }
        };

        println!(
            "persisting node temp_dir after panic: {}",
            logs_dir.display()
        );

        if let Some(dir) = temp_dir {
            let options = CopyOptions {
                content_only: true,
                ..Default::default()
            };
            move_dir(dir.path(), &logs_dir, &options)
                .map(|_| ())
                .unwrap_or_else(|e| eprintln!("Could not move files to new dir: {}", e));
        }

        for (filename, content) in additional_contents {
            std::fs::write(logs_dir.join(filename.as_ref()), content.as_ref()).unwrap_or_else(
                |e| eprint!("Could not write {} to disk: {}", filename.as_ref(), e),
            );
        }
    }
}