away modes

irc/channel.go

@@ -55,11 +55,13 @@ func (channel *Channel) Destroy() error {
 	return nil
 }
 
-func (channel *Channel) Reply(reply Reply) error {
+func (channel *Channel) Reply(replies ...Reply) error {
 	if channel.replies == nil {
 		return ErrAlreadyDestroyed
 	}
+	for _, reply := range replies {
 		channel.replies <- reply
+	}
 	return nil
 }
 

irc/client.go

@@ -12,6 +12,7 @@ import (
 
 type Client struct {
 	away        bool
+	awayMessage string
 	channels    ChannelSet
 	conn        net.Conn
 	hostname    string

irc/commands.go

@@ -18,6 +18,7 @@ var (
 	NotEnoughArgsError = errors.New("not enough arguments")
 	ErrParseCommand    = errors.New("failed to parse message")
 	parseCommandFuncs  = map[string]parseCommandFunc{
+		"AWAY":    NewAwayCommand,
 		"CAP":     NewCapCommand,
 		"JOIN":    NewJoinCommand,
 		"MODE":    NewModeCommand,
@@ -609,3 +610,24 @@ func NewProxyCommand(args []string) (editableCommand, error) {
 		destPort:   args[4],
 	}, nil
 }
+
+type AwayCommand struct {
+	BaseCommand
+	text string
+	away bool
+}
+
+func (msg *AwayCommand) String() string {
+	return fmt.Sprintf("AWAY(%s)", msg.text)
+}
+
+func NewAwayCommand(args []string) (editableCommand, error) {
+	cmd := &AwayCommand{}
+
+	if len(args) > 0 {
+		cmd.text = args[0]
+		cmd.away = true
+	}
+
+	return cmd, nil
+}

irc/reply.go

@@ -252,8 +252,23 @@ func RplBanList(channel *Channel, ban UserMask) Reply {
 }
 
 func RplEndOfBanList(channel *Channel) Reply {
-	return NewNumericReply(channel.server, RPL_ENDOFBANLIST, "%s :End of channel ban list",
+	return NewNumericReply(channel.server, RPL_ENDOFBANLIST,
-		channel.name)
+		"%s :End of channel ban list", channel.name)
+}
+
+func RplNowAway(server *Server) Reply {
+	return NewNumericReply(server, RPL_NOWAWAY,
+		":You have been marked as being away")
+}
+
+func RplUnAway(server *Server) Reply {
+	return NewNumericReply(server, RPL_UNAWAY,
+		":You are no longer marked as being away")
+}
+
+func RplAway(server *Server, client *Client) Reply {
+	return NewNumericReply(server, RPL_AWAY,
+		"%s :%s", client.nick, client.awayMessage)
 }
 
 // errors (also numeric)

irc/server.go

@@ -147,12 +147,12 @@ func (s *Server) tryRegister(c *Client) {
 			RplYourHost(s),
 			RplCreated(s),
 			RplMyInfo(s))
-		server.MOTD(c)
+		s.MOTD(c)
 	}
 }
 
 func (server *Server) MOTD(client *Client) {
-	c.Reply(ErrNoMOTD(server))
+	client.Reply(ErrNoMOTD(server))
 }
 
 func (s *Server) Id() string {
@@ -305,6 +305,9 @@ func (m *PrivMsgCommand) HandleServer(s *Server) {
 		return
 	}
 	target.Reply(RplPrivMsg(m.Client(), target, m.message))
+	if target.away {
+		m.Client().Reply(RplAway(s, target))
+	}
 }
 
 func (m *ModeCommand) HandleServer(s *Server) {
@@ -407,3 +410,15 @@ func (msg *CapCommand) HandleServer(server *Server) {
 func (msg *ProxyCommand) HandleServer(server *Server) {
 	msg.Client().hostname = LookupHostname(msg.sourceIP)
 }
+
+func (msg *AwayCommand) HandleServer(server *Server) {
+	client := msg.Client()
+	client.away = msg.away
+	client.awayMessage = msg.text
+
+	if client.away {
+		client.Reply(RplNowAway(server))
+	} else {
+		client.Reply(RplUnAway(server))
+	}
+}

irc/types.go

@@ -108,7 +108,7 @@ type Identifier interface {
 }
 
 type Replier interface {
-	Reply(Reply) error
+	Reply(...Reply) error
 }
 
 type Reply interface {