oku_fs/
config.rs

1use crate::fs::FS_PATH;
2use iroh_docs::NamespaceId;
3use log::error;
4use miette::{miette, IntoDiagnostic};
5use serde::{Deserialize, Serialize};
6use std::{
7    path::PathBuf,
8    sync::{Arc, LazyLock, Mutex},
9};
10
11pub(crate) static CONFIG_PATH: LazyLock<PathBuf> =
12    LazyLock::new(|| PathBuf::from(FS_PATH).join("config.toml"));
13
14#[derive(Clone, Debug, Serialize, Deserialize)]
15/// Configuration of an Oku file system node.
16pub struct OkuFsConfig {
17    /// The home replica of the node.
18    home_replica: Arc<Mutex<Option<NamespaceId>>>,
19}
20
21impl OkuFsConfig {
22    /// Loads the configuration of the file system from disk, or creates a new configuration if none exists.
23    ///
24    /// # Returns
25    ///
26    /// The configuration of the file system.
27    pub fn load_or_create_config() -> miette::Result<Self> {
28        let config_file_contents = std::fs::read_to_string(&*CONFIG_PATH);
29        match config_file_contents {
30            Ok(config_file_toml) => match toml::from_str(&config_file_toml) {
31                Ok(config) => Ok(config),
32                Err(e) => {
33                    error!("{}", e);
34                    let config = Self {
35                        home_replica: Arc::new(Mutex::new(None)),
36                    };
37                    Ok(config)
38                }
39            },
40            Err(e) => {
41                error!("{}", e);
42                let config = Self {
43                    home_replica: Arc::new(Mutex::new(None)),
44                };
45                let config_toml = toml::to_string_pretty(&config).into_diagnostic()?;
46                std::fs::write(&*CONFIG_PATH, config_toml).into_diagnostic()?;
47                Ok(config)
48            }
49        }
50    }
51
52    /// The home replica of the node.
53    pub fn home_replica(&self) -> miette::Result<Option<NamespaceId>> {
54        Ok(*self.home_replica.try_lock().map_err(|e| miette!("{}", e))?)
55    }
56
57    /// Sets the home replica of the node.
58    ///
59    /// # Arguments
60    ///
61    /// * `home_replica` - The home replica of the node.
62    pub fn set_home_replica(&self, home_replica: &Option<NamespaceId>) -> miette::Result<()> {
63        *self.home_replica.try_lock().map_err(|e| miette!("{}", e))? = *home_replica;
64        Ok(())
65    }
66
67    /// Writes the configuration to disk.
68    pub fn save(&self) -> miette::Result<()> {
69        let config_toml = toml::to_string_pretty(&self).into_diagnostic()?;
70        std::fs::write(&*CONFIG_PATH, config_toml).into_diagnostic()?;
71        Ok(())
72    }
73}
oku_fs/config.rs

oku_fs/
config.rs
