package handlers

import (
	"fmt"
	"net/http"
	"os"
	"path/filepath"
	"strings"
	"time"

	"software-download-center/database"
	"software-download-center/models"
	"software-download-center/utils"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

// LogEntry 日志条目
type LogEntry struct {
	Time    string `json:"time"`
	Level   string `json:"level"`
	Message string `json:"message"`
}

var logBuffer []LogEntry
var maxLogEntries = 1000

// AddLog 添加日志到缓冲区
func AddLog(level, message string) {
	entry := LogEntry{
		Time:    time.Now().Format("2006-01-02 15:04:05"),
		Level:   level,
		Message: message,
	}
	
	logBuffer = append(logBuffer, entry)
	
	// 同时输出到控制台（使用 utils.Logger）
	logger := utils.NewLogger()
	switch level {
	case "ERROR":
		logger.Error(message)
	case "WARN":
		logger.Warn(message)
	case "INFO":
		logger.Info(message)
	default:
		logger.System(message)
	}
	
	// 保持缓冲区大小
	if len(logBuffer) > maxLogEntries {
		logBuffer = logBuffer[len(logBuffer)-maxLogEntries:]
	}
}

// GetLogBuffer 获取日志缓冲区（用于外部访问）
func GetLogBuffer() []LogEntry {
	return logBuffer
}

// GetLogs 获取日志
func GetLogs(c *gin.Context) {
	limit := 100
	if limitStr := c.Query("limit"); limitStr != "" {
		fmt.Sscanf(limitStr, "%d", &limit)
	}

	start := len(logBuffer) - limit
	if start < 0 {
		start = 0
	}

	logs := logBuffer[start:]
	c.JSON(http.StatusOK, gin.H{
		"logs": logs,
		"total": len(logBuffer),
	})
}

// GetRoutes 获取所有路由
func GetRoutes(c *gin.Context) {
	var routes []models.Route
	if err := database.DB.Order("`order` ASC, id ASC").Find(&routes).Error; err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "获取路由失败",
		})
		return
	}
	c.JSON(http.StatusOK, gin.H{
		"routes": routes,
	})
}

// CreateRoute 创建路由
func CreateRoute(c *gin.Context) {
	var route models.Route
	if err := c.ShouldBindJSON(&route); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "请求参数错误: " + err.Error(),
		})
		return
	}

	// 验证路径是否已存在
	var existingRoute models.Route
	if err := database.DB.Where("path = ? AND method = ?", route.Path, route.Method).First(&existingRoute).Error; err == nil {
		AddLog("WARN", fmt.Sprintf("创建路由失败（已存在）: %s %s (IP: %s)", route.Method, route.Path, c.ClientIP()))
		c.JSON(http.StatusConflict, gin.H{
			"error": "路由已存在",
		})
		return
	}

	if err := database.DB.Create(&route).Error; err != nil {
		AddLog("ERROR", fmt.Sprintf("创建路由数据库错误: %s %s - %s (IP: %s)", route.Method, route.Path, err.Error(), c.ClientIP()))
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "创建路由失败",
		})
		return
	}

	AddLog("INFO", fmt.Sprintf("创建路由成功: %s %s (类型: %s, IP: %s)", route.Method, route.Path, route.Type, c.ClientIP()))
	c.JSON(http.StatusOK, gin.H{
		"message": "路由创建成功",
		"route":   route,
	})
}

// UpdateRoute 更新路由
func UpdateRoute(c *gin.Context) {
	id := c.Param("id")
	var route models.Route
	oldPath := route.Path
	oldMethod := route.Method

	if err := database.DB.First(&route, id).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			AddLog("WARN", fmt.Sprintf("更新路由失败（不存在）: ID=%s (IP: %s)", id, c.ClientIP()))
			c.JSON(http.StatusNotFound, gin.H{
				"error": "路由不存在",
			})
			return
		}
		AddLog("ERROR", fmt.Sprintf("查询路由失败: ID=%s - %s (IP: %s)", id, err.Error(), c.ClientIP()))
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "查询路由失败",
		})
		return
	}

	oldPath = route.Path
	oldMethod = route.Method

	if err := c.ShouldBindJSON(&route); err != nil {
		AddLog("WARN", fmt.Sprintf("更新路由请求参数错误: ID=%s - %s (IP: %s)", id, err.Error(), c.ClientIP()))
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "请求参数错误",
		})
		return
	}

	if err := database.DB.Save(&route).Error; err != nil {
		AddLog("ERROR", fmt.Sprintf("更新路由数据库错误: %s %s - %s (IP: %s)", route.Method, route.Path, err.Error(), c.ClientIP()))
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "更新路由失败",
		})
		return
	}

	AddLog("INFO", fmt.Sprintf("更新路由成功: %s %s -> %s %s (IP: %s)", oldMethod, oldPath, route.Method, route.Path, c.ClientIP()))
	c.JSON(http.StatusOK, gin.H{
		"message": "路由更新成功",
		"route":   route,
	})
}

// DeleteRoute 删除路由
func DeleteRoute(c *gin.Context) {
	id := c.Param("id")
	var route models.Route
	if err := database.DB.First(&route, id).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			AddLog("WARN", fmt.Sprintf("删除路由失败（不存在）: ID=%s (IP: %s)", id, c.ClientIP()))
			c.JSON(http.StatusNotFound, gin.H{
				"error": "路由不存在",
			})
			return
		}
		AddLog("ERROR", fmt.Sprintf("查询路由失败: ID=%s - %s (IP: %s)", id, err.Error(), c.ClientIP()))
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "查询路由失败",
		})
		return
	}

	routeInfo := fmt.Sprintf("%s %s", route.Method, route.Path)

	if err := database.DB.Delete(&route).Error; err != nil {
		AddLog("ERROR", fmt.Sprintf("删除路由数据库错误: %s - %s (IP: %s)", routeInfo, err.Error(), c.ClientIP()))
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "删除路由失败",
		})
		return
	}

	AddLog("INFO", fmt.Sprintf("删除路由成功: %s (IP: %s)", routeInfo, c.ClientIP()))
	c.JSON(http.StatusOK, gin.H{
		"message": "路由删除成功",
	})
}

// GetFiles 获取文件列表
func GetFiles(c *gin.Context) {
	dir := c.Query("dir")
	if dir == "" {
		dir = "public/downloads"
	}

	// 安全检查：防止目录遍历
	if strings.Contains(dir, "..") {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "无效的目录路径",
		})
		return
	}

	files, err := os.ReadDir(dir)
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "读取目录失败",
		})
		return
	}

	var fileList []gin.H
	for _, file := range files {
		info, _ := file.Info()
		fileList = append(fileList, gin.H{
			"name":    file.Name(),
			"size":    info.Size(),
			"mod_time": info.ModTime().Format("2006-01-02 15:04:05"),
			"is_dir":  file.IsDir(),
		})
	}

	c.JSON(http.StatusOK, gin.H{
		"files": fileList,
		"path":  dir,
	})
}

// SaveFile 保存文件
func SaveFile(c *gin.Context) {
	var req struct {
		Path    string `json:"path" binding:"required"`
		Content string `json:"content" binding:"required"`
	}

	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "请求参数错误",
		})
		return
	}

	// 安全检查
	if strings.Contains(req.Path, "..") {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "无效的文件路径",
		})
		return
	}

	// 确保目录存在
	dir := filepath.Dir(req.Path)
	if err := os.MkdirAll(dir, 0755); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "创建目录失败",
		})
		return
	}

	// 保存文件
	if err := os.WriteFile(req.Path, []byte(req.Content), 0644); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "保存文件失败",
		})
		return
	}

	AddLog("INFO", fmt.Sprintf("保存文件: %s", req.Path))
	c.JSON(http.StatusOK, gin.H{
		"message": "文件保存成功",
	})
}

// ReadFile 读取文件
func ReadFile(c *gin.Context) {
	path := c.Query("path")
	if path == "" {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "缺少文件路径参数",
		})
		return
	}

	// 安全检查
	if strings.Contains(path, "..") {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "无效的文件路径",
		})
		return
	}

	content, err := os.ReadFile(path)
	if err != nil {
		AddLog("ERROR", fmt.Sprintf("读取文件失败: %s - %s (IP: %s)", path, err.Error(), c.ClientIP()))
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "读取文件失败",
		})
		return
	}

	AddLog("INFO", fmt.Sprintf("读取文件: %s (大小: %d 字节, IP: %s)", path, len(content), c.ClientIP()))
	c.JSON(http.StatusOK, gin.H{
		"content": string(content),
		"path":    path,
		"size":    len(content),
	})
}

// ReloadRoutes 热重载路由和配置
func ReloadRoutes(c *gin.Context) {
	var req struct {
		Type string `json:"type"` // "config" 或 "all"
	}
	
	if err := c.ShouldBindJSON(&req); err != nil {
		req.Type = "all"
	}
	
	if req.Type == "config" || req.Type == "all" {
		// 重新加载所有配置文件
		files := []string{"tool-status.json", "update-info.json", "media-types.json"}
		successCount := 0
		for _, file := range files {
			if _, err := os.Stat(filepath.Join("public", file)); err == nil {
				if err := utils.ReloadConfig(file); err == nil {
					successCount++
				}
			}
		}
		AddLog("INFO", fmt.Sprintf("重新加载配置文件: %d/%d 成功 (IP: %s)", successCount, len(files), c.ClientIP()))
	}
	
	// 注意：Gin 不支持动态路由重载，路由更改需要重启服务器
	if req.Type == "routes" || req.Type == "all" {
		AddLog("INFO", fmt.Sprintf("路由热重载请求（需要重启服务器才能生效）(IP: %s)", c.ClientIP()))
		c.JSON(http.StatusOK, gin.H{
			"message": "配置已重新加载",
			"note":    "路由更改需要重启服务器才能生效，建议使用进程管理器（如 systemd、supervisor）",
		})
		return
	}
	
	c.JSON(http.StatusOK, gin.H{
		"message": "配置已重新加载",
	})
}

// GetSystemInfo 获取系统信息
func GetSystemInfo(c *gin.Context) {
	var userCount int64
	var routeCount int64
	database.DB.Model(&models.User{}).Count(&userCount)
	database.DB.Model(&models.Route{}).Count(&routeCount)

	c.JSON(http.StatusOK, gin.H{
		"users":      userCount,
		"routes":    routeCount,
		"logs":       len(logBuffer),
		"version":    "1.0.0",
		"server_time": time.Now().Format("2006-01-02 15:04:05"),
	})
}

// UpdateJSONConfig 更新 JSON 配置文件
func UpdateJSONConfig(c *gin.Context) {
	var req struct {
		File    string                 `json:"file" binding:"required"`
		Content map[string]interface{} `json:"content" binding:"required"`
		Reload  bool                   `json:"reload"` // 是否立即重新加载
	}

	if err := c.ShouldBindJSON(&req); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "请求参数错误",
		})
		return
	}

	// 验证文件路径
	allowedFiles := []string{"tool-status.json", "update-info.json", "media-types.json"}
	allowed := false
	for _, f := range allowedFiles {
		if req.File == f {
			allowed = true
			break
		}
	}

	if !allowed {
		c.JSON(http.StatusBadRequest, gin.H{
			"error": "不允许修改此文件",
		})
		return
	}

	// 使用配置工具保存并更新缓存
	if err := utils.SaveConfig(req.File, req.Content); err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{
			"error": "保存文件失败: " + err.Error(),
		})
		return
	}

	message := "配置文件更新成功"
	if req.Reload {
		// 重新加载配置到缓存
		if err := utils.ReloadConfig(req.File); err != nil {
			AddLog("WARN", fmt.Sprintf("配置文件已保存但重新加载失败: %s - %s", req.File, err.Error()))
			message = "配置文件已保存，但重新加载时出现警告"
		} else {
			AddLog("INFO", fmt.Sprintf("配置文件已保存并立即加载: %s", req.File))
			message = "配置文件已保存并立即生效"
		}
	} else {
		AddLog("INFO", fmt.Sprintf("更新配置文件: %s", req.File))
	}

	c.JSON(http.StatusOK, gin.H{
		"message": message,
		"file":    req.File,
	})
}