package main

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"strings"
	"time"
)

// API测试结构
type APITest struct {
	Name     string
	Method   string
	URL      string
	Headers  map[string]string
	Body     string
	Expected int // 期望的HTTP状态码
}

// 响应结构
type Response struct {
	Code    int         `json:"code"`
	Message string      `json:"message"`
	Data    interface{} `json:"data"`
}

func main() {
	baseURL := "http://localhost:16001"
	
	// 定义所有需要测试的API
	tests := []APITest{
		// 用户端API
		{Name: "用户注册", Method: "POST", URL: "/api/v1/auth/auth/register", Expected: 200},
		{Name: "用户登录", Method: "POST", URL: "/api/v1/auth/auth/login", Expected: 200},
		{Name: "获取用户信息", Method: "GET", URL: "/api/v1/auth/auth/profile", Expected: 200},
		{Name: "更新用户信息", Method: "POST", URL: "/api/v1/auth/auth/profile", Expected: 200},
		{Name: "刷新令牌", Method: "POST", URL: "/api/v1/auth/auth/refresh", Expected: 200},
		{Name: "用户退出", Method: "POST", URL: "/api/v1/auth/auth/logout", Expected: 200},
		
		// 影片相关API
		{Name: "获取影片列表", Method: "GET", URL: "/api/v1/movies", Expected: 200},
		{Name: "获取影片详情", Method: "GET", URL: "/api/v1/movies/1", Expected: 200},
		{Name: "搜索影片", Method: "GET", URL: "/api/v1/movies/search?keyword=test", Expected: 200},
		{Name: "获取推荐影片", Method: "GET", URL: "/api/v1/movies/recommend", Expected: 200},
		{Name: "获取热门影片", Method: "GET", URL: "/api/v1/movies/hot", Expected: 200},
		{Name: "获取最新影片", Method: "GET", URL: "/api/v1/movies/new", Expected: 200},
		{Name: "按分类获取影片", Method: "GET", URL: "/api/v1/movies/category/1", Expected: 200},
		
		// 集数相关API
		{Name: "获取影片集数", Method: "GET", URL: "/api/v1/episodes/movie/1", Expected: 200},
		{Name: "获取集数详情", Method: "GET", URL: "/api/v1/episodes/1", Expected: 200},
		{Name: "获取视频信息", Method: "GET", URL: "/api/v1/episodes/video/info", Expected: 200},
		
		// 用户收藏API
		{Name: "添加收藏", Method: "POST", URL: "/api/v1/user/collect/add", Expected: 200},
		{Name: "取消收藏", Method: "POST", URL: "/api/v1/user/collect/remove", Expected: 200},
		{Name: "获取收藏列表", Method: "GET", URL: "/api/v1/user/collect/list", Expected: 200},
		{Name: "检查收藏状态", Method: "GET", URL: "/api/v1/user/collect/check", Expected: 200},
		
		// 观看历史API
		{Name: "添加观看记录", Method: "POST", URL: "/api/v1/user/history/add", Expected: 200},
		{Name: "获取观看历史", Method: "GET", URL: "/api/v1/user/history/list", Expected: 200},
		{Name: "获取观看记录", Method: "GET", URL: "/api/v1/user/history/get", Expected: 200},
		{Name: "删除观看记录", Method: "POST", URL: "/api/v1/user/history/delete", Expected: 200},
		{Name: "清空观看历史", Method: "POST", URL: "/api/v1/user/history/clear", Expected: 200},
		
		// 轮播图API
		{Name: "获取轮播图", Method: "GET", URL: "/api/v1/banner/list", Expected: 200},
		
		// VIP相关API
		{Name: "获取VIP等级", Method: "GET", URL: "/api/v1/vip/levels", Expected: 200},
		{Name: "获取我的VIP", Method: "GET", URL: "/api/v1/vip/my", Expected: 200},
		
		// 订单相关API
		{Name: "创建订单", Method: "POST", URL: "/api/v1/payment/create", Expected: 200},
		{Name: "获取订单列表", Method: "GET", URL: "/api/v1/payment/list", Expected: 200},
		{Name: "获取订单详情", Method: "GET", URL: "/api/v1/payment/1", Expected: 200},
		
		// 附件相关API
		{Name: "上传附件", Method: "POST", URL: "/api/v1/attachment/upload", Expected: 200},
		{Name: "获取附件列表", Method: "GET", URL: "/api/v1/attachment/list", Expected: 200},
		
		// 配置相关API
		{Name: "获取公共配置", Method: "GET", URL: "/api/v1/config/public", Expected: 200},
		
		// 日志相关API
		{Name: "获取我的日志", Method: "GET", URL: "/api/v1/log/my", Expected: 200},
		
		// 管理员端API
		{Name: "管理员注册", Method: "POST", URL: "/api/v1/admin/register", Expected: 200},
		{Name: "管理员登录", Method: "POST", URL: "/api/v1/admin/login", Expected: 200},
		{Name: "获取管理员信息", Method: "GET", URL: "/api/v1/admin/profile", Expected: 200},
		{Name: "更新管理员信息", Method: "POST", URL: "/api/v1/admin/profile", Expected: 200},
		{Name: "刷新管理员令牌", Method: "POST", URL: "/api/v1/admin/refresh", Expected: 200},
		{Name: "管理员退出", Method: "POST", URL: "/api/v1/admin/logout", Expected: 200},
		
		// 管理员-影片管理
		{Name: "管理员创建影片", Method: "POST", URL: "/api/v1/movies", Expected: 200},
		{Name: "管理员更新影片", Method: "POST", URL: "/api/v1/movies/update", Expected: 200},
		{Name: "管理员批量更新影片", Method: "POST", URL: "/api/v1/movies/batch-update", Expected: 200},
		{Name: "管理员删除影片", Method: "DELETE", URL: "/api/v1/movies/1", Expected: 200},
		{Name: "管理员上传海报", Method: "POST", URL: "/api/v1/movies/upload/poster", Expected: 200},
		{Name: "管理员上传视频", Method: "POST", URL: "/api/v1/movies/upload/video", Expected: 200},
		
		// 管理员-集数管理
		{Name: "管理员创建集数", Method: "POST", URL: "/api/v1/episodes", Expected: 200},
		{Name: "管理员更新集数", Method: "POST", URL: "/api/v1/episodes/update", Expected: 200},
		{Name: "管理员批量创建集数", Method: "POST", URL: "/api/v1/episodes/batch", Expected: 200},
		{Name: "管理员批量更新状态", Method: "POST", URL: "/api/v1/episodes/batch-status", Expected: 200},
		{Name: "管理员删除集数", Method: "DELETE", URL: "/api/v1/episodes/1", Expected: 200},
		{Name: "管理员上传视频", Method: "POST", URL: "/api/v1/episodes/upload/video", Expected: 200},
		{Name: "管理员生成缩略图", Method: "POST", URL: "/api/v1/episodes/thumbnail", Expected: 200},
	}
	
	fmt.Println("开始测试API接口...")
	fmt.Println(strings.Repeat("=", 60))
	
	successCount := 0
	failCount := 0
	
	for i, test := range tests {
		fmt.Printf("[%d/%d] 测试: %s\n", i+1, len(tests), test.Name)
		
		success := testAPI(baseURL+test.URL, test.Method, test.Headers, test.Body, test.Expected)
		if success {
			fmt.Printf("✅ 成功\n")
			successCount++
		} else {
			fmt.Printf("❌ 失败\n")
			failCount++
		}
		fmt.Println()
		
		// 避免请求过快
		time.Sleep(100 * time.Millisecond)
	}
	
	fmt.Println(strings.Repeat("=", 60))
	fmt.Printf("测试完成！成功: %d, 失败: %d, 总计: %d\n", successCount, failCount, len(tests))
	
	if failCount == 0 {
		fmt.Println("🎉 所有API接口测试通过！")
	} else {
		fmt.Printf("⚠️  有 %d 个接口测试失败，请检查相关代码\n", failCount)
	}
}

func testAPI(url, method string, headers map[string]string, body string, expectedStatus int) bool {
	client := &http.Client{
		Timeout: 10 * time.Second,
	}
	
	var req *http.Request
	var err error
	
	if body != "" {
		req, err = http.NewRequest(method, url, strings.NewReader(body))
	} else {
		req, err = http.NewRequest(method, url, nil)
	}
	
	if err != nil {
		fmt.Printf("创建请求失败: %v\n", err)
		return false
	}
	
	// 设置默认头部
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("User-Agent", "API-Test-Client/1.0")
	
	// 设置自定义头部
	for key, value := range headers {
		req.Header.Set(key, value)
	}
	
	resp, err := client.Do(req)
	if err != nil {
		fmt.Printf("请求失败: %v\n", err)
		return false
	}
	defer resp.Body.Close()
	
	// 读取响应体
	respBody, err := io.ReadAll(resp.Body)
	if err != nil {
		fmt.Printf("读取响应失败: %v\n", err)
		return false
	}
	
	// 检查HTTP状态码
	if resp.StatusCode != expectedStatus {
		fmt.Printf("HTTP状态码不匹配: 期望 %d, 实际 %d\n", expectedStatus, resp.StatusCode)
		fmt.Printf("响应内容: %s\n", string(respBody))
		return false
	}
	
	// 尝试解析JSON响应
	var response Response
	if err := json.Unmarshal(respBody, &response); err != nil {
		fmt.Printf("JSON解析失败: %v\n", err)
		fmt.Printf("响应内容: %s\n", string(respBody))
		return false
	}
	
	// 检查业务逻辑响应码
	if response.Code != 0 && response.Code != 200 {
		fmt.Printf("业务错误码: %d, 消息: %s\n", response.Code, response.Message)
		// 对于某些预期的错误（如未登录、数据不存在等），不算作失败
		if isExpectedError(response.Code) {
			return true
		}
		return false
	}
	
	fmt.Printf("✅ 响应正常: %s\n", response.Message)
	return true
}

// 判断是否为预期的错误码
func isExpectedError(code int) bool {
	expectedErrors := []int{
		401, // 未登录
		404, // 数据不存在
		403, // 权限不足
		400, // 参数错误
	}
	
	for _, expectedCode := range expectedErrors {
		if code == expectedCode {
			return true
		}
	}
	return false
}