package camera

import (
	"github.com/long/test/httpclient"
	"github.com/long/test/sdk"

	"nanomsg.org/go-mangos"
	"nanomsg.org/go-mangos/protocol/pair"
	"nanomsg.org/go-mangos/transport/ipc"
	"nanomsg.org/go-mangos/transport/tcp"

	"context"
	"encoding/json"
	"fmt"
	"time"
)

/*
* 1.  获取 cid
* 2.  获取 cid和 taskid关系
* 3.  获取 cid ipc communication
 */
var SocketManage = make(map[string]SocketContext)
var Initchannel = make(chan Camerdata)
var UrlPort = 7000

type Camerdata struct {
	Cameraid string
	Rtsp     string
}

type SocketContext struct {
	Sock    mangos.Socket
	Context context.Context
	Cancel  context.CancelFunc
}

func Taskdolist(cid string, taskid string, data []byte) {

	fmt.Println("======================================")
	fmt.Println()
	//  数据加工(打标签)
	sdkmsg := sdk.SdkData(cid, taskid, data)
	if sdkmsg.Tasklab == nil {
		fmt.Println("cid:%s 没有任务%s", cid, taskid)
		return
	}

	//  计算分发的主题
	SendTopic := sdk.SdkSendTopic(sdkmsg)
	if _, ok := sdk.SdkMap[SendTopic]; ok {
		fmt.Println("分发的主题存在")
		sdk.SdkMap[SendTopic] <- sdkmsg
		//fmt.Println("重新开始循环: ", sdk.SdkMap)
	} else {
		fmt.Println("分发的主题不存在")
	}

}

func Init() {

	CameraRelative()
	fmt.Println()
	fmt.Println("cid,taskid , sid: ", CtsId)

	_, socket, err := NewCamerSocketListen("init", "tcp", "192.168.1.124", 0)
	if err != nil {
		return
	}

	go SendRecv(socket, Initchannel)

	var cameratodecode Camerdata
	for {
		time.Sleep(2 * time.Second)
		i := 0
		for _, cam := range CtsId {

			if i < 2 {
				cameratodecode.Cameraid = cam.Cameraid
				cameratodecode.Rtsp = cam.RtspUrl
				Initchannel <- cameratodecode
			}
			i++
		}
	}
}

func Recv(cameraid string, socket SocketContext) {
	socket.Sock.SetOption(mangos.OptionRecvDeadline, 5*time.Second)
	var msg []byte
	var err error
	for {
		select {
		case <-socket.Context.Done():
			fmt.Println("listen recv quit")
			return
		default:
			if msg, err = socket.Sock.Recv(); err != nil {
				//fmt.Printf("%s ", err)
				continue
			} else {
				for _, taskid := range GetAlltask(cameraid) {
					go Taskdolist(cameraid, taskid, msg)
					fmt.Println("receive: ", len(msg), "cameraid: ", cameraid, "taskid: ", taskid)
				}
			}
		}
	}
}

func send(socket SocketContext, cam Camerdata) {
	b, err := json.Marshal(cam)
	if err != nil {
		fmt.Println("can not json convert !", cam)
	}
	if err := socket.Sock.Send(b); err != nil {
		fmt.Println("failed send")
	}
}

func SendRecv(socket SocketContext, camera chan Camerdata) {
	socket.Sock.SetOption(mangos.OptionRecvDeadline, 5*time.Second)
	for cam := range camera {
		send(socket, cam)
		if _, ok := SocketManage[cam.Cameraid]; !ok {
			go func(id string) {
				fmt.Println("create cid server: ", id)
				cid, socketlisten, _ := NewCamerSocketListen(id, "tcp", "192.168.1.124", 0)
				fmt.Println("input id: ", id, " output id :", cid)
				Recv(cid, socketlisten)

			}(cam.Cameraid)
		}
	}
}

// new camera server
func NewCamerSocketListen(cameraid string, protocol string, ip string, port int) (cid string, socket SocketContext, err error) {

	ctx, cancel := context.WithCancel(context.Background())

	var url string

	socket.Context = ctx
	socket.Cancel = cancel

	switch protocol {
	case "tcp":
		if port == 0 {
			port = UrlPort
			UrlPort++
		}
		url = fmt.Sprintf("%s://%s:%d", protocol, ip, port)
	case "ipc":
		url = fmt.Sprintf("%s://%s", cameraid)
	}
	fmt.Println(url)

	if socket.Sock, err = pair.NewSocket(); err != nil {
		fmt.Println(cameraid, "can't get new pair socket: ", err.Error())
		return cameraid, socket, err
	}
	socket.Sock.AddTransport(tcp.NewTransport())
	socket.Sock.AddTransport(ipc.NewTransport())

	socket.Sock.SetOption(mangos.OptionMaxRecvSize, 32*1024*1024)
	socket.Sock.SetOption(mangos.OptionWriteQLen, 10)
	socket.Sock.SetOption(mangos.OptionReadQLen, 10)

	if err = socket.Sock.Listen(url); err != nil {
		fmt.Println("socket lisnte error ", cameraid)
	}
	fmt.Println(cameraid, "success")

	SocketManage[cameraid] = socket

	return cameraid, socket, err
}

// 获取 cid ， taskid, sdkid 关系
var CtsId []httpclient.Camerasingle

func CameraRelative() {
	CtsId = httpclient.GetEsDataReq("http://127.0.0.1:8000/data/api-v/camera/queryCameraAndTaskInfo")
}

//   根据cid 获取 所有的任务
func GetAlltask(cid string) (tasks []string) {
	for _, camsingle := range CtsId {
		if cid == camsingle.Cameraid {
			for _, tasksingle := range camsingle.TaskList {
				tasks = append(tasks, tasksingle.Taskid)
			}
			return
		}
	}
	return
}