shenjack/openbmclapi-rs
2
2
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

把serve改好了（

Browse Source
This commit is contained in:
shenjack 2024-01-26 17:10:02 +08:00
parent 33f49505a6
commit 9632ff96e7
Signed by: shenjack
GPG Key ID: 7B1134A979775551
1 changed files with 21 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
src/serve.rs
View File

@ -1,4 +1,4 @@
use axum::{extract::Path, http::header::HeaderMap, http::StatusCode, response::IntoResponse};
use axum::{extract::Path, http::header::HeaderMap, http::StatusCode, response::Response};
/// ```ts
/// import express from 'express'
@ -22,22 +22,35 @@ use axum::{extract::Path, http::header::HeaderMap, http::StatusCode, response::I
///
/// export default MeasureRoute
/// ```
async fn measure(header: HeaderMap, Path(size): Path<u32>) -> impl IntoResponse {
async fn measure(header: HeaderMap, Path(size): Path<u32>) -> Response<Vec<u8>> {
let mut data: Vec<u8> = Vec::new();
match header.get("x-openbmclapi-secret") {
Some(secret) => {
if secret != "secret" {
return (StatusCode::FORBIDDEN, Vec::new());
return Response::builder()
.status(StatusCode::FORBIDDEN)
.body(data)
.unwrap();
}
if size > 200 {
return (StatusCode::BAD_REQUEST, Vec::new());
return Response::builder()
.status(StatusCode::BAD_REQUEST)
.body(data)
.unwrap();
}
let buffer: Vec<u8> = vec![0x00, 0x66, 0xcc, 0xff];
let mut response: Vec<u8> = Vec::new();
for _ in 0..size {
response.extend(&buffer);
data.extend(&buffer);
}
return (StatusCode::OK, response);
// return (StatusCode::OK, response);
return Response::builder()
.status(StatusCode::OK)
.body(data)
.unwrap();
}
None => return (StatusCode::FORBIDDEN, Vec::new()),
None => Response::builder()
.status(StatusCode::FORBIDDEN)
.body(data)
.unwrap(),
}
}