package api

// ==========================================================================
// 生成日期：2023-04-10 11:26:49 +0800 CST
// 生成路径: apps/visual/api/visual_screen.go
// 生成人：panda
// ==========================================================================
import (
	"github.com/XM-GO/PandaKit/model"
	"github.com/XM-GO/PandaKit/restfulx"
	"github.com/emicklei/go-restful/v3"
	"github.com/kakuilan/kgo"
	"pandax/pkg/global"
	"strings"

	"pandax/apps/visual/entity"
	"pandax/apps/visual/services"
	pxSocket "pandax/pkg/websocket"
)

type VisualScreenApi struct {
	VisualScreenApp services.VisualScreenModel
}

// GetVisualScreenList Screen列表数据
func (p *VisualScreenApi) GetVisualScreenList(rc *restfulx.ReqCtx) {
	data := entity.VisualScreen{}
	pageNum := restfulx.QueryInt(rc, "pageNum", 1)
	pageSize := restfulx.QueryInt(rc, "pageSize", 10)
	data.ScreenName = restfulx.QueryParam(rc, "screenName")
	data.Status = restfulx.QueryParam(rc, "status")

	data.GroupId = int64(restfulx.QueryInt(rc, "groupId", 0))

	list, total := p.VisualScreenApp.FindListPage(pageNum, pageSize, data)

	rc.ResData = model.ResultPage{
		Total:    total,
		PageNum:  int64(pageNum),
		PageSize: int64(pageNum),
		Data:     list,
	}
}

// GetVisualScreen 获取Screen
func (p *VisualScreenApi) GetVisualScreen(rc *restfulx.ReqCtx) {
	screenId := restfulx.PathParam(rc, "screenId")
	rc.ResData = p.VisualScreenApp.FindOne(screenId)
}

// InsertVisualScreen 添加Screen
func (p *VisualScreenApi) InsertVisualScreen(rc *restfulx.ReqCtx) {
	var data entity.VisualScreen
	restfulx.BindQuery(rc, &data)
	data.UserId = rc.LoginAccount.UserId
	data.ScreenId = kgo.KStr.Uniqid("px")
	data.Creator = rc.LoginAccount.UserName
	p.VisualScreenApp.Insert(data)
}

// UpdateVisualScreen 修改Screen
func (p *VisualScreenApi) UpdateVisualScreen(rc *restfulx.ReqCtx) {
	var data entity.VisualScreen
	restfulx.BindQuery(rc, &data)

	p.VisualScreenApp.Update(data)
}

// DeleteVisualScreen 删除Screen
func (p *VisualScreenApi) DeleteVisualScreen(rc *restfulx.ReqCtx) {
	screenId := restfulx.PathParam(rc, "screenId")
	screenIds := strings.Split(screenId, ",")
	p.VisualScreenApp.Delete(screenIds)
}

// UpdateScreenStatus 修改状态
func (p *VisualScreenApi) UpdateScreenStatus(rc *restfulx.ReqCtx) {
	var screen entity.VisualScreen
	restfulx.BindQuery(rc, &screen)
	p.VisualScreenApp.Update(screen)
}

func (p *VisualScreenApi) ScreenTwin(request *restful.Request, response *restful.Response) {
	screenId := request.QueryParameter("screenId")
	if screenId == "" {
		restfulx.ErrorRes(response, "请传组态Id")
		return
	}
	newWebsocket, err := pxSocket.NewWebsocket(response.ResponseWriter, request.Request, nil)
	if err != nil {
		restfulx.ErrorRes(response, "创建Websocket失败")
		return
	}
	go func() {
		for {
			_, message, err := newWebsocket.Conn.ReadMessage()
			if err != nil {
				return
			}
			pxSocket.OnMessage(newWebsocket, string(message))
		}
	}()
	global.Log.Info("Websocket连接成功")
}