xfy/candy
1
0
Fork 0
mirror of https://github.com/DefectingCat/candy synced 2025-07-15 16:51:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(lua): impl set body to lua

Browse Source
This commit is contained in:
xfy
2025-07-02 01:11:48 +08:00
parent 14196c937c
commit e92b191360
2 changed files with 31 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
src/http/lua.rs
View File

@ -1,11 +1,13 @@
use anyhow::Context; use std::str::FromStr;
use anyhow::{Context, anyhow};
use axum::{ use axum::{
body::Body, body::Body,
extract::{Path, Request}, extract::{Path, Request},
response::{IntoResponse, Response}, response::{IntoResponse, Response},
}; };
use axum_extra::extract::Host; use axum_extra::extract::Host;
use http::Uri; use http::{HeaderMap, HeaderName, HeaderValue, Uri};
use mlua::{UserData, UserDataMethods, UserDataRef}; use mlua::{UserData, UserDataMethods, UserDataRef};
use tokio::fs::{self}; use tokio::fs::{self};
use tracing::error; use tracing::error;
@ -17,13 +19,19 @@ use crate::{
use super::error::RouteResult; use super::error::RouteResult;
/// 为 Lua 脚本提供 HTTP 请求上下文
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
struct CandyRequest { struct CandyRequest {
#[allow(dead_code)]
method: String,
/// Uri 在路由中被添加到上下文中
uri: Uri, uri: Uri,
} }
/// 为 Lua 脚本提供 HTTP 响应上下文
#[derive(Clone, Debug)] #[derive(Clone, Debug)]
struct CandyResponse { struct CandyResponse {
status: u16, status: u16,
headers: HeaderMap,
body: String, body: String,
} }
// HTTP 请求上下文，可在 Lua 中使用 // HTTP 请求上下文，可在 Lua 中使用
@ -40,6 +48,9 @@ impl UserData for RequestContext {
Ok(this.req.uri.path().to_string()) Ok(this.req.uri.path().to_string())
}); });
// 获取请求方法
methods.add_method("get_method", |_, this, ()| Ok(this.req.method.to_string()));
// 设置响应状态码 // 设置响应状态码
methods.add_method_mut("set_status", |_, this, status: u16| { methods.add_method_mut("set_status", |_, this, status: u16| {
this.res.status = status; this.res.status = status;
@ -48,7 +59,17 @@ impl UserData for RequestContext {
// 设置响应内容 // 设置响应内容
methods.add_method_mut("set_body", |_, this, body: String| { methods.add_method_mut("set_body", |_, this, body: String| {
this.res.body = body; this.res.body = format!("{}{}", this.res.body, body);
Ok(())
});
// 设置响应头
methods.add_method_mut("set_header", |_, this, (key, value): (String, String)| {
this.res.headers.insert(
HeaderName::from_str(&key).map_err(|err| anyhow!("header name error: {err}"))?,
HeaderValue::from_str(&value)
.map_err(|err| anyhow!("header value error: {err}"))?,
);
Ok(()) Ok(())
}); });
} }
@ -60,8 +81,6 @@ pub async fn lua(
Host(host): Host, Host(host): Host,
req: Request<Body>, req: Request<Body>,
) -> RouteResult<impl IntoResponse> { ) -> RouteResult<impl IntoResponse> {
// let req_path = req.uri().path();
let scheme = req.uri().scheme_str().unwrap_or("http"); let scheme = req.uri().scheme_str().unwrap_or("http");
let port = parse_port_from_host(&host, scheme).ok_or(RouteError::BadRequest())?; let port = parse_port_from_host(&host, scheme).ok_or(RouteError::BadRequest())?;
let route_map = &HOSTS let route_map = &HOSTS
@ -84,6 +103,8 @@ pub async fn lua(
.ok_or(RouteError::InternalError()) .ok_or(RouteError::InternalError())
.with_context(|| "lua script not found")?; .with_context(|| "lua script not found")?;
let method = req.method().to_string();
let lua = &LUA_ENGINE.lua; let lua = &LUA_ENGINE.lua;
let script = fs::read_to_string(lua_script) let script = fs::read_to_string(lua_script)
.await .await
@ -92,9 +113,13 @@ pub async fn lua(
.set( .set(
"ctx", "ctx",
RequestContext { RequestContext {
req: CandyRequest { uri: req_uri }, req: CandyRequest {
method,
uri: req_uri,
},
res: CandyResponse { res: CandyResponse {
status: 200, status: 200,
headers: HeaderMap::new(),
body: "".to_string(), body: "".to_string(),
}, },
}, },

1
src/middlewares/mod.rs
View File

@ -71,7 +71,6 @@ pub async fn add_headers(Host(host): Host, req: Request, next: Next) -> impl Int
debug!("port {:?}", port); debug!("port {:?}", port);
let mut res = next.run(req).await; let mut res = next.run(req).await;
let req_headers = res.headers_mut(); let req_headers = res.headers_mut();
// let host = HOSTS.read().await;
let Some(host) = HOSTS.get(&port) else { let Some(host) = HOSTS.get(&port) else {
return res; return res;
}; };