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
2026-01-04 11:11:15 +08:00
parent 44ba82c194
commit 4e2c6a90c3
23 changed files with 7156 additions and 89 deletions
@@ -0,0 +1,320 @@
+import meiseayoung.hono
+import net.http
+
+// Cookie Helper 功能测试
+// 测试 Cookie 的 get/set/delete 和签名功能
+
+struct TestStats {
+mut:
+	total_tests  int
+	passed_tests int
+	failed_tests int
+}
+
+fn (mut stats TestStats) run_test(test_name string, test_func fn () bool) {
+	stats.total_tests++
+	print('🧪 ${test_name}... ')
+
+	if test_func() {
+		stats.passed_tests++
+		println('✅')
+	} else {
+		stats.failed_tests++
+		println('❌')
+	}
+}
+
+fn (stats TestStats) print_summary() {
+	println('\n=== Cookie Helper 测试总结 ===')
+	println('总测试数: ${stats.total_tests}')
+	println('通过: ${stats.passed_tests}')
+	println('失败: ${stats.failed_tests}')
+
+	if stats.failed_tests == 0 {
+		println('🎉 所有测试通过！')
+	} else {
+		println('⚠️  有 ${stats.failed_tests} 个测试失败')
+	}
+}
+
+// 创建带 Cookie 头的测试 Context
+fn create_test_context_with_cookies(cookie_header string) hono.Context {
+	mut headers := http.new_header()
+	if cookie_header.len > 0 {
+		headers.add_custom('Cookie', cookie_header) or {}
+	}
+	
+	req := http.Request{
+		method: .get
+		url: '/test'
+		header: headers
+	}
+	return hono.Context.new(req, map[string]string{}, map[string]string{}, '')
+}
+
+// 创建空的测试 Context
+fn create_empty_context() hono.Context {
+	req := http.Request{
+		method: .get
+		url: '/test'
+	}
+	return hono.Context.new(req, map[string]string{}, map[string]string{}, '')
+}
+
+
+// 测试 1: 获取单个 Cookie
+fn test_get_single_cookie() bool {
+	ctx := create_test_context_with_cookies('session_id=abc123')
+	
+	if value := hono.get_cookie(ctx, 'session_id') {
+		return value == 'abc123'
+	}
+	return false
+}
+
+// 测试 2: 获取不存在的 Cookie 返回 none
+fn test_get_nonexistent_cookie() bool {
+	ctx := create_test_context_with_cookies('session_id=abc123')
+	
+	if _ := hono.get_cookie(ctx, 'nonexistent') {
+		return false // 应该返回 none
+	}
+	return true
+}
+
+// 测试 3: 获取多个 Cookie
+fn test_get_multiple_cookies() bool {
+	ctx := create_test_context_with_cookies('session_id=abc123; user_id=456; theme=dark')
+	
+	session := hono.get_cookie(ctx, 'session_id') or { return false }
+	user := hono.get_cookie(ctx, 'user_id') or { return false }
+	theme := hono.get_cookie(ctx, 'theme') or { return false }
+	
+	return session == 'abc123' && user == '456' && theme == 'dark'
+}
+
+// 测试 4: 获取所有 Cookie
+fn test_get_all_cookies() bool {
+	ctx := create_test_context_with_cookies('a=1; b=2; c=3')
+	
+	cookies := hono.get_all_cookies(ctx)
+	
+	return cookies.len == 3 && 
+		cookies['a'] == '1' && 
+		cookies['b'] == '2' && 
+		cookies['c'] == '3'
+}
+
+// 测试 5: 空 Cookie 头返回空 map
+fn test_get_all_cookies_empty() bool {
+	ctx := create_empty_context()
+	
+	cookies := hono.get_all_cookies(ctx)
+	
+	return cookies.len == 0
+}
+
+// 测试 6: 设置 Cookie
+fn test_set_cookie_basic() bool {
+	mut ctx := create_empty_context()
+	
+	hono.set_cookie(mut ctx, 'session', 'xyz789')
+	
+	// 检查 Set-Cookie 头是否被设置
+	if set_cookie := ctx.headers['Set-Cookie'] {
+		return set_cookie.contains('session=xyz789')
+	}
+	return false
+}
+
+// 测试 7: 设置 Cookie 带选项
+fn test_set_cookie_with_options() bool {
+	mut ctx := create_empty_context()
+	
+	hono.set_cookie(mut ctx, 'token', 'secret123', hono.CookieOptions{
+		path: '/api'
+		http_only: true
+		secure: true
+		max_age: 3600
+		same_site: .strict
+	})
+	
+	if set_cookie := ctx.headers['Set-Cookie'] {
+		return set_cookie.contains('token=secret123') &&
+			set_cookie.contains('Path=/api') &&
+			set_cookie.contains('HttpOnly') &&
+			set_cookie.contains('Secure') &&
+			set_cookie.contains('Max-Age=3600') &&
+			set_cookie.contains('SameSite=Strict')
+	}
+	return false
+}
+
+
+// 测试 8: 删除 Cookie
+fn test_delete_cookie() bool {
+	mut ctx := create_empty_context()
+	
+	hono.delete_cookie(mut ctx, 'session')
+	
+	if set_cookie := ctx.headers['Set-Cookie'] {
+		// 删除 Cookie 应该设置 Max-Age=0 或过期时间
+		return set_cookie.contains('session=') &&
+			(set_cookie.contains('Max-Age=0') || set_cookie.contains('Expires='))
+	}
+	return false
+}
+
+// 测试 9: Cookie 值带空格
+fn test_cookie_with_spaces() bool {
+	ctx := create_test_context_with_cookies('name=John Doe')
+	
+	if value := hono.get_cookie(ctx, 'name') {
+		return value == 'John Doe'
+	}
+	return false
+}
+
+// 测试 10: Cookie 值带引号
+fn test_cookie_with_quotes() bool {
+	ctx := create_test_context_with_cookies('data="hello world"')
+	
+	if value := hono.get_cookie(ctx, 'data') {
+		return value == 'hello world'
+	}
+	return false
+}
+
+// 测试 11: 签名 Cookie 设置和获取
+fn test_signed_cookie_roundtrip() bool {
+	mut ctx := create_empty_context()
+	secret := 'my-secret-key-12345'
+	
+	// 设置签名 Cookie
+	hono.set_signed_cookie(mut ctx, 'auth', 'user123', secret) or {
+		println('Failed to set signed cookie: ${err}')
+		return false
+	}
+	
+	// 获取 Set-Cookie 头中的值
+	set_cookie_header := ctx.headers['Set-Cookie'] or { return false }
+	
+	// 提取 Cookie 值（格式: auth=value.signature; ...）
+	mut cookie_value := ''
+	parts := set_cookie_header.split(';')
+	if parts.len > 0 {
+		name_value := parts[0].trim_space()
+		eq_pos := name_value.index('=') or { return false }
+		cookie_value = name_value[eq_pos + 1..]
+	}
+	
+	// 创建一个带有这个 Cookie 的新 Context 来验证
+	verify_ctx := create_test_context_with_cookies('auth=${cookie_value}')
+	
+	// 验证签名 Cookie
+	if value := hono.get_signed_cookie(verify_ctx, 'auth', secret) {
+		return value == 'user123'
+	} else {
+		println('Failed to get signed cookie: ${err}')
+		return false
+	}
+}
+
+// 测试 12: 签名 Cookie 篡改检测
+fn test_signed_cookie_tamper_detection() bool {
+	mut ctx := create_empty_context()
+	secret := 'my-secret-key-12345'
+	
+	// 设置签名 Cookie
+	hono.set_signed_cookie(mut ctx, 'auth', 'user123', secret) or { return false }
+	
+	// 获取 Set-Cookie 头中的值
+	set_cookie_header := ctx.headers['Set-Cookie'] or { return false }
+	
+	// 提取 Cookie 值
+	mut cookie_value := ''
+	parts := set_cookie_header.split(';')
+	if parts.len > 0 {
+		name_value := parts[0].trim_space()
+		eq_pos := name_value.index('=') or { return false }
+		cookie_value = name_value[eq_pos + 1..]
+	}
+	
+	// 篡改 Cookie 值
+	tampered_value := 'tampered' + cookie_value[8..] // 修改值部分
+	
+	// 创建带篡改 Cookie 的 Context
+	verify_ctx := create_test_context_with_cookies('auth=${tampered_value}')
+	
+	// 验证应该失败
+	if _ := hono.get_signed_cookie(verify_ctx, 'auth', secret) {
+		return false // 不应该成功
+	}
+	return true // 验证失败是预期的
+}
+
+// 测试 13: 签名 Cookie 错误密钥
+fn test_signed_cookie_wrong_secret() bool {
+	mut ctx := create_empty_context()
+	
+	// 使用一个密钥设置
+	hono.set_signed_cookie(mut ctx, 'auth', 'user123', 'secret1') or { return false }
+	
+	// 获取 Set-Cookie 头中的值
+	set_cookie_header := ctx.headers['Set-Cookie'] or { return false }
+	
+	// 提取 Cookie 值
+	mut cookie_value := ''
+	parts := set_cookie_header.split(';')
+	if parts.len > 0 {
+		name_value := parts[0].trim_space()
+		eq_pos := name_value.index('=') or { return false }
+		cookie_value = name_value[eq_pos + 1..]
+	}
+	
+	// 创建带 Cookie 的 Context
+	verify_ctx := create_test_context_with_cookies('auth=${cookie_value}')
+	
+	// 使用不同的密钥验证应该失败
+	if _ := hono.get_signed_cookie(verify_ctx, 'auth', 'secret2') {
+		return false // 不应该成功
+	}
+	return true // 验证失败是预期的
+}
+
+
+// 测试 14: 空密钥应该返回错误
+fn test_signed_cookie_empty_secret() bool {
+	mut ctx := create_empty_context()
+	
+	// 空密钥应该返回错误
+	if _ := hono.set_signed_cookie(mut ctx, 'auth', 'value', '') {
+		return false // 不应该成功
+	}
+	return true
+}
+
+fn main() {
+	println('🚀 开始 Cookie Helper 功能测试...\n')
+
+	mut stats := TestStats{}
+
+	// 运行所有测试
+	stats.run_test('获取单个 Cookie', test_get_single_cookie)
+	stats.run_test('获取不存在的 Cookie', test_get_nonexistent_cookie)
+	stats.run_test('获取多个 Cookie', test_get_multiple_cookies)
+	stats.run_test('获取所有 Cookie', test_get_all_cookies)
+	stats.run_test('空 Cookie 头返回空 map', test_get_all_cookies_empty)
+	stats.run_test('设置 Cookie 基本功能', test_set_cookie_basic)
+	stats.run_test('设置 Cookie 带选项', test_set_cookie_with_options)
+	stats.run_test('删除 Cookie', test_delete_cookie)
+	stats.run_test('Cookie 值带空格', test_cookie_with_spaces)
+	stats.run_test('Cookie 值带引号', test_cookie_with_quotes)
+	stats.run_test('签名 Cookie 往返一致性', test_signed_cookie_roundtrip)
+	stats.run_test('签名 Cookie 篡改检测', test_signed_cookie_tamper_detection)
+	stats.run_test('签名 Cookie 错误密钥', test_signed_cookie_wrong_secret)
+	stats.run_test('签名 Cookie 空密钥错误', test_signed_cookie_empty_secret)
+
+	// 打印测试总结
+	stats.print_summary()
+}