package server

import (
	"fmt"
	"net"
	"net/http"
	"os"

	searchagent "aichat/agents/search"
	sqlquery "aichat/agents/sql"
	"aichat/config"
	"aichat/conversation"
	"aichat/llm"
	"aichat/toolrouter"

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

type Server struct {
	cfg             *config.Config
	aiState         *llm.State
	searchState     *searchagent.State
	sqlState        *sqlquery.State
	toolRouterState *toolrouter.State
	store           *conversation.Store
	router          *gin.Engine
}

func New(cfg *config.Config, aiState *llm.State, searchState *searchagent.State, sqlState *sqlquery.State, toolRouterState *toolrouter.State, store *conversation.Store) *Server {
	s := &Server{
		cfg:             cfg,
		aiState:         aiState,
		searchState:     searchState,
		sqlState:        sqlState,
		toolRouterState: toolRouterState,
		store:           store,
		router:          gin.Default(),
	}
	s.router.LoadHTMLGlob("templates/*")
	s.router.Static("/static", "./static")
	s.registerRoutes()
	return s
}

func (s *Server) Run() error {
	switch s.cfg.Server.Mode {
	case "unix":
		return s.runUnix(s.cfg.Server.Address)
	default:
		fmt.Println("服务已启动，监听 TCP:", s.cfg.Server.Address)
		return s.router.Run(s.cfg.Server.Address)
	}
}

func (s *Server) runUnix(socketPath string) error {
	if _, statErr := os.Stat(socketPath); statErr == nil {
		os.Remove(socketPath)
	}
	ln, err := net.Listen("unix", socketPath)
	if err != nil {
		return fmt.Errorf("监听 Unix socket 失败: %w", err)
	}
	fmt.Println("服务已启动，监听 Unix socket:", socketPath)
	return http.Serve(ln, s.router)
}