feat: add 7 built-in middleware components inspired by Hono.js
New Middleware: - CORS: Cross-Origin Resource Sharing with full configuration support - Cookie Helper: Cookie management with signed cookie support (HMAC-SHA256) - JWT: JSON Web Token authentication (HS256/384/512) - Bearer Auth: Simple Bearer token authentication - Compression: Gzip and deflate response compression - Rate Limiting: Request rate limiting with customizable stores - Validator: Schema-based request validation Features: - Context store for middleware data sharing (get/set/get_client_ip) - Middleware export module with utility functions - Comprehensive property-based tests for all middleware - Example file demonstrating all middleware usage Files Added: - cors.v, cookie.v, jwt.v, bearer_auth.v, compress.v, rate_limit.v, validator.v - middleware.v (unified exports) - examples/middleware_demo.v - tests/test_*.v (9 new test files) Documentation: - Updated README.md with detailed middleware usage guide - Added Schema Builder API documentation
This commit is contained in:
meiseayoung
+249
@@ -0,0 +1,249 @@
|
||||
module hono
|
||||
|
||||
import net.http
|
||||
import compress.gzip
|
||||
import compress.zlib
|
||||
|
||||
// CompressEncoding 枚举 - 支持的压缩编码类型
|
||||
pub enum CompressEncoding {
|
||||
gzip
|
||||
deflate
|
||||
}
|
||||
|
||||
// CompressOptions 结构体 - 压缩配置选项
|
||||
pub struct CompressOptions {
|
||||
pub:
|
||||
encoding ?CompressEncoding // 指定编码，none 表示自动选择
|
||||
threshold int = 1024 // 最小压缩大小（字节），默认 1KB
|
||||
level int = 128 // 压缩级别 (0-4095 for gzip)
|
||||
}
|
||||
|
||||
// compress - 压缩中间件工厂函数
|
||||
// 返回一个 ContextMiddleware，用于压缩响应体
|
||||
pub fn compress(options ...CompressOptions) ContextMiddleware {
|
||||
opts := if options.len > 0 { options[0] } else { CompressOptions{} }
|
||||
|
||||
return fn [opts] (mut c Context, next fn (mut Context) http.Response) http.Response {
|
||||
// 先执行后续处理器获取响应
|
||||
mut response := next(mut c)
|
||||
|
||||
// 检查 Cache-Control: no-transform
|
||||
if has_no_transform(response) {
|
||||
return response
|
||||
}
|
||||
|
||||
// 检查响应体大小是否达到阈值
|
||||
if response.body.len < opts.threshold {
|
||||
return response
|
||||
}
|
||||
|
||||
// 检查 Content-Type 是否可压缩
|
||||
content_type := get_response_content_type(response)
|
||||
if !is_compressible_content_type(content_type) {
|
||||
return response
|
||||
}
|
||||
|
||||
// 检查响应是否已经被压缩
|
||||
if is_already_compressed(response) {
|
||||
return response
|
||||
}
|
||||
|
||||
// 获取客户端支持的编码
|
||||
accept_encoding := c.req.header.get_custom('Accept-Encoding') or { '' }
|
||||
if accept_encoding.len == 0 {
|
||||
return response
|
||||
}
|
||||
|
||||
// 选择压缩编码
|
||||
encoding := select_encoding(accept_encoding, opts.encoding)
|
||||
if encoding == none {
|
||||
return response
|
||||
}
|
||||
|
||||
// 执行压缩
|
||||
selected_encoding := encoding or { return response }
|
||||
compressed_body := compress_body(response.body, selected_encoding, opts.level) or {
|
||||
// 压缩失败，返回原始响应
|
||||
return response
|
||||
}
|
||||
|
||||
// 如果压缩后更大，返回原始响应
|
||||
if compressed_body.len >= response.body.len {
|
||||
return response
|
||||
}
|
||||
|
||||
// 构建新的响应头
|
||||
mut new_header := http.new_header()
|
||||
|
||||
// 复制原有头部
|
||||
for key in response.header.keys() {
|
||||
values := response.header.custom_values(key)
|
||||
for value in values {
|
||||
// 跳过 Content-Length，稍后重新设置
|
||||
if key.to_lower() != 'content-length' {
|
||||
new_header.add_custom(key, value) or { continue }
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 设置 Content-Encoding
|
||||
encoding_str := if selected_encoding == .gzip { 'gzip' } else { 'deflate' }
|
||||
new_header.add_custom('Content-Encoding', encoding_str) or {}
|
||||
|
||||
// 设置新的 Content-Length
|
||||
new_header.add_custom('Content-Length', compressed_body.len.str()) or {}
|
||||
|
||||
// 添加 Vary 头，表示响应根据 Accept-Encoding 变化
|
||||
new_header.add_custom('Vary', 'Accept-Encoding') or {}
|
||||
|
||||
return http.Response{
|
||||
status_code: response.status_code
|
||||
header: new_header
|
||||
body: compressed_body.bytestr()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// has_no_transform - 检查响应是否包含 Cache-Control: no-transform
|
||||
fn has_no_transform(response http.Response) bool {
|
||||
cache_control := response.header.get_custom('Cache-Control') or { '' }
|
||||
return cache_control.contains('no-transform')
|
||||
}
|
||||
|
||||
// get_response_content_type - 获取响应的 Content-Type
|
||||
fn get_response_content_type(response http.Response) string {
|
||||
return response.header.get_custom('Content-Type') or { '' }
|
||||
}
|
||||
|
||||
// is_compressible_content_type - 检查 Content-Type 是否可压缩
|
||||
fn is_compressible_content_type(content_type string) bool {
|
||||
if content_type.len == 0 {
|
||||
return true // 默认可压缩
|
||||
}
|
||||
|
||||
ct_lower := content_type.to_lower()
|
||||
|
||||
// 可压缩的 MIME 类型
|
||||
compressible_types := [
|
||||
'text/',
|
||||
'application/json',
|
||||
'application/javascript',
|
||||
'application/xml',
|
||||
'application/xhtml+xml',
|
||||
'application/rss+xml',
|
||||
'application/atom+xml',
|
||||
'application/x-javascript',
|
||||
'application/x-font-ttf',
|
||||
'font/opentype',
|
||||
'font/ttf',
|
||||
'font/eot',
|
||||
'image/svg+xml',
|
||||
'image/x-icon',
|
||||
'image/vnd.microsoft.icon',
|
||||
]
|
||||
|
||||
for ct in compressible_types {
|
||||
if ct_lower.starts_with(ct) || ct_lower.contains(ct) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
// 不可压缩的类型（已压缩的格式）
|
||||
non_compressible_types := [
|
||||
'image/png',
|
||||
'image/jpeg',
|
||||
'image/gif',
|
||||
'image/webp',
|
||||
'video/',
|
||||
'audio/',
|
||||
'application/zip',
|
||||
'application/gzip',
|
||||
'application/x-gzip',
|
||||
'application/x-compress',
|
||||
'application/x-compressed',
|
||||
'application/x-bzip',
|
||||
'application/x-bzip2',
|
||||
'application/x-rar-compressed',
|
||||
'application/x-7z-compressed',
|
||||
]
|
||||
|
||||
for ct in non_compressible_types {
|
||||
if ct_lower.starts_with(ct) || ct_lower.contains(ct) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
return true
|
||||
}
|
||||
|
||||
// is_already_compressed - 检查响应是否已经被压缩
|
||||
fn is_already_compressed(response http.Response) bool {
|
||||
content_encoding := response.header.get_custom('Content-Encoding') or { '' }
|
||||
return content_encoding.len > 0
|
||||
}
|
||||
|
||||
// select_encoding - 选择最佳压缩编码
|
||||
fn select_encoding(accept_encoding string, preferred ?CompressEncoding) ?CompressEncoding {
|
||||
// 如果指定了首选编码，检查客户端是否支持
|
||||
if pref := preferred {
|
||||
match pref {
|
||||
.gzip {
|
||||
if accepts_gzip(accept_encoding) {
|
||||
return .gzip
|
||||
}
|
||||
}
|
||||
.deflate {
|
||||
if accepts_deflate(accept_encoding) {
|
||||
return .deflate
|
||||
}
|
||||
}
|
||||
}
|
||||
return none
|
||||
}
|
||||
|
||||
// 自动选择：优先 gzip，其次 deflate
|
||||
if accepts_gzip(accept_encoding) {
|
||||
return .gzip
|
||||
}
|
||||
if accepts_deflate(accept_encoding) {
|
||||
return .deflate
|
||||
}
|
||||
|
||||
return none
|
||||
}
|
||||
|
||||
// accepts_gzip - 检查是否接受 gzip 编码
|
||||
fn accepts_gzip(accept_encoding string) bool {
|
||||
return accept_encoding.contains('gzip') || accept_encoding.contains('*')
|
||||
}
|
||||
|
||||
// accepts_deflate - 检查是否接受 deflate 编码
|
||||
fn accepts_deflate(accept_encoding string) bool {
|
||||
return accept_encoding.contains('deflate') || accept_encoding.contains('*')
|
||||
}
|
||||
|
||||
// compress_body - 压缩响应体
|
||||
fn compress_body(body string, encoding CompressEncoding, level int) ![]u8 {
|
||||
data := body.bytes()
|
||||
|
||||
match encoding {
|
||||
.gzip {
|
||||
return gzip.compress(data, gzip.CompressParams{
|
||||
compression_level: level
|
||||
})
|
||||
}
|
||||
.deflate {
|
||||
return zlib.compress(data)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// decompress_gzip - 解压 gzip 数据（用于测试）
|
||||
pub fn decompress_gzip(data []u8) ![]u8 {
|
||||
return gzip.decompress(data)
|
||||
}
|
||||
|
||||
// decompress_deflate - 解压 deflate 数据（用于测试）
|
||||
pub fn decompress_deflate(data []u8) ![]u8 {
|
||||
return zlib.decompress(data)
|
||||
}
|
||||
Reference in New Issue
Block a user