package emqx

import (
	mqtt "github.com/eclipse/paho.mqtt.golang"
	"time"
)

var mq *MQTT

type PublishHook func()

type MQTT struct {
	*Options
	client      mqtt.Client
	publishHook PublishHook
}

func GetConnect() *MQTT {
	return mq
}

func InitClient(opt *Options) *MQTT {
	opts := mqtt.NewClientOptions().AddBroker("tcp://" + opt.Addr).SetClientID(opt.ID)
	opts.SetAutoReconnect(true)
	opts.SetConnectRetryInterval(time.Second * 5)
	opts.SetUsername(opt.Username)
	opts.SetPassword(opt.Password)
	opts.SetKeepAlive(60 * time.Second)
	opts.SetPingTimeout(1 * time.Second)
	opts.OnConnect = mqtt.OnConnectHandler(opt.ConnectHandler)
	opts.OnConnectionLost = mqtt.ConnectionLostHandler(opt.ConnectLostHandler)
	mq = &MQTT{
		Options: opt,
		client:  mqtt.NewClient(opts),
	}
	return mq
}

func (m *MQTT) Connect() error {
	if token := m.client.Connect(); token.Wait() && token.Error() != nil {
		m.Logger.Error(token.Error())
	}
	return nil
}

func (m *MQTT) Publish(topic string, data []byte) error {
	if token := m.client.Publish(topic, 1, false, data); token != nil {
		return token.Error()
	}
	return nil
}

// PublishShare 发布共享消息
func (m *MQTT) PublishShare(topic string, data []byte, group string) error {
	if group == "" {
		group = "default"
	}
	if token := m.client.Publish("$share/"+group+"/"+topic, 1, false, data); token != nil {
		return token.Error()
	}
	return nil
}