junhong_cmp_fiber/internal/handler/admin/asset.go

package admin

import (
	"strconv"

	"github.com/gofiber/fiber/v2"

	dto "github.com/break/junhong_cmp_fiber/internal/model/dto"
	assetService "github.com/break/junhong_cmp_fiber/internal/service/asset"
	deviceService "github.com/break/junhong_cmp_fiber/internal/service/device"
	iotCardService "github.com/break/junhong_cmp_fiber/internal/service/iot_card"
	pollingSvc "github.com/break/junhong_cmp_fiber/internal/service/polling"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
	"github.com/break/junhong_cmp_fiber/pkg/errors"
	"github.com/break/junhong_cmp_fiber/pkg/middleware"
	"github.com/break/junhong_cmp_fiber/pkg/response"
)

// AssetHandler 资产管理处理器
// 提供统一的资产解析、实时状态、套餐查询、停复机、轮询管控等接口
type AssetHandler struct {
	assetService      *assetService.Service
	deviceService     *deviceService.Service
	iotCardStopResume *iotCardService.StopResumeService
	assetPolling      *pollingSvc.AssetPollingService
}

// NewAssetHandler 创建资产管理处理器
func NewAssetHandler(
	assetSvc *assetService.Service,
	deviceSvc *deviceService.Service,
	iotCardStopResume *iotCardService.StopResumeService,
	assetPolling *pollingSvc.AssetPollingService,
) *AssetHandler {
	return &AssetHandler{
		assetService:      assetSvc,
		deviceService:     deviceSvc,
		iotCardStopResume: iotCardStopResume,
		assetPolling:      assetPolling,
	}
}

// Resolve 通过任意标识符解析资产（设备或卡）
// GET /api/admin/assets/resolve/:identifier
func (h *AssetHandler) Resolve(c *fiber.Ctx) error {
	identifier := c.Params("identifier")
	if identifier == "" {
		return errors.New(errors.CodeInvalidParam, "标识符不能为空")
	}

	result, err := h.assetService.Resolve(c.UserContext(), identifier)
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

// RealtimeStatus 获取资产实时状态
// GET /api/admin/assets/:asset_type/:id/realtime-status
func (h *AssetHandler) RealtimeStatus(c *fiber.Ctx) error {
	assetType := c.Params("asset_type")
	id, err := strconv.ParseUint(c.Params("id"), 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的资产ID")
	}

	result, err := h.assetService.GetRealtimeStatus(c.UserContext(), assetType, uint(id))
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

// Refresh 刷新资产状态（调网关同步）
// POST /api/admin/assets/:asset_type/:id/refresh
func (h *AssetHandler) Refresh(c *fiber.Ctx) error {
	// 企业账号只读，不允许主动触发运营商刷新
	userType := middleware.GetUserTypeFromContext(c.UserContext())
	if userType == constants.UserTypeEnterprise {
		return errors.New(errors.CodeForbidden, "企业账号无权主动刷新资产状态")
	}
	assetType := c.Params("asset_type")
	id, err := strconv.ParseUint(c.Params("id"), 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的资产ID")
	}

	result, err := h.assetService.Refresh(c.UserContext(), assetType, uint(id))
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

// Packages 获取资产所有套餐列表（支持分页，默认 page=1 pageSize=50）
// GET /api/admin/assets/:asset_type/:id/packages
func (h *AssetHandler) Packages(c *fiber.Ctx) error {
	assetType := c.Params("asset_type")
	id, err := strconv.ParseUint(c.Params("id"), 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的资产ID")
	}

	page := c.QueryInt("page", 1)
	pageSize := c.QueryInt("page_size", 50)

	result, err := h.assetService.GetPackages(c.UserContext(), assetType, uint(id), page, pageSize)
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

// CurrentPackage 获取资产当前生效套餐
// GET /api/admin/assets/:asset_type/:id/current-package
func (h *AssetHandler) CurrentPackage(c *fiber.Ctx) error {
	assetType := c.Params("asset_type")
	id, err := strconv.ParseUint(c.Params("id"), 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的资产ID")
	}

	result, err := h.assetService.GetCurrentPackage(c.UserContext(), assetType, uint(id))
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

// StopDevice 设备停机（批量停机设备下所有已实名卡）
// POST /api/admin/assets/device/:device_id/stop
func (h *AssetHandler) StopDevice(c *fiber.Ctx) error {
	deviceID, err := strconv.ParseUint(c.Params("device_id"), 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的设备ID")
	}

	result, err := h.deviceService.StopDevice(c.UserContext(), uint(deviceID))
	if err != nil {
		return err
	}

	return response.Success(c, result)
}

// StartDevice 设备复机（批量复机设备下所有已实名卡）
// POST /api/admin/assets/device/:device_id/start
func (h *AssetHandler) StartDevice(c *fiber.Ctx) error {
	deviceID, err := strconv.ParseUint(c.Params("device_id"), 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam, "无效的设备ID")
	}

	if err := h.deviceService.StartDevice(c.UserContext(), uint(deviceID)); err != nil {
		return err
	}

	return response.Success(c, nil)
}

// StopCard 单卡停机（通过ICCID）
// POST /api/admin/assets/card/:iccid/stop
func (h *AssetHandler) StopCard(c *fiber.Ctx) error {
	iccid := c.Params("iccid")
	if iccid == "" {
		return errors.New(errors.CodeInvalidParam, "ICCID不能为空")
	}

	if err := h.iotCardStopResume.ManualStopCard(c.UserContext(), iccid); err != nil {
		return err
	}

	return response.Success(c, nil)
}

// StartCard 单卡复机（通过ICCID）
// POST /api/admin/assets/card/:iccid/start
func (h *AssetHandler) StartCard(c *fiber.Ctx) error {
	iccid := c.Params("iccid")
	if iccid == "" {
		return errors.New(errors.CodeInvalidParam, "ICCID不能为空")
	}

	if err := h.iotCardStopResume.ManualStartCard(c.UserContext(), iccid); err != nil {
		return err
	}

	return response.Success(c, nil)
}

// UpdatePollingStatus 更新资产轮询状态
// PATCH /api/admin/assets/:asset_type/:id/polling-status
func (h *AssetHandler) UpdatePollingStatus(c *fiber.Ctx) error {
	assetType := c.Params("asset_type")
	idStr := c.Params("id")

	if assetType == "" || idStr == "" {
		return errors.New(errors.CodeInvalidParam)
	}

	id, err := strconv.ParseUint(idStr, 10, 64)
	if err != nil {
		return errors.New(errors.CodeInvalidParam)
	}

	var req dto.UpdateAssetPollingStatusRequest
	if err := c.BodyParser(&req); err != nil {
		return errors.New(errors.CodeInvalidParam)
	}

	_ = middleware.GetUserIDFromContext(c.UserContext())

	if h.assetPolling == nil {
		return errors.New(errors.CodeInternalError, "轮询管控服务未配置")
	}

	if err := h.assetPolling.UpdatePollingStatus(c.UserContext(), assetType, uint(id), req.EnablePolling); err != nil {
		return err
	}

	return response.Success(c, &dto.UpdateAssetPollingStatusResponse{
		AssetType:     assetType,
		AssetID:       uint(id),
		EnablePolling: req.EnablePolling,
	})
}