mirror of
https://github.com/amtoaer/bili-sync.git
synced 2026-05-06 20:42:48 +08:00
99 lines
3.4 KiB
Rust
99 lines
3.4 KiB
Rust
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).head(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");
|
||
let default_headers = [
|
||
(header::CONTENT_TYPE, content_type),
|
||
(header::CACHE_CONTROL, "no-cache"),
|
||
(header::ETAG, &content.hash()),
|
||
];
|
||
if let Some(if_none_match) = request.headers().get(header::IF_NONE_MATCH)
|
||
&& let Ok(client_etag) = if_none_match.to_str()
|
||
&& client_etag == content.hash()
|
||
{
|
||
return (StatusCode::NOT_MODIFIED, default_headers).into_response();
|
||
}
|
||
|
||
if request.method() == axum::http::Method::HEAD {
|
||
return (StatusCode::OK, default_headers).into_response();
|
||
}
|
||
if cfg!(debug_assertions) {
|
||
// safety: `RustEmbed` returns uncompressed files directly from the filesystem in debug mode
|
||
return (StatusCode::OK, default_headers, content.data().unwrap()).into_response();
|
||
}
|
||
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)
|
||
&& let Some(data) = data
|
||
{
|
||
return (
|
||
StatusCode::OK,
|
||
[
|
||
(header::CONTENT_TYPE, content_type),
|
||
(header::CACHE_CONTROL, "no-cache"),
|
||
(header::ETAG, &content.hash()),
|
||
(header::CONTENT_ENCODING, encoding),
|
||
],
|
||
data,
|
||
)
|
||
.into_response();
|
||
}
|
||
}
|
||
(
|
||
StatusCode::NOT_ACCEPTABLE,
|
||
"Client must support gzip or brotli compression",
|
||
)
|
||
.into_response()
|
||
}
|