0.34.6_repo_cleanup

src/admin/public/mod.rs

@@ -0,0 +1,65 @@
+use crate::{
+    auth::Session,
+    runtime::{ApiError, App},
+    settings::PanelCompression,
+};
+use axum::{
+    extract::{rejection::JsonRejection, State},
+    response::{Html, IntoResponse, Response},
+    Extension,
+};
+use http::{HeaderMap, StatusCode};
+use std::sync::{Arc, OnceLock};
+
+use super::panel_content::PanelContent;
+
+static ADMIN_INDEX: OnceLock<PanelContent> = OnceLock::new();
+
+fn panel_response(headers: &HeaderMap, compression: &PanelCompression) -> Response {
+    let content = ADMIN_INDEX.get_or_init(|| {
+        let (metrics, web_metrics) = include_str!("../web/partials/overview-metrics.html")
+            .split_once("<!-- RP_METRICS_SPLIT -->")
+            .expect("invalid Overview metrics partial");
+        let (charts, web_charts) = include_str!("../web/partials/overview-charts.html")
+            .split_once("<!-- RP_CHARTS_SPLIT -->")
+            .expect("invalid Overview charts partial");
+        let html = include_str!("../web/index.html")
+            .replace("<!-- RP_METRICS_MAIN -->", metrics)
+            .replace("<!-- RP_CHARTS_MAIN -->", charts)
+            .replace("<!-- RP_METRICS_WEB -->", web_metrics)
+            .replace("<!-- RP_CHARTS_WEB -->", web_charts)
+            .replace("__RUSTPROXY_ASSET_ID__", super::asset_build_id())
+            .replace("__RUSTPROXY_VERSION__", env!("CARGO_PKG_VERSION"));
+        PanelContent::new(Box::leak(html.into_boxed_str()))
+    });
+    content.response(headers, compression, "text/html; charset=utf-8")
+}
+
+pub(super) async fn index(
+    State(state): State<Arc<App>>,
+    session: Option<Extension<Session>>,
+    headers: HeaderMap,
+) -> Response {
+    if session.is_none() {
+        let html = include_str!("../web/login.html")
+            .replace("__RUSTPROXY_ASSET_ID__", super::asset_build_id());
+        return Html(html).into_response();
+    }
+    let snapshot = state.config.load();
+    panel_response(
+        &headers,
+        &snapshot.envelope.config.administration.panel_compression,
+    )
+}
+
+pub(super) fn invalid_json(error: JsonRejection) -> ApiError {
+    let status = match error.status() {
+        StatusCode::PAYLOAD_TOO_LARGE => StatusCode::PAYLOAD_TOO_LARGE,
+        StatusCode::UNSUPPORTED_MEDIA_TYPE => StatusCode::UNSUPPORTED_MEDIA_TYPE,
+        _ => StatusCode::BAD_REQUEST,
+    };
+    ApiError(status, "auth.invalid_request".into())
+}
+
+#[cfg(test)]
+mod tests;