ergo/src/irc/channel.go

package irc

import (
	"sort"
)

type Channel struct {
	name       string
	key        string
	topic      string
	members    ClientSet
	inviteOnly bool
	invites    map[string]bool
	server     *Server
}

type ChannelSet map[*Channel]bool

// NewChannel creates a new channel from a `Server` and a `name` string, which
// must be unique on the server.
func NewChannel(s *Server, name string) *Channel {
	return &Channel{
		name:    name,
		members: make(ClientSet),
		invites: make(map[string]bool),
		server:  s,
	}
}

// Send a `Reply` to all `Client`s of the `Channel`. Skip `fromClient`, if it is
// provided.
func (ch *Channel) Send(reply Reply, fromClient *Client) {
	for client := range ch.members {
		if client != fromClient {
			client.send <- reply
		}
	}
}

func (ch *Channel) Nicks() []string {
	nicks := make([]string, len(ch.members))
	i := 0
	for member := range ch.members {
		nicks[i] = member.Nick()
		i++
	}
	sort.Strings(nicks)
	return nicks
}

//
// channel functionality
//

func (ch *Channel) Join(cl *Client, key string) {
	if ch.key != key {
		cl.send <- ErrInviteOnlyChannel(ch)
		return
	}

	if ch.inviteOnly && !ch.invites[cl.nick] {
		cl.send <- ErrBadChannelKey(ch)
		return
	}

	ch.members[cl] = true
	cl.channels[ch] = true

	ch.Send(RplJoin(ch, cl), nil)
	ch.GetTopic(cl)
	cl.send <- RplNamReply(ch)
	cl.send <- RplEndOfNames(ch.server)
}

func (ch *Channel) Part(cl *Client, message string) {
	if !ch.members[cl] {
		cl.send <- ErrNotOnChannel(ch)
		return
	}

	if message == "" {
		message = cl.Nick()
	}

	ch.Send(RplPart(ch, cl, message), nil)

	delete(ch.members, cl)
	delete(cl.channels, ch)
}

func (ch *Channel) PrivMsg(cl *Client, message string) {
	ch.Send(RplPrivMsgChannel(ch, cl, message), cl)
}

func (ch *Channel) GetTopic(cl *Client) {
	if !ch.members[cl] {
		cl.send <- ErrNotOnChannel(ch)
		return
	}

	if ch.topic != "" {
		cl.send <- RplTopic(ch)
	} else {
		cl.send <- RplNoTopic(ch)
	}
}

func (ch *Channel) ChangeTopic(cl *Client, newTopic string) {
	if !ch.members[cl] {
		cl.send <- ErrNotOnChannel(ch)
		return
	}

	ch.topic = newTopic

	if ch.topic != "" {
		ch.Send(RplTopic(ch), nil)
	} else {
		ch.Send(RplNoTopic(ch), nil)
	}
}

func (ch *Channel) Invite(inviter *Client, invitee *Client) {
	if !ch.members[inviter] {
		inviter.send <- ErrNotOnChannel(ch)
		return
	}

	if ch.members[invitee] {
		inviter.send <- ErrUserOnChannel(ch, invitee)
		return
	}

	ch.invites[invitee.nick] = true

	invitee.send <- RplInviteMsg(ch, inviter)
	inviter.send <- RplInvitingMsg(ch, invitee)
}