MeBox/internal/handler/site_handler.go

// Package handler — PT 站点管理 HTTP 处理。
package handler

import (
	"net/http"

	"github.com/gin-gonic/gin"

	"github.com/ShukeBta/MediaStationGo/internal/model"
	"github.com/ShukeBta/MediaStationGo/internal/service"
)

// SiteHandler 站点管理 CRUD。
type SiteHandler struct {
	svc *service.Container
}

// NewSiteHandler 创建站点管理 Handler。
func NewSiteHandler(svc *service.Container) *SiteHandler {
	return &SiteHandler{svc: svc}
}

// ListSites 列出所有站点。
func (h *SiteHandler) ListSites(c *gin.Context) {
	sites, err := h.svc.Site.List(c.Request.Context())
	if err != nil {
		c.JSON(http.StatusInternalServerError, gin.H{"code": 1, "message": err.Error()})
		return
	}
	if sites == nil {
		sites = []model.Site{}
	}
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok", "data": sites})
}

// GetSite 获取单个站点详情（解密敏感字段）。
func (h *SiteHandler) GetSite(c *gin.Context) {
	site, err := h.svc.Site.GetByID(c.Request.Context(), c.Param("id"))
	if err != nil {
		if err == service.ErrSiteNotFound {
			c.JSON(http.StatusNotFound, gin.H{"code": 1, "message": "site not found"})
			return
		}
		c.JSON(http.StatusInternalServerError, gin.H{"code": 1, "message": err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok", "data": site})
}

// CreateSite 创建站点。
func (h *SiteHandler) CreateSite(c *gin.Context) {
	var site model.Site
	if err := c.ShouldBindJSON(&site); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "message": err.Error()})
		return
	}
	created, err := h.svc.Site.Create(c.Request.Context(), &site)
	if err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "message": err.Error()})
		return
	}
	c.JSON(http.StatusCreated, gin.H{"code": 0, "message": "ok", "data": created})
}

// UpdateSite 更新站点。
func (h *SiteHandler) UpdateSite(c *gin.Context) {
	var site model.Site
	if err := c.ShouldBindJSON(&site); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "message": err.Error()})
		return
	}
	site.ID = c.Param("id")
	updated, err := h.svc.Site.Update(c.Request.Context(), &site)
	if err != nil {
		if err == service.ErrSiteNotFound {
			c.JSON(http.StatusNotFound, gin.H{"code": 1, "message": "site not found"})
			return
		}
		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "message": err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok", "data": updated})
}

// DeleteSite 删除站点。
func (h *SiteHandler) DeleteSite(c *gin.Context) {
	if err := h.svc.Site.Delete(c.Request.Context(), c.Param("id")); err != nil {
		if err == service.ErrSiteNotFound {
			c.JSON(http.StatusNotFound, gin.H{"code": 1, "message": "site not found"})
			return
		}
		c.JSON(http.StatusInternalServerError, gin.H{"code": 1, "message": err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok"})
}

// TestSite 测试站点连通性。
func (h *SiteHandler) TestSite(c *gin.Context) {
	if err := h.svc.Site.Authenticate(c.Request.Context(), c.Param("id")); err != nil {
		c.JSON(http.StatusBadRequest, gin.H{"code": 1, "message": err.Error()})
		return
	}
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok"})
}

// GetSiteTypes 返回支持的站点类型列表。
func (h *SiteHandler) GetSiteTypes(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok", "data": model.SiteTypes()})
}

// GetAuthTypes 返回支持的认证方式列表。
func (h *SiteHandler) GetAuthTypes(c *gin.Context) {
	c.JSON(http.StatusOK, gin.H{"code": 0, "message": "ok", "data": model.AuthTypes()})
}