package http

import (
	"encoding/json"
	"net/http"
	"time"
)

// Response 标准响应结构
type Response struct {
	Code      int         `json:"code"`      // 业务状态码，0表示成功
	Message   string      `json:"message"`   // 响应消息
	Timestamp int64       `json:"timestamp"` // 时间戳
	Data      interface{} `json:"data"`      // 响应数据
}

// PageResponse 分页响应结构
type PageResponse struct {
	Code      int       `json:"code"`
	Message   string    `json:"message"`
	Timestamp int64     `json:"timestamp"`
	Data      *PageData `json:"data"`
}

// PageData 分页数据
type PageData struct {
	List     interface{} `json:"list"`     // 数据列表
	Total    int64       `json:"total"`    // 总记录数
	Page     int         `json:"page"`     // 当前页码
	PageSize int         `json:"pageSize"` // 每页大小
}

// Success 成功响应
// data: 响应数据，可以为nil
func Success(w http.ResponseWriter, data interface{}) {
	WriteJSON(w, http.StatusOK, 0, "success", data)
}

// SuccessWithMessage 带消息的成功响应
func SuccessWithMessage(w http.ResponseWriter, message string, data interface{}) {
	WriteJSON(w, http.StatusOK, 0, message, data)
}

// Error 错误响应
// code: 业务错误码，非0表示业务错误
// message: 错误消息
func Error(w http.ResponseWriter, code int, message string) {
	WriteJSON(w, http.StatusOK, code, message, nil)
}

// SystemError 系统错误响应（返回HTTP 500）
// message: 错误消息
func SystemError(w http.ResponseWriter, message string) {
	WriteJSON(w, http.StatusInternalServerError, 500, message, nil)
}

// BadRequest 请求错误响应（HTTP 400）
func BadRequest(w http.ResponseWriter, message string) {
	WriteJSON(w, http.StatusBadRequest, 400, message, nil)
}

// Unauthorized 未授权响应（HTTP 401）
func Unauthorized(w http.ResponseWriter, message string) {
	WriteJSON(w, http.StatusUnauthorized, 401, message, nil)
}

// Forbidden 禁止访问响应（HTTP 403）
func Forbidden(w http.ResponseWriter, message string) {
	WriteJSON(w, http.StatusForbidden, 403, message, nil)
}

// NotFound 未找到响应（HTTP 404）
func NotFound(w http.ResponseWriter, message string) {
	WriteJSON(w, http.StatusNotFound, 404, message, nil)
}

// WriteJSON 写入JSON响应
// httpCode: HTTP状态码（200表示正常，500表示系统错误等）
// code: 业务状态码（0表示成功，非0表示业务错误）
// message: 响应消息
// data: 响应数据
func WriteJSON(w http.ResponseWriter, httpCode, code int, message string, data interface{}) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(httpCode)

	response := Response{
		Code:      code,
		Message:   message,
		Timestamp: time.Now().Unix(),
		Data:      data,
	}

	json.NewEncoder(w).Encode(response)
}

// SuccessPage 分页成功响应
// list: 数据列表
// total: 总记录数
// page: 当前页码
// pageSize: 每页大小
func SuccessPage(w http.ResponseWriter, list interface{}, total int64, page, pageSize int) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)

	response := PageResponse{
		Code:      0,
		Message:   "success",
		Timestamp: time.Now().Unix(),
		Data: &PageData{
			List:     list,
			Total:    total,
			Page:     page,
			PageSize: pageSize,
		},
	}

	json.NewEncoder(w).Encode(response)
}

// SuccessPageWithMessage 带消息的分页成功响应
func SuccessPageWithMessage(w http.ResponseWriter, message string, list interface{}, total int64, page, pageSize int) {
	w.Header().Set("Content-Type", "application/json; charset=utf-8")
	w.WriteHeader(http.StatusOK)

	response := PageResponse{
		Code:      0,
		Message:   message,
		Timestamp: time.Now().Unix(),
		Data: &PageData{
			List:     list,
			Total:    total,
			Page:     page,
			PageSize: pageSize,
		},
	}

	json.NewEncoder(w).Encode(response)
}