package mtwmapi

import (
	"encoding/json"
	"fmt"
	"git.rosy.net.cn/baseapi/utils"
	"github.com/gazeboxu/mapstructure"
	"github.com/go-redis/redis"
	"github.com/gorilla/websocket"
	"sync"
	"testing"
	"time"
)

const (
	MTIMPushUrl = "wss://wpush.meituan.com/websocket"
	TestAppID   = "589_WMOPEN"
	TestToken   = "wo589i4VsZHFH2fh4uVsr6Dtc3k6vG8Xu0vxpreBQFy6QAvg"

	TestMTIMPushUrl = "wss://wpush.meituan.com/websocket/589_WMOPEN/wo589i4VsZHFH2fh4uVsr6Dtc3k6vG8Xu0vxpreBQFy6QAvg"
)

type ClientManager struct {
	ClientIdMap     map[string]*Client // 全部的连接
	ClientIdMapLock sync.RWMutex       // 读写锁

	Connect    chan *Client // 连接处理
	DisConnect chan *Client // 断开连接处理

	GroupLock sync.RWMutex
	Groups    map[string][]string

	//SystemClientsLock sync.RWMutex
	//SystemClients     map[string][]string
	Clients  map[string]*Client  // 保存连接
	Accounts map[string][]string // 账号和连接关系,map的key是账号id即：AccountId，这里主要考虑到一个账号多个连接
	mu       *sync.Mutex
}

var Manager = NewClientManager()

func NewClientManager() (clientManager *ClientManager) {
	clientManager = &ClientManager{
		Accounts:    make(map[string][]string),
		ClientIdMap: make(map[string]*Client, 100),
		Connect:     make(chan *Client, 10000),
		DisConnect:  make(chan *Client, 10000),
		mu:          new(sync.Mutex),
	}

	return
}

var RegisterChan = make(chan *Client, 100)

type Client struct {
	ID            string          // 连接ID
	AccountId     string          // 账号id, 一个账号可能有多个连接
	Socket        *websocket.Conn // 连接
	HeartbeatTime int64           // 前一次心跳时间
}

var rdb = redis.NewClient(&redis.Options{
	//Addr: "www.jxc4.com:6379",
	//Password: "",
	Addr:     "127.0.0.1:6379",
	Password: "123456",
	DB:       0,
})

//测试心跳
func TestHeartCheck(t *testing.T) {
	//go func() {
	//	ticker := time.NewTicker(5 * time.Second)
	//	defer ticker.Stop()
	//	for {
	//		<-ticker.C
	//发送心跳
	conn, resp, err := websocket.DefaultDialer.Dial(TestMTIMPushUrl, nil)
	fmt.Println(resp, err)
	err1 := conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(time.Second))
	for {
		_, msg, err := conn.ReadMessage()
		if err != nil {
			break
		}
		fmt.Printf("%s receive: %s\n", conn.RemoteAddr(), string(msg))
	}
	fmt.Println(err1)
	//}
	//}()
}

func TestGetConnectionToken(t *testing.T) {
	resp, err := api.GetConnectionToken()
	if err != nil {
		t.Fatal(err)
	}
	retVal := GetConnTokenResp{}
	err = mapstructure.Decode(resp, &retVal)
	fmt.Println(err)
	fmt.Println(utils.Format4Output(retVal, false))
}

//解密测试
func TestAesCBCDecrypt(t *testing.T) {
	secret := "a81eb3df418d83d6a1a4b7c572156d2f"
	key := secret[:16]
	str := "qodhoVd4IGtgPKrvYwq6QrzBecJZkeSUPYR88iGRUsCRFmCFxDHpUhqsbBztNXQx"
	//str := "Vv+Y/K8vfS42W+P7xq26aIb6uoaG/nL0ZoMMXpitc5QQ3XJm3Roh10NuSoojYrG/3JZwbzgtYA+kBvodoY2eJV00f9MBY+kLkxToP+aSofsYva9tHbipvjVtexebc+eP7aQMtzbwU4BNNnuRG6e7TkXP+BLdtiGsyvHolGfky+p2fZgWes9R6JIxkuRCXW/yBhUo8F+wWCZ2YQl/szp5lHJ3cmneD6cwem36E0FBcvxZNB9an4pRkBrqi1p43V8QBLO719oXkQ+dqTqJMi1/xDSBrCDYN8QORnARP8+j1oDuqE34Kklcse4WL9rwTJ2sOmOu/O2h6Gx3ZaFaMaWRXBDYv8JpzTZjCbRrLSENlEHTof29BmvXTJ0QZ7qi6iAD"
	data, err := Decrypt(str, []byte(key))
	//data, err := DecryptAES(key, str)
	fmt.Println(data)
	fmt.Println(err)
}

var wsList []*websocket.Conn

func sendmsg() {
	for _, conn := range wsList {
		if err := conn.WriteMessage(websocket.TextMessage, []byte("~#HHHBBB#~")); err != nil {
			fmt.Printf("%s", err) //"use of closed network connection"
		}
	}
}
func TestPut(t *testing.T) {
	fmt.Println(wsList)
}

func TestWebSocketClient(t *testing.T) {
	//发送webSocket请求
	conn, resp, err := websocket.DefaultDialer.Dial(TestMTIMPushUrl, nil)
	if err != nil {
		fmt.Printf("连接失败：%v", err)
	}
	fmt.Printf("响应:%s", fmt.Sprint(resp))
	//wsList = append(wsList, conn)
	//关闭
	conn.SetCloseHandler(func(code int, text string) error {
		fmt.Printf("WebSocket connection closed with code %d and text: %s\n", code, text)
		return nil
	})
	defer func(conn *websocket.Conn) {
		err := conn.Close()
		if err != nil {
			return
		}
	}(conn)

	//赋入全局变量
	//Default(conn)
	//生成clientID
	clientID := GenClientId()

	//创建实例连接
	client := &Client{
		ID: clientID,
		//AccountId:conn. ,
		Socket:        conn,
		HeartbeatTime: time.Now().Unix(),
	}
	//rdb.Set("testPush", client, 0)

	//注册到连接管理
	RegisterChan <- client
	//todo 暂时不确定放哪
	//go Start()

	done := make(chan SingleChat)
	//err = conn.WriteControl(websocket.PingMessage, []byte{}, time.Now().Add(time.Second))
	err = conn.WriteMessage(websocket.TextMessage, []byte("~#HHHBBB#~"))
	if err != nil {
		fmt.Println(err)
	}

	for {
		_, msg, err := conn.ReadMessage()
		if err != nil {
			//log.Fatal(err)
			break
		}
		fmt.Printf("%s receive: %s\n", conn.RemoteAddr(), string(msg))
	}
	<-done

}

func TestPUSH(t *testing.T) {
	key := "589:7954977:10"
	//rdb.RPush(key, "1111111111")
	//rdb.RPush(key, "{\"vendorID\":10,\"userID\":11158569333,\"NewMessageNum\":3,\"latestMsg\":\"hhhhhhhhhhh\",\"latestTime\":1681983980}")
	//rdb.RPush(key, "{\"vendorID\":10,\"userID\":11158569333,\"NewMessageNum\":3,\"latestMsg\":\"oooooooooo\",\"latestTime\":1681983980}")
	//rdb.RPush(key, "2222222222222")
	rdb.RPush(key, "{\"vendorID\":10,\"userID\":11158569333,\"NewMessageNum\":4,\"latestMsg\":\"成功插入新数据，看下cnt\",\"latestTime\":1681983980}")
	rdb.RPush(key, "{\"vendorID\":10,\"userID\":11158569333,\"NewMessageNum\":5,\"latestMsg\":\"成功插入新数据，看下cnt\",\"latestTime\":1681983980}")
}

//用户消息列表
type UserMessageList struct {
	VendorID      int    `json:"vendorID"`      //平台品牌 10-美团 11-饿了么
	UserID        int    `json:"userID"`        //用户ID
	NewMessageNum int    `json:"NewMessageNum"` //新消息数量
	LatestMsg     string `json:"latestMsg"`     //最新一条消息
	LatestTime    int    `json:"latestTime"`    //最新一条消息发送时间
}

func TestNewRedis(t *testing.T) {
	var flag = 11158569333
	var key = "589:7954977:10"
	s2 := rdb.LRange(key, 0, -1).Val()
	fmt.Printf("before len %d\n", len(s2))
	fmt.Printf("before ans %s\n", s2)
	cnt := 0
	n := rdb.Exists(key).Val()
	if n > 0 {
		for i := 0; i < len(s2); i++ {
			v := UserMessageList{}
			_ = json.Unmarshal([]byte(s2[i]), &v)
			if v.UserID == flag {
				rdb.LSet(key, int64(i), "del")
				rdb.LRem(key, 0, "del")
				s2 = append(s2[:i], s2[i+1:]...)
				i--
				if v.NewMessageNum == 0 { //目前为首条
					cnt++ //赋值1
				} else {
					cnt = v.NewMessageNum
				}
			}
		}
	}
	fmt.Printf("after cnt %d\n", cnt)
	fmt.Printf("after len %d\n", len(s2))
	fmt.Printf("after ans %s\n", s2)
	//存入flag数据
	ans := UserMessageList{
		VendorID:      10,
		UserID:        11158569333,
		NewMessageNum: cnt,
		LatestMsg:     "成功插入新数据，看下cnt",
		LatestTime:    1681983980,
	}
	param, _ := json.Marshal(ans)
	rdb.RPush(key, param)
}

// 根据账号获取连接
func TestGetClient(t *testing.T) {
	accountId := "QW+r2FtsRKGGLJnlgyDNlChzcKcSZ8Kfgh0qw//ONuQCDKzky4x+nlbnx3k1JX13"
	clients := make([]*Client, 0)

	Manager.mu.Lock()
	defer Manager.mu.Unlock()

	if len(Manager.Accounts[accountId]) > 0 {
		for _, clientId := range Manager.Accounts[accountId] {
			if c, ok := Manager.Clients[clientId]; ok {
				clients = append(clients, c)
			}
		}
	}

	fmt.Printf(utils.Format4Output(clients, false))
}
func TestMal(t *testing.T) {

}

func TestGetWayBillFee(t *testing.T) {
	//order, _ := api.OrderGetOrderDetail(1100486451772280163, false)
	//globals.SugarLogger.Debugf("order:==%s", utils.Format4Output(order, false)) 2002
	api.GetWayBillFee("1100486451772280163", 2)
	api.GetWayBillFee("1100486451772280163", 1)
}

// 商家没有接入众包配送,无法进行众包配送相关操作
func TestGetShippingFeeList(t *testing.T) {
	api.GetShippingFeeList("1300486314174032613,1100487300210228389", 1)
	api.GetShippingFeeList("1300486314174032613,1100487300210228389", 2)
}