use std::sync::Arc;

use erp_core::events::EventBus;
use erp_core::health_provider::HealthDataProvider;
use sea_orm::DatabaseConnection;

use crate::provider::registry::ProviderRegistry;
use crate::service::analysis::AnalysisService;
use crate::service::cache::CacheService;
use crate::service::chat_message::ChatMessageService;
use crate::service::chat_session::ChatSessionService;
use crate::service::feature_flag_service::FeatureFlagService;
use crate::service::insight_service::InsightService;
use crate::service::knowledge::KnowledgeService;
use crate::service::knowledge_v2::KnowledgeV2Service;
use crate::service::prompt::PromptService;
use crate::service::quota::QuotaService;
use crate::service::risk_service::RiskService;
use crate::service::suggestion::SuggestionService;
use crate::service::usage::UsageService;

#[derive(Clone)]
pub struct AiState {
    pub db: DatabaseConnection,
    pub event_bus: EventBus,
    pub analysis: Arc<AnalysisService>,
    pub prompt: Arc<PromptService>,
    pub usage: Arc<UsageService>,
    pub suggestion: Arc<SuggestionService>,
    pub health_provider: Arc<dyn HealthDataProvider>,
    pub provider_registry: Arc<ProviderRegistry>,
    pub quota: Arc<QuotaService>,
    pub cache: Arc<CacheService>,
    pub risk_service: Arc<RiskService>,
    pub insight_service: Arc<InsightService>,
    pub feature_flags: Arc<FeatureFlagService>,
    pub knowledge: Arc<KnowledgeService>,
    pub knowledge_v2: Arc<KnowledgeV2Service>,
    pub chat_session: Arc<ChatSessionService>,
    pub chat_message: Arc<ChatMessageService>,
}