bili-sync/crates/bili_sync/src/task/http_server.rs

use std::collections::HashSet;
use std::sync::Arc;

use anyhow::{Context, Result};
use axum::extract::Request;
use axum::http::header;
use axum::response::IntoResponse;
use axum::routing::get;
use axum::{Extension, ServiceExt};
use reqwest::StatusCode;
use rust_embed_for_web::{EmbedableFile, RustEmbed};
use sea_orm::DatabaseConnection;

use crate::api::{LogHelper, router};
use crate::bilibili::BiliClient;
use crate::config::VersionedConfig;

#[derive(RustEmbed)]
#[preserve_source = false]
#[folder = "../../web/build"]
struct Asset;

pub async fn http_server(
    database_connection: DatabaseConnection,
    bili_client: Arc<BiliClient>,
    log_writer: LogHelper,
) -> Result<()> {
    let app = router()
        .fallback_service(get(frontend_files))
        .layer(Extension(database_connection))
        .layer(Extension(bili_client))
        .layer(Extension(log_writer));
    let config = VersionedConfig::get().load_full();
    let listener = tokio::net::TcpListener::bind(&config.bind_address)
        .await
        .context("bind address failed")?;
    info!("开始运行管理页: http://{}", config.bind_address);
    Ok(axum::serve(listener, ServiceExt::<Request>::into_make_service(app)).await?)
}

async fn frontend_files(request: Request) -> impl IntoResponse {
    let mut path = request.uri().path().trim_start_matches('/');
    if path.is_empty() || Asset::get(path).is_none() {
        path = "index.html";
    }
    let Some(content) = Asset::get(path) else {
        return (StatusCode::NOT_FOUND, "404 Not Found").into_response();
    };
    let mime_type = content.mime_type();
    let content_type = mime_type.as_deref().unwrap_or("application/octet-stream");
    if cfg!(debug_assertions) {
        (
            [(header::CONTENT_TYPE, content_type)],
            // safety: `RustEmbed` returns uncompressed files directly from the filesystem in debug mode
            content.data().unwrap(),
        )
            .into_response()
    } else {
        let accepted_encodings = request
            .headers()
            .get(header::ACCEPT_ENCODING)
            .and_then(|v| v.to_str().ok())
            .map(|s| s.split(',').map(str::trim).collect::<HashSet<_>>())
            .unwrap_or_default();
        for (encoding, data) in [("br", content.data_br()), ("gzip", content.data_gzip())] {
            if accepted_encodings.contains(encoding) {
                if let Some(data) = data {
                    return (
                        [
                            (header::CONTENT_TYPE, content_type),
                            (header::CONTENT_ENCODING, encoding),
                        ],
                        data,
                    )
                        .into_response();
                }
            }
        }
        "Unsupported Encoding".into_response()
    }
}