435 lines
9.0 KiB
Go
435 lines
9.0 KiB
Go
package api
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
|
|
"strconv"
|
|
|
|
"licserver/internal/model"
|
|
"licserver/internal/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
type DeviceHandler struct {
|
|
deviceService *service.DeviceService
|
|
}
|
|
|
|
func NewDeviceHandler(deviceService *service.DeviceService) *DeviceHandler {
|
|
|
|
return &DeviceHandler{deviceService: deviceService}
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) CreateDevice(c *gin.Context) {
|
|
|
|
var input service.DeviceCreateInput
|
|
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if err := h.deviceService.CreateDevice(&input); err != nil {
|
|
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "设备创建成功"})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) GetDevices(c *gin.Context) {
|
|
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("pageSize", "10"))
|
|
|
|
params := &service.DeviceQueryParams{
|
|
|
|
UID: c.Query("uid"),
|
|
|
|
DeviceType: c.Query("deviceType"),
|
|
|
|
Company: c.Query("company"),
|
|
|
|
LicenseType: c.Query("licenseType"),
|
|
|
|
Status: c.Query("status"),
|
|
|
|
Page: page,
|
|
|
|
PageSize: pageSize,
|
|
}
|
|
|
|
devices, total, err := h.deviceService.GetDevices(params)
|
|
|
|
if err != nil {
|
|
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "获取设备列表成功",
|
|
"count": total,
|
|
"data": devices,
|
|
})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) UpdateStartCount(c *gin.Context) {
|
|
|
|
uid := c.Param("uid")
|
|
|
|
if err := h.deviceService.UpdateStartCount(uid); err != nil {
|
|
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "启动次数更新成功"})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) UpdateDevice(c *gin.Context) {
|
|
|
|
uid := c.Param("uid")
|
|
|
|
var updates map[string]interface{}
|
|
|
|
if err := c.ShouldBindJSON(&updates); err != nil {
|
|
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
if err := h.deviceService.UpdateDevice(uid, updates); err != nil {
|
|
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "设备更新成功"})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) DeleteDevice(c *gin.Context) {
|
|
|
|
uid := c.Param("uid")
|
|
|
|
if err := h.deviceService.DeleteDevice(uid); err != nil {
|
|
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "设备删除成功"})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) RegisterDevice(c *gin.Context) {
|
|
var input service.DeviceRegisterInput
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := h.deviceService.RegisterDevice(&input, c.ClientIP()); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
status := "未激活"
|
|
if input.LicenseCode != "" {
|
|
status = "已激活"
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": fmt.Sprintf("设备注册成功，当前状态：%s", status),
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) ValidateDevice(c *gin.Context) {
|
|
|
|
uid := c.Param("uid")
|
|
|
|
if err := h.deviceService.ValidateDevice(uid); err != nil {
|
|
|
|
c.JSON(http.StatusUnauthorized, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{"message": "设备验证通过"})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) BindLicense(c *gin.Context) {
|
|
uid := c.Param("uid")
|
|
var input struct {
|
|
LicenseCode string `json:"license_code" binding:"required"`
|
|
}
|
|
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := h.deviceService.BindLicense(uid, input.LicenseCode); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "授权码绑定成功",
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) UnbindLicense(c *gin.Context) {
|
|
uid := c.Param("uid")
|
|
|
|
if err := h.deviceService.UnbindLicense(uid); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "授权码解绑成功",
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) GetLicenseInfo(c *gin.Context) {
|
|
|
|
deviceUID := c.Param("uid")
|
|
|
|
device, err := h.deviceService.GetLicenseInfo(deviceUID)
|
|
|
|
if err != nil {
|
|
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, device)
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) CheckLicenseStatus(c *gin.Context) {
|
|
|
|
deviceUID := c.Param("uid")
|
|
|
|
status, err := h.deviceService.CheckLicenseStatus(deviceUID)
|
|
|
|
if err != nil {
|
|
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
|
|
return
|
|
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
|
|
"status": status,
|
|
})
|
|
|
|
}
|
|
|
|
func (h *DeviceHandler) CheckUpdate(c *gin.Context) {
|
|
deviceUID := c.Param("uid")
|
|
currentVersion := c.Query("version")
|
|
|
|
update, err := h.deviceService.CheckUpdate(deviceUID, currentVersion)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, update)
|
|
}
|
|
|
|
func (h *DeviceHandler) CreateDeviceModel(c *gin.Context) {
|
|
var model model.DeviceModel
|
|
if err := c.ShouldBindJSON(&model); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
model.CreatedBy = c.GetUint("userID")
|
|
model.Status = "active"
|
|
|
|
if err := h.deviceService.CreateDeviceModel(&model); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "设备型号创建成功",
|
|
"data": model,
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) GetDeviceModels(c *gin.Context) {
|
|
modelName := c.Query("model_name")
|
|
deviceType := c.Query("device_type")
|
|
company := c.Query("company")
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("limit", "10"))
|
|
|
|
models, total, err := h.deviceService.GetDeviceModels(modelName, deviceType, company, page, pageSize)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "",
|
|
"count": total,
|
|
"data": models,
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) UpdateDeviceModel(c *gin.Context) {
|
|
id, err := strconv.ParseUint(c.Param("id"), 10, 32)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "无效的ID"})
|
|
return
|
|
}
|
|
|
|
var model model.DeviceModel
|
|
if err := c.ShouldBindJSON(&model); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := h.deviceService.UpdateDeviceModel(uint(id), &model); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "设备型号更新成功",
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) DeleteDeviceModel(c *gin.Context) {
|
|
id, err := strconv.ParseUint(c.Param("id"), 10, 32)
|
|
if err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": "无效的ID"})
|
|
return
|
|
}
|
|
|
|
if err := h.deviceService.DeleteDeviceModel(uint(id)); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "设备型号删除成功",
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) BatchDeleteDeviceModels(c *gin.Context) {
|
|
var input struct {
|
|
IDs []uint `json:"ids" binding:"required"`
|
|
}
|
|
|
|
if err := c.ShouldBindJSON(&input); err != nil {
|
|
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
if err := h.deviceService.BatchDeleteDeviceModels(input.IDs); err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"message": "设备型号批量删除成功",
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) GetRegisteredDevices(c *gin.Context) {
|
|
uid := c.Query("uid")
|
|
deviceModel := c.Query("device_model")
|
|
status := c.Query("status")
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("limit", "10"))
|
|
|
|
devices, total, err := h.deviceService.GetRegisteredDevices(uid, deviceModel, status, page, pageSize)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "",
|
|
"count": total,
|
|
"data": devices,
|
|
})
|
|
}
|
|
|
|
func (h *DeviceHandler) GetDeviceLogs(c *gin.Context) {
|
|
uid := c.Param("uid")
|
|
page, _ := strconv.Atoi(c.DefaultQuery("page", "1"))
|
|
pageSize, _ := strconv.Atoi(c.DefaultQuery("limit", "10"))
|
|
|
|
logs, total, err := h.deviceService.GetDeviceLogs(uid, page, pageSize)
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"msg": "",
|
|
"count": total,
|
|
"data": logs,
|
|
})
|
|
}
|
|
|
|
// GetDashboardStats 获取仪表盘统计数据
|
|
func (h *DeviceHandler) GetDashboardStats(c *gin.Context) {
|
|
stats, err := h.deviceService.GetDashboardStats()
|
|
if err != nil {
|
|
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
|
return
|
|
}
|
|
|
|
c.JSON(http.StatusOK, gin.H{
|
|
"code": 0,
|
|
"data": stats,
|
|
})
|
|
}
|