package app

import (
	. "code.hoteas.com/golang/hotime"
	. "code.hoteas.com/golang/hotime/common"
	"golang.org/x/net/websocket"
	"time"
)

type WSClient struct {
	ID string
	*websocket.Conn
	time.Time
	DeadTime time.Time
	IsDead   bool
}

//websocket链接池
var WsUserMap = map[string][]*WSClient{}
var WSMasterID = ""

func WsSendMsg(ws *WSClient, data Map) bool {
	if WsUserMap[ws.ID] == nil {
		return false
	}

	for _, v := range WsUserMap[ws.ID] {
		if v.IsDead || v == ws {
			//WsUserMap[ws.ID]=WsUserMap[ws.ID][:k]
			continue
		}

		str := data.ToJsonString()
		v.Conn.Write([]byte(str))

	}

	return true
}

var WebsocketCtr = Ctr{
	"conn": func(this *Context) {
		id := this.SessionId
		if WsUserMap[id] == nil {
			WsUserMap[id] = []*WSClient{}
		}
		hdler := websocket.Handler(func(ws *websocket.Conn) {

			client := &WSClient{ID: id, Conn: ws, Time: time.Now(), DeadTime: time.Now(), IsDead: false}

			WsUserMap[id] = append(WsUserMap[id], client)
			var message string
			for true {
				err := websocket.Message.Receive(ws, &message)
				if err != nil {
					client.DeadTime = time.Now()
					client.IsDead = true
					return
				}

				data := Map{}
				data.JsonToMap(message)
				WsSendMsg(client, data)
				//switch data.GetString("type") {
				//
				//}

			}
		})
		hdler.ServeHTTP(this.Resp, this.Req)
	},
}