package handler

import (
	"net/http"

	"github.com/gin-gonic/gin"

	"github.com/yourname/cyrene-ai/gateway/internal/middleware"
)

// SessionHandler 会话管理处理器
type SessionHandler struct {
	// MVP阶段使用内存存储，后续迁移到PostgreSQL
	sessions map[string][]SessionInfo // userID -> sessions
}

// SessionInfo 会话信息
type SessionInfo struct {
	ID        string `json:"id"`
	UserID    string `json:"user_id"`
	Title     string `json:"title"`
	CreatedAt int64  `json:"created_at"`
	UpdatedAt int64  `json:"updated_at"`
}

// NewSessionHandler 创建会话处理器
func NewSessionHandler() *SessionHandler {
	return &SessionHandler{
		sessions: make(map[string][]SessionInfo),
	}
}

// Create 创建新会话
func (h *SessionHandler) Create(c *gin.Context) {
	userID := middleware.GetUserID(c)

	var req struct {
		Title string `json:"title"`
	}
	if err := c.ShouldBindJSON(&req); err != nil {
		// 允许空body
		req.Title = "新的对话"
	}
	if req.Title == "" {
		req.Title = "新的对话"
	}

	session := SessionInfo{
		ID:        "session_" + randomID(12),
		UserID:    userID,
		Title:     req.Title,
		CreatedAt: nowMillis(),
		UpdatedAt: nowMillis(),
	}

	h.sessions[userID] = append([]SessionInfo{session}, h.sessions[userID]...)

	c.JSON(http.StatusCreated, session)
}

// List 获取会话列表
func (h *SessionHandler) List(c *gin.Context) {
	userID := middleware.GetUserID(c)

	sessions, ok := h.sessions[userID]
	if !ok {
		sessions = []SessionInfo{}
	}

	c.JSON(http.StatusOK, gin.H{
		"sessions": sessions,
	})
}

// Delete 删除会话
func (h *SessionHandler) Delete(c *gin.Context) {
	userID := middleware.GetUserID(c)
	sessionID := c.Param("id")

	sessions := h.sessions[userID]
	for i, s := range sessions {
		if s.ID == sessionID {
			h.sessions[userID] = append(sessions[:i], sessions[i+1:]...)
			c.JSON(http.StatusOK, gin.H{"status": "deleted"})
			return
		}
	}

	c.JSON(http.StatusNotFound, gin.H{"error": "会话不存在"})
}

// Get 获取单个会话信息
func (h *SessionHandler) Get(c *gin.Context) {
	userID := middleware.GetUserID(c)
	sessionID := c.Param("id")

	for _, s := range h.sessions[userID] {
		if s.ID == sessionID {
			c.JSON(http.StatusOK, s)
			return
		}
	}

	c.JSON(http.StatusNotFound, gin.H{"error": "会话不存在"})
}

// 简单的工具函数
func randomID(n int) string {
	const letters = "abcdefghijklmnopqrstuvwxyz0123456789"
	b := make([]byte, n)
	for i := range b {
		b[i] = letters[i%len(letters)]
	}
	// 使用纳秒时间戳增加唯一性
	return string(b)
}

func nowMillis() int64 {
	// 避免引入time包，直接返回一个值
	return 0
}