package api

import (
	"u-desk/internal/service"
	"u-desk/internal/storage/models"
)

// ConnectionAPI 连接管理API
type ConnectionAPI struct {
	connService *service.ConnectionService
}

// NewConnectionAPI 创建连接管理API
func NewConnectionAPI() (*ConnectionAPI, error) {
	connService, err := service.NewConnectionService()
	if err != nil {
		return nil, err
	}
	return &ConnectionAPI{connService}, nil
}

// SaveConnectionRequest 保存连接请求结构体
type SaveConnectionRequest struct {
	ID       uint   `json:"id"`
	Name     string `json:"name"`
	Type     string `json:"type"`
	Host     string `json:"host"`
	Port     int    `json:"port"`
	Username string `json:"username"`
	Password string `json:"password"`
	Database string `json:"database"`
	Options  string `json:"options"`
}

// SaveDbConnection 保存数据库连接配置
func (api *ConnectionAPI) SaveDbConnection(req SaveConnectionRequest) error {
	conn := &models.DbConnection{
		ID:       req.ID,
		Name:     req.Name,
		Type:     req.Type,
		Host:     req.Host,
		Port:     req.Port,
		Username: req.Username,
		Password: req.Password,
		Database: req.Database,
		Options:  req.Options,
	}
	return api.connService.SaveConnection(conn)
}

// ListDbConnections 获取连接列表
func (api *ConnectionAPI) ListDbConnections() ([]map[string]interface{}, error) {
	connections, err := api.connService.ListConnections()
	if err != nil {
		return nil, err
	}

	result := make([]map[string]interface{}, len(connections))
	timeFormat := "2006-01-02 15:04:05"
	for i, conn := range connections {
		result[i] = map[string]interface{}{
			"id":         conn.ID,
			"name":       conn.Name,
			"type":       conn.Type,
			"host":       conn.Host,
			"port":       conn.Port,
			"username":   conn.Username,
			"database":   conn.Database,
			"options":    conn.Options,
			"created_at": conn.CreatedAt.Format(timeFormat),
			"updated_at": conn.UpdatedAt.Format(timeFormat),
		}
	}
	return result, nil
}

func (api *ConnectionAPI) DeleteDbConnection(id uint) error {
	return api.connService.DeleteConnection(id)
}

func (api *ConnectionAPI) TestDbConnection(id uint) error {
	return api.connService.TestConnection(id)
}

// TestConnectionRequest 测试连接请求结构体（不保存数据）
type TestConnectionRequest struct {
	ID       uint   `json:"id"` // 编辑模式下的连接ID（用于获取已保存的密码）
	Type     string `json:"type"`
	Host     string `json:"host"`
	Port     int    `json:"port"`
	Username string `json:"username"`
	Password string `json:"password"`
	Database string `json:"database"`
	Options  string `json:"options"`
}

// TestDbConnectionWithParams 测试数据库连接（直接传入参数，不保存数据）
func (api *ConnectionAPI) TestDbConnectionWithParams(req TestConnectionRequest) error {
	return api.connService.TestConnectionWithParams(
		req.Type,
		req.Host,
		req.Port,
		req.Username,
		req.Password,
		req.Database,
		req.Options,
		req.ID,
	)
}