mirror of
https://github.com/mofeng-git/One-KVM.git
synced 2026-03-15 07:26:44 +08:00
init
This commit is contained in:
mofeng-git
204
src/web/static_files.rs
Normal file
204
src/web/static_files.rs
Normal file
@@ -0,0 +1,204 @@
|
||||
use axum::{
|
||||
body::Body,
|
||||
http::{header, Response, StatusCode, Uri},
|
||||
routing::get,
|
||||
Router,
|
||||
};
|
||||
use std::path::PathBuf;
|
||||
use std::sync::OnceLock;
|
||||
use tracing;
|
||||
|
||||
// Only embed assets in release mode
|
||||
#[cfg(not(debug_assertions))]
|
||||
use rust_embed::Embed;
|
||||
|
||||
#[cfg(not(debug_assertions))]
|
||||
/// Embedded static assets (frontend files) - only in release mode
|
||||
#[derive(Embed)]
|
||||
#[folder = "web/dist"]
|
||||
#[prefix = ""]
|
||||
pub struct StaticAssets;
|
||||
|
||||
/// Get the base directory for static files
|
||||
/// In debug mode: relative to executable directory
|
||||
/// In release mode: not used (embedded assets)
|
||||
fn get_static_base_dir() -> PathBuf {
|
||||
static BASE_DIR: OnceLock<PathBuf> = OnceLock::new();
|
||||
BASE_DIR.get_or_init(|| {
|
||||
// Try to get executable directory
|
||||
if let Ok(exe_path) = std::env::current_exe() {
|
||||
if let Some(exe_dir) = exe_path.parent() {
|
||||
return exe_dir.join("web").join("dist");
|
||||
}
|
||||
}
|
||||
// Fallback to current directory
|
||||
PathBuf::from("web/dist")
|
||||
}).clone()
|
||||
}
|
||||
|
||||
/// Create router for static file serving
|
||||
pub fn static_file_router<S>() -> Router<S>
|
||||
where
|
||||
S: Clone + Send + Sync + 'static,
|
||||
{
|
||||
Router::new()
|
||||
.route("/", get(index_handler))
|
||||
.route("/*path", get(static_handler))
|
||||
}
|
||||
|
||||
/// Serve index.html for root path
|
||||
async fn index_handler() -> Response<Body> {
|
||||
serve_file("index.html")
|
||||
}
|
||||
|
||||
/// Serve static files
|
||||
async fn static_handler(uri: Uri) -> Response<Body> {
|
||||
let path = uri.path().trim_start_matches('/');
|
||||
|
||||
// Try to serve the exact file
|
||||
if let Some(response) = try_serve_file(path) {
|
||||
return response;
|
||||
}
|
||||
|
||||
// For SPA routing, serve index.html for non-asset paths
|
||||
if !path.contains('.') {
|
||||
if let Some(response) = try_serve_file("index.html") {
|
||||
return response;
|
||||
}
|
||||
}
|
||||
|
||||
// If no embedded assets found, return placeholder page
|
||||
// This happens when web/dist was not built before compilation
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header(header::CONTENT_TYPE, "text/html; charset=utf-8")
|
||||
.body(Body::from(placeholder_html()))
|
||||
.unwrap()
|
||||
}
|
||||
|
||||
fn serve_file(path: &str) -> Response<Body> {
|
||||
try_serve_file(path).unwrap_or_else(|| {
|
||||
// If index.html not found in embedded assets, return placeholder
|
||||
if path == "index.html" {
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header(header::CONTENT_TYPE, "text/html; charset=utf-8")
|
||||
.body(Body::from(placeholder_html()))
|
||||
.unwrap()
|
||||
} else {
|
||||
Response::builder()
|
||||
.status(StatusCode::NOT_FOUND)
|
||||
.body(Body::from("Not Found"))
|
||||
.unwrap()
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
fn try_serve_file(path: &str) -> Option<Response<Body>> {
|
||||
#[cfg(debug_assertions)]
|
||||
{
|
||||
// Debug mode: read from file system
|
||||
let base_dir = get_static_base_dir();
|
||||
let file_path = base_dir.join(path);
|
||||
|
||||
// Check if file exists and is within base directory (prevent directory traversal)
|
||||
if !file_path.starts_with(&base_dir) {
|
||||
tracing::warn!("Path traversal attempt blocked: {}", path);
|
||||
return None;
|
||||
}
|
||||
|
||||
// Normalize path to prevent directory traversal (only if file exists)
|
||||
if let (Ok(normalized_path), Ok(normalized_base)) =
|
||||
(file_path.canonicalize(), base_dir.canonicalize())
|
||||
{
|
||||
if !normalized_path.starts_with(&normalized_base) {
|
||||
tracing::warn!("Path traversal attempt blocked (canonicalized): {}", path);
|
||||
return None;
|
||||
}
|
||||
}
|
||||
|
||||
match std::fs::read(&file_path) {
|
||||
Ok(data) => {
|
||||
let mime = mime_guess::from_path(path)
|
||||
.first_or_octet_stream()
|
||||
.to_string();
|
||||
|
||||
return Some(
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header(header::CONTENT_TYPE, mime)
|
||||
.header(header::CACHE_CONTROL, "public, max-age=86400")
|
||||
.body(Body::from(data))
|
||||
.unwrap(),
|
||||
);
|
||||
}
|
||||
Err(e) => {
|
||||
tracing::debug!(
|
||||
"Failed to read static file '{}' from '{}': {}",
|
||||
path,
|
||||
file_path.display(),
|
||||
e
|
||||
);
|
||||
return None;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(not(debug_assertions))]
|
||||
{
|
||||
// Release mode: use embedded assets
|
||||
let asset = StaticAssets::get(path)?;
|
||||
|
||||
let mime = mime_guess::from_path(path)
|
||||
.first_or_octet_stream()
|
||||
.to_string();
|
||||
|
||||
Some(
|
||||
Response::builder()
|
||||
.status(StatusCode::OK)
|
||||
.header(header::CONTENT_TYPE, mime)
|
||||
.header(header::CACHE_CONTROL, "public, max-age=86400")
|
||||
.body(Body::from(asset.data.to_vec()))
|
||||
.unwrap(),
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
/// Placeholder index.html when frontend is not built
|
||||
pub fn placeholder_html() -> &'static str {
|
||||
r#"<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
||||
<title>One-KVM</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
min-height: 100vh;
|
||||
margin: 0;
|
||||
background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
|
||||
color: #fff;
|
||||
}
|
||||
.container {
|
||||
text-align: center;
|
||||
padding: 2rem;
|
||||
}
|
||||
h1 { font-size: 2.5rem; margin-bottom: 1rem; }
|
||||
p { color: #888; font-size: 1.1rem; }
|
||||
.version { color: #666; margin-top: 2rem; font-size: 0.9rem; }
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<div class="container">
|
||||
<h1>One-KVM</h1>
|
||||
<p>Frontend not built yet.</p>
|
||||
<p>Please build the frontend or access the API directly.</p>
|
||||
<div class="version">v0.1.0</div>
|
||||
</div>
|
||||
</body>
|
||||
</html>"#
|
||||
}
|
||||
Reference in New Issue
Block a user