mod events;
mod stream;
use anyhow::Result;
use axum::Router;
use axum::routing::{get, post};
use std::sync::Arc;
use systemprompt_analytics::{AnalyticsEventsRepository, EngagementRepository};
use systemprompt_content::ContentRepository;
use systemprompt_runtime::AppContext;
pub use events::AnalyticsState;
pub fn router(ctx: &AppContext) -> Result<Router> {
let state = AnalyticsState {
events: Arc::new(AnalyticsEventsRepository::new(ctx.db_pool())?),
content: Arc::new(ContentRepository::new(ctx.db_pool())?),
engagement: Arc::new(EngagementRepository::new(ctx.db_pool())?),
content_routing: ctx.content_routing(),
};
Ok(Router::new()
.route("/events", post(events::record_event))
.route("/events/batch", post(events::record_events_batch))
.route("/stream", get(stream::analytics_stream))
.with_state(state))
}