From 9b777922ae7bc72c746b433009334922de324f1e Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Mon, 16 Mar 2020 15:11:08 +0100
Subject: [PATCH] Allow CAP command when registered

---
 downstream.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/downstream.go b/downstream.go
index c5aa1e0..9b2e6b1 100644
--- a/downstream.go
+++ b/downstream.go
@@ -339,7 +339,6 @@ func (dc *downstreamConn) handleMessageUnregistered(msg *irc.Message) error {
 		if err := parseMessageParams(msg, &subCmd); err != nil {
 			return err
 		}
-		subCmd = strings.ToUpper(subCmd)
 		if err := dc.handleCapCommand(subCmd, msg.Params[1:]); err != nil {
 			return err
 		}
@@ -354,6 +353,8 @@ func (dc *downstreamConn) handleMessageUnregistered(msg *irc.Message) error {
 }
 
 func (dc *downstreamConn) handleCapCommand(cmd string, args []string) error {
+	cmd = strings.ToUpper(cmd)
+
 	replyTo := dc.nick
 	if !dc.registered {
 		replyTo = "*"
@@ -620,6 +621,14 @@ func (dc *downstreamConn) runUntilRegistered() error {
 
 func (dc *downstreamConn) handleMessageRegistered(msg *irc.Message) error {
 	switch msg.Command {
+	case "CAP":
+		var subCmd string
+		if err := parseMessageParams(msg, &subCmd); err != nil {
+			return err
+		}
+		if err := dc.handleCapCommand(subCmd, msg.Params[1:]); err != nil {
+			return err
+		}
 	case "PING":
 		dc.SendMessage(&irc.Message{
 			Prefix:  dc.srv.prefix(),