adding theater-mode, fixes #15

irc/theater.go

@@ -0,0 +1,121 @@
+package irc
+
+import (
+	"fmt"
+)
+
+type TheaterClient Name
+
+func (c TheaterClient) Id() Name {
+	return Name(c)
+}
+
+func (c TheaterClient) Nick() Name {
+	return Name(c)
+}
+
+type TheaterSubCommand string
+
+type theaterSubCommand interface {
+	String() string
+}
+
+type TheaterIdentifyCommand struct {
+	PassCommand
+	channel Name
+}
+
+func (m *TheaterIdentifyCommand) LoadPassword(s *Server) {
+	m.hash = s.theaters[m.channel]
+}
+
+func (cmd *TheaterIdentifyCommand) String() string {
+	return fmt.Sprintf("THEATER_IDENTIFY(channel=%s)", cmd.channel)
+}
+
+func (m *TheaterIdentifyCommand) HandleServer(s *Server) {
+	client := m.Client()
+	if !m.channel.IsChannel() {
+		client.ErrNoSuchChannel(m.channel)
+		return
+	}
+
+	channel := s.channels.Get(m.channel)
+	if channel == nil {
+		client.ErrNoSuchChannel(m.channel)
+		return
+	}
+
+	if (m.hash == nil) || (m.err != nil) {
+		client.ErrPasswdMismatch()
+		return
+	}
+
+	if channel.theaterUser == nil {
+		client.theaterChannels = append(client.theaterChannels, channel)
+		channel.flags[Theater] = true
+		channel.theaterUser = client
+	}
+}
+
+type TheaterPrivMsgCommand struct {
+	BaseCommand
+	channel Name
+	asNick  Name
+	message Text
+}
+
+func (cmd *TheaterPrivMsgCommand) String() string {
+	return fmt.Sprintf("THEATER_PRIVMSG(channel=%s, asNick=%s, message=%s)", cmd.channel, cmd.asNick, cmd.message)
+
+}
+func (m *TheaterPrivMsgCommand) HandleServer(s *Server) {
+	client := m.Client()
+	if !m.channel.IsChannel() {
+		client.ErrNoSuchChannel(m.channel)
+		return
+	}
+
+	channel := s.channels.Get(m.channel)
+	if channel == nil {
+		client.ErrNoSuchChannel(m.channel)
+		return
+	}
+
+	if channel.theaterUser == client {
+		for member := range channel.members {
+			member.Reply(RplPrivMsg(TheaterClient(m.asNick), channel, m.message))
+		}
+	}
+}
+
+type TheaterActionCommand struct {
+	BaseCommand
+	channel Name
+	asNick  Name
+	action  Text
+}
+
+func (cmd *TheaterActionCommand) String() string {
+	return fmt.Sprintf("THEATER_ACTION(channel=%s, asNick=%s, action=%s)", cmd.channel, cmd.asNick, cmd.action)
+}
+
+func (m *TheaterActionCommand) HandleServer(s *Server) {
+	client := m.Client()
+	if m.channel.IsChannel() {
+		client.ErrNoSuchChannel(m.channel)
+		return
+	}
+
+	channel := s.channels.Get(m.channel)
+	if channel == nil {
+		client.ErrNoSuchChannel(m.channel)
+		return
+	}
+
+	if channel.theaterUser == client {
+		for member := range channel.members {
+			member.Reply(RplPrivMsg(TheaterClient(m.asNick), channel, NewText(fmt.Sprintf("\001ACTION %s\001", m.action))))
+		}
+	}
+}