opencrabs 0.3.58

The autonomous, self-improving AI agent. Single Rust binary. Every channel. Install with: cargo install opencrabs
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

//! Service context and service manager.

use crate::db::Pool;
use std::sync::Arc;

use super::{FileService, MessageService, PlanService, ProjectService, SessionService};

/// Service context that holds shared resources
#[derive(Clone)]
pub struct ServiceContext {
    /// Database connection pool
    pub pool: Arc<Pool>,
}

impl ServiceContext {
    /// Create a new service context
    pub fn new(pool: Pool) -> Self {
        Self {
            pool: Arc::new(pool),
        }
    }

    /// Get a clone of the database pool (cheap operation)
    pub fn pool(&self) -> Pool {
        (*self.pool).clone()
    }
}

/// Service manager that holds all services
pub struct ServiceManager {
    context: ServiceContext,
    session_service: SessionService,
    message_service: MessageService,
    file_service: FileService,
    plan_service: PlanService,
    project_service: ProjectService,
}

impl ServiceManager {
    /// Create a new service manager
    pub fn new(pool: Pool) -> Self {
        let context = ServiceContext::new(pool);

        Self {
            session_service: SessionService::new(context.clone()),
            message_service: MessageService::new(context.clone()),
            file_service: FileService::new(context.clone()),
            plan_service: PlanService::new(context.clone()),
            project_service: ProjectService::new(context.clone()),
            context,
        }
    }

    /// Get the session service
    pub fn sessions(&self) -> &SessionService {
        &self.session_service
    }

    /// Get the message service
    pub fn messages(&self) -> &MessageService {
        &self.message_service
    }

    /// Get the file service
    pub fn files(&self) -> &FileService {
        &self.file_service
    }

    /// Get the plan service
    pub fn plans(&self) -> &PlanService {
        &self.plan_service
    }

    /// Get the project service
    pub fn projects(&self) -> &ProjectService {
        &self.project_service
    }

    /// Get the service context
    pub fn context(&self) -> &ServiceContext {
        &self.context
    }
}