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, multi-strategy auth (bearer token + session cookie), and a
5//! WebSocket for real-time events.
6
7pub mod account;
8pub mod auth;
9pub mod error;
10pub mod routes;
11pub mod state;
12pub mod ws;
13
14use std::sync::Arc;
15
16use axum::middleware;
17use axum::routing::{delete, get, patch, post};
18use axum::Router;
19use tower_http::cors::CorsLayer;
20use tower_http::trace::TraceLayer;
21
22use crate::state::AppState;
23
24/// Build the complete axum router with all API routes and middleware.
25pub fn build_router(state: Arc<AppState>) -> Router {
26    let api = Router::new()
27        .route("/health", get(routes::health::health))
28        .route("/health/detailed", get(routes::health::health_detailed))
29        // Auth
30        .route("/auth/login", post(auth::routes::login))
31        .route("/auth/logout", post(auth::routes::logout))
32        .route("/auth/status", get(auth::routes::status))
33        // Analytics
34        .route("/analytics/summary", get(routes::analytics::summary))
35        .route("/analytics/followers", get(routes::analytics::followers))
36        .route(
37            "/analytics/performance",
38            get(routes::analytics::performance),
39        )
40        .route("/analytics/topics", get(routes::analytics::topics))
41        .route(
42            "/analytics/recent-performance",
43            get(routes::analytics::recent_performance),
44        )
45        // Approval
46        .route("/approval/export", get(routes::approval::export_items))
47        .route("/approval", get(routes::approval::list_items))
48        .route("/approval/stats", get(routes::approval::stats))
49        .route("/approval/approve-all", post(routes::approval::approve_all))
50        .route(
51            "/approval/{id}/history",
52            get(routes::approval::get_edit_history),
53        )
54        .route("/approval/{id}", patch(routes::approval::edit_item))
55        .route(
56            "/approval/{id}/approve",
57            post(routes::approval::approve_item),
58        )
59        .route("/approval/{id}/reject", post(routes::approval::reject_item))
60        // Activity
61        .route("/activity/export", get(routes::activity::export_activity))
62        .route("/activity", get(routes::activity::list_activity))
63        .route(
64            "/activity/rate-limits",
65            get(routes::activity::rate_limit_usage),
66        )
67        // Replies
68        .route("/replies", get(routes::replies::list_replies))
69        // Content
70        .route(
71            "/content/tweets",
72            get(routes::content::list_tweets).post(routes::content::compose_tweet),
73        )
74        .route(
75            "/content/threads",
76            get(routes::content::list_threads).post(routes::content::compose_thread),
77        )
78        .route("/content/calendar", get(routes::content::calendar))
79        .route("/content/schedule", get(routes::content::schedule))
80        .route("/content/compose", post(routes::content::compose))
81        .route(
82            "/content/scheduled/{id}",
83            patch(routes::content::edit_scheduled).delete(routes::content::cancel_scheduled),
84        )
85        // Drafts
86        .route(
87            "/content/drafts",
88            get(routes::content::list_drafts).post(routes::content::create_draft),
89        )
90        .route(
91            "/content/drafts/{id}",
92            patch(routes::content::edit_draft).delete(routes::content::delete_draft),
93        )
94        .route(
95            "/content/drafts/{id}/schedule",
96            post(routes::content::schedule_draft),
97        )
98        .route(
99            "/content/drafts/{id}/publish",
100            post(routes::content::publish_draft),
101        )
102        // Targets
103        .route(
104            "/targets",
105            get(routes::targets::list_targets).post(routes::targets::add_target),
106        )
107        .route(
108            "/targets/{username}/timeline",
109            get(routes::targets::target_timeline),
110        )
111        .route(
112            "/targets/{username}/stats",
113            get(routes::targets::target_stats),
114        )
115        .route(
116            "/targets/{username}",
117            delete(routes::targets::remove_target),
118        )
119        // Strategy
120        .route("/strategy/current", get(routes::strategy::current))
121        .route("/strategy/history", get(routes::strategy::history))
122        .route("/strategy/refresh", post(routes::strategy::refresh))
123        .route("/strategy/inputs", get(routes::strategy::inputs))
124        // Costs — LLM
125        .route("/costs/summary", get(routes::costs::summary))
126        .route("/costs/daily", get(routes::costs::daily))
127        .route("/costs/by-model", get(routes::costs::by_model))
128        .route("/costs/by-type", get(routes::costs::by_type))
129        // Costs — X API
130        .route("/costs/x-api/summary", get(routes::costs::x_api_summary))
131        .route("/costs/x-api/daily", get(routes::costs::x_api_daily))
132        .route(
133            "/costs/x-api/by-endpoint",
134            get(routes::costs::x_api_by_endpoint),
135        )
136        // AI Assist
137        .route("/assist/tweet", post(routes::assist::assist_tweet))
138        .route("/assist/reply", post(routes::assist::assist_reply))
139        .route("/assist/thread", post(routes::assist::assist_thread))
140        .route("/assist/improve", post(routes::assist::assist_improve))
141        .route("/assist/topics", get(routes::assist::assist_topics))
142        .route(
143            "/assist/optimal-times",
144            get(routes::assist::assist_optimal_times),
145        )
146        .route("/assist/mode", get(routes::assist::get_mode))
147        // Discovery feed
148        .route("/discovery/feed", get(routes::discovery::feed))
149        .route("/discovery/keywords", get(routes::discovery::keywords))
150        .route(
151            "/discovery/{tweet_id}/compose-reply",
152            post(routes::discovery::compose_reply),
153        )
154        .route(
155            "/discovery/{tweet_id}/queue-reply",
156            post(routes::discovery::queue_reply),
157        )
158        // Media
159        .route("/media/upload", post(routes::media::upload))
160        .route("/media/file", get(routes::media::serve_file))
161        // LAN settings
162        .route(
163            "/settings/lan",
164            get(routes::lan::get_status).patch(routes::lan::toggle_lan),
165        )
166        .route(
167            "/settings/lan/reset-passphrase",
168            post(routes::lan::reset_passphrase),
169        )
170        // Settings
171        .route("/settings/status", get(routes::settings::config_status))
172        .route("/settings/init", post(routes::settings::init_settings))
173        .route(
174            "/settings/validate",
175            post(routes::settings::validate_settings),
176        )
177        .route("/settings/defaults", get(routes::settings::get_defaults))
178        .route("/settings/test-llm", post(routes::settings::test_llm))
179        .route(
180            "/settings",
181            get(routes::settings::get_settings).patch(routes::settings::patch_settings),
182        )
183        // MCP governance
184        .route(
185            "/mcp/policy",
186            get(routes::mcp::get_policy).patch(routes::mcp::patch_policy),
187        )
188        .route("/mcp/policy/templates", get(routes::mcp::list_templates))
189        .route(
190            "/mcp/policy/templates/{name}",
191            post(routes::mcp::apply_template),
192        )
193        .route(
194            "/mcp/telemetry/summary",
195            get(routes::mcp::telemetry_summary),
196        )
197        .route(
198            "/mcp/telemetry/metrics",
199            get(routes::mcp::telemetry_metrics),
200        )
201        .route("/mcp/telemetry/errors", get(routes::mcp::telemetry_errors))
202        .route("/mcp/telemetry/recent", get(routes::mcp::telemetry_recent))
203        // Runtime
204        .route("/runtime/status", get(routes::runtime::status))
205        .route("/runtime/start", post(routes::runtime::start))
206        .route("/runtime/stop", post(routes::runtime::stop))
207        // Accounts
208        .route(
209            "/accounts",
210            get(routes::accounts::list_accounts).post(routes::accounts::create_account),
211        )
212        .route(
213            "/accounts/{id}/roles",
214            get(routes::accounts::list_roles)
215                .post(routes::accounts::set_role)
216                .delete(routes::accounts::remove_role),
217        )
218        .route(
219            "/accounts/{id}",
220            get(routes::accounts::get_account)
221                .patch(routes::accounts::update_account)
222                .delete(routes::accounts::delete_account),
223        )
224        // WebSocket
225        .route("/ws", get(ws::ws_handler))
226        // Auth middleware — applied to all routes; exempt paths handled internally.
227        .layer(middleware::from_fn_with_state(
228            state.clone(),
229            auth::auth_middleware,
230        ));
231
232    Router::new()
233        .nest("/api", api)
234        .layer(CorsLayer::permissive())
235        .layer(TraceLayer::new_for_http())
236        .with_state(state)
237}