windows_lock_go/app/admin/apis/mm_machine.go

package apis

import (
	"encoding/json"
	"fmt"
	"net/http"

	"github.com/gin-gonic/gin"
	"github.com/gin-gonic/gin/binding"
	"github.com/go-admin-team/go-admin-core/logger"
	"github.com/go-admin-team/go-admin-core/sdk/api"
	"github.com/go-admin-team/go-admin-core/sdk/pkg/jwtauth/user"
	_ "github.com/go-admin-team/go-admin-core/sdk/pkg/response"

	"go-admin/app/admin/models"
	"go-admin/app/admin/service"
	"go-admin/app/admin/service/dto"
	"go-admin/common/actions"
)

type MmMachine struct {
	api.Api
}

// GetPage 获取设备白名单配置列表
// @Summary 获取设备白名单配置列表
// @Description 获取设备白名单配置列表
// @Tags 设备白名单配置
// @Param status query string false "状态 0-掉线 1-在线"
// @Param pageSize query int false "页条数"
// @Param pageIndex query int false "页码"
// @Success 200 {object} response.Response{data=response.Page{list=[]models.MmMachine}} "{"code": 200, "data": [...]}"
// @Router /api/v1/mm-machine [get]
// @Security Bearer
func (e MmMachine) GetPage(c *gin.Context) {
	req := dto.MmMachineGetPageReq{}
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	p := actions.GetPermissionFromContext(c)
	list := make([]models.MmMachine, 0)
	var count int64

	err = s.GetPage(&req, p, &list, &count)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("获取设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.PageOK(list, int(count), req.GetPageIndex(), req.GetPageSize(), "查询成功")
}

// Get 获取设备白名单配置
// @Summary 获取设备白名单配置
// @Description 获取设备白名单配置
// @Tags 设备白名单配置
// @Param id path int false "id"
// @Success 200 {object} response.Response{data=models.MmMachine} "{"code": 200, "data": [...]}"
// @Router /api/v1/mm-machine/{id} [get]
// @Security Bearer
func (e MmMachine) Get(c *gin.Context) {
	req := dto.MmMachineGetReq{}
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	var object models.MmMachine

	p := actions.GetPermissionFromContext(c)
	err = s.Get(&req, p, &object)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("获取设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(object, "查询成功")
}

// Insert 创建设备白名单配置
// @Summary 创建设备白名单配置
// @Description 创建设备白名单配置
// @Tags 设备白名单配置
// @Accept application/json
// @Product application/json
// @Param data body dto.MmMachineInsertReq true "data"
// @Success 200 {object} response.Response	"{"code": 200, "message": "添加成功"}"
// @Router /api/v1/mm-machine [post]
// @Security Bearer
func (e MmMachine) Insert(c *gin.Context) {
	req := dto.MmMachineInsertReq{}
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	// 设置创建人
	req.SetCreateBy(user.GetUserId(c))

	err = s.Insert(&req)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("创建设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(req.GetId(), "创建成功")
}

// Update 修改设备白名单配置
// @Summary 修改设备白名单配置
// @Description 修改设备白名单配置
// @Tags 设备白名单配置
// @Accept application/json
// @Product application/json
// @Param id path int true "id"
// @Param data body dto.MmMachineUpdateReq true "body"
// @Success 200 {object} response.Response	"{"code": 200, "message": "修改成功"}"
// @Router /api/v1/mm-machine/{id} [put]
// @Security Bearer
func (e MmMachine) Update(c *gin.Context) {
	req := dto.MmMachineUpdateReq{}
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}
	req.SetUpdateBy(user.GetUserId(c))
	p := actions.GetPermissionFromContext(c)

	err = s.Update(&req, p)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}
	e.OK(req.GetId(), "修改成功")
}

// UpdateLimitBatch 批量修改设备白名单配置
func (e MmMachine) UpdateLimitBatch(c *gin.Context) {
	req := dto.MmMachineUpdateLimitBatchReq{}
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	if err := req.Validate(); err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	p := actions.GetPermissionFromContext(c)

	err = s.UpdateLimitBatch(&req, p)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(nil, "批量修改成功")
}

// 批量修改
func (e MmMachine) UpdateBatch(c *gin.Context) {
	req := dto.MmMachineUpdateBatchReq{}
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	if err := req.Validate(); err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	req.SetUpdateBy(user.GetUserId(c))
	p := actions.GetPermissionFromContext(c)

	err = s.UpdateBatch(&req, p)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}
	e.OK(nil, "批量修改成功")
}

// Delete 删除设备白名单配置
// @Summary 删除设备白名单配置
// @Description 删除设备白名单配置
// @Tags 设备白名单配置
// @Param data body dto.MmMachineDeleteReq true "body"
// @Success 200 {object} response.Response	"{"code": 200, "message": "删除成功"}"
// @Router /api/v1/mm-machine [delete]
// @Security Bearer
func (e MmMachine) Delete(c *gin.Context) {
	s := service.MmMachine{}
	req := dto.MmMachineDeleteReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	// req.SetUpdateBy(user.GetUserId(c))
	p := actions.GetPermissionFromContext(c)

	err = s.Remove(&req, p)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("删除设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}
	e.OK(req.GetId(), "删除成功")
}

// GetPassword 获取设备密码
func (e MmMachine) GetPassword(c *gin.Context) {
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
	}

	pass, err := s.GetPassword()

	if err != nil {
		c.String(500, err.Error())
		return
	}

	c.String(http.StatusOK, pass)
}

// 获取设备白名单数据
func (e MmMachine) GetWhiteProcess(c *gin.Context) {
	logger.Info("获取设备白名单数据")
	s := service.MmMachine{}
	req := dto.MmMachineQueryWhiteListReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.Query, binding.Form).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
	}

	whiteProcess, err := s.GetWhiteProcess(&req, false)

	if err != nil {
		c.String(500, err.Error())
		return
	}

	resp, _ := json.Marshal(whiteProcess)

	c.String(http.StatusOK, string(resp))
}

// 修改设备白名单数据
func (e MmMachine) UpdateMachineWhite(c *gin.Context) {
	s := service.MmMachine{}
	req := dto.MmMachineUpdateWhiteReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.Form, binding.JSON).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		c.Set("status", 500)
	}

	err = s.UpdateMachineWhite(&req)
	if err != nil {
		c.String(500, err.Error())
		return
	}

	c.String(http.StatusOK, "")
}

// 占用设备编号 1-成功 2-失败（已被占用）
func (e MmMachine) CheckMachine(c *gin.Context) {
	s := service.MmMachine{}
	req := dto.MmMachineCheckMachineReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.Query, binding.Form, binding.JSON).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		c.Set("status", 500)
	}

	resp := s.CheckMachine(&req)
	// if err != nil {
	// 	e.Logger.Errorf("CheckMachine error:%s", err.Error())
	// 	if strings.Contains(err.Error(), "设备编号已被占用") {
	// 		c.String(http.StatusOK, err.Error())
	// 	} else {
	// 		c.String(500, "0")
	// 	}
	// 	return
	// }

	byts, _ := json.Marshal(resp)

	c.String(http.StatusOK, string(byts))
}

func (e MmMachine) GetMachineList(c *gin.Context) {
	logger.Info("获取设备列表")
	s := service.MmMachine{}
	// req := dto.MmMachineQueryWhiteListReq{}
	err := e.MakeContext(c).
		MakeOrm().
		// Bind(&req, binding.Query, binding.Form).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
	}

	p := actions.GetPermissionFromContext(c)
	p.RoleId = user.GetRoleId(c)
	p.UserId = user.GetUserId(c)
	machineList, err := s.GetMachineList(p)

	if err != nil {
		e.Error(500, err, fmt.Sprintf("获取设备白名单配置失败，\r\n失败信息 %s", err.Error()))
	}
	e.OK(machineList, "查询成功")
}

// 批量修改设备分组
func (e MmMachine) UpdateGroupBatch(c *gin.Context) {
	req := dto.MmMachineUpdateGroupBatchReq{}
	s := service.MmMachine{}

	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	if err := req.Validate(); err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	p := actions.GetPermissionFromContext(c)

	err = s.UpdateGroupBatch(&req, p)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(nil, "批量修改成功")
}

// 清除所有设备
func (e MmMachine) ClearAll(c *gin.Context) {
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	err = s.ClearAll()
	if err != nil {
		e.Error(500, err, fmt.Sprintf("清除所有设备失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(nil, "清除成功")
}

// 解绑所有设备
func (e MmMachine) Uninstall(c *gin.Context) {
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	err = s.Uninstall()
	if err != nil {
		e.Error(500, err, fmt.Sprintf("解绑所有设备失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(nil, "解绑成功")
}

// 批量修改设备描述
func (e MmMachine) UpdateDescriptionBatch(c *gin.Context) {
	req := dto.MmMachineUpdateDescriptionBatchReq{}
	s := service.MmMachine{}

	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors
	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	if err := req.Validate(); err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	p := actions.GetPermissionFromContext(c)

	err = s.UpdateDescriptionBatch(&req, p)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("批量修改设备白名单配置失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(nil, "批量修改成功")
}

// 查询设备间隔账号
func (e MmMachine) QueryIntervalAccount(c *gin.Context) {
	s := service.MmMachine{}
	err := e.MakeContext(c).
		MakeOrm().
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	err = s.QueryIntervalAccount(c)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("查询设备间隔账号失败，\r\n失败信息 %s", err.Error()))
		return
	}
}

// 重启设备
func (e MmMachine) RebootMachine(c *gin.Context) {
	s := service.MmMachine{}
	req := dto.MmMachineRebootReq{}
	err := e.MakeContext(c).
		MakeOrm().
		Bind(&req, binding.JSON).
		MakeService(&s.Service).
		Errors

	if err != nil {
		e.Logger.Error(err)
		e.Error(500, err, err.Error())
		return
	}

	err = s.RebootMachine(&req)
	if err != nil {
		e.Error(500, err, fmt.Sprintf("重启设备失败，\r\n失败信息 %s", err.Error()))
		return
	}

	e.OK(nil, "重启成功")
}