Add more parseMessage() test cases, handle a prefix edge case

irc/message.go

@@ -24,6 +24,8 @@ func parseMessage(line string) *Message {
 
 		if i := strings.Index(msg.Prefix, "!"); i > 0 {
 			msg.Nick = msg.Prefix[:i]
+		} else if i := strings.Index(msg.Prefix, "@"); i > 0 {
+			msg.Nick = msg.Prefix[:i]
 		} else {
 			msg.Nick = msg.Prefix
 		}

irc/message_test.go

@@ -34,6 +34,38 @@ func TestParseMessage(t *testing.T) {
 				Command: "CMD",
 				Params:  []string{"a", "b"},
 			},
+		}, {
+			"CMD a b\r\n",
+			&Message{
+				Command: "CMD",
+				Params:  []string{"a", "b"},
+			},
+		}, {
+			"CMD\r\n",
+			&Message{
+				Command: "CMD",
+			},
+		}, {
+			"CMD :tests and stuff\r\n",
+			&Message{
+				Command:  "CMD",
+				Params:   []string{"tests and stuff"},
+				Trailing: "tests and stuff",
+			},
+		}, {
+			":nick@host.com CMD\r\n",
+			&Message{
+				Prefix:  "nick@host.com",
+				Nick:    "nick",
+				Command: "CMD",
+			},
+		}, {
+			":ni@ck!user!name@host!.com CMD\r\n",
+			&Message{
+				Prefix:  "ni@ck!user!name@host!.com",
+				Nick:    "ni@ck",
+				Command: "CMD",
+			},
 		},
 	}