Spark/client/common/common.go

package common

import (
	"Spark/client/config"
	"Spark/modules"
	"Spark/utils"
	"encoding/hex"
	ws "github.com/gorilla/websocket"
	"github.com/imroc/req/v3"
	"sync"
	"time"
)

type Conn struct {
	*ws.Conn
	Secret []byte
}

var WSConn *Conn
var lock = sync.Mutex{}

func SendPack(pack interface{}, wsConn *Conn) error {
	lock.Lock()
	defer lock.Unlock()
	data, err := utils.JSON.Marshal(pack)
	if err != nil {
		return err
	}
	data, err = utils.Encrypt(data, wsConn.Secret)
	if err != nil {
		return err
	}
	if len(data) > 1024 {
		_, err = req.C().R().
			SetBody(data).
			SetHeader(`Secret`, hex.EncodeToString(wsConn.Secret)).
			Send(`POST`, config.GetBaseURL(false)+`/ws`)
		return err
	}
	wsConn.SetWriteDeadline(time.Now().Add(5 * time.Second))
	defer wsConn.SetWriteDeadline(time.Time{})
	return wsConn.WriteMessage(ws.BinaryMessage, data)
}

func SendCb(pack, prev modules.Packet, wsConn *Conn) error {
	if prev.Data != nil {
		trigger, ok := prev.Data[`event`]
		if ok {
			if pack.Data == nil {
				pack.Data = map[string]interface{}{
					`callback`: trigger,
				}
			} else {
				pack.Data[`callback`] = trigger
			}
		}
	}
	return SendPack(pack, wsConn)
}