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

package device

import (
	"context"
	"errors"
	"strconv"

	"git.huangwc.com/pig/pig-farm-controller/internal/app/controller"
	"git.huangwc.com/pig/pig-farm-controller/internal/app/dto"
	"git.huangwc.com/pig/pig-farm-controller/internal/app/service"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"

	"github.com/labstack/echo/v4"
	"gorm.io/gorm"
)

// AreaControllerController 区域主控控制器
type AreaControllerController struct {
	ctx                   context.Context
	areaControllerService service.AreaControllerService
}

// NewAreaControllerController 创建一个新的区域主控控制器实例
func NewAreaControllerController(
	ctx context.Context,
	areaControllerService service.AreaControllerService,
) *AreaControllerController {
	return &AreaControllerController{
		ctx:                   ctx,
		areaControllerService: areaControllerService,
	}
}

// CreateAreaController godoc
// @Summary      创建新区域主控
// @Description  根据提供的信息创建一个新区域主控
// @Tags         区域主控管理
// @Security     BearerAuth
// @Accept       json
// @Produce      json
// @Param        areaController body dto.CreateAreaControllerRequest true "区域主控信息"
// @Success      200 {object} controller.Response{data=dto.AreaControllerResponse}
// @Router       /api/v1/area-controllers [post]
func (c *AreaControllerController) CreateAreaController(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "CreateAreaController")

	const actionType = "创建区域主控"
	var req dto.CreateAreaControllerRequest
	if err := ctx.Bind(&req); err != nil {
		logger.Errorf("%s: 参数绑定失败: %v", actionType, err)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体: "+err.Error(), actionType, "请求体绑定失败", req)
	}

	resp, err := c.areaControllerService.CreateAreaController(reqCtx, &req)
	if err != nil {
		logger.Errorf("%s: 服务层创建失败: %v", actionType, err)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "创建区域主控失败: "+err.Error(), actionType, "服务层创建失败", req)
	}

	logger.Infof("%s: 区域主控创建成功, ID: %d", actionType, resp.ID)
	return controller.SendSuccessWithAudit(ctx, controller.CodeCreated, "区域主控创建成功", resp, actionType, "区域主控创建成功", resp)
}

// GetAreaController godoc
// @Summary      获取区域主控信息
// @Description  根据ID获取单个区域主控的详细信息
// @Tags         区域主控管理
// @Security     BearerAuth
// @Produce      json
// @Param        id path string true "区域主控ID"
// @Success      200 {object} controller.Response{data=dto.AreaControllerResponse}
// @Router       /api/v1/area-controllers/{id} [get]
func (c *AreaControllerController) GetAreaController(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "GetAreaController")

	const actionType = "获取区域主控"
	acID := ctx.Param("id")

	id, err := strconv.ParseUint(acID, 10, 64)
	if err != nil {
		logger.Errorf("%s: 无效的ID: %s", actionType, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的ID: "+acID, actionType, "无效的ID", acID)
	}

	resp, err := c.areaControllerService.GetAreaController(reqCtx, uint32(id))
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			logger.Warnf("%s: 区域主控不存在, ID: %s", actionType, acID)
			return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, "区域主控未找到", actionType, "区域主控不存在", acID)
		}
		logger.Errorf("%s: 服务层获取失败: %v, ID: %s", actionType, err, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "获取区域主控信息失败: "+err.Error(), actionType, "服务层获取失败", acID)
	}

	logger.Infof("%s: 获取区域主控信息成功, ID: %d", actionType, resp.ID)
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取区域主控信息成功", resp, actionType, "获取区域主控信息成功", resp)
}

// ListAreaControllers godoc
// @Summary      获取所有区域主控列表
// @Description  获取系统中所有区域主控的列表
// @Tags         区域主控管理
// @Security     BearerAuth
// @Produce      json
// @Success      200 {object} controller.Response{data=[]dto.AreaControllerResponse}
// @Router       /api/v1/area-controllers [get]
func (c *AreaControllerController) ListAreaControllers(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "ListAreaControllers")

	const actionType = "获取区域主控列表"
	resp, err := c.areaControllerService.ListAreaControllers(reqCtx)
	if err != nil {
		logger.Errorf("%s: 服务层获取列表失败: %v", actionType, err)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "获取区域主控列表失败: "+err.Error(), actionType, "服务层获取列表失败", nil)
	}

	logger.Infof("%s: 获取区域主控列表成功, 数量: %d", actionType, len(resp))
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "获取区域主控列表成功", resp, actionType, "获取区域主控列表成功", resp)
}

// UpdateAreaController godoc
// @Summary      更新区域主控信息
// @Description  根据ID更新一个已存在的区域主控信息
// @Tags         区域主控管理
// @Security     BearerAuth
// @Accept       json
// @Produce      json
// @Param        id path string true "区域主控ID"
// @Param        areaController body dto.UpdateAreaControllerRequest true "要更新的区域主控信息"
// @Success      200 {object} controller.Response{data=dto.AreaControllerResponse}
// @Router       /api/v1/area-controllers/{id} [put]
func (c *AreaControllerController) UpdateAreaController(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "UpdateAreaController")

	const actionType = "更新区域主控"
	acID := ctx.Param("id")

	var req dto.UpdateAreaControllerRequest
	if err := ctx.Bind(&req); err != nil {
		logger.Errorf("%s: 参数绑定失败: %v", actionType, err)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求体: "+err.Error(), actionType, "请求体绑定失败", req)
	}
	id, err := strconv.ParseUint(acID, 10, 64)
	if err != nil {
		logger.Errorf("%s: 无效的ID: %s", actionType, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的ID: "+acID, actionType, "无效的ID", acID)
	}

	resp, err := c.areaControllerService.UpdateAreaController(reqCtx, uint32(id), &req)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			logger.Warnf("%s: 区域主控不存在, ID: %s", actionType, acID)
			return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, "区域主控未找到", actionType, "区域主控不存在", acID)
		}
		logger.Errorf("%s: 服务层更新失败: %v, ID: %s", actionType, err, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "更新区域主控失败: "+err.Error(), actionType, "服务层更新失败", acID)
	}

	logger.Infof("%s: 区域主控更新成功, ID: %d", actionType, resp.ID)
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "区域主控更新成功", resp, actionType, "区域主控更新成功", resp)
}

// DeleteAreaController godoc
// @Summary      删除区域主控
// @Description  根据ID删除一个区域主控（软删除）
// @Tags         区域主控管理
// @Security     BearerAuth
// @Produce      json
// @Param        id path string true "区域主控ID"
// @Success      200 {object} controller.Response
// @Router       /api/v1/area-controllers/{id} [delete]
func (c *AreaControllerController) DeleteAreaController(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "DeleteAreaController")

	const actionType = "删除区域主控"
	acID := ctx.Param("id")

	id, err := strconv.ParseUint(acID, 10, 64)
	if err != nil {
		logger.Errorf("%s: 无效的ID: %s", actionType, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的ID: "+acID, actionType, "无效的ID", acID)
	}

	if err := c.areaControllerService.DeleteAreaController(reqCtx, uint32(id)); err != nil {
		switch {
		case errors.Is(err, gorm.ErrRecordNotFound):
			logger.Warnf("%s: 区域主控不存在, ID: %s", actionType, acID)
			return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, "区域主控未找到", actionType, "区域主控不存在", acID)
		case errors.Is(err, service.ErrAreaControllerInUse):
			logger.Warnf("%s: 尝试删除正在被使用的主控, ID: %s", actionType, acID)
			return controller.SendErrorWithAudit(ctx, controller.CodeConflict, err.Error(), actionType, "主控正在被使用", acID)
		default:
			logger.Errorf("%s: 服务层删除失败: %v, ID: %s", actionType, err, acID)
			return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "删除区域主控失败: "+err.Error(), actionType, "服务层删除失败", acID)
		}
	}

	logger.Infof("%s: 区域主控删除成功, ID: %s", actionType, acID)
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "区域主控删除成功", nil, actionType, "区域主控删除成功", acID)
}

// StartUpgrade godoc
// @Summary      启动区域主控OTA升级
// @Description  为指定的区域主控上传固件并启动一个OTA升级任务
// @Tags         区域主控管理
// @Security     BearerAuth
// @Accept       mpfd
// @Produce      json
// @Param        id   path      string  true  "区域主控ID"
// @Param        firmware_file formData file true "固件压缩包文件"
// @Success      200  {object}  controller.Response{data=dto.OtaUpgradeResponse}
// @Router       /api/v1/area-controllers/{id}/ota/start [post]
func (c *AreaControllerController) StartUpgrade(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "StartUpgrade")
	const actionType = "启动区域主控OTA升级"

	acID := ctx.Param("id")
	id, err := strconv.ParseUint(acID, 10, 64)
	if err != nil {
		logger.Errorf("%s: 无效的区域主控ID: %s", actionType, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的区域主控ID: "+acID, actionType, "无效的ID", acID)
	}

	var req dto.OtaUpgradeRequest
	if err := ctx.Bind(&req); err != nil {
		logger.Errorf("%s: 参数绑定失败: %v", actionType, err)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的请求: "+err.Error(), actionType, "请求绑定失败", req)
	}

	resp, err := c.areaControllerService.StartUpgrade(reqCtx, uint32(id), &req)
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			logger.Warnf("%s: 区域主控不存在, ID: %s", actionType, acID)
			return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, "区域主控未找到", actionType, "区域主控不存在", acID)
		}
		logger.Errorf("%s: 服务层启动升级失败: %v, ID: %s", actionType, err, acID)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "启动升级失败: "+err.Error(), actionType, "服务层启动升级失败", acID)
	}

	logger.Infof("%s: 升级任务启动成功, 任务ID: %d", actionType, resp.TaskID)
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "升级任务启动成功", resp, actionType, "升级任务启动成功", resp)
}

// GetUpgradeProgress godoc
// @Summary      查询OTA升级进度
// @Description  根据任务ID查询指定OTA升级任务的当前进度
// @Tags         区域主控管理
// @Security     BearerAuth
// @Produce      json
// @Param        taskId   path      string  true  "OTA任务ID"
// @Success      200      {object}  controller.Response{data=dto.OtaUpgradeProgressResponse}
// @Router       /api/v1/area-controllers/ota/progress/{taskId} [get]
func (c *AreaControllerController) GetUpgradeProgress(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "GetUpgradeProgress")
	const actionType = "查询OTA升级进度"

	taskIDStr := ctx.Param("taskId")
	taskID, err := strconv.ParseUint(taskIDStr, 10, 64)
	if err != nil {
		logger.Errorf("%s: 无效的任务ID: %s", actionType, taskIDStr)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的任务ID: "+taskIDStr, actionType, "无效的ID", taskIDStr)
	}

	resp, err := c.areaControllerService.GetUpgradeProgress(reqCtx, uint32(taskID))
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			logger.Warnf("%s: 升级任务不存在, ID: %s", actionType, taskIDStr)
			return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, "升级任务未找到", actionType, "升级任务不存在", taskIDStr)
		}
		logger.Errorf("%s: 服务层查询进度失败: %v, ID: %s", actionType, err, taskIDStr)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "查询进度失败: "+err.Error(), actionType, "服务层查询进度失败", taskIDStr)
	}

	logger.Infof("%s: 查询进度成功, 任务ID: %d", actionType, resp.TaskID)
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "查询进度成功", resp, actionType, "查询进度成功", resp)
}

// StopUpgrade godoc
// @Summary      停止OTA升级任务
// @Description  根据任务ID请求停止一个正在进行的OTA升级任务
// @Tags         区域主控管理
// @Security     BearerAuth
// @Produce      json
// @Param        taskId   path      string  true  "OTA任务ID"
// @Success      200      {object}  controller.Response
// @Router       /api/v1/area-controllers/ota/tasks/{taskId}/stop [post]
func (c *AreaControllerController) StopUpgrade(ctx echo.Context) error {
	reqCtx, logger := logs.Trace(ctx.Request().Context(), c.ctx, "StopUpgrade")
	const actionType = "停止OTA升级任务"

	taskIDStr := ctx.Param("taskId")
	taskID, err := strconv.ParseUint(taskIDStr, 10, 64)
	if err != nil {
		logger.Errorf("%s: 无效的任务ID: %s", actionType, taskIDStr)
		return controller.SendErrorWithAudit(ctx, controller.CodeBadRequest, "无效的任务ID: "+taskIDStr, actionType, "无效的ID", taskIDStr)
	}

	err = c.areaControllerService.StopUpgrade(reqCtx, uint32(taskID))
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			logger.Warnf("%s: 升级任务不存在, ID: %s", actionType, taskIDStr)
			return controller.SendErrorWithAudit(ctx, controller.CodeNotFound, "升级任务未找到", actionType, "升级任务不存在", taskIDStr)
		}
		logger.Errorf("%s: 服务层停止任务失败: %v, ID: %s", actionType, err, taskIDStr)
		return controller.SendErrorWithAudit(ctx, controller.CodeInternalError, "停止任务失败: "+err.Error(), actionType, "服务层停止任务失败", taskIDStr)
	}

	logger.Infof("%s: 停止任务请求成功, 任务ID: %s", actionType, taskIDStr)
	return controller.SendSuccessWithAudit(ctx, controller.CodeSuccess, "停止任务请求成功", nil, actionType, "停止任务请求成功", taskIDStr)
}