package service

import (
	"basic.com/pubsub/protomsg.git"
	"basic.com/valib/deliver.git"
	"fmt"
	"github.com/gogo/protobuf/proto"
	"github.com/pierrec/lz4"
	"github.com/satori/go.uuid"
	"image"
	"io/ioutil"
	"os"
	"time"
)

const (

	Url_Service_PUSH = "ipc:///tmp///virtual-faceextract-sdk-pull_2.ipc"
	Url_Service_PULL = "ipc:///tmp///virtual-faceextract-sdk-pull_1.ipc"
)
var imgPushChan chan protomsg.Recvmsg
var client_push deliver.Deliver
var client_pull deliver.Deliver

func TestPushImgMsg() {
	InitService()
	proImg := readImgFile()

	if b, err := proto.Marshal(&proImg);err !=nil{
		fmt.Println("protoImage marshal err")
		return
	} else {
		bc := make([]byte,len(b))
		ht := make([]int, 64<<10)
		n,err := lz4.CompressBlock(b,bc,ht)
		if err !=nil {
			fmt.Println(err)
		}
		if n >= len(b){
			fmt.Println("image is not compressible")
		}
		bc = bc[:n]
		//for {
			PushImgMsg(protomsg.Recvmsg{
				Id:uuid.NewV4().String(),
				Addr:"",
				Picdata:bc,
			})
			fmt.Println("pushed img")
			//time.Sleep(10*time.Second)
		//}

	}
}

func readImgFile() protomsg.Image{
	var i protomsg.Image
	timeUnix := time.Now().Unix()
	formatTimeStr := time.Unix(timeUnix, 0).Format("2006-01-02 15:04:05")
	filePath := "/home/user/workspace/timg.jpg"

	file, err := os.Open(filePath)
	defer file.Close()
	if err !=nil{
		fmt.Println("image not exist")
		return i
	} else {
		img, _, err := image.Decode(file)
		bytes, err := ioutil.ReadFile(filePath)
		if err !=nil {
			return i
		}
		b := img.Bounds()
		width := b.Max.X
		height := b.Max.Y
		i = protomsg.Image{
			Width:int32(width),
			Height:int32(height),
			Timestamp:formatTimeStr,
			Data:bytes,
		}
		return i
	}
}

func PushImgMsg(is protomsg.Recvmsg){
	imgPushChan <- is
}

var resultMap map[string]protomsg.SdkMessage


func InitService(){
	fmt.Println("service init!")
	imgPushChan = make(chan protomsg.Recvmsg)
	client_push = deliver.NewClient(deliver.PushPull, Url_Service_PUSH)
	client_pull = deliver.NewClient(deliver.PushPull, Url_Service_PULL)
	defer func() {
		client_push.Close()
		client_pull.Close()
	}()
	go thSend()

	go thRecv()
}

func thSend(){
	for {
		select {
		case is := <- imgPushChan:
			fmt.Println("imgPushChan in")
			b, _ := proto.Marshal(&is)
			err := client_push.Send(b)
			if err !=nil {
				fmt.Println("img Send err:",err)
			}
		default:
			//fmt.Println("no img in")
		}
	}
}

func thRecv(){
	for {
		resultBytes, err := client_pull.Recv()
		if err !=nil{
			fmt.Println("pull err:",err)
			continue
		}
		rMsg := protomsg.SdkMessage{}
		if err := proto.Unmarshal(resultBytes, &rMsg);err ==nil{
			fmt.Println("recv MSG:",rMsg)
			resultMap[rMsg.Cid] = rMsg
		} else {
			fmt.Println("recv msg Err:",err)
		}

	}
}