// Package verify 校验
package verify

import (
	"encoding/json"
	"fmt"
	"go_dreamfactory/comm"
	"go_dreamfactory/pb"
	"go_dreamfactory/stress/model"

	"google.golang.org/protobuf/proto"
)

// WebSocketResponseJSON 返回数据结构体，返回值为json
type WebSocketResponseJSON struct {
	Seq      string `json:"seq"`
	Cmd      string `json:"cmd"`
	Response struct {
		Code    int         `json:"code"`
		CodeMsg string      `json:"codeMsg"`
		Data    interface{} `json:"data"`
	} `json:"response"`
}

// WebSocketJSON 通过返回的Body 判断
// 返回示例: {"seq":"1566276523281-585638","cmd":"heartbeat","response":{"code":200,"codeMsg":"Success","data":null}}
// code 取body中的返回code
func WebSocketJSON(request *model.Request, seq string, msg []byte) (code int, isSucceed bool) {
	responseJSON := &WebSocketResponseJSON{}
	err := json.Unmarshal(msg, responseJSON)
	if err != nil {
		code = model.ParseError
		fmt.Printf("请求结果 json.Unmarshal msg:%s err:%v", string(msg), err)
	} else {

		if seq != responseJSON.Seq {
			code = model.ParseError
			fmt.Println("请求和返回seq不一致 ~请求:", seq, responseJSON.Seq, string(msg))
		} else {
			code = responseJSON.Response.Code
			// body 中code返回200为返回数据成功
			if code == 200 {
				isSucceed = true
			}
		}
	}
	// 开启调试模式
	if request.GetDebug() {
		fmt.Printf("请求结果 seq:%s body:%s \n", seq, string(msg))
	}
	return
}

func WebSocketProto(request *model.Request, seq string, data []byte) (code int, isSucceed bool) {

	var msg *pb.UserMessage = &pb.UserMessage{}

	if err := proto.Unmarshal(data, msg); err != nil {
		fmt.Printf("unmarshal err:%v", err)
	}
	//fmt.Printf("接收消息=====msg.MainType = %s,msg.SubType = %s\n", msg.MainType, msg.SubType)
	if msg.MainType == "chat" && msg.SubType == "message" {
		resp := &pb.ChatMessagePush{}
		if !comm.ProtoUnmarshal(msg, resp) {
			code = model.ParseError
		}
		return
		//fmt.Printf("接收消息=====resp:%v\n", resp.Chat)
	} else if msg.MainType == "user" && msg.SubType == "login" {
		resp := &pb.UserLoginResp{}
		if !comm.ProtoUnmarshal(msg, resp) {
			code = model.ParseError
		}
		//fmt.Printf("接收消息=====loginData:%v,userExpand:%v\n", resp.Data, resp.Ex)
		return
	}
	return
}