agent-diva-core 0.4.10

Core types and traits for agent-diva
Documentation 
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131

//! Soul state helpers.

use chrono::{DateTime, Utc};
use serde::{Deserialize, Serialize};
use std::path::{Path, PathBuf};

/// Runtime state for soul/bootstrap lifecycle.
#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct SoulState {
    /// Timestamp when bootstrap was first seeded.
    pub bootstrap_seeded_at: Option<DateTime<Utc>>,
    /// Timestamp when bootstrap was marked as completed.
    pub bootstrap_completed_at: Option<DateTime<Utc>>,
}

/// Small persistence helper for soul lifecycle state.
pub struct SoulStateStore {
    path: PathBuf,
}

impl SoulStateStore {
    /// Create a state store under `<workspace>/.agent-diva/soul-state.json`.
    pub fn new(workspace: impl AsRef<Path>) -> Self {
        let path = workspace
            .as_ref()
            .join(".agent-diva")
            .join("soul-state.json");
        Self { path }
    }

    /// Read state from disk. Missing file returns defaults.
    pub fn load(&self) -> std::io::Result<SoulState> {
        if !self.path.exists() {
            return Ok(SoulState::default());
        }

        let raw = std::fs::read_to_string(&self.path)?;
        let state: SoulState = serde_json::from_str(&raw).unwrap_or_default();
        Ok(state)
    }

    /// Persist state to disk.
    pub fn save(&self, state: &SoulState) -> std::io::Result<()> {
        if let Some(parent) = self.path.parent() {
            std::fs::create_dir_all(parent)?;
        }
        let content = serde_json::to_string_pretty(state)?;
        std::fs::write(&self.path, content)
    }

    /// Returns true if bootstrap completion was already recorded.
    pub fn is_bootstrap_completed(&self) -> bool {
        self.load()
            .ok()
            .and_then(|state| state.bootstrap_completed_at)
            .is_some()
    }

    /// Mark bootstrap as seeded once.
    pub fn mark_bootstrap_seeded(&self) -> std::io::Result<()> {
        let mut state = self.load().unwrap_or_default();
        if state.bootstrap_seeded_at.is_none() {
            state.bootstrap_seeded_at = Some(Utc::now());
            self.save(&state)?;
        }
        Ok(())
    }

    /// Mark bootstrap as completed.
    pub fn mark_bootstrap_completed(&self) -> std::io::Result<()> {
        let mut state = self.load().unwrap_or_default();
        if state.bootstrap_seeded_at.is_none() {
            state.bootstrap_seeded_at = Some(Utc::now());
        }
        if state.bootstrap_completed_at.is_none() {
            state.bootstrap_completed_at = Some(Utc::now());
            self.save(&state)?;
        }
        Ok(())
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_load_missing_returns_default() {
        let temp = tempfile::tempdir().unwrap();
        let store = SoulStateStore::new(temp.path());
        let state = store.load().unwrap();
        assert!(state.bootstrap_seeded_at.is_none());
        assert!(state.bootstrap_completed_at.is_none());
    }

    #[test]
    fn test_mark_bootstrap_seeded_writes_once() {
        let temp = tempfile::tempdir().unwrap();
        let store = SoulStateStore::new(temp.path());

        store.mark_bootstrap_seeded().unwrap();
        let first = store.load().unwrap();
        assert!(first.bootstrap_seeded_at.is_some());

        store.mark_bootstrap_seeded().unwrap();
        let second = store.load().unwrap();
        assert_eq!(first.bootstrap_seeded_at, second.bootstrap_seeded_at);
    }

    #[test]
    fn test_is_bootstrap_completed() {
        let temp = tempfile::tempdir().unwrap();
        let store = SoulStateStore::new(temp.path());
        assert!(!store.is_bootstrap_completed());

        let mut state = SoulState::default();
        state.bootstrap_completed_at = Some(Utc::now());
        store.save(&state).unwrap();
        assert!(store.is_bootstrap_completed());
    }

    #[test]
    fn test_mark_bootstrap_completed_sets_completed_at() {
        let temp = tempfile::tempdir().unwrap();
        let store = SoulStateStore::new(temp.path());
        store.mark_bootstrap_completed().unwrap();
        let state = store.load().unwrap();
        assert!(state.bootstrap_seeded_at.is_some());
        assert!(state.bootstrap_completed_at.is_some());
    }
}