nil_core/
savedata.rs

1// Copyright (C) Call of Nil contributors
2// SPDX-License-Identifier: AGPL-3.0-only
3
4use crate::chat::Chat;
5use crate::continent::Continent;
6use crate::error::{AnyResult, Error, Result};
7use crate::military::Military;
8use crate::npc::bot::BotManager;
9use crate::npc::precursor::PrecursorManager;
10use crate::player::{Player, PlayerManager};
11use crate::ranking::Ranking;
12use crate::report::ReportManager;
13use crate::round::{Round, RoundId};
14use crate::world::config::{WorldConfig, WorldName};
15use crate::world::stats::WorldStats;
16use anyhow::bail;
17use flate2::Compression;
18use flate2::bufread::GzDecoder;
19use flate2::write::GzEncoder;
20use jiff::Zoned;
21use semver::Version;
22use serde::de::DeserializeOwned;
23use serde::{Deserialize, Serialize};
24use std::fmt;
25use std::io::{Read, Write};
26use tar::Archive;
27
28const MINIFY: bool = cfg!(any(
29  not(debug_assertions),
30  target_os = "android",
31  target_os = "ios"
32));
33
34type TarBuilder<'a> = tar::Builder<GzEncoder<&'a mut Vec<u8>>>;
35
36#[derive(Clone, Debug, Deserialize, Serialize)]
37#[serde(rename_all = "camelCase")]
38pub struct Savedata {
39  pub(crate) round: Round,
40  pub(crate) continent: Continent,
41  pub(crate) player_manager: PlayerManager,
42  pub(crate) bot_manager: BotManager,
43  pub(crate) precursor_manager: PrecursorManager,
44  pub(crate) military: Military,
45  pub(crate) ranking: Ranking,
46  pub(crate) report: ReportManager,
47  pub(crate) config: WorldConfig,
48  pub(crate) stats: WorldStats,
49  pub(crate) chat: Chat,
50}
51
52impl Savedata {
53  pub fn read(bytes: &[u8]) -> Result<Self> {
54    read_tar(bytes, "world")
55      .inspect_err(|err| tracing::error!(message = %err, error = ?err))
56      .map_err(|_| Error::FailedToReadSavedata)
57  }
58
59  pub(crate) fn write(self, buffer: &mut Vec<u8>) -> Result<()> {
60    write_tar(buffer, &self)
61      .inspect_err(|err| tracing::error!(message = %err, error = ?err))
62      .map_err(|_| Error::FailedToWriteSavedata)
63  }
64
65  pub fn players(&self) -> impl Iterator<Item = &Player> {
66    self.player_manager.players()
67  }
68}
69
70#[derive(Clone, Debug, Deserialize, Serialize)]
71#[serde(rename_all = "camelCase")]
72pub struct SavedataInfo {
73  world_name: WorldName,
74  round: RoundId,
75  version: Version,
76  saved_at: Zoned,
77}
78
79impl SavedataInfo {
80  fn new(data: &Savedata) -> AnyResult<Self> {
81    Ok(Self {
82      world_name: data.config.name(),
83      round: data.round.id(),
84      version: Version::parse(env!("CARGO_PKG_VERSION"))?,
85      saved_at: Zoned::now(),
86    })
87  }
88
89  pub fn read(bytes: &[u8]) -> Result<Self> {
90    read_tar(bytes, "info")
91      .inspect_err(|err| tracing::error!(message = %err, error = ?err))
92      .map_err(|_| Error::FailedToReadSavedata)
93  }
94}
95
96fn read_tar<T>(bytes: &[u8], entry_name: &str) -> AnyResult<T>
97where
98  T: DeserializeOwned,
99{
100  let decoder = GzDecoder::new(bytes);
101  let mut archive = Archive::new(decoder);
102
103  for entry in archive.entries()? {
104    let mut entry = entry?;
105    if let Ok(entry_path) = entry.path()
106      && let Some(path) = entry_path.to_str()
107      && path == entry_name
108    {
109      let size = entry.size().try_into()?;
110      let mut buffer = Vec::with_capacity(size);
111      entry.read_to_end(&mut buffer)?;
112      return Ok(serde_json::from_slice(&buffer)?);
113    }
114  }
115
116  bail!("Entry not found: {entry_name}");
117}
118
119fn write_tar(buffer: &mut Vec<u8>, data: &Savedata) -> AnyResult<()> {
120  let encoder = GzEncoder::new(buffer, Compression::best());
121  let mut tar_builder = TarBuilder::new(encoder);
122
123  let info = SavedataInfo::new(data)?;
124  append(&mut tar_builder, &info, "info")?;
125  append(&mut tar_builder, data, "world")?;
126
127  tar_builder.into_inner()?.finish()?.flush()?;
128
129  Ok(())
130}
131
132fn append<T>(builder: &mut TarBuilder, value: &T, path: &str) -> AnyResult<()>
133where
134  T: ?Sized + Serialize,
135{
136  let bytes = if MINIFY {
137    serde_json::to_vec(value)?
138  } else {
139    serde_json::to_vec_pretty(value)?
140  };
141
142  let mut header = tar::Header::new_gnu();
143  header.set_size(bytes.len().try_into()?);
144
145  builder.append_data(&mut header, path, bytes.as_slice())?;
146
147  Ok(())
148}
149
150pub struct SaveHandle(Box<dyn FnOnce(Vec<u8>) + Send + Sync>);
151
152impl SaveHandle {
153  pub fn new<F>(f: F) -> Self
154  where
155    F: FnOnce(Vec<u8>) + Send + Sync + 'static,
156  {
157    Self(Box::new(f))
158  }
159
160  #[inline]
161  pub fn save(self, data: Vec<u8>) {
162    (self.0)(data);
163  }
164}
165
166impl fmt::Debug for SaveHandle {
167  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
168    f.debug_tuple("SaveHandle")
169      .finish_non_exhaustive()
170  }
171}
nil_core/savedata.rs

nil_core/
savedata.rs
