package webstocetmsg

import (
	"fmt"
	"log"
	"time"

	"github.com/gin-gonic/gin"
	"github.com/gorilla/websocket"
	"github.com/sirupsen/logrus"
)

func (a *ApiMethod) WebsocketDemo(c *gin.Context) {
	ws, err := upGraderes.Upgrade(c.Writer, c.Request, nil)
	fmt.Printf("err--1->%v\n", err)
	if err != nil {
		return
	}
	defer func(ws *websocket.Conn) {
		err = ws.Close()
		if err != nil {
			log.Fatalln(err)
		}
	}(ws)
	MsgHandler(c, ws)
}

func MsgHandler(c *gin.Context, ws *websocket.Conn) {
	// fmt.Printf("err--2->%s\n", ws)
	for {

		msg, err := Subscribe(c, PublishKey)
		fmt.Printf("err--2->%v--2->%v\n", msg, err)
		if err != nil {
			logrus.Errorf("Subscribe error: %s", err.Error())
			fmt.Printf("Subscribe error: %s\n", err.Error())
		}

		tm := time.Now().Format("2006-01-02 15:04:05")
		fmt.Printf("tm--->%v\n", tm)
		m := fmt.Sprintf("[ws][%s]:%s", tm, msg)
		err = ws.WriteMessage(1, []byte(m))
		fmt.Printf("m--->%v\n", m)
		fmt.Printf("err--->%v\n", err)
		if err != nil {
			log.Fatalln(err)
		}
	}
}