rspin 0.1.0

A terminal-based slot machine simulator with animated reels
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

use std::{error::Error, fs, path::PathBuf};

use directories::ProjectDirs;
use serde::{Deserialize, Serialize};

#[derive(Deserialize, Serialize, Debug)]
pub struct Database {
    pub balance: i32,
    pub total_spins: i32,
}

// TODO add a display command to use for `rspin stats`
impl Default for Database {
    fn default() -> Self {
        Self {
            balance: 1000,
            total_spins: 0,
        }
    }
}

impl Database {
    pub fn load() -> Result<Self, Box<dyn Error>> {
        let path = Self::get_path()?;

        if !path.exists() {
            let default_data = Self::default();
            default_data.save()?;
            return Ok(default_data);
        }

        let file_str = fs::read_to_string(path)?;
        let data: Self = toml::from_str(&file_str)?;
        Ok(data)
    }

    pub fn save(&self) -> Result<(), Box<dyn Error>> {
        let path = Self::get_path()?;
        let toml_str = toml::to_string(self)?;
        fs::write(path, toml_str)?;
        Ok(())
    }

    pub fn get_path() -> Result<PathBuf, Box<dyn Error>> {
        let proj_dirs = ProjectDirs::from("com", "rspin-database", "rspin")
            .ok_or("Could not determine home directory")?;

        let data_dir = proj_dirs.data_dir();
        if !data_dir.exists() {
            fs::create_dir_all(data_dir)?;
        }

        Ok(data_dir.join("state.toml"))
    }
}