Skip to main content

tuitbot_server/
lib.rs

1//! Tuitbot HTTP API server.
2//!
3//! Exposes `tuitbot-core`'s storage layer as a REST API with read + write
4//! endpoints, local bearer-token auth, and a WebSocket for real-time events.
5
6pub mod auth;
7pub mod error;
8pub mod routes;
9pub mod state;
10pub mod ws;
11
12use std::sync::Arc;
13
14use axum::middleware;
15use axum::routing::{delete, get, patch, post};
16use axum::Router;
17use tower_http::cors::CorsLayer;
18use tower_http::trace::TraceLayer;
19
20use crate::state::AppState;
21
22/// Build the complete axum router with all API routes and middleware.
23pub fn build_router(state: Arc<AppState>) -> Router {
24    let api = Router::new()
25        .route("/health", get(routes::health::health))
26        // Analytics
27        .route("/analytics/summary", get(routes::analytics::summary))
28        .route("/analytics/followers", get(routes::analytics::followers))
29        .route(
30            "/analytics/performance",
31            get(routes::analytics::performance),
32        )
33        .route("/analytics/topics", get(routes::analytics::topics))
34        .route(
35            "/analytics/recent-performance",
36            get(routes::analytics::recent_performance),
37        )
38        // Approval
39        .route("/approval", get(routes::approval::list_items))
40        .route("/approval/stats", get(routes::approval::stats))
41        .route("/approval/approve-all", post(routes::approval::approve_all))
42        .route("/approval/{id}", patch(routes::approval::edit_item))
43        .route(
44            "/approval/{id}/approve",
45            post(routes::approval::approve_item),
46        )
47        .route("/approval/{id}/reject", post(routes::approval::reject_item))
48        // Activity
49        .route("/activity", get(routes::activity::list_activity))
50        .route(
51            "/activity/rate-limits",
52            get(routes::activity::rate_limit_usage),
53        )
54        // Replies
55        .route("/replies", get(routes::replies::list_replies))
56        // Content
57        .route(
58            "/content/tweets",
59            get(routes::content::list_tweets).post(routes::content::compose_tweet),
60        )
61        .route(
62            "/content/threads",
63            get(routes::content::list_threads).post(routes::content::compose_thread),
64        )
65        .route("/content/calendar", get(routes::content::calendar))
66        .route("/content/schedule", get(routes::content::schedule))
67        .route("/content/compose", post(routes::content::compose))
68        .route(
69            "/content/scheduled/{id}",
70            patch(routes::content::edit_scheduled).delete(routes::content::cancel_scheduled),
71        )
72        // Targets
73        .route(
74            "/targets",
75            get(routes::targets::list_targets).post(routes::targets::add_target),
76        )
77        .route(
78            "/targets/{username}/timeline",
79            get(routes::targets::target_timeline),
80        )
81        .route(
82            "/targets/{username}/stats",
83            get(routes::targets::target_stats),
84        )
85        .route(
86            "/targets/{username}",
87            delete(routes::targets::remove_target),
88        )
89        // Strategy
90        .route("/strategy/current", get(routes::strategy::current))
91        .route("/strategy/history", get(routes::strategy::history))
92        .route("/strategy/refresh", post(routes::strategy::refresh))
93        .route("/strategy/inputs", get(routes::strategy::inputs))
94        // Costs
95        .route("/costs/summary", get(routes::costs::summary))
96        .route("/costs/daily", get(routes::costs::daily))
97        .route("/costs/by-model", get(routes::costs::by_model))
98        .route("/costs/by-type", get(routes::costs::by_type))
99        // Settings
100        .route("/settings/status", get(routes::settings::config_status))
101        .route("/settings/init", post(routes::settings::init_settings))
102        .route(
103            "/settings/validate",
104            post(routes::settings::validate_settings),
105        )
106        .route("/settings/defaults", get(routes::settings::get_defaults))
107        .route("/settings/test-llm", post(routes::settings::test_llm))
108        .route(
109            "/settings",
110            get(routes::settings::get_settings).patch(routes::settings::patch_settings),
111        )
112        // Runtime
113        .route("/runtime/status", get(routes::runtime::status))
114        .route("/runtime/start", post(routes::runtime::start))
115        .route("/runtime/stop", post(routes::runtime::stop))
116        // WebSocket
117        .route("/ws", get(ws::ws_handler))
118        // Auth middleware — applied to all routes; health is exempted internally.
119        .layer(middleware::from_fn_with_state(
120            state.clone(),
121            auth::auth_middleware,
122        ));
123
124    Router::new()
125        .nest("/api", api)
126        .layer(CorsLayer::permissive())
127        .layer(TraceLayer::new_for_http())
128        .with_state(state)
129}