bestool-alertd 14.3.0

//! HTTP server for alertd daemon control and metrics.

use std::{collections::HashMap, sync::Arc, time::Duration};

use axum::{
	Router,
	routing::{get, post},
};
use jiff::Timestamp;
use tower_http::trace::{DefaultMakeSpan, DefaultOnResponse, TraceLayer};
use tracing::{Level, error, info, warn};

use crate::{
	context::InternalContext,
	daemon::DaemonControl,
	tasks::{BackgroundTask, TaskEndpointHandler},
};

mod endpoints;
mod state;
#[cfg(test)]
mod test_utils;
mod types;

pub use endpoints::*;
pub use state::ServerState;
pub use types::*;

pub async fn start_server(
	internal_context: Arc<InternalContext>,
	addrs: Vec<std::net::SocketAddr>,
	watchdog_timeout: Option<Duration>,
	background_tasks: &[Arc<dyn BackgroundTask>],
	control: DaemonControl,
	backups: Option<Arc<crate::BackupRegistry>>,
) {
	let started_at = Timestamp::now();
	let pid = std::process::id();

	let task_endpoints = collect_task_endpoints(background_tasks);

	let state = ServerState {
		started_at,
		pid,
		internal_context,
		watchdog_timeout,
		task_endpoints: Arc::new(task_endpoints),
		control,
		backups,
	};

	let app = Router::new()
		.route("/", get(handle_index))
		.route("/metrics", get(handle_metrics))
		.route("/status", get(handle_status))
		.route("/health", get(handle_health))
		.route("/reload", post(handle_reload))
		.route("/restart", post(handle_restart))
		.route("/tasks/{task}/{endpoint}", get(handle_task_endpoint))
		.layer(
			TraceLayer::new_for_http()
				.make_span_with(
					DefaultMakeSpan::new()
						.level(Level::INFO)
						.include_headers(false),
				)
				.on_request(|request: &axum::http::Request<_>, _span: &tracing::Span| {
					info!(
						method = %request.method(),
						uri = %request.uri(),
						"HTTP request"
					);
				})
				.on_response(
					DefaultOnResponse::new()
						.level(Level::INFO)
						.include_headers(false),
				),
		)
		.with_state(Arc::new(state));

	// Use default if no addresses provided
	let addrs_to_try = if addrs.is_empty() {
		vec![
			"[::1]:8271".parse().unwrap(),
			"127.0.0.1:8271".parse().unwrap(),
		]
	} else {
		addrs
	};

	let mut listener = None;
	let mut last_error = None;

	// Try each address in order until one succeeds
	for addr in &addrs_to_try {
		match tokio::net::TcpListener::bind(addr).await {
			Ok(l) => {
				info!("HTTP server listening on http://{}", addr);
				listener = Some(l);
				break;
			}
			Err(e) => {
				warn!("failed to bind HTTP server to {}: {}", addr, e);
				last_error = Some(e);
			}
		}
	}

	let listener = match listener {
		Some(l) => l,
		None => {
			if let Some(e) = last_error {
				warn!("failed to bind HTTP server to any address: {}", e);
			} else {
				warn!("no addresses provided for HTTP server");
			}
			warn!("waiting 10 seconds before continuing without");
			warn!("use --no-server to disable the HTTP server and this warning");

			tokio::time::sleep(tokio::time::Duration::from_secs(10)).await;

			info!("continuing without HTTP server");
			return;
		}
	};

	if let Err(e) = axum::serve(listener, app).await {
		error!("HTTP server error: {}", e);
	}
}

fn collect_task_endpoints(
	tasks: &[Arc<dyn BackgroundTask>],
) -> HashMap<(String, String), TaskEndpointHandler> {
	let mut map = HashMap::new();
	for task in tasks {
		let task_name = task.name();
		for endpoint in task.http_endpoints() {
			let key = (task_name.to_string(), endpoint.name.to_string());
			if map.contains_key(&key) {
				warn!(
					task = task_name,
					endpoint = endpoint.name,
					"duplicate task endpoint name; later registration wins"
				);
			}
			info!(
				task = task_name,
				endpoint = endpoint.name,
				path = %format!("/tasks/{task_name}/{}", endpoint.name),
				"mounting task endpoint"
			);
			map.insert(key, endpoint.handler);
		}
	}
	map
}