pig-farm-controller/internal/controller/device/device.go

// Package device 提供设备控制相关功能的控制器
// 实现设备控制、查询等操作
package device

import (
	"net/http"

	"git.huangwc.com/pig/pig-farm-controller/internal/api/middleware"
	"git.huangwc.com/pig/pig-farm-controller/internal/logs"
	"git.huangwc.com/pig/pig-farm-controller/internal/model"
	"git.huangwc.com/pig/pig-farm-controller/internal/storage/repository"
	"github.com/gin-gonic/gin"
)

// Controller 设备控制控制器
type Controller struct {
	deviceControlRepo repository.DeviceControlRepo
	deviceRepo        repository.DeviceRepo
	logger            *logs.Logger
}

// NewController 创建设备控制控制器实例
func NewController(deviceControlRepo repository.DeviceControlRepo, deviceRepo repository.DeviceRepo) *Controller {
	return &Controller{
		deviceControlRepo: deviceControlRepo,
		deviceRepo:        deviceRepo,
		logger:            logs.NewLogger(),
	}
}

// ControlRequest 设备控制请求结构体
type ControlRequest struct {
	ParentID   *uint  `json:"parent_id"` // 区域主控ID
	DeviceType string `json:"device_type" binding:"required,oneof=fan water_curtain"`
	DeviceID   string `json:"device_id" binding:"required"`
	Action     string `json:"action" binding:"required,oneof=on off"`
}

// Switch 设备控制接口
func (c *Controller) Switch(ctx *gin.Context) {
	// 从上下文中获取用户信息
	userValue, exists := ctx.Get("user")
	if !exists {
		ctx.JSON(http.StatusUnauthorized, gin.H{"error": "无法获取用户信息"})
		return
	}

	user, ok := userValue.(*middleware.AuthUser)
	if !ok {
		ctx.JSON(http.StatusInternalServerError, gin.H{"error": "用户信息格式错误"})
		return
	}

	var req ControlRequest
	if err := ctx.ShouldBindJSON(&req); err != nil {
		ctx.JSON(http.StatusBadRequest, gin.H{"error": "请求参数错误"})
		return
	}

	// 获取区域主控设备信息（如果提供了ParentID）
	var location string
	if req.ParentID != nil {
		parentDevice, err := c.deviceRepo.FindByID(*req.ParentID)
		if err != nil {
			c.logger.Error("查找区域主控设备失败: " + err.Error())
			ctx.JSON(http.StatusBadRequest, gin.H{"error": "无效的区域主控ID"})
			return
		}
		location = parentDevice.Name
	} else {
		location = "未知区域"
	}

	// TODO: 实际的设备控制逻辑
	// 这里暂时用TODO代替具体逻辑

	// 创建设备控制记录
	control := &model.DeviceControl{
		UserID:     user.ID,
		Location:   location,
		DeviceType: model.DeviceType(req.DeviceType),
		DeviceID:   req.DeviceID,
		Action:     req.Action,
		Status:     "success", // 假设控制成功
		Result:     "设备控制成功",
	}

	if err := c.deviceControlRepo.Create(control); err != nil {
		c.logger.Error("创建设备控制记录失败: " + err.Error())
		ctx.JSON(http.StatusInternalServerError, gin.H{"error": "设备控制失败"})
		return
	}

	ctx.JSON(http.StatusOK, gin.H{
		"message": "设备控制成功",
		"data": map[string]interface{}{
			"id":          control.ID,
			"location":    control.Location,
			"device_type": control.DeviceType,
			"device_id":   control.DeviceID,
			"action":      control.Action,
			"status":      control.Status,
			"result":      control.Result,
			"created_at":  control.CreatedAt,
		},
	})
}