agent-file-tools 0.42.0

Agent File Tools — tree-sitter powered code analysis for AI agents
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

use std::sync::Arc;

use crate::context::{App, AppContext};

pub type ProjectRuntime = AppContext;

pub struct RuntimeRegistry {
    app: Arc<App>,
    single: ProjectRuntime,
}

impl RuntimeRegistry {
    pub fn standalone(app: Arc<App>, rt: ProjectRuntime) -> Self {
        Self { app, single: rt }
    }

    pub fn app(&self) -> Arc<App> {
        Arc::clone(&self.app)
    }

    pub fn current(&self) -> &ProjectRuntime {
        &self.single
    }

    pub fn current_mut(&mut self) -> &mut ProjectRuntime {
        &mut self.single
    }

    pub fn iter(&self) -> impl Iterator<Item = &ProjectRuntime> {
        std::iter::once(&self.single)
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::{config::Config, parser::TreeSitterProvider};

    #[test]
    fn standalone_current_and_iter_return_single_runtime() {
        let ctx = AppContext::new(Box::new(TreeSitterProvider::new()), Config::default());
        let app = ctx.app();
        let mut registry = RuntimeRegistry::standalone(Arc::clone(&app), ctx);
        assert!(Arc::ptr_eq(&app, &registry.app()));
        assert!(Arc::ptr_eq(&app, &registry.current().app()));

        let current_ptr = registry.current() as *const ProjectRuntime;
        let iter_ptrs = registry
            .iter()
            .map(|runtime| runtime as *const ProjectRuntime)
            .collect::<Vec<_>>();

        assert_eq!(iter_ptrs, vec![current_ptr]);

        let current_mut_ptr = registry.current_mut() as *mut ProjectRuntime;
        assert_eq!(current_mut_ptr as *const ProjectRuntime, current_ptr);
    }
}