package hub

import (
	"log"
)

var h *Hub = nil

func InitHub() {
	if h == nil {
		h = NewHub()
	}
}

func NewHub() *Hub {
	return &Hub{
		clientMap:  make(map[ClientID]*Client),
		register:   make(chan *Client, 1024),
		unregister: make(chan *Client, 1024),
		message:    make(chan *AgentCmd, 1024),
	}
}

func Run() {
	InitHub()
	for {
		select {
		case client := <-h.register:
			h.clientMap[client.Id] = client
		case client := <-h.unregister:
			if _, ok := h.clientMap[client.Id]; ok {
				log.Println("unregister agent:", client.Id)
				delete(h.clientMap, client.Id)
				close(client.Send)
				client.Conn.Close()
			}
		case msg := <-h.message:
			if _, ok := h.clientMap[msg.AgentId]; ok {
				h.clientMap[msg.AgentId].Send <- []byte(msg.Cmd)
				log.Println("forward:", msg)
			}
		}
	}
}

func GetClientById(agentId string) *Client {
	if c, ok := h.clientMap[agentId]; ok {
		return c
	}
	return nil
}

func DeleteClient(c *Client) bool {
	h.unregister <- c
	return true
}

func AddClient(c *Client) bool {
	h.register <- c
	return true
}

func SendMessage(cmd *AgentCmd) {
	h.message <- cmd
}