stama 1.0.2

A terminal user interface for monitoring and managing slurm jobs.
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

use color_eyre::eyre::{self, Result};
use serde::{Deserialize, Serialize};
use std::fs::{self, File};
use std::io::prelude::*;

#[derive(Debug, Deserialize, Serialize, Clone)]
pub struct UserOptions {
    pub refresh_rate: usize,       // Refresh rate in milliseconds
    pub show_completed_jobs: bool, // Show completed jobs
    pub confirm_before_quit: bool, // Confirm before quitting
    pub confirm_before_kill: bool, // Confirm before killing a job
    pub external_editor: String,   // External editor command (e.g. "vim")
}

impl Default for UserOptions {
    fn default() -> Self {
        Self {
            refresh_rate: 250,
            show_completed_jobs: true,
            confirm_before_quit: false,
            confirm_before_kill: true,
            external_editor: "vim".to_string(),
        }
    }
}

// ====================================================================
//  LOADING AND SAVING
// ====================================================================

impl UserOptions {
    pub fn load_from_file() -> Result<Self> {
        let file_dir = get_file_dir()?;
        let file_path = get_file_path(&file_dir);

        let contents = fs::read_to_string(file_path)?;
        let user_options = toml::from_str(&contents)?;
        Ok(user_options)
    }

    pub fn load() -> Self {
        if !file_exists() {
            return Self::default();
        }
        match Self::load_from_file() {
            Ok(user_options) => user_options,
            Err(_) => Self::default(),
        }
    }

    pub fn save(&self) {
        match self.try_save() {
            Ok(_) => (),
            Err(_) => (),
        }
    }

    pub fn try_save(&self) -> Result<()> {
        let file_dir = get_file_dir()?;
        touch_dir(&file_dir)?;
        let file_path = get_file_path(&file_dir);

        let toml = toml::to_string(self)?;
        let mut file = File::create(file_path)?;
        file.write_all(toml.as_bytes())?;
        Ok(())
    }
}

fn get_file_dir() -> Result<String> {
    let home = std::env::var("HOME");
    let home = match home {
        Ok(path) => path,
        Err(_) => return Err(eyre::eyre!("Could not find HOME environment variable")),
    };
    Ok(format!("{}/.config/stama", home))
}

fn get_file_path(file_dir: &str) -> String {
    format!("{}/config.toml", file_dir)
}

fn file_exists() -> bool {
    let file_dir = match get_file_dir() {
        Ok(file_dir) => file_dir,
        Err(_) => return false,
    };
    let file_path = get_file_path(&file_dir);
    std::path::Path::new(&file_path).exists()
}

fn touch_dir(file_dir: &str) -> Result<()> {
    // create the directory
    match fs::create_dir_all(file_dir) {
        Ok(_) => Ok(()),
        Err(_) => Err(eyre::eyre!("Could not create directory")),
    }
}