webfind 0.5.0

Self-hosted native search engine for AI agents — pure CLI, zero cost
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

use std::path::PathBuf;
use std::sync::Arc;

use tokio::sync::RwLock;

use crate::api::ApiState;
use crate::engine::categories::CategoryService;
use crate::engine::crawl_graph::CrawlGraphStore;
use crate::engine::query_log::QueryLogService;
use crate::engine::search_engine::SearchEngine;

/// Shared state for the GUI server.
#[derive(Clone)]
pub struct GuiState {
    pub indexer: Arc<RwLock<Arc<dyn SearchEngine + Send + Sync>>>,
    pub graph_store: Option<Arc<dyn CrawlGraphStore + Send + Sync>>,
    pub data_dir: PathBuf,
    pub query_log: Option<Arc<QueryLogService>>,
    pub categories: Option<Arc<CategoryService>>,
}

impl GuiState {
    pub fn new(
        indexer: Arc<dyn SearchEngine + Send + Sync>,
        graph_store: Option<Arc<dyn CrawlGraphStore + Send + Sync>>,
        data_dir: PathBuf,
    ) -> Self {
        Self {
            indexer: Arc::new(RwLock::new(indexer)),
            graph_store,
            data_dir,
            query_log: None,
            categories: None,
        }
    }

    pub fn from_api_state(state: Arc<ApiState>) -> Self {
        Self {
            indexer: state.indexer(),
            graph_store: state.graph_store(),
            data_dir: state.data_dir().clone(),
            query_log: state.query_log(),
            categories: state.categories(),
        }
    }
}