hub.go

package main

import (
	"fmt"
	"strconv"
	"strings"
)

type hub struct {
	channels        map[string]*channel
	clients         map[string]*client
	commands        chan command
	deregistrations chan *client
	registrations   chan *client
}

func newHub() *hub {
	return &hub{
		registrations:   make(chan *client),
		deregistrations: make(chan *client),
		clients:         make(map[string]*client),
		channels:        make(map[string]*channel),
		commands:        make(chan command),
	}
}

func (h *hub) run() {
	for {
		select {
		case client := <-h.registrations:
			h.register(client)
		case client := <-h.deregistrations:
			h.deregister(client)
		case cmd := <-h.commands:
			switch cmd.id {
			case JOIN:
				h.joinChannel(cmd.sender, cmd.recipient)
			case LEAVE:
				h.leaveChannel(cmd.sender, cmd.recipient)
			case MSG:
				h.message(cmd.sender, cmd.recipient, cmd.body)
			case DATA:
				h.data(cmd.sender, cmd.recipient, cmd.body)
			case USRS:
				h.listUsers(cmd.sender)
			case CHNS:
				h.listChannels(cmd.sender)
			default:
				// Freak out?
			}
		}
	}
}

func (h *hub) register(c *client) {
	//if _, exists := h.clients[c.username]; exists {
	//	c.username = ""
	//	c.conn.Write([]byte("ERR username taken\n"))
	//} else {
	//}

	player := newPlayer(c.username)
	if player != nil {
		player.read()

		h.clients[c.username] = c
		c.conn.Write([]byte("REGOK\n"))

		gender := append([]byte("GENDER "), []byte(player.Gender)...)
		gender = append(gender, []byte("\n")...)
		c.conn.Write([]byte(gender))
	}
}

func (h *hub) deregister(c *client) {
	if _, exists := h.clients[c.username]; exists {
		delete(h.clients, c.username)
		fmt.Println("User unregistered.")
		for _, channel := range h.channels {
			delete(channel.clients, c)
		}
	}
}

func (h *hub) joinChannel(u string, c string) {
	if client, ok := h.clients[u]; ok {
		if channel, ok := h.channels[c]; ok {
			// Channel exists, join
			channel.clients[client] = true
		} else {
			// Channel doesn't exists, create and join
			ch := newChannel(c)
			ch.clients[client] = true
			h.channels[c] = ch
		}
		client.conn.Write([]byte("CHOK\n"))
	}
}

func (h *hub) leaveChannel(u string, c string) {
	if client, ok := h.clients[u]; ok {
		if channel, ok := h.channels[c]; ok {
			delete(channel.clients, client)
		}
	}
}

func (h *hub) message(u string, r string, m []byte) {
	if sender, ok := h.clients[u]; ok {
		switch r[0] {
		case '#':
			if channel, ok := h.channels[r]; ok {
				if _, ok := channel.clients[sender]; ok {
					channel.broadcast(sender.username, m)
				}
			} else {
				sender.conn.Write([]byte("ERR no such channel"))
			}
		case '@':
			if user, ok := h.clients[r]; ok {

				player := newPlayer(sender.username)
				if player != nil {
					player.read()

					msg := append([]byte("!"), []byte(player.ShoutColor)...)
					msg = append(msg, []byte(sender.username)...)
					msg = append(msg, ": "...)
					msg = append(msg, m...)
					msg = append(msg, '\n')

					user.conn.Write(msg)
				}

			} else {
				sender.conn.Write([]byte("ERR no such user"))
			}
		default:
			sender.conn.Write([]byte("ERR MSG command"))
		}
	}
}

func (h *hub) data(u string, r string, m []byte) {
	if sender, ok := h.clients[u]; ok {

		switch r {
		case "#INFO":
			player := newPlayer(sender.username)
			if player != nil {
				player.readscore()
				player.read()

				msg := append([]byte("UINFO "), []byte(sender.username)...)
				msg = append(msg, " "...)
				msg = append(msg, []byte(strconv.Itoa(player.TotalScore))...)
				msg = append(msg, " "...)
				msg = append(msg, []byte(strconv.Itoa(player.SeasonScore))...)
				msg = append(msg, " "...)
				msg = append(msg, []byte(strconv.Itoa(player.TotalRank))...)
				msg = append(msg, " "...)
				msg = append(msg, []byte(strconv.Itoa(player.SeasonRank))...)
				msg = append(msg, " "...)
				msg = append(msg, []byte(strconv.Itoa(player.ShoutCount))...)
				msg = append(msg, " "...)
				msg = append(msg, []byte(strconv.Itoa(player.Country))...)
				msg = append(msg, '\n')

				sender.conn.Write(msg)
			}
		default:
			sender.conn.Write([]byte("ERR DATA command"))
		}
	}
}

func (h *hub) listChannels(u string) {
	if client, ok := h.clients[u]; ok {
		var names []string

		if len(h.channels) == 0 {
			client.conn.Write([]byte("ERR no channels found\n"))
		}

		for c := range h.channels {
			names = append(names, "#"+c+" ")
		}

		resp := strings.Join(names, ", ")

		client.conn.Write([]byte(resp + "\n"))
	}
}

func (h *hub) listUsers(u string) {
	if client, ok := h.clients[u]; ok {
		var names []string

		for c := range h.clients {
			names = append(names, "@"+c+" ")
		}

		resp := strings.Join(names, ", ")

		client.conn.Write([]byte(resp + "\n"))
	}
}