canic_host/install_root/
state.rs1use crate::release_set::icp_root;
2use serde::{Deserialize, Serialize};
3use std::{fs, path::Path, path::PathBuf};
4
5pub(super) const INSTALL_STATE_SCHEMA_VERSION: u32 = 1;
6
7#[derive(Clone, Debug, Deserialize, Eq, PartialEq, Serialize)]
12pub struct InstallState {
13 pub schema_version: u32,
14 pub fleet: String,
15 pub installed_at_unix_secs: u64,
16 pub network: String,
17 pub root_target: String,
18 pub root_canister_id: String,
19 pub root_build_target: String,
20 pub workspace_root: String,
21 pub icp_root: String,
22 pub config_path: String,
23 pub release_set_manifest_path: String,
24}
25
26pub(super) fn read_fleet_install_state(
28 icp_root: &Path,
29 network: &str,
30 fleet: &str,
31) -> Result<Option<InstallState>, Box<dyn std::error::Error>> {
32 validate_network_name(network)?;
33 validate_fleet_name(fleet)?;
34 let path = fleet_install_state_path(icp_root, network, fleet);
35 if !path.is_file() {
36 return Ok(None);
37 }
38
39 let bytes = fs::read(&path)?;
40 let state: InstallState = serde_json::from_slice(&bytes)?;
41 Ok(Some(state))
42}
43
44pub fn read_named_fleet_install_state(
46 network: &str,
47 fleet: &str,
48) -> Result<Option<InstallState>, Box<dyn std::error::Error>> {
49 let icp_root = icp_root()?;
50 read_fleet_install_state(&icp_root, network, fleet)
51}
52
53#[must_use]
55pub(super) fn fleet_install_state_path(icp_root: &Path, network: &str, fleet: &str) -> PathBuf {
56 fleets_dir(icp_root, network).join(format!("{fleet}.json"))
57}
58
59fn fleets_dir(icp_root: &Path, network: &str) -> PathBuf {
61 icp_root.join(".canic").join(network).join("fleets")
62}
63
64pub(super) fn write_install_state(
66 icp_root: &Path,
67 network: &str,
68 state: &InstallState,
69) -> Result<PathBuf, Box<dyn std::error::Error>> {
70 validate_network_name(network)?;
71 validate_fleet_name(&state.fleet)?;
72 let path = fleet_install_state_path(icp_root, network, &state.fleet);
73 if let Some(parent) = path.parent() {
74 fs::create_dir_all(parent)?;
75 }
76 fs::write(&path, serde_json::to_vec_pretty(state)?)?;
77 Ok(path)
78}
79
80pub(super) fn validate_fleet_name(name: &str) -> Result<(), Box<dyn std::error::Error>> {
82 let valid = !name.is_empty()
83 && name
84 .bytes()
85 .all(|byte| byte.is_ascii_alphanumeric() || matches!(byte, b'-' | b'_'));
86 if valid {
87 Ok(())
88 } else {
89 Err(format!("invalid fleet name {name:?}; use letters, numbers, '-' or '_'").into())
90 }
91}
92
93fn validate_network_name(name: &str) -> Result<(), Box<dyn std::error::Error>> {
95 let valid = !name.is_empty()
96 && name
97 .bytes()
98 .all(|byte| byte.is_ascii_alphanumeric() || matches!(byte, b'-' | b'_'));
99 if valid {
100 Ok(())
101 } else {
102 Err(format!("invalid network name {name:?}; use letters, numbers, '-' or '_'").into())
103 }
104}