backend-go/controllers/employees_controller.go

package controllers

import (
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"

	"git.beifan.cn/trace-system/backend-go/models"
	"git.beifan.cn/trace-system/backend-go/services"
)

// EmployeeSerialsController 员工序列号控制器
type EmployeeSerialsController struct {
	employeeSerialsService services.EmployeeSerialsService
}

// NewEmployeeSerialsController 创建员工序列号控制器实例
func NewEmployeeSerialsController() *EmployeeSerialsController {
	return &EmployeeSerialsController{
		employeeSerialsService: services.EmployeeSerialsService{},
	}
}

// Generate 生成员工序列号
// @Summary 生成员工序列号
// @Description 生成指定数量的员工序列号（无有效期）
// @Tags 员工赋码管理
// @Accept json
// @Produce json
// @Security BearerAuth
// @Param generateData body models.GenerateEmployeeSerialDTO true "生成数据"
// @Success 200 {object} models.DataResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Failure 500 {object} models.ErrorResponse
// @Router /employee-serials/generate [post]
func (c *EmployeeSerialsController) Generate(ctx *gin.Context) {
	userModel, ok := GetCurrentUser(ctx)
	if !ok {
		return
	}

	var generateData models.GenerateEmployeeSerialDTO
	if !BindJSON(ctx, &generateData) {
		return
	}

	serials, err := c.employeeSerialsService.Generate(
		generateData.CompanyName,
		generateData.Department,
		generateData.EmployeeName,
		generateData.Quantity,
		userModel.ID,
	)
	if err != nil {
		ErrorResponse(ctx, http.StatusInternalServerError, err.Error())
		return
	}

	SuccessResponse(ctx, "成功生成 "+strconv.Itoa(len(serials))+" 个员工序列号", gin.H{
		"serials": serials,
	})
}

// GenerateQRCode 生成员工二维码
// @Summary 生成员工二维码
// @Description 为指定员工序列号生成查询二维码
// @Tags 员工赋码管理
// @Accept json
// @Produce json
// @Security BearerAuth
// @Param serialNumber path string true "序列号"
// @Param qrCodeData body models.QRCodeDTO false "二维码数据"
// @Success 200 {object} models.QRCodeResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Failure 404 {object} models.ErrorResponse
// @Failure 500 {object} models.ErrorResponse
// @Router /employee-serials/{serialNumber}/qrcode [post]
func (c *EmployeeSerialsController) GenerateQRCode(ctx *gin.Context) {
	serialNumber := ctx.Param("serialNumber")

	var qrCodeData models.QRCodeDTO
	if !BindJSON(ctx, &qrCodeData) {
		return
	}

	protocol := "http"
	if ctx.Request.TLS != nil {
		protocol = "https"
	}

	qrCodeBase64, queryUrl, err := c.employeeSerialsService.GenerateQRCode(
		serialNumber,
		qrCodeData.BaseUrl,
		ctx.Request.Host,
		protocol,
	)
	if err != nil {
		ErrorResponse(ctx, http.StatusBadRequest, err.Error())
		return
	}

	SuccessResponse(ctx, "员工二维码生成成功", gin.H{
		"qrCodeData": qrCodeBase64,
		"queryUrl":   queryUrl,
	})
}

// Query 查询员工序列号信息
// @Summary 查询员工序列号信息
// @Description 查询指定员工序列号的详细信息
// @Tags 员工赋码查询
// @Produce json
// @Param serialNumber path string true "序列号"
// @Success 200 {object} models.DataResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 404 {object} models.ErrorResponse
// @Failure 500 {object} models.ErrorResponse
// @Router /employee-serials/{serialNumber}/query [get]
func (c *EmployeeSerialsController) Query(ctx *gin.Context) {
	serialNumber := ctx.Param("serialNumber")

	serial, err := c.employeeSerialsService.Query(serialNumber)
	if err != nil {
		ErrorResponse(ctx, http.StatusBadRequest, err.Error())
		return
	}

	SuccessResponse(ctx, "查询成功", gin.H{
		"serial": serial,
	})
}

// FindAll 获取员工序列号列表
// @Summary 获取员工序列号列表
// @Description 获取员工序列号列表，支持分页和搜索
// @Tags 员工赋码管理
// @Produce json
// @Security BearerAuth
// @Param page query int false "页码"
// @Param limit query int false "每页数量"
// @Param search query string false "搜索关键词"
// @Success 200 {object} models.PaginationResponse
// @Failure 401 {object} models.ErrorResponse
// @Failure 500 {object} models.ErrorResponse
// @Router /employee-serials [get]
func (c *EmployeeSerialsController) FindAll(ctx *gin.Context) {
	page, _ := strconv.Atoi(ctx.DefaultQuery("page", "1"))
	limit, _ := strconv.Atoi(ctx.DefaultQuery("limit", "20"))
	search := ctx.DefaultQuery("search", "")

	serials, total, totalPages, err := c.employeeSerialsService.FindAll(page, limit, search)
	if err != nil {
		ErrorResponse(ctx, http.StatusInternalServerError, err.Error())
		return
	}

	SuccessResponse(ctx, "获取员工序列号列表成功", gin.H{
		"data": serials,
		"pagination": gin.H{
			"page":       page,
			"limit":      limit,
			"total":      total,
			"totalPages": totalPages,
		},
	})
}

// Update 更新员工序列号信息
// @Summary 更新员工序列号信息
// @Description 更新指定员工序列号的信息
// @Tags 员工赋码管理
// @Accept json
// @Produce json
// @Security BearerAuth
// @Param serialNumber path string true "序列号"
// @Param updateData body models.UpdateEmployeeSerialDTO true "更新数据"
// @Success 200 {object} models.DataResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Failure 404 {object} models.ErrorResponse
// @Failure 500 {object} models.ErrorResponse
// @Router /employee-serials/{serialNumber} [put]
func (c *EmployeeSerialsController) Update(ctx *gin.Context) {
	serialNumber := ctx.Param("serialNumber")

	var updateData models.UpdateEmployeeSerialDTO
	if !BindJSON(ctx, &updateData) {
		return
	}

	serial, err := c.employeeSerialsService.Update(serialNumber, updateData)
	if err != nil {
		ErrorResponse(ctx, http.StatusInternalServerError, err.Error())
		return
	}

	SuccessResponse(ctx, "员工序列号信息更新成功", gin.H{
		"serial": serial,
	})
}

// Revoke 吊销员工序列号
// @Summary 吊销员工序列号
// @Description 吊销指定员工序列号
// @Tags 员工赋码管理
// @Produce json
// @Security BearerAuth
// @Param serialNumber path string true "序列号"
// @Success 200 {object} models.BaseResponse
// @Failure 400 {object} models.ErrorResponse
// @Failure 401 {object} models.ErrorResponse
// @Failure 404 {object} models.ErrorResponse
// @Failure 500 {object} models.ErrorResponse
// @Router /employee-serials/{serialNumber}/revoke [post]
func (c *EmployeeSerialsController) Revoke(ctx *gin.Context) {
	serialNumber := ctx.Param("serialNumber")

	err := c.employeeSerialsService.Revoke(serialNumber)
	if err != nil {
		ErrorResponse(ctx, http.StatusBadRequest, err.Error())
		return
	}

	SuccessResponse(ctx, "员工序列号吊销成功")
}