Accusys
54 lines
1.9 KiB
Rust
54 lines
1.9 KiB
Rust
use axum::{extract::Path, http::StatusCode, routing::get, Router};
|
|
|
|
use super::types::AppState;
|
|
|
|
async fn doc_redirect() -> axum::response::Redirect {
|
|
axum::response::Redirect::to("/doc-wasm")
|
|
}
|
|
|
|
async fn wasm_doc_handler() -> Result<impl axum::response::IntoResponse, (StatusCode, &'static str)>
|
|
{
|
|
let path = std::path::Path::new("/Users/accusys/momentry_core/docs_v1.0/doc_wasm/index.html");
|
|
match tokio::fs::read_to_string(path).await {
|
|
Ok(html) => Ok(([("content-type", "text/html; charset=utf-8")], html)),
|
|
Err(_) => Err((StatusCode::NOT_FOUND, "Doc not found")),
|
|
}
|
|
}
|
|
|
|
async fn wasm_doc_file_handler(
|
|
Path(file): Path<String>,
|
|
) -> Result<impl axum::response::IntoResponse, (StatusCode, &'static str)> {
|
|
if file.contains("..") || file.contains("//") {
|
|
return Err((StatusCode::NOT_FOUND, "Invalid path"));
|
|
}
|
|
let base = std::path::Path::new("/Users/accusys/momentry_core/docs_v1.0/doc_wasm");
|
|
let path = base.join(&file);
|
|
if !path.exists() || !path.starts_with(base) {
|
|
return Err((StatusCode::NOT_FOUND, "File not found"));
|
|
}
|
|
let data = tokio::fs::read(&path)
|
|
.await
|
|
.map_err(|_| (StatusCode::NOT_FOUND, "Read error"))?;
|
|
let mime = if file.ends_with(".wasm") {
|
|
"application/wasm"
|
|
} else if file.ends_with(".js") {
|
|
"application/javascript"
|
|
} else if file.ends_with(".md") {
|
|
"text/markdown; charset=utf-8"
|
|
} else if file.ends_with(".css") {
|
|
"text/css"
|
|
} else {
|
|
"application/octet-stream"
|
|
};
|
|
Ok(([("content-type", mime)], data))
|
|
}
|
|
|
|
pub fn doc_routes() -> Router<AppState> {
|
|
Router::new()
|
|
.route("/doc", get(doc_redirect))
|
|
.route("/doc/*file", get(doc_redirect))
|
|
.route("/dev-doc", get(doc_redirect))
|
|
.route("/doc-wasm", get(wasm_doc_handler))
|
|
.route("/doc-wasm/*file", get(wasm_doc_file_handler))
|
|
}
|