package websocket

import (
	"fmt"
	"github.com/gorilla/websocket"
	"log"
	"net/http"
	"pandax/pkg/global"
	"strings"
)

var upGrader = websocket.Upgrader{
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

type Websocket struct {
	Conn *websocket.Conn
}

func NewWebsocket(writer http.ResponseWriter, r *http.Request, header http.Header) (*Websocket, error) {
	ws, err := upGrader.Upgrade(writer, r, header)
	if err != nil {
		return nil, err
	}
	ws.SetCloseHandler(func(code int, text string) error {
		global.Log.Info(fmt.Sprintf("websocket 连接关闭,code: %d, text: %s", code, text))
		return ws.Close()
	})

	webs := &Websocket{Conn: ws}
	return webs, nil
}

// OnMessage 消息
//发送消息消息类型 01:发送的设备数据  02:收到指令回复  03: 心跳回复
func OnMessage(ws *Websocket, message string) {
	log.Println(message)
	if message != "" && strings.Index(message, "ONLINE") != -1 {
		AddWebSocketByScreenId(strings.Split(message, "ONLINE")[0], ws)
	}
	//画布离开
	if message != "" && strings.Index(message, "LEAVE") != -1 {
		RemoveWebSocket(strings.Split(message, "LEAVE")[0])
	}
	//客户端传来了控制命令 格式   场景控制代码CONTROLCMD控制命令CONTROLCMD传感器id
	if message != "" && strings.Index(message, "CONTROLCMD") != -1 {
		split := strings.Split(message, "CONTROLCMD")
		if len(split) < 2 {
			return
		}
		screenId, controlCMD := split[0], split[1] //指令cmd : {key: '', value: 3} k:v形式
		log.Println(screenId, controlCMD)
		//TODO 在这里编写代码，将命令发送到现场设备 这里已经拿到了 按钮命令和画布id
		//1. 根据组态Id查询设备Id，及设备模型
		//2. 根据设备下发CMD指令

		sendMessages("02", "'命令发送成功'", screenId)
	}
	//心跳处理
	if message != "" && strings.Index(message, "HEARTCMD") != -1 {
		split := strings.Split(message, "HEARTCMD")
		if len(split) < 1 {
			return
		}
		screenId := split[0]
		sendMessages("03", "'心跳正常'", screenId)
	}

}

func sendMessages(messageType, messageContent, screenId string) {
	msg := fmt.Sprintf(`{'MESSAGETYPE':'%s','MESSAGECONTENT':%s}`, messageType, messageContent)
	SendMessage(msg, screenId)
}