package trace

import (
	"strings"
	"sync"

	"github.com/VolantMQ/volantmq/util"
)

var gTraceSwitch *Trace

type Trace struct {
	clientID string
	topic    string
	rwlock   sync.RWMutex
}

func init() {
	gTraceSwitch = &Trace{}

	return
}

func GetInstance() *Trace {
	return gTraceSwitch
}

func (t *Trace) SetClientID(id string) {
	t.rwlock.Lock()
	defer t.rwlock.Unlock()

	t.clientID = id
	t.topic = ""

	return
}

func (t *Trace) SetTopic(topic string) {
	t.rwlock.Lock()
	defer t.rwlock.Unlock()

	t.topic = topic
	t.clientID = ""

	return
}

func (t *Trace) SetOff() {
	t.rwlock.Lock()
	defer t.rwlock.Unlock()

	t.clientID = ""
	t.topic = ""

	return
}

func (t *Trace) Status(clientID string, topic string) bool {
	t.rwlock.RLock()
	defer t.rwlock.RUnlock()

	if t.clientID == "" && t.topic == "" {
		return false
	}

	if t.clientID != "" && clientID != "" {
		return strings.EqualFold(t.clientID, clientID)
	}

	if t.topic != "" && topic != "" {
		return util.MatchTopic(strings.Split(t.topic, "/"), strings.Split(topic, "/"))
	}

	return false
}