1use std::path::PathBuf;
2
3use anyhow::Context;
4use serde::{Deserialize, Serialize};
5
6const CONFIG_DIR_NAME: &str = "gitdot";
7const CONFIG_FILE_NAME: &str = "config.toml";
8
9#[derive(Debug, Clone, Serialize, Deserialize, Default)]
10pub struct Config {
11    #[serde(default)]
12    pub user_name: String,
13
14    #[serde(default)]
15    pub user_email: String,
16}
17
18pub enum AuthStatus {
19    LoggedIn { user_name: String },
20    LoggedOut,
21}
22
23impl Config {
24    pub async fn load() -> anyhow::Result<Self> {
25        let config_path = Self::get_config_path()?;
26
27        if !config_path.exists() {
28            return Ok(Self::default());
29        }
30
31        let contents = tokio::fs::read_to_string(&config_path)
32            .await
33            .with_context(|| format!("Failed to read config file: {}", config_path.display()))?;
34
35        let config: Config = toml::from_str(&contents)
36            .with_context(|| format!("Failed to parse config file: {}", config_path.display()))?;
37
38        Ok(config)
39    }
40
41    pub async fn save(&self) -> anyhow::Result<()> {
42        let config_path = Self::get_config_path()?;
43
44        if let Some(parent) = config_path.parent() {
45            tokio::fs::create_dir_all(parent).await.with_context(|| {
46                format!("Failed to create config directory: {}", parent.display())
47            })?;
48        }
49
50        let contents = toml::to_string_pretty(self).context("Failed to serialize config")?;
51
52        tokio::fs::write(&config_path, contents)
53            .await
54            .with_context(|| format!("Failed to write config file: {}", config_path.display()))?;
55
56        Ok(())
57    }
58
59    pub fn get_auth_status(&self) -> AuthStatus {
60        if self.user_name.is_empty() {
61            AuthStatus::LoggedOut
62        } else {
63            AuthStatus::LoggedIn {
64                user_name: self.user_name.clone(),
65            }
66        }
67    }
68
69    fn get_config_path() -> anyhow::Result<PathBuf> {
70        let base_dir = dirs_next::config_dir().context("Could not determine config directory")?;
71        let config_dir = base_dir.join(CONFIG_DIR_NAME);
72        Ok(config_dir.join(CONFIG_FILE_NAME))
73    }
74}
75