Skald-Circle/src/core/skald/bundles.rs

//! Domain bundles: the managers, grouped by cohesion, that make up `Skald`.
//!
//! Each bundle owns a `build()` that constructs its managers (plus their startup
//! logging and non-fatal `seed_*` calls) from the shared [`Runtime`] and whatever
//! sibling bundles it depends on at construction time. Cross-bundle *cycles* are
//! not expressed here — they are resolved by the managers' `OnceLock` setters,
//! called in one place by [`super::wiring::wire`]. Bundle structs never hold
//! references to each other.

use std::sync::Arc;

use anyhow::Result;
use tracing::{debug, info, warn};

use core_api::remote::RemoteAccess;

use crate::core::approval::ApprovalManager;
use crate::core::chat_hub::ChatHub;
use crate::core::clarification::ClarificationManager;
use crate::core::command::LlmCommandManager;
use crate::core::compactor::ContextCompactor;
use crate::core::config::{CoreConfig, DatetimeConfig};
use crate::core::cron::TaskManager;
use crate::core::elicitation::ElicitationManager;
use crate::core::image_generate::ImageGeneratorManager;
use crate::core::inbox::Inbox;
use crate::core::latex::LatexCompiler;
use crate::core::llm::LlmManager;
use crate::core::location::LocationManager;
use crate::core::mcp::McpManager;
use crate::core::memory::MemoryManager;
use crate::core::plugin::PluginManager;
use crate::core::projects::tickets::ProjectTicketManager;
use crate::core::projects::ProjectManager;
use crate::core::provider::ProviderRegistry;
use crate::core::run_context::RunContextManager;
use crate::core::secrets::SecretsStore;
use crate::core::session::handler::{DEFAULT_MAX_PARALLEL_SUBAGENTS, DEFAULT_MAX_TOOL_ROUNDS};
use crate::core::session::manager::ChatSessionManager;
use crate::core::tic::TicManager;
use crate::core::tool_catalog::ToolCatalog;
use crate::core::tool_discovery::ToolDiscovery;
use crate::core::tools::ToolRegistry;
use crate::core::transcribe::TranscribeManager;
use crate::core::tts::TtsManager;

use tokio::sync::RwLock;

use core_api::plugin::Plugin;

use super::runtime::Runtime;

// ── Models: LLM/provider stack ──────────────────────────────────────────────

pub(super) struct Models {
    pub(super) provider_registry: Arc<ProviderRegistry>,
    pub(super) llm_manager:       Arc<LlmManager>,
    pub(super) secrets:           Arc<SecretsStore>,
    pub(super) memory_manager:    Arc<MemoryManager>,
}

impl Models {
    pub(super) async fn build(rt: &Runtime, config: &CoreConfig) -> Result<Self> {
        let mut provider_registry = ProviderRegistry::new(Arc::clone(&rt.system_bus));
        provider_registry.register_builtin(crate::core::llm::providers::openai::OpenAiProvider);
        provider_registry.register_builtin(crate::core::llm::providers::anthropic::AnthropicProvider::new());
        provider_registry.register_builtin(crate::core::llm::providers::openrouter::OpenRouterProvider::new());
        provider_registry.register_builtin(crate::core::llm::providers::ollama::OllamaProvider::new());
        provider_registry.register_builtin(crate::core::llm::providers::lm_studio::LmStudioProvider::new());
        provider_registry.register_builtin(crate::core::llm::providers::deepseek::DeepSeekProvider::new());
        provider_registry.register_builtin(crate::core::llm::providers::zai::ZaiProvider::new());
        let provider_registry = Arc::new(provider_registry);
        info!("provider registry ready ({} built-in providers)", provider_registry.all().len());

        let log_flags = config.llm.requests_log.as_ref().filter(|r| r.enabled).map(|r| {
            use crate::core::chatbot::logging::LogSaveFlags;
            LogSaveFlags {
                request_payload:  r.request_payload_save,
                response_payload: r.response_payload_save,
                request_headers:  r.request_header_save,
                response_headers: r.response_header_save,
            }
        });
        let llm_manager = LlmManager::new(Arc::clone(&rt.db), Arc::clone(&provider_registry), log_flags).await?;
        let client_count = llm_manager.client_names().await.len().saturating_sub(1);
        let default_client = llm_manager.default_name().await;
        info!(clients = client_count, default = %default_client, "LLM clients loaded");

        let secrets = SecretsStore::new(Arc::clone(&rt.db));
        info!("secrets store ready");

        let memory_manager = Arc::new(MemoryManager::new());
        info!("memory manager ready");

        Ok(Models { provider_registry, llm_manager, secrets, memory_manager })
    }
}

// ── Media: transcription / TTS / image generation ───────────────────────────

pub(super) struct Media {
    pub(super) image_generator_manager: Arc<ImageGeneratorManager>,
    pub(super) transcribe_manager:      Arc<TranscribeManager>,
    pub(super) tts_manager:             Arc<TtsManager>,
}

impl Media {
    pub(super) async fn build(rt: &Runtime, models: &Models) -> Result<Self> {
        let image_generator_manager = ImageGeneratorManager::new(
            Arc::clone(&rt.db),
            Arc::clone(&models.provider_registry),
            "data",
        ).await?;
        // Evaluate the await outside the `info!` macro: leaving the temporary
        // `tracing::Value` from the field expression alive across the await
        // makes the surrounding future non-Send, which Tauri's runtime rejects.
        let image_generator_models = image_generator_manager.list_models_info().await.len();
        info!(
            db_backed = image_generator_models,
            "image generator manager ready",
        );

        let transcribe_manager = TranscribeManager::new(
            Arc::clone(&rt.db),
            Arc::clone(&models.provider_registry),
            Arc::clone(&rt.system_bus),
            rt.shutdown_token.clone(),
        ).await?;
        let transcribe_models = transcribe_manager.list_models_info().await.len();
        info!(
            db_backed = transcribe_models,
            "transcribe manager ready",
        );

        let tts_manager = TtsManager::new(
            Arc::clone(&rt.db),
            Arc::clone(&models.provider_registry),
            Arc::clone(&rt.system_bus),
            rt.shutdown_token.clone(),
        ).await?;
        let tts_models = tts_manager.list_models_info().await.len();
        info!(
            db_backed = tts_models,
            "tts manager ready",
        );

        Ok(Media { image_generator_manager, transcribe_manager, tts_manager })
    }
}

// ── Integrations: MCP + plugins ─────────────────────────────────────────────

pub(super) struct Integrations {
    pub(super) mcp:            Arc<McpManager>,
    pub(super) plugin_manager: Arc<PluginManager>,
}

impl Integrations {
    /// Builds the MCP manager (its `initialize()` is deferred to `spawn_background`,
    /// after the elicitation handler is wired) and the plugin manager (plugins are
    /// injected by `main.rs`; `start_enabled()` runs later, from `WebFrontend`).
    pub(super) fn build(rt: &Runtime, plugins: Vec<Arc<dyn Plugin>>) -> Self {
        let mcp = Arc::new(McpManager::new(Arc::clone(&rt.db), rt.shutdown_token.clone(), "data"));

        let mut plugin_manager = PluginManager::new(Arc::clone(&rt.db));
        for plugin in plugins {
            plugin_manager.register_arc(plugin);
        }
        info!("plugins registered");
        let plugin_manager = Arc::new(plugin_manager);

        Integrations { mcp, plugin_manager }
    }
}

// ── Tasks: cron + projects/tickets ──────────────────────────────────────────

pub(super) struct Tasks {
    pub(super) cron:           Arc<TaskManager>,
    pub(super) projects:       Arc<ProjectManager>,
    pub(super) ticket_manager: Arc<ProjectTicketManager>,
}

impl Tasks {
    /// Built before `Tools` so cron tools can capture the `TaskManager`.
    pub(super) fn build(rt: &Runtime, config: &CoreConfig) -> Self {
        let cron_tz = config.timezone.as_deref().and_then(|s| {
            match s.parse::<chrono_tz::Tz>() {
                Ok(tz)  => { info!("timezone: using {s}"); Some(tz) }
                Err(_)  => { warn!("timezone: unknown value '{s}', falling back to local time"); None }
            }
        });
        let cron = TaskManager::new(Arc::clone(&rt.db), cron_tz, Arc::clone(&rt.system_bus));

        let ticket_manager = ProjectTicketManager::new(Arc::clone(&rt.db));
        let projects       = Arc::new(ProjectManager::new(Arc::clone(&rt.db)));
        info!("project manager ready");

        Tasks { cron, projects, ticket_manager }
    }
}

// ── Tools: registry + catalog + slash commands ──────────────────────────────

pub(super) struct Tools {
    pub(super) tools:           Arc<ToolRegistry>,
    pub(super) catalog:         ToolCatalog,
    pub(super) command_manager: Arc<LlmCommandManager>,
}

impl Tools {
    /// Captures sibling managers (mcp, plugins, cron, secrets) into the tool
    /// registry. `execute_task` is deliberately NOT registered here — it is injected
    /// per interactive session by `ChatHub::send_message`.
    pub(super) fn build(integrations: &Integrations, tasks: &Tasks, models: &Models) -> Self {
        let mut tool_registry = ToolRegistry::new();
        crate::core::tools::fs::register_all(&mut tool_registry);
        tool_registry.register(crate::core::tools::ast_outline::AstOutline::new());
        tool_registry.register(crate::core::tools::exec::ExecuteCmd);
        tool_registry.register(crate::core::tools::read_notification::ReadNotification);
        tool_registry.register(crate::core::tools::restart::Restart);
        // Unified listing / toggling across mcp, plugins, cron (+ agents for list).
        tool_registry.register(crate::core::tools::list_items::ListItems::new(
            Arc::clone(&integrations.mcp), Arc::clone(&integrations.plugin_manager), Arc::clone(&tasks.cron)));
        tool_registry.register(crate::core::tools::toggle_item::ToggleItem::new(
            Arc::clone(&integrations.mcp), Arc::clone(&integrations.plugin_manager), Arc::clone(&tasks.cron)));
        tool_registry.register(crate::core::tools::register_mcp::RegisterMcp::new(Arc::clone(&integrations.mcp)));
        tool_registry.register(crate::core::tools::register_mcp::DeleteMcp::new(Arc::clone(&integrations.mcp)));
        tool_registry.register(crate::core::tools::cron_jobs::DeleteCronJob(Arc::clone(&tasks.cron)));
        tool_registry.register(crate::core::tools::set_secret::SetSecret(Arc::clone(&models.secrets)));
        tool_registry.register(crate::core::tools::list_secrets::ListSecrets(Arc::clone(&models.secrets)));
        tool_registry.register(crate::core::tools::configure_plugin::ConfigurePlugin(Arc::clone(&integrations.plugin_manager)));

        // Mobile-connector control tools (plugin.md §11). The plugin Arc itself
        // implements `RelayAgent`; we look it up and bind the tools to it. The tools
        // call into the plugin lazily, so registering them before the plugin's
        // runloop starts is fine (they fail gracefully while stopped).
        if let Some(mc) = integrations.plugin_manager
            .get_plugin_typed::<plugin_mobile_connector::MobileConnectorPlugin>("mobile-connector")
        {
            let agent: Arc<dyn plugin_mobile_connector::RelayAgent> = mc;
            for tool in plugin_mobile_connector::mobile_tools(agent) {
                tool_registry.register_arc(tool);
            }
            info!("mobile-connector tools registered");
        }
        debug!("tool registry built");

        let tools = Arc::new(tool_registry);
        let catalog = ToolCatalog::new(Arc::clone(&tools), Arc::clone(&integrations.mcp));
        let command_manager = Arc::new(LlmCommandManager::new());

        Tools { tools, catalog, command_manager }
    }
}

// ── Interaction: approval + inbox + clarification + elicitation ─────────────

pub(super) struct Interaction {
    pub(super) approval:      Arc<ApprovalManager>,
    pub(super) inbox:         Inbox,
    pub(super) clarification: Arc<ClarificationManager>,
    pub(super) elicitation:   Arc<ElicitationManager>,
}

impl Interaction {
    pub(super) async fn build(rt: &Runtime, tools: &Tools) -> Result<Self> {
        let approval = Arc::new(ApprovalManager::new(Arc::clone(&rt.db), rt.global_tx.clone()));
        if let Err(e) = approval.seed_defaults().await {
            warn!(error = %e, "failed to seed default approval rules (non-fatal)");
        }
        if let Err(e) = approval.migrate_legacy_fs_rules().await {
            warn!(error = %e, "failed to migrate legacy filesystem rules (non-fatal)");
        }
        if let Err(e) = approval.seed_fs_path_rules().await {
            warn!(error = %e, "failed to seed File System path rules (non-fatal)");
        }
        if let Err(e) = approval.seed_default_catch_all().await {
            warn!(error = %e, "failed to seed default catch-all rule (non-fatal)");
        }
        info!("approval manager ready");

        let clarification = ClarificationManager::new(rt.global_tx.clone());
        let elicitation = ElicitationManager::new(rt.global_tx.clone());

        let inbox = Inbox::new(
            Arc::clone(&approval),
            Arc::clone(&clarification),
            Arc::clone(&elicitation),
            Arc::clone(&tools.tools),
        );

        Ok(Interaction { approval, inbox, clarification, elicitation })
    }
}

// ── Conversation: session manager + chat hub + run context + TIC ────────────

pub(super) struct Conversation {
    pub(super) manager:             Arc<ChatSessionManager>,
    pub(super) chat_hub:            Arc<ChatHub>,
    pub(super) run_context_manager: Arc<RunContextManager>,
    /// TIC lives here (rather than in `Tasks`) because it is constructed from and
    /// drives the conversation stack (session manager + chat hub + run context);
    /// this keeps every bundle a single-shot `build()` with no two-phase init.
    pub(super) tic_manager:         Arc<TicManager>,
}

impl Conversation {
    #[allow(clippy::too_many_arguments)]
    pub(super) async fn build(
        rt: &Runtime,
        models: &Models,
        media: &Media,
        tools: &Tools,
        integrations: &Integrations,
        interaction: &Interaction,
        config: &CoreConfig,
    ) -> Result<Self> {
        let run_context_manager =
            Arc::new(RunContextManager::new(Arc::clone(&rt.db), Arc::clone(&interaction.approval)));
        if let Err(e) = run_context_manager.seed_defaults().await {
            warn!(error = %e, "failed to seed default permission group (non-fatal)");
        }
        info!("run_context manager ready");

        let compactor = config.llm.compaction.as_ref().map(|cfg| {
            info!(
                threshold_tokens = cfg.threshold_tokens,
                keep_recent      = cfg.keep_recent,
                ?cfg.strength,
                "context compactor enabled"
            );
            Arc::new(ContextCompactor::new(
                cfg.clone(),
                Arc::clone(&models.llm_manager),
                Arc::clone(&rt.event_bus),
            ))
        });
        if compactor.is_none() {
            info!("context compactor disabled (no compaction config)");
        }

        let manager = Arc::new(ChatSessionManager::new(
            Arc::clone(&rt.db),
            Arc::clone(&models.llm_manager),
            config.llm.max_history_messages,
            config.llm.max_tool_rounds.unwrap_or(DEFAULT_MAX_TOOL_ROUNDS),
            config.llm.max_parallel_subagents.unwrap_or(DEFAULT_MAX_PARALLEL_SUBAGENTS),
            config.llm.max_tool_result_chars,
            DatetimeConfig { timezone: config.timezone.clone(), ..config.llm.datetime },
            Arc::clone(&tools.tools),
            Arc::clone(&integrations.mcp),
            Arc::clone(&interaction.approval),
            Arc::clone(&interaction.clarification),
            Arc::clone(&rt.event_bus),
            Arc::clone(&models.memory_manager),
            Arc::clone(&media.image_generator_manager),
            compactor,
            Arc::clone(&run_context_manager),
            Arc::new(ToolDiscovery::new(Arc::clone(&rt.db))),
        ));

        let chat_hub = ChatHub::new(
            Arc::clone(&rt.db),
            Arc::clone(&manager),
            Arc::clone(&interaction.approval),
            rt.global_tx.clone(),
            rt.shutdown_token.clone(),
        );
        chat_hub.register("web").await;
        chat_hub.register("talk").await;

        let tic_manager = TicManager::new(
            Arc::clone(&rt.db),
            Arc::clone(&manager),
            Arc::clone(&chat_hub),
            config.tic.clone(),
            Arc::clone(&rt.config),
            Arc::clone(&run_context_manager),
            Arc::clone(&rt.system_bus),
        );

        Ok(Conversation { manager, chat_hub, run_context_manager, tic_manager })
    }
}

// ── Infra: leftover singletons ──────────────────────────────────────────────

pub(super) struct Infra {
    pub(super) latex_compiler:   LatexCompiler,
    pub(super) location_manager: Arc<LocationManager>,
    pub(super) remote:           Arc<RwLock<Option<Arc<dyn RemoteAccess>>>>,
}

impl Infra {
    pub(super) fn build() -> Self {
        Infra {
            latex_compiler:   LatexCompiler::new(),
            location_manager: Arc::new(LocationManager::new()),
            remote:           Arc::new(RwLock::new(None)),
        }
    }
}