CAP: Primitive negotiation

1 files changed, 34 insertions, 1 deletions

diff --git a/cmd_cap.go b/cmd_cap.go
index 328bce6..c1ddb1d 100644
--- a/cmd_cap.go
+++ b/cmd_cap.go
@@ -16,13 +16,46 @@ func handleClientCap(msg RMsg, client *Client) error {
 		}
 		return nil
 	}
+	if client.State == ClientStateRemote {
+		return ErrRemoteClient
+	}
 	switch strings.ToUpper(msg.Params[0]) {
 	case "LS":
-		err := client.Send(MakeMsg(self, "CAP", client.Nick, "LS", "sasl=PLAIN,EXTERNAL"))
+		if client.State == ClientStatePreRegistration {
+			client.State = ClientStateCapabilities
+		}
+		err := client.Send(MakeMsg(self, "CAP", client.Nick, "LS", capls))
+		// TODO: Split when too long
 		if err != nil {
 			return err
 		}
 	case "REQ":
+		if client.State == ClientStatePreRegistration {
+			client.State = ClientStateCapabilities
+		}
+		caps := strings.Split(msg.Params[1], " ")
+		for _, c := range caps {
+			if c[0] == '-' {
+				// TODO: Remove capability
+				delete(client.Caps, c)
+				continue
+			}
+			_, ok := Caps[c]
+			if ok {
+				client.Send(MakeMsg(self, "CAP", client.Nick, "ACK", c))
+				client.Caps[c] = struct{}{}
+				// TODO: This is terrible
+			} else {
+				client.Send(MakeMsg(self, "CAP", client.Nick, "NAK", c))
+			}
+		}
+	case "END":
+		if client.State != ClientStateCapabilities {
+			// Just ignore it
+			return nil
+		}
+		client.State = ClientStateCapabilitiesFinished
+		client.checkRegistration()
 	}
 	return nil
 }