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
105
106
107
108
109
110
111
|
use std::path::{PathBuf, Path};
use std::fs;
use std::env;
use serde_json;
use serde_json::Value;
use crate::{
PlayerId,
RoomId,
savestate::SaveState,
playerstate::PlayerState,
Result
};
pub trait PersistentStorage {
fn load_room(&self, id: RoomId) -> Result<SaveState>;
fn load_player(&self, id: PlayerId) -> Result<PlayerState>;
fn save_room(&self, id: RoomId, state: SaveState) -> Result<()>;
fn save_player(&self, id: PlayerId, sate: PlayerState) -> Result<()>;
}
pub struct FileStorage {
directory: PathBuf
}
impl FileStorage {
pub fn new(path: PathBuf) -> Self {
Self {
directory: path
}
}
pub fn savedir() -> Option<PathBuf> {
if let Some(pathname) = env::var_os("ASCIIFARM_SAVE_DIR") {
Some(PathBuf::from(pathname))
} else if let Some(pathname) = env::var_os("XDG_DATA_HOME") {
let mut path = PathBuf::from(pathname);
path.push("asciifarm");
path.push("saves");
Some(path)
} else if let Some(pathname) = env::var_os("HOME") {
let mut path = PathBuf::from(pathname);
path.push(".asciifarm");
path.push("saves");
Some(path)
} else {
None
}
}
}
impl PersistentStorage for FileStorage {
fn load_room(&self, id: RoomId) -> Result<SaveState> {
let mut path = self.directory.clone();
path.push("rooms");
let fname = id.to_string() + ".save.json";
path.push(fname);
let text = fs::read_to_string(path)?;
let json: Value = serde_json::from_str(&text)?;
SaveState::from_json(&json)
}
fn load_player(&self, id: PlayerId) -> Result<PlayerState> {
let mut path = self.directory.clone();
path.push("players");
let fname = id.to_string() + ".save.json";
path.push(fname);
let text = fs::read_to_string(path)?;
let json: Value = serde_json::from_str(&text)?;
PlayerState::from_json(&json)
}
fn save_room(&self, id: RoomId, state: SaveState) -> Result<()> {
let mut path = self.directory.clone();
path.push("rooms");
fs::create_dir_all(&path)?;
let fname = id.to_string() + ".save.json";
path.push(fname);
let text = state.to_json().to_string();
// todo: write to a temp file first
write_file_safe(path, text)?;
Ok(())
}
fn save_player(&self, id: PlayerId, state: PlayerState) -> Result<()> {
let mut path = self.directory.clone();
path.push("players");
fs::create_dir_all(&path)?;
let fname = id.to_string() + ".save.json";
path.push(fname);
let text = state.to_json().to_string();
// todo: write to a temp file first
write_file_safe(path, text)?;
Ok(())
}
}
fn write_file_safe<P: AsRef<Path>, C: AsRef<[u8]>>(path: P, contents: C) -> Result<()> {
let temppath = path.as_ref().with_file_name(format!("tempfile_{}.tmp", rand::random::<u64>()));
fs::write(&temppath, contents)?;
fs::rename(&temppath, path)?;
Ok(())
}
|
