agentdir 0.1.5

Virtual filesystem for agent-optimized file exploration using CoW reflinks
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
132
133
134
135
136
137
138
139
140
141
142
143
144
145

use std::sync::Arc;
use std::time::Duration;

use tokio::sync::mpsc;
use tokio::time;

use crate::backend::{Backend, SourceEvent, WatchHandle};
use crate::error::Result;
use crate::types::SourcePath;

pub struct FileWatcher {
    backend: Arc<dyn Backend>,
    roots: Vec<SourcePath>,
    poll_interval: Duration,
}

impl FileWatcher {
    pub fn new(backend: Arc<dyn Backend>, roots: Vec<SourcePath>) -> Self {
        Self {
            backend,
            roots,
            poll_interval: Duration::from_secs(60),
        }
    }

    pub fn with_poll_interval(mut self, interval: Duration) -> Self {
        self.poll_interval = interval;
        self
    }

    pub async fn start(&self) -> Result<(mpsc::Receiver<SourceEvent>, WatchHandle)> {
        let (tx, rx) = mpsc::channel(256);

        let handle = self.backend.watch(&self.roots, tx.clone()).await?;

        let poll_tx = tx.clone();
        let poll_interval = self.poll_interval;
        let cancel = handle.cancel_token();

        tokio::spawn(async move {
            let mut interval = time::interval(poll_interval);
            interval.tick().await;

            loop {
                tokio::select! {
                    _ = interval.tick() => {
                        let _ = poll_tx.send(SourceEvent::RescanNeeded).await;
                    }
                    _ = cancel.cancelled() => {
                        break;
                    }
                }
            }
        });

        Ok((rx, handle))
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::backend::local::LocalBackend;
    use std::time::Duration;
    use tempfile::TempDir;
    use tokio::time::timeout;

    #[tokio::test]
    async fn test_detect_file_creation() {
        let dir = TempDir::new().unwrap();
        let backend: Arc<dyn Backend> = Arc::new(LocalBackend);
        let roots = vec![SourcePath::new(dir.path().to_path_buf())];

        let watcher = FileWatcher::new(backend, roots);
        let (mut rx, _handle) = watcher.start().await.unwrap();

        tokio::time::sleep(Duration::from_millis(200)).await;

        std::fs::write(dir.path().join("newfile.txt"), b"hello").unwrap();

        let event = timeout(Duration::from_secs(5), rx.recv()).await;
        assert!(event.is_ok(), "Timed out waiting for file creation event");
        assert!(event.unwrap().is_some());
    }

    #[tokio::test]
    async fn test_start_waits_until_roots_are_watched() {
        let dir = TempDir::new().unwrap();
        let backend: Arc<dyn Backend> = Arc::new(LocalBackend);
        let roots = vec![SourcePath::new(dir.path().to_path_buf())];

        let watcher = FileWatcher::new(backend, roots);
        let (mut rx, _handle) = watcher.start().await.unwrap();

        std::fs::write(dir.path().join("immediate.txt"), b"hello").unwrap();

        let event = timeout(Duration::from_secs(5), rx.recv()).await;
        assert!(
            event.is_ok(),
            "watcher missed a file created immediately after start"
        );
        assert!(event.unwrap().is_some());
    }

    #[tokio::test]
    async fn test_watcher_cleanup() {
        let dir = TempDir::new().unwrap();
        let backend: Arc<dyn Backend> = Arc::new(LocalBackend);
        let roots = vec![SourcePath::new(dir.path().to_path_buf())];

        let watcher = FileWatcher::new(backend, roots);
        let (_rx, handle) = watcher.start().await.unwrap();

        drop(handle);
    }

    #[tokio::test]
    async fn test_periodic_polling_emits_rescan() {
        let dir = TempDir::new().unwrap();
        let backend: Arc<dyn Backend> = Arc::new(LocalBackend);
        let roots = vec![SourcePath::new(dir.path().to_path_buf())];

        let watcher =
            FileWatcher::new(backend, roots).with_poll_interval(Duration::from_millis(200));
        let (mut rx, _handle) = watcher.start().await.unwrap();

        let found_rescan = timeout(Duration::from_secs(3), async {
            loop {
                if let Some(event) = rx.recv().await {
                    if matches!(event, SourceEvent::RescanNeeded) {
                        return true;
                    }
                } else {
                    return false;
                }
            }
        })
        .await;

        assert!(
            found_rescan.is_ok() && found_rescan.unwrap(),
            "Expected RescanNeeded event from periodic polling"
        );
    }
}