67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
package irc
|
|
|
|
import (
|
|
"strings"
|
|
)
|
|
|
|
type Message struct {
|
|
Prefix string
|
|
Nick string
|
|
Command string
|
|
Params []string
|
|
}
|
|
|
|
func (m *Message) LastParam() string {
|
|
if len(m.Params) > 0 {
|
|
return m.Params[len(m.Params)-1]
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func parseMessage(line string) *Message {
|
|
line = strings.Trim(line, "\r\n")
|
|
msg := Message{}
|
|
cmdStart := 0
|
|
cmdEnd := len(line)
|
|
|
|
if strings.HasPrefix(line, ":") {
|
|
cmdStart = strings.Index(line, " ") + 1
|
|
|
|
if cmdStart > 0 {
|
|
msg.Prefix = line[1 : cmdStart-1]
|
|
} else {
|
|
// Invalid message
|
|
return &msg
|
|
}
|
|
|
|
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
|
|
}
|
|
}
|
|
|
|
var usesTrailing bool
|
|
var trailing string
|
|
|
|
if i := strings.Index(line, " :"); i > 0 {
|
|
cmdEnd = i
|
|
trailing = line[i+2:]
|
|
usesTrailing = true
|
|
}
|
|
|
|
cmd := strings.Split(line[cmdStart:cmdEnd], " ")
|
|
msg.Command = cmd[0]
|
|
if len(cmd) > 1 {
|
|
msg.Params = cmd[1:]
|
|
}
|
|
|
|
if usesTrailing {
|
|
msg.Params = append(msg.Params, trailing)
|
|
}
|
|
|
|
return &msg
|
|
}
|