Something went wrong. Try again.
A lexicon-driven AppView for ATProto. forked from gamesgamesgamesgames.games/happyview
Something went wrong. Try again.
Rust
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281use axum::Json;use axum::extract::{Path, State};use axum::http::StatusCode;
use crate::AppState;use crate::db::{adapt_sql, now_rfc3339};use crate::error::AppError;use crate::event_log::{EventLog, Severity, log_event};
use super::auth::UserAuth;use super::permissions::Permission;use super::types::{ AddAllowlistBody, AllowlistEntry, RateLimitsResponse, SetEnabledBody, UpsertRateLimitBody,};
/// GET /admin/rate-limits — list rate limit config.pub(super) async fn list( State(state): State<AppState>, auth: UserAuth,) -> Result<Json<RateLimitsResponse>, AppError> { auth.require(Permission::RateLimitsRead).await?;
let backend = state.db_backend;
let enabled_sql = adapt_sql( "SELECT value FROM rate_limit_settings WHERE key = 'enabled'", backend, ); let enabled: String = sqlx::query_scalar(&enabled_sql) .fetch_optional(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to read rate limit settings: {e}")))? .unwrap_or_else(|| "true".to_string());
let limits_sql = adapt_sql( "SELECT capacity, refill_rate, default_query_cost, default_procedure_cost, default_proxy_cost FROM rate_limits WHERE method IS NULL", backend, ); let row: Option<(i32, f64, i32, i32, i32)> = sqlx::query_as(&limits_sql) .fetch_optional(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to read rate limits: {e}")))?;
let (capacity, refill_rate, default_query_cost, default_procedure_cost, default_proxy_cost) = row.unwrap_or((100, 2.0, 1, 1, 1));
let allowlist_sql = adapt_sql( "SELECT id, cidr, note, created_at FROM rate_limit_allowlist ORDER BY id", backend, ); let allowlist_rows: Vec<(i32, String, Option<String>, String)> = sqlx::query_as(&allowlist_sql) .fetch_all(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to list allowlist: {e}")))?;
let allowlist: Vec<AllowlistEntry> = allowlist_rows .into_iter() .map(|(id, cidr, note, created_at)| AllowlistEntry { id, cidr, note, created_at, }) .collect();
Ok(Json(RateLimitsResponse { enabled: enabled == "true", capacity, refill_rate, default_query_cost, default_procedure_cost, default_proxy_cost, allowlist, }))}
/// POST /admin/rate-limits — upsert the global rate limit config.pub(super) async fn upsert( State(state): State<AppState>, auth: UserAuth, Json(body): Json<UpsertRateLimitBody>,) -> Result<StatusCode, AppError> { auth.require(Permission::RateLimitsCreate).await?;
let backend = state.db_backend; let now = now_rfc3339(); let sql = adapt_sql( r#" INSERT INTO rate_limits (method, capacity, refill_rate, default_query_cost, default_procedure_cost, default_proxy_cost, created_at) VALUES (NULL, ?, ?, ?, ?, ?, ?) ON CONFLICT (method) DO UPDATE SET capacity = EXCLUDED.capacity, refill_rate = EXCLUDED.refill_rate, default_query_cost = EXCLUDED.default_query_cost, default_procedure_cost = EXCLUDED.default_procedure_cost, default_proxy_cost = EXCLUDED.default_proxy_cost, updated_at = ? "#, backend, ); sqlx::query(&sql) .bind(body.capacity as i32) .bind(body.refill_rate) .bind(body.default_query_cost as i32) .bind(body.default_procedure_cost as i32) .bind(body.default_proxy_cost as i32) .bind(&now) .bind(&now) .execute(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to upsert rate limit: {e}")))?;
state.rate_limiter.reload_from_db(&state.db).await;
log_event( &state.db, EventLog { event_type: "rate_limit.upserted".to_string(), severity: Severity::Info, actor_did: Some(auth.did.clone()), subject: None, detail: serde_json::json!({ "capacity": body.capacity, "refill_rate": body.refill_rate, "default_query_cost": body.default_query_cost, "default_procedure_cost": body.default_procedure_cost, "default_proxy_cost": body.default_proxy_cost, }), }, state.db_backend, ) .await;
Ok(StatusCode::CREATED)}
/// PUT /admin/rate-limits/enabled — toggle rate limiting.pub(super) async fn set_enabled( State(state): State<AppState>, auth: UserAuth, Json(body): Json<SetEnabledBody>,) -> Result<StatusCode, AppError> { auth.require(Permission::RateLimitsCreate).await?;
let value = if body.enabled { "true" } else { "false" };
let backend = state.db_backend; let now = now_rfc3339(); let sql = adapt_sql( r#" INSERT INTO rate_limit_settings (key, value) VALUES ('enabled', ?) ON CONFLICT (key) DO UPDATE SET value = EXCLUDED.value, updated_at = ? "#, backend, ); sqlx::query(&sql) .bind(value) .bind(&now) .execute(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to update rate limit settings: {e}")))?;
state.rate_limiter.set_enabled(body.enabled);
log_event( &state.db, EventLog { event_type: "rate_limit.toggled".to_string(), severity: Severity::Info, actor_did: Some(auth.did.clone()), subject: None, detail: serde_json::json!({ "enabled": body.enabled }), }, state.db_backend, ) .await;
Ok(StatusCode::NO_CONTENT)}
/// POST /admin/rate-limits/allowlist — add an IP/CIDR to the allowlist.pub(super) async fn add_allowlist( State(state): State<AppState>, auth: UserAuth, Json(body): Json<AddAllowlistBody>,) -> Result<StatusCode, AppError> { auth.require(Permission::RateLimitsCreate).await?;
// Validate CIDR syntax; if it's a bare IP, append /32 or /128 let cidr_str = if body.cidr.contains('/') { body.cidr.clone() } else if let Ok(ip) = body.cidr.parse::<std::net::IpAddr>() { match ip { std::net::IpAddr::V4(_) => format!("{}/32", body.cidr), std::net::IpAddr::V6(_) => format!("{}/128", body.cidr), } } else { return Err(AppError::BadRequest(format!( "invalid IP or CIDR: {}", body.cidr ))); };
// Validate it parses as IpNet if cidr_str.parse::<ipnet::IpNet>().is_err() { return Err(AppError::BadRequest(format!("invalid CIDR: {}", cidr_str))); }
let backend = state.db_backend; let now = now_rfc3339(); let sql = adapt_sql( "INSERT INTO rate_limit_allowlist (cidr, note, created_at) VALUES (?, ?, ?)", backend, ); sqlx::query(&sql) .bind(&cidr_str) .bind(&body.note) .bind(&now) .execute(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to add allowlist entry: {e}")))?;
state.rate_limiter.reload_from_db(&state.db).await;
log_event( &state.db, EventLog { event_type: "rate_limit.allowlist_added".to_string(), severity: Severity::Info, actor_did: Some(auth.did.clone()), subject: Some(cidr_str), detail: serde_json::json!({ "note": body.note }), }, state.db_backend, ) .await;
Ok(StatusCode::CREATED)}
/// DELETE /admin/rate-limits/allowlist/{id} — remove an allowlist entry.pub(super) async fn remove_allowlist( State(state): State<AppState>, auth: UserAuth, Path(id): Path<i32>,) -> Result<StatusCode, AppError> { auth.require(Permission::RateLimitsDelete).await?;
let backend = state.db_backend; let sql = adapt_sql("DELETE FROM rate_limit_allowlist WHERE id = ?", backend); let result = sqlx::query(&sql) .bind(id) .execute(&state.db) .await .map_err(|e| AppError::Internal(format!("failed to delete allowlist entry: {e}")))?;
if result.rows_affected() == 0 { return Err(AppError::NotFound(format!( "allowlist entry {id} not found" ))); }
state.rate_limiter.reload_from_db(&state.db).await;
log_event( &state.db, EventLog { event_type: "rate_limit.allowlist_removed".to_string(), severity: Severity::Info, actor_did: Some(auth.did.clone()), subject: Some(id.to_string()), detail: serde_json::json!({}), }, state.db_backend, ) .await;
Ok(StatusCode::NO_CONTENT)}