import hono
import net.http
import os
import time

fn main() {
	println('=== 文件流式传输功能测试 ===')
	
	// 创建测试文件
	create_test_files()
	
	// 创建Context进行测试
	test_file_streaming()
	
	println('\n功能测试完成!')
	
	// 清理测试文件
	cleanup_test_files()
}

fn create_test_files() {
	println('创建测试文件...')
	
	// 创建小文件 (1KB)
	small_content := 'Hello World! '.repeat(80)
	os.write_file('test_small.txt', small_content) or { 
		println('  ❌ 创建小文件失败: $err')
		return
	}
	println('  ✅ 创建小文件: test_small.txt (${small_content.len} bytes)')
	
	// 创建中等文件 (约100KB)
	mut medium_content := ''
	for i in 0 .. 1000 {
		medium_content += 'This is line ${i:04d} with some content to make it longer and test streaming.\n'
	}
	os.write_file('test_medium.txt', medium_content) or { 
		println('  ❌ 创建中等文件失败: $err')
		return
	}
	println('  ✅ 创建中等文件: test_medium.txt (${medium_content.len} bytes)')
	
	// 创建大文件 (约1MB)
	mut large_content := ''
	base_line := 'This is a long line of text that will be repeated many times to create a large file for testing streaming functionality. '
	for i in 0 .. 10000 {
		large_content += '${i:05d}: $base_line\n'
	}
	os.write_file('test_large.txt', large_content) or { 
		println('  ❌ 创建大文件失败: $err')
		return
	}
	println('  ✅ 创建大文件: test_large.txt (${large_content.len} bytes)')
}

fn test_file_streaming() {
	println('\n开始功能测试...')
	
	// 创建模拟请求
	test_req := http.Request{
		method: http.Method.get
		url: '/test'
		data: ''
		header: http.new_header()
	}
	
	// 测试1：传统文件服务
	println('\n--- 测试1: 传统文件服务 ---')
	check_traditional_file_serving(test_req)
	
	// 测试2：流式文件服务  
	println('\n--- 测试2: 流式文件服务 ---')
	check_stream_file_serving(test_req)
	
	// 测试3：智能文件服务
	println('\n--- 测试3: 智能文件服务 ---')
	check_smart_file_serving(test_req)
	
	// 测试4：Range请求
	println('\n--- 测试4: Range请求测试 ---')
	check_range_requests()
	
	// 测试5：自定义选项测试
	println('\n--- 测试5: 自定义选项测试 ---')
	check_custom_options(test_req)
}

fn check_traditional_file_serving(req http.Request) {
	files_to_test := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
	
	for file in files_to_test {
		if !os.exists(file) {
			println('  ❌ 文件不存在: $file')
			continue
		}
		
		// 创建Context
		mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
		
		start := time.now()
		response := ctx.file(file)
		duration := time.now() - start
		
		mut original_size := i64(0)
		if stat := os.stat(file) {
			original_size = stat.size
		}
		response_size := response.body.len
		
		if response.status_code == 200 && response_size == int(original_size) {
			println('  ✅ $file: 成功 (${response_size} bytes, ${duration})')
		} else {
			println('  ❌ $file: 失败 (状态码: ${response.status_code}, 大小: ${response_size}/${original_size})')
		}
	}
}

fn check_stream_file_serving(req http.Request) {
	files_to_test := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
	
	for file in files_to_test {
		if !os.exists(file) {
			println('  ❌ 文件不存在: $file')
			continue
		}
		
		// 创建Context
		mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
		
		start := time.now()
		response := ctx.file_stream(file)
		duration := time.now() - start
		
		mut original_size := i64(0)
		if stat := os.stat(file) {
			original_size = stat.size
		}
		response_size := response.body.len
		
		if response.status_code == 200 && response_size == int(original_size) {
			println('  ✅ $file (流式): 成功 (${response_size} bytes, ${duration})')
		} else {
			println('  ❌ $file (流式): 失败 (状态码: ${response.status_code}, 大小: ${response_size}/${original_size})')
		}
	}
}

fn check_smart_file_serving(req http.Request) {
	files_to_test := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
	
	for file in files_to_test {
		if !os.exists(file) {
			println('  ❌ 文件不存在: $file')
			continue
		}
		
		// 创建Context
		mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
		
		start := time.now()
		response := ctx.file_smart(file)
		duration := time.now() - start
		
		mut original_size := i64(0)
		if stat := os.stat(file) {
			original_size = stat.size
		}
		response_size := response.body.len
		
		// 检查是否正确选择了传输方式
		mut expected_method := "内存"
		if original_size > 50 * 1024 * 1024 { // 默认阈值50MB
			expected_method = "流式"
		}
		
		if response.status_code == 200 && response_size == int(original_size) {
			println('  ✅ $file (智能-$expected_method): 成功 (${response_size} bytes, ${duration})')
		} else {
			println('  ❌ $file (智能): 失败 (状态码: ${response.status_code}, 大小: ${response_size}/${original_size})')
		}
	}
}

fn check_range_requests() {
	file := 'test_medium.txt'
	if !os.exists(file) {
		println('  ❌ 测试文件不存在: $file')
		return
	}
	
	// 创建带Range头的请求
	mut range_header := http.new_header()
	range_header.add_custom('Range', 'bytes=0-99') or { }
	
	range_req := http.Request{
		method: http.Method.get
		url: '/test'
		data: ''
		header: range_header
	}
	
	// 创建Context
	mut ctx := hono.Context.new(range_req, map[string]string{}, map[string]string{}, '')
	
	// 使用支持Range的选项
	options := hono.FileOptions{
		enable_range: true
		stream_threshold: 1024  // 强制使用流式传输来测试Range功能
	}
	
	response := ctx.file_stream_with_options(file, options)
	
	if response.status_code == 206 {  // Partial Content
		println('  ✅ Range请求: 成功 (状态码: 206, 内容长度: ${response.body.len})')
		
		// 检查Content-Range头
		if content_range := response.header.get_custom('Content-Range') {
			println('  📊 Content-Range: $content_range')
		}
	} else {
		println('  ❌ Range请求: 失败 (状态码: ${response.status_code})')
	}
}

fn check_custom_options(req http.Request) {
	file := 'test_medium.txt'
	if !os.exists(file) {
		println('  ❌ 测试文件不存在: $file')
		return
	}
	
	// 测试自定义选项
	custom_options := hono.FileOptions{
		stream_threshold: 10 * 1024  // 10KB阈值
		buffer_size: 2048            // 2KB缓冲区
		enable_range: true
		max_age: 7200
		content_type: 'text/plain; charset=utf-8'
		headers: {
			'X-Custom-Header': 'streaming-test'
			'X-Buffer-Size': '2048'
		}
	}
	
	// 创建Context
	mut ctx := hono.Context.new(req, map[string]string{}, map[string]string{}, '')
	
	response := ctx.file_stream_with_options(file, custom_options)
	
	if response.status_code == 200 {
		println('  ✅ 自定义选项: 成功')
		
		// 检查自定义头部
		if custom_header := response.header.get_custom('X-Custom-Header') {
			println('  📋 自定义头部: $custom_header')
		}
		
		if cache_control := response.header.get_custom('Cache-Control') {
			println('  🕒 缓存控制: $cache_control')
		}
		
		if content_type := response.header.get_custom('Content-Type') {
			println('  📝 内容类型: $content_type')
		}
	} else {
		println('  ❌ 自定义选项: 失败 (状态码: ${response.status_code})')
	}
}

fn cleanup_test_files() {
	println('\n清理测试文件...')
	test_files := ['test_small.txt', 'test_medium.txt', 'test_large.txt']
	
	for file in test_files {
		if os.exists(file) {
			os.rm(file) or {
				println('  警告: 无法删除 $file')
			}
		}
	}
	println('清理完成')
}