mirror of https://github.com/fluffle/goirc
Use the internal event handling for initial pass/user/nick messages.
Added constants for internal and named events. gofmt'ed all files and updated client.go/documentation.
This commit is contained in:
parent
a674267128
commit
8e16f59b5f
|
@ -24,11 +24,11 @@ Synopsis:
|
||||||
|
|
||||||
// Add handlers to do things here!
|
// Add handlers to do things here!
|
||||||
// e.g. join a channel on connect.
|
// e.g. join a channel on connect.
|
||||||
c.AddHandler("connected",
|
c.HandleFunc(irc.CONNECTED,
|
||||||
func(conn *irc.Conn, line *irc.Line) { conn.Join("#channel") })
|
func(conn *irc.Conn, line *irc.Line) { conn.Join("#channel") })
|
||||||
// And a signal on disconnect
|
// And a signal on disconnect
|
||||||
quit := make(chan bool)
|
quit := make(chan bool)
|
||||||
c.AddHandler("disconnected",
|
c.HandleFunc(irc.DISCONNECTED,
|
||||||
func(conn *irc.Conn, line *irc.Line) { quit <- true })
|
func(conn *irc.Conn, line *irc.Line) { quit <- true })
|
||||||
|
|
||||||
// Tell client to connect
|
// Tell client to connect
|
||||||
|
|
12
client.go
12
client.go
|
@ -1,11 +1,11 @@
|
||||||
package main
|
package main
|
||||||
|
|
||||||
import (
|
import (
|
||||||
irc "github.com/fluffle/goirc/client"
|
"bufio"
|
||||||
"flag"
|
"flag"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
irc "github.com/fluffle/goirc/client"
|
||||||
"os"
|
"os"
|
||||||
"bufio"
|
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -16,14 +16,14 @@ func main() {
|
||||||
flag.Parse()
|
flag.Parse()
|
||||||
|
|
||||||
// create new IRC connection
|
// create new IRC connection
|
||||||
c := irc.SimpleClient("GoTest", "gotest")
|
c := irc.Client("GoTest", "gotest")
|
||||||
c.EnableStateTracking()
|
c.EnableStateTracking()
|
||||||
c.AddHandler("connected",
|
c.HandleFunc(irc.CONNECTED,
|
||||||
func(conn *irc.Conn, line *irc.Line) { conn.Join(*channel) })
|
func(conn *irc.Conn, line *irc.Line) { conn.Join(*channel) })
|
||||||
|
|
||||||
// Set up a handler to notify of disconnect events.
|
// Set up a handler to notify of disconnect events.
|
||||||
quit := make(chan bool)
|
quit := make(chan bool)
|
||||||
c.AddHandler("disconnected",
|
c.HandleFunc(irc.DISCONNECTED,
|
||||||
func(conn *irc.Conn, line *irc.Line) { quit <- true })
|
func(conn *irc.Conn, line *irc.Line) { quit <- true })
|
||||||
|
|
||||||
// set up a goroutine to read commands from stdin
|
// set up a goroutine to read commands from stdin
|
||||||
|
@ -36,6 +36,8 @@ func main() {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
// wha?, maybe ctrl-D...
|
// wha?, maybe ctrl-D...
|
||||||
close(in)
|
close(in)
|
||||||
|
reallyquit = true
|
||||||
|
c.Quit("")
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
// no point in sending empty lines down the channel
|
// no point in sending empty lines down the channel
|
||||||
|
|
|
@ -2,6 +2,34 @@ package client
|
||||||
|
|
||||||
import "strings"
|
import "strings"
|
||||||
|
|
||||||
|
const (
|
||||||
|
INIT = "init"
|
||||||
|
CONNECTED = "connected"
|
||||||
|
DISCONNECTED = "disconnected"
|
||||||
|
ACTION = "ACTION"
|
||||||
|
AWAY = "AWAY"
|
||||||
|
CTCP = "CTCP"
|
||||||
|
CTCPREPLY = "CTCPREPLY"
|
||||||
|
INVITE = "INVITE"
|
||||||
|
JOIN = "JOIN"
|
||||||
|
KICK = "KICK"
|
||||||
|
MODE = "MODE"
|
||||||
|
NICK = "NICK"
|
||||||
|
NOTICE = "NOTICE"
|
||||||
|
OPER = "OPER"
|
||||||
|
PART = "PART"
|
||||||
|
PASS = "PASS"
|
||||||
|
PING = "PING"
|
||||||
|
PONG = "PONG"
|
||||||
|
PRIVMSG = "PRIVMSG"
|
||||||
|
QUIT = "QUIT"
|
||||||
|
TOPIC = "TOPIC"
|
||||||
|
USER = "USER"
|
||||||
|
VERSION = "VERSION"
|
||||||
|
WHO = "WHO"
|
||||||
|
WHOIS = "WHOIS"
|
||||||
|
)
|
||||||
|
|
||||||
// this file contains the various commands you can
|
// this file contains the various commands you can
|
||||||
// send to the server using an Conn connection
|
// send to the server using an Conn connection
|
||||||
|
|
||||||
|
@ -14,18 +42,18 @@ import "strings"
|
||||||
func (conn *Conn) Raw(rawline string) { conn.out <- rawline }
|
func (conn *Conn) Raw(rawline string) { conn.out <- rawline }
|
||||||
|
|
||||||
// Pass() sends a PASS command to the server
|
// Pass() sends a PASS command to the server
|
||||||
func (conn *Conn) Pass(password string) { conn.out <- "PASS " + password }
|
func (conn *Conn) Pass(password string) { conn.out <- PASS + " " + password }
|
||||||
|
|
||||||
// Nick() sends a NICK command to the server
|
// Nick() sends a NICK command to the server
|
||||||
func (conn *Conn) Nick(nick string) { conn.out <- "NICK " + nick }
|
func (conn *Conn) Nick(nick string) { conn.out <- NICK + " " + nick }
|
||||||
|
|
||||||
// User() sends a USER command to the server
|
// User() sends a USER command to the server
|
||||||
func (conn *Conn) User(ident, name string) {
|
func (conn *Conn) User(ident, name string) {
|
||||||
conn.out <- "USER " + ident + " 12 * :" + name
|
conn.out <- USER + " " + ident + " 12 * :" + name
|
||||||
}
|
}
|
||||||
|
|
||||||
// Join() sends a JOIN command to the server
|
// Join() sends a JOIN command to the server
|
||||||
func (conn *Conn) Join(channel string) { conn.out <- "JOIN " + channel }
|
func (conn *Conn) Join(channel string) { conn.out <- JOIN + " " + channel }
|
||||||
|
|
||||||
// Part() sends a PART command to the server with an optional part message
|
// Part() sends a PART command to the server with an optional part message
|
||||||
func (conn *Conn) Part(channel string, message ...string) {
|
func (conn *Conn) Part(channel string, message ...string) {
|
||||||
|
@ -33,7 +61,7 @@ func (conn *Conn) Part(channel string, message ...string) {
|
||||||
if msg != "" {
|
if msg != "" {
|
||||||
msg = " :" + msg
|
msg = " :" + msg
|
||||||
}
|
}
|
||||||
conn.out <- "PART " + channel + msg
|
conn.out <- PART + " " + channel + msg
|
||||||
}
|
}
|
||||||
|
|
||||||
// Kick() sends a KICK command to remove a nick from a channel
|
// Kick() sends a KICK command to remove a nick from a channel
|
||||||
|
@ -42,7 +70,7 @@ func (conn *Conn) Kick(channel, nick string, message ...string) {
|
||||||
if msg != "" {
|
if msg != "" {
|
||||||
msg = " :" + msg
|
msg = " :" + msg
|
||||||
}
|
}
|
||||||
conn.out <- "KICK " + channel + " " + nick + msg
|
conn.out <- KICK + " " + channel + " " + nick + msg
|
||||||
}
|
}
|
||||||
|
|
||||||
// Quit() sends a QUIT command to the server with an optional quit message
|
// Quit() sends a QUIT command to the server with an optional quit message
|
||||||
|
@ -51,20 +79,20 @@ func (conn *Conn) Quit(message ...string) {
|
||||||
if msg == "" {
|
if msg == "" {
|
||||||
msg = "GoBye!"
|
msg = "GoBye!"
|
||||||
}
|
}
|
||||||
conn.out <- "QUIT :" + msg
|
conn.out <- QUIT + " :" + msg
|
||||||
}
|
}
|
||||||
|
|
||||||
// Whois() sends a WHOIS command to the server
|
// Whois() sends a WHOIS command to the server
|
||||||
func (conn *Conn) Whois(nick string) { conn.out <- "WHOIS " + nick }
|
func (conn *Conn) Whois(nick string) { conn.out <- WHOIS + " " + nick }
|
||||||
|
|
||||||
//Who() sends a WHO command to the server
|
//Who() sends a WHO command to the server
|
||||||
func (conn *Conn) Who(nick string) { conn.out <- "WHO " + nick }
|
func (conn *Conn) Who(nick string) { conn.out <- WHO + " " + nick }
|
||||||
|
|
||||||
// Privmsg() sends a PRIVMSG to the target t
|
// Privmsg() sends a PRIVMSG to the target t
|
||||||
func (conn *Conn) Privmsg(t, msg string) { conn.out <- "PRIVMSG " + t + " :" + msg }
|
func (conn *Conn) Privmsg(t, msg string) { conn.out <- PRIVMSG + " " + t + " :" + msg }
|
||||||
|
|
||||||
// Notice() sends a NOTICE to the target t
|
// Notice() sends a NOTICE to the target t
|
||||||
func (conn *Conn) Notice(t, msg string) { conn.out <- "NOTICE " + t + " :" + msg }
|
func (conn *Conn) Notice(t, msg string) { conn.out <- NOTICE + " " + t + " :" + msg }
|
||||||
|
|
||||||
// Ctcp() sends a (generic) CTCP message to the target t
|
// Ctcp() sends a (generic) CTCP message to the target t
|
||||||
// with an optional argument
|
// with an optional argument
|
||||||
|
@ -87,10 +115,10 @@ func (conn *Conn) CtcpReply(t, ctcp string, arg ...string) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Version() sends a CTCP "VERSION" to the target t
|
// Version() sends a CTCP "VERSION" to the target t
|
||||||
func (conn *Conn) Version(t string) { conn.Ctcp(t, "VERSION") }
|
func (conn *Conn) Version(t string) { conn.Ctcp(t, VERSION) }
|
||||||
|
|
||||||
// Action() sends a CTCP "ACTION" to the target t
|
// Action() sends a CTCP "ACTION" to the target t
|
||||||
func (conn *Conn) Action(t, msg string) { conn.Ctcp(t, "ACTION", msg) }
|
func (conn *Conn) Action(t, msg string) { conn.Ctcp(t, ACTION, msg) }
|
||||||
|
|
||||||
// Topic() sends a TOPIC command to the channel
|
// Topic() sends a TOPIC command to the channel
|
||||||
// Topic(channel) retrieves the current channel topic (see "332" handler)
|
// Topic(channel) retrieves the current channel topic (see "332" handler)
|
||||||
|
@ -100,7 +128,7 @@ func (conn *Conn) Topic(channel string, topic ...string) {
|
||||||
if t != "" {
|
if t != "" {
|
||||||
t = " :" + t
|
t = " :" + t
|
||||||
}
|
}
|
||||||
conn.out <- "TOPIC " + channel + t
|
conn.out <- TOPIC + " " + channel + t
|
||||||
}
|
}
|
||||||
|
|
||||||
// Mode() sends a MODE command to the server. This one can get complicated if
|
// Mode() sends a MODE command to the server. This one can get complicated if
|
||||||
|
@ -115,7 +143,7 @@ func (conn *Conn) Mode(t string, modestring ...string) {
|
||||||
if mode != "" {
|
if mode != "" {
|
||||||
mode = " " + mode
|
mode = " " + mode
|
||||||
}
|
}
|
||||||
conn.out <- "MODE " + t + mode
|
conn.out <- MODE + " " + t + mode
|
||||||
}
|
}
|
||||||
|
|
||||||
// Away() sends an AWAY command to the server
|
// Away() sends an AWAY command to the server
|
||||||
|
@ -126,15 +154,18 @@ func (conn *Conn) Away(message ...string) {
|
||||||
if msg != "" {
|
if msg != "" {
|
||||||
msg = " :" + msg
|
msg = " :" + msg
|
||||||
}
|
}
|
||||||
conn.out <- "AWAY" + msg
|
conn.out <- AWAY + msg
|
||||||
}
|
}
|
||||||
|
|
||||||
// Invite() sends an INVITE command to the server
|
// Invite() sends an INVITE command to the server
|
||||||
func (conn *Conn) Invite(nick, channel string) {
|
func (conn *Conn) Invite(nick, channel string) { conn.out <- INVITE + " " + nick + " " + channel }
|
||||||
conn.out <- "INVITE " + nick + " " + channel
|
|
||||||
}
|
|
||||||
|
|
||||||
// Oper() sends an OPER command to the server
|
// Oper() sends an OPER command to the server
|
||||||
func (conn *Conn) Oper(user, pass string) {
|
func (conn *Conn) Oper(user, pass string) { conn.out <- OPER + " " + user + " " + pass }
|
||||||
conn.out <- "OPER " + user + " " + pass
|
|
||||||
}
|
// Ping() sends a PING command to the server
|
||||||
|
// A PONG response is to be expected afterwards
|
||||||
|
func (conn *Conn) Ping(message string) { conn.out <- PING + " :" + message }
|
||||||
|
|
||||||
|
// Pong() sends a PONG command to the server
|
||||||
|
func (conn *Conn) Pong(message string) { conn.out <- PONG + " :" + message }
|
||||||
|
|
|
@ -75,4 +75,10 @@ func TestClientCommands(t *testing.T) {
|
||||||
|
|
||||||
c.Oper("user", "pass")
|
c.Oper("user", "pass")
|
||||||
s.nc.Expect("OPER user pass")
|
s.nc.Expect("OPER user pass")
|
||||||
|
|
||||||
|
c.Ping("woot")
|
||||||
|
s.nc.Expect("PING :woot")
|
||||||
|
|
||||||
|
c.Pong("pwoot")
|
||||||
|
s.nc.Expect("PONG :pwoot")
|
||||||
}
|
}
|
||||||
|
|
|
@ -15,9 +15,10 @@ import (
|
||||||
// An IRC connection is represented by this struct.
|
// An IRC connection is represented by this struct.
|
||||||
type Conn struct {
|
type Conn struct {
|
||||||
// Connection Hostname and Nickname
|
// Connection Hostname and Nickname
|
||||||
Host string
|
Host string
|
||||||
Me *state.Nick
|
Me *state.Nick
|
||||||
Network string
|
Network string
|
||||||
|
password string
|
||||||
|
|
||||||
// Handlers and Commands
|
// Handlers and Commands
|
||||||
handlers *hSet
|
handlers *hSet
|
||||||
|
@ -60,7 +61,7 @@ type Conn struct {
|
||||||
Flood bool
|
Flood bool
|
||||||
|
|
||||||
// Internal counters for flood protection
|
// Internal counters for flood protection
|
||||||
badness time.Duration
|
badness time.Duration
|
||||||
lastsent time.Time
|
lastsent time.Time
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -164,13 +165,12 @@ func (conn *Conn) Connect(host string, pass ...string) error {
|
||||||
}
|
}
|
||||||
conn.Host = host
|
conn.Host = host
|
||||||
conn.Connected = true
|
conn.Connected = true
|
||||||
conn.postConnect()
|
|
||||||
|
|
||||||
if len(pass) > 0 {
|
if len(pass) > 0 {
|
||||||
conn.Pass(pass[0])
|
conn.password = pass[0]
|
||||||
|
} else {
|
||||||
|
conn.password = ""
|
||||||
}
|
}
|
||||||
conn.Nick(conn.Me.Nick)
|
conn.postConnect()
|
||||||
conn.User(conn.Me.Ident, conn.Me.Name)
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -188,6 +188,7 @@ func (conn *Conn) postConnect() {
|
||||||
go func() { <-conn.cPing }()
|
go func() { <-conn.cPing }()
|
||||||
}
|
}
|
||||||
go conn.runLoop()
|
go conn.runLoop()
|
||||||
|
conn.dispatch(&Line{Cmd: INIT})
|
||||||
}
|
}
|
||||||
|
|
||||||
// copied from http.client for great justice
|
// copied from http.client for great justice
|
||||||
|
@ -235,7 +236,7 @@ func (conn *Conn) ping() {
|
||||||
for {
|
for {
|
||||||
select {
|
select {
|
||||||
case <-tick.C:
|
case <-tick.C:
|
||||||
conn.Raw(fmt.Sprintf("PING :%d", time.Now().UnixNano()))
|
conn.Ping(fmt.Sprintf("%d", time.Now().UnixNano()))
|
||||||
case <-conn.cPing:
|
case <-conn.cPing:
|
||||||
tick.Stop()
|
tick.Stop()
|
||||||
return
|
return
|
||||||
|
@ -305,7 +306,7 @@ func (conn *Conn) shutdown() {
|
||||||
// as calling sock.Close() will cause recv() to recieve EOF in readstring()
|
// as calling sock.Close() will cause recv() to recieve EOF in readstring()
|
||||||
if conn.Connected {
|
if conn.Connected {
|
||||||
logging.Info("irc.shutdown(): Disconnected from server.")
|
logging.Info("irc.shutdown(): Disconnected from server.")
|
||||||
conn.dispatch(&Line{Cmd: "disconnected"})
|
conn.dispatch(&Line{Cmd: DISCONNECTED})
|
||||||
conn.Connected = false
|
conn.Connected = false
|
||||||
conn.sock.Close()
|
conn.sock.Close()
|
||||||
conn.cSend <- true
|
conn.cSend <- true
|
||||||
|
|
|
@ -3,8 +3,8 @@ package client
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
"code.google.com/p/gomock/gomock"
|
"code.google.com/p/gomock/gomock"
|
||||||
"github.com/fluffle/golog/logging"
|
|
||||||
"github.com/fluffle/goirc/state"
|
"github.com/fluffle/goirc/state"
|
||||||
|
"github.com/fluffle/golog/logging"
|
||||||
"strings"
|
"strings"
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
@ -33,6 +33,10 @@ func setUp(t *testing.T, start ...bool) (*Conn, *testState) {
|
||||||
// Hack to allow tests of send, recv, write etc.
|
// Hack to allow tests of send, recv, write etc.
|
||||||
// NOTE: the value of the boolean doesn't matter.
|
// NOTE: the value of the boolean doesn't matter.
|
||||||
c.postConnect()
|
c.postConnect()
|
||||||
|
// All connections start with NICK/USER expect these.
|
||||||
|
nc.Expect("NICK test")
|
||||||
|
nc.Expect("USER test 12 * :Testing IRC")
|
||||||
|
|
||||||
// Sleep 1ms to allow background routines to start.
|
// Sleep 1ms to allow background routines to start.
|
||||||
<-time.After(1e6)
|
<-time.After(1e6)
|
||||||
}
|
}
|
||||||
|
@ -57,7 +61,7 @@ func TestEOF(t *testing.T) {
|
||||||
|
|
||||||
// Set up a handler to detect whether disconnected handlers are called
|
// Set up a handler to detect whether disconnected handlers are called
|
||||||
dcon := false
|
dcon := false
|
||||||
c.HandleFunc("disconnected", func (conn *Conn, line *Line) {
|
c.HandleFunc(DISCONNECTED, func(conn *Conn, line *Line) {
|
||||||
dcon = true
|
dcon = true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -356,12 +360,12 @@ func TestRunLoop(t *testing.T) {
|
||||||
|
|
||||||
// Set up a handler to detect whether 001 handler is called
|
// Set up a handler to detect whether 001 handler is called
|
||||||
h001 := false
|
h001 := false
|
||||||
c.HandleFunc("001", func (conn *Conn, line *Line) {
|
c.HandleFunc("001", func(conn *Conn, line *Line) {
|
||||||
h001 = true
|
h001 = true
|
||||||
})
|
})
|
||||||
// Set up a handler to detect whether 002 handler is called
|
// Set up a handler to detect whether 002 handler is called
|
||||||
h002 := false
|
h002 := false
|
||||||
c.HandleFunc("002", func (conn *Conn, line *Line) {
|
c.HandleFunc("002", func(conn *Conn, line *Line) {
|
||||||
h002 = true
|
h002 = true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -470,7 +474,7 @@ func TestRateLimit(t *testing.T) {
|
||||||
|
|
||||||
// We'll be needing this later...
|
// We'll be needing this later...
|
||||||
abs := func(i time.Duration) time.Duration {
|
abs := func(i time.Duration) time.Duration {
|
||||||
if (i < 0) {
|
if i < 0 {
|
||||||
return -i
|
return -i
|
||||||
}
|
}
|
||||||
return i
|
return i
|
||||||
|
@ -491,13 +495,13 @@ func TestRateLimit(t *testing.T) {
|
||||||
// 2.5 seconds minus the delta between the two ratelimit calls. This should
|
// 2.5 seconds minus the delta between the two ratelimit calls. This should
|
||||||
// be minimal but it's guaranteed that it won't be zero. Use 10us as a fuzz.
|
// be minimal but it's guaranteed that it won't be zero. Use 10us as a fuzz.
|
||||||
if l := c.rateLimit(60); l != 0 ||
|
if l := c.rateLimit(60); l != 0 ||
|
||||||
abs(c.badness - 2500*time.Millisecond) > 10 * time.Microsecond {
|
abs(c.badness-2500*time.Millisecond) > 10*time.Microsecond {
|
||||||
t.Errorf("Rate limit calculating badness incorrectly.")
|
t.Errorf("Rate limit calculating badness incorrectly.")
|
||||||
}
|
}
|
||||||
// At this point, we can tip over the badness scale, with a bit of help.
|
// At this point, we can tip over the badness scale, with a bit of help.
|
||||||
// 720 chars => +8 seconds of badness => 10.5 seconds => ratelimit
|
// 720 chars => +8 seconds of badness => 10.5 seconds => ratelimit
|
||||||
if l := c.rateLimit(720); l != 8 * time.Second ||
|
if l := c.rateLimit(720); l != 8*time.Second ||
|
||||||
abs(c.badness - 10500*time.Millisecond) > 10 * time.Microsecond {
|
abs(c.badness-10500*time.Millisecond) > 10*time.Microsecond {
|
||||||
t.Errorf("Rate limit failed to return correct limiting values.")
|
t.Errorf("Rate limit failed to return correct limiting values.")
|
||||||
t.Errorf("l=%d, badness=%d", l, c.badness)
|
t.Errorf("l=%d, badness=%d", l, c.badness)
|
||||||
}
|
}
|
||||||
|
|
|
@ -105,7 +105,9 @@ func (hs *hSet) dispatch(conn *Conn, line *Line) {
|
||||||
defer hs.RUnlock()
|
defer hs.RUnlock()
|
||||||
ev := strings.ToLower(line.Cmd)
|
ev := strings.ToLower(line.Cmd)
|
||||||
list, ok := hs.set[ev]
|
list, ok := hs.set[ev]
|
||||||
if !ok { return }
|
if !ok {
|
||||||
|
return
|
||||||
|
}
|
||||||
for hn := list.start; hn != nil; hn = hn.next {
|
for hn := list.start; hn != nil; hn = hn.next {
|
||||||
go hn.Handle(conn, line)
|
go hn.Handle(conn, line)
|
||||||
}
|
}
|
||||||
|
|
|
@ -83,7 +83,7 @@ func TestHandlerSet(t *testing.T) {
|
||||||
if callcount != 0 {
|
if callcount != 0 {
|
||||||
t.Errorf("Something incremented call count before we were expecting it.")
|
t.Errorf("Something incremented call count before we were expecting it.")
|
||||||
}
|
}
|
||||||
hs.dispatch(nil, &Line{Cmd:"One"})
|
hs.dispatch(nil, &Line{Cmd: "One"})
|
||||||
<-time.After(time.Millisecond)
|
<-time.After(time.Millisecond)
|
||||||
if callcount != 4 {
|
if callcount != 4 {
|
||||||
t.Errorf("Our handler wasn't called four times :-(")
|
t.Errorf("Our handler wasn't called four times :-(")
|
||||||
|
@ -107,7 +107,7 @@ func TestHandlerSet(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Dispatch should result in 3 additions.
|
// Dispatch should result in 3 additions.
|
||||||
hs.dispatch(nil, &Line{Cmd:"One"})
|
hs.dispatch(nil, &Line{Cmd: "One"})
|
||||||
<-time.After(time.Millisecond)
|
<-time.After(time.Millisecond)
|
||||||
if callcount != 7 {
|
if callcount != 7 {
|
||||||
t.Errorf("Our handler wasn't called three times :-(")
|
t.Errorf("Our handler wasn't called three times :-(")
|
||||||
|
@ -129,7 +129,7 @@ func TestHandlerSet(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Dispatch should result in 2 additions.
|
// Dispatch should result in 2 additions.
|
||||||
hs.dispatch(nil, &Line{Cmd:"One"})
|
hs.dispatch(nil, &Line{Cmd: "One"})
|
||||||
<-time.After(time.Millisecond)
|
<-time.After(time.Millisecond)
|
||||||
if callcount != 9 {
|
if callcount != 9 {
|
||||||
t.Errorf("Our handler wasn't called two times :-(")
|
t.Errorf("Our handler wasn't called two times :-(")
|
||||||
|
@ -151,7 +151,7 @@ func TestHandlerSet(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Dispatch should result in 1 addition.
|
// Dispatch should result in 1 addition.
|
||||||
hs.dispatch(nil, &Line{Cmd:"One"})
|
hs.dispatch(nil, &Line{Cmd: "One"})
|
||||||
<-time.After(time.Millisecond)
|
<-time.After(time.Millisecond)
|
||||||
if callcount != 10 {
|
if callcount != 10 {
|
||||||
t.Errorf("Our handler wasn't called once :-(")
|
t.Errorf("Our handler wasn't called once :-(")
|
||||||
|
@ -170,7 +170,7 @@ func TestHandlerSet(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
// Dispatch should result in NO additions.
|
// Dispatch should result in NO additions.
|
||||||
hs.dispatch(nil, &Line{Cmd:"One"})
|
hs.dispatch(nil, &Line{Cmd: "One"})
|
||||||
<-time.After(time.Millisecond)
|
<-time.After(time.Millisecond)
|
||||||
if callcount != 10 {
|
if callcount != 10 {
|
||||||
t.Errorf("Our handler was called?")
|
t.Errorf("Our handler was called?")
|
||||||
|
@ -184,7 +184,7 @@ func TestCommandSet(t *testing.T) {
|
||||||
}
|
}
|
||||||
|
|
||||||
c := &command{
|
c := &command{
|
||||||
fn: func(c *Conn, l *Line) {},
|
fn: func(c *Conn, l *Line) {},
|
||||||
help: "wtf?",
|
help: "wtf?",
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -196,7 +196,7 @@ func TestCommandSet(t *testing.T) {
|
||||||
if fail := cs.add("one", c); fail != nil {
|
if fail := cs.add("one", c); fail != nil {
|
||||||
t.Errorf("Adding a second 'one' command did not fail as expected.")
|
t.Errorf("Adding a second 'one' command did not fail as expected.")
|
||||||
}
|
}
|
||||||
|
|
||||||
cn2 := cs.add("One Two", c).(*cNode)
|
cn2 := cs.add("One Two", c).(*cNode)
|
||||||
if _, ok := cs.set["one two"]; !ok || cn2.set != cs || cn2.prefix != "one two" {
|
if _, ok := cs.set["one two"]; !ok || cn2.set != cs || cn2.prefix != "one two" {
|
||||||
t.Errorf("Command 'one two' not added to set correctly.")
|
t.Errorf("Command 'one two' not added to set correctly.")
|
||||||
|
@ -208,7 +208,7 @@ func TestCommandSet(t *testing.T) {
|
||||||
if c, l := cs.match("one"); c.(*cNode) != cn1 || l != 3 {
|
if c, l := cs.match("one"); c.(*cNode) != cn1 || l != 3 {
|
||||||
t.Errorf("Didn't match 'one' when we should have.")
|
t.Errorf("Didn't match 'one' when we should have.")
|
||||||
}
|
}
|
||||||
if c, l := cs.match ("one two three"); c.(*cNode) != cn2 || l != 7 {
|
if c, l := cs.match("one two three"); c.(*cNode) != cn2 || l != 7 {
|
||||||
t.Errorf("Didn't match 'one two' when we should have.")
|
t.Errorf("Didn't match 'one two' when we should have.")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -216,14 +216,14 @@ func TestCommandSet(t *testing.T) {
|
||||||
if _, ok := cs.set["one two"]; ok || cn2.set != nil {
|
if _, ok := cs.set["one two"]; ok || cn2.set != nil {
|
||||||
t.Errorf("Command 'one two' not removed correctly.")
|
t.Errorf("Command 'one two' not removed correctly.")
|
||||||
}
|
}
|
||||||
if c, l := cs.match ("one two three"); c.(*cNode) != cn1 || l != 3 {
|
if c, l := cs.match("one two three"); c.(*cNode) != cn1 || l != 3 {
|
||||||
t.Errorf("Didn't match 'one' when we should have.")
|
t.Errorf("Didn't match 'one' when we should have.")
|
||||||
}
|
}
|
||||||
cn1.Remove()
|
cn1.Remove()
|
||||||
if _, ok := cs.set["one"]; ok || cn1.set != nil {
|
if _, ok := cs.set["one"]; ok || cn1.set != nil {
|
||||||
t.Errorf("Command 'one' not removed correctly.")
|
t.Errorf("Command 'one' not removed correctly.")
|
||||||
}
|
}
|
||||||
if c, l := cs.match ("one two three"); c != nil || l != 0 {
|
if c, l := cs.match("one two three"); c != nil || l != 0 {
|
||||||
t.Errorf("Matched 'one' when we shouldn't have.")
|
t.Errorf("Matched 'one' when we shouldn't have.")
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -9,11 +9,12 @@ import (
|
||||||
|
|
||||||
// sets up the internal event handlers to do essential IRC protocol things
|
// sets up the internal event handlers to do essential IRC protocol things
|
||||||
var intHandlers = map[string]HandlerFunc{
|
var intHandlers = map[string]HandlerFunc{
|
||||||
|
INIT: (*Conn).h_init,
|
||||||
"001": (*Conn).h_001,
|
"001": (*Conn).h_001,
|
||||||
"433": (*Conn).h_433,
|
"433": (*Conn).h_433,
|
||||||
"CTCP": (*Conn).h_CTCP,
|
CTCP: (*Conn).h_CTCP,
|
||||||
"NICK": (*Conn).h_NICK,
|
NICK: (*Conn).h_NICK,
|
||||||
"PING": (*Conn).h_PING,
|
PING: (*Conn).h_PING,
|
||||||
}
|
}
|
||||||
|
|
||||||
func (conn *Conn) addIntHandlers() {
|
func (conn *Conn) addIntHandlers() {
|
||||||
|
@ -24,15 +25,24 @@ func (conn *Conn) addIntHandlers() {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Password/User/Nick broadcast on connection.
|
||||||
|
func (conn *Conn) h_init(line *Line) {
|
||||||
|
if conn.password != "" {
|
||||||
|
conn.Pass(conn.password)
|
||||||
|
}
|
||||||
|
conn.Nick(conn.Me.Nick)
|
||||||
|
conn.User(conn.Me.Ident, conn.Me.Name)
|
||||||
|
}
|
||||||
|
|
||||||
// Basic ping/pong handler
|
// Basic ping/pong handler
|
||||||
func (conn *Conn) h_PING(line *Line) {
|
func (conn *Conn) h_PING(line *Line) {
|
||||||
conn.Raw("PONG :" + line.Args[0])
|
conn.Pong(line.Args[0])
|
||||||
}
|
}
|
||||||
|
|
||||||
// Handler to trigger a "CONNECTED" event on receipt of numeric 001
|
// Handler to trigger a "CONNECTED" event on receipt of numeric 001
|
||||||
func (conn *Conn) h_001(line *Line) {
|
func (conn *Conn) h_001(line *Line) {
|
||||||
// we're connected!
|
// we're connected!
|
||||||
conn.dispatch(&Line{Cmd: "connected"})
|
conn.dispatch(&Line{Cmd: CONNECTED})
|
||||||
// and we're being given our hostname (from the server's perspective)
|
// and we're being given our hostname (from the server's perspective)
|
||||||
t := line.Args[len(line.Args)-1]
|
t := line.Args[len(line.Args)-1]
|
||||||
if idx := strings.LastIndex(t, " "); idx != -1 {
|
if idx := strings.LastIndex(t, " "); idx != -1 {
|
||||||
|
@ -70,10 +80,10 @@ func (conn *Conn) h_433(line *Line) {
|
||||||
|
|
||||||
// Handle VERSION requests and CTCP PING
|
// Handle VERSION requests and CTCP PING
|
||||||
func (conn *Conn) h_CTCP(line *Line) {
|
func (conn *Conn) h_CTCP(line *Line) {
|
||||||
if line.Args[0] == "VERSION" {
|
if line.Args[0] == VERSION {
|
||||||
conn.CtcpReply(line.Nick, "VERSION", "powered by goirc...")
|
conn.CtcpReply(line.Nick, VERSION, "powered by goirc...")
|
||||||
} else if line.Args[0] == "PING" {
|
} else if line.Args[0] == PING {
|
||||||
conn.CtcpReply(line.Nick, "PING", line.Args[2])
|
conn.CtcpReply(line.Nick, PING, line.Args[2])
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -99,7 +109,9 @@ func (conn *Conn) h_PRIVMSG(line *Line) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
cmd, l := conn.cmdMatch(txt)
|
cmd, l := conn.cmdMatch(txt)
|
||||||
if cmd == nil { return }
|
if cmd == nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
if conn.CommandStripPrefix {
|
if conn.CommandStripPrefix {
|
||||||
txt = strings.TrimSpace(txt[l:])
|
txt = strings.TrimSpace(txt[l:])
|
||||||
}
|
}
|
||||||
|
|
|
@ -27,7 +27,7 @@ func Test001(t *testing.T) {
|
||||||
l := parseLine(":irc.server.org 001 test :Welcome to IRC test!ident@somehost.com")
|
l := parseLine(":irc.server.org 001 test :Welcome to IRC test!ident@somehost.com")
|
||||||
// Set up a handler to detect whether connected handler is called from 001
|
// Set up a handler to detect whether connected handler is called from 001
|
||||||
hcon := false
|
hcon := false
|
||||||
c.HandleFunc("connected", func (conn *Conn, line *Line) {
|
c.HandleFunc(CONNECTED, func(conn *Conn, line *Line) {
|
||||||
hcon = true
|
hcon = true
|
||||||
})
|
})
|
||||||
|
|
||||||
|
@ -139,11 +139,11 @@ func TestCTCP(t *testing.T) {
|
||||||
c.h_CTCP(parseLine(":blah!moo@cows.com PRIVMSG test :\001UNKNOWN ctcp\001"))
|
c.h_CTCP(parseLine(":blah!moo@cows.com PRIVMSG test :\001UNKNOWN ctcp\001"))
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestPRIVMSG(t *testing.T){
|
func TestPRIVMSG(t *testing.T) {
|
||||||
c, s := setUp(t)
|
c, s := setUp(t)
|
||||||
defer s.tearDown()
|
defer s.tearDown()
|
||||||
|
|
||||||
f := func (conn *Conn, line *Line) {
|
f := func(conn *Conn, line *Line) {
|
||||||
conn.Privmsg(line.Args[0], line.Args[1])
|
conn.Privmsg(line.Args[0], line.Args[1])
|
||||||
}
|
}
|
||||||
c.CommandFunc("prefix", f, "")
|
c.CommandFunc("prefix", f, "")
|
||||||
|
@ -188,7 +188,6 @@ func TestPRIVMSG(t *testing.T){
|
||||||
c.h_PRIVMSG(parseLine(":blah!moo@cows.com PRIVMSG #foo :test! prefix bar"))
|
c.h_PRIVMSG(parseLine(":blah!moo@cows.com PRIVMSG #foo :test! prefix bar"))
|
||||||
s.nc.ExpectNothing()
|
s.nc.ExpectNothing()
|
||||||
|
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Test the handler for JOIN messages
|
// Test the handler for JOIN messages
|
||||||
|
@ -317,7 +316,6 @@ func TestMODE(t *testing.T) {
|
||||||
t.Errorf("Channel.ParseModes() not called correctly.")
|
t.Errorf("Channel.ParseModes() not called correctly.")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
// Send a nick mode line, returning Me
|
// Send a nick mode line, returning Me
|
||||||
gomock.InOrder(
|
gomock.InOrder(
|
||||||
s.st.EXPECT().GetChannel("test").Return(nil),
|
s.st.EXPECT().GetChannel("test").Return(nil),
|
||||||
|
|
|
@ -62,7 +62,7 @@ func parseLine(s string) *Line {
|
||||||
// So, I think CTCP and (in particular) CTCP ACTION are better handled as
|
// So, I think CTCP and (in particular) CTCP ACTION are better handled as
|
||||||
// separate events as opposed to forcing people to have gargantuan
|
// separate events as opposed to forcing people to have gargantuan
|
||||||
// handlers to cope with the possibilities.
|
// handlers to cope with the possibilities.
|
||||||
if (line.Cmd == "PRIVMSG" || line.Cmd == "NOTICE") &&
|
if (line.Cmd == PRIVMSG || line.Cmd == NOTICE) &&
|
||||||
len(line.Args[1]) > 2 &&
|
len(line.Args[1]) > 2 &&
|
||||||
strings.HasPrefix(line.Args[1], "\001") &&
|
strings.HasPrefix(line.Args[1], "\001") &&
|
||||||
strings.HasSuffix(line.Args[1], "\001") {
|
strings.HasSuffix(line.Args[1], "\001") {
|
||||||
|
@ -72,16 +72,16 @@ func parseLine(s string) *Line {
|
||||||
// Replace the line with the unwrapped CTCP
|
// Replace the line with the unwrapped CTCP
|
||||||
line.Args[1] = t[1]
|
line.Args[1] = t[1]
|
||||||
}
|
}
|
||||||
if c := strings.ToUpper(t[0]); c == "ACTION" && line.Cmd == "PRIVMSG" {
|
if c := strings.ToUpper(t[0]); c == ACTION && line.Cmd == PRIVMSG {
|
||||||
// make a CTCP ACTION it's own event a-la PRIVMSG
|
// make a CTCP ACTION it's own event a-la PRIVMSG
|
||||||
line.Cmd = c
|
line.Cmd = c
|
||||||
} else {
|
} else {
|
||||||
// otherwise, dispatch a generic CTCP/CTCPREPLY event that
|
// otherwise, dispatch a generic CTCP/CTCPREPLY event that
|
||||||
// contains the type of CTCP in line.Args[0]
|
// contains the type of CTCP in line.Args[0]
|
||||||
if line.Cmd == "PRIVMSG" {
|
if line.Cmd == PRIVMSG {
|
||||||
line.Cmd = "CTCP"
|
line.Cmd = CTCP
|
||||||
} else {
|
} else {
|
||||||
line.Cmd = "CTCPREPLY"
|
line.Cmd = CTCPREPLY
|
||||||
}
|
}
|
||||||
line.Args = append([]string{c}, line.Args...)
|
line.Args = append([]string{c}, line.Args...)
|
||||||
}
|
}
|
||||||
|
|
|
@ -7,14 +7,14 @@ import (
|
||||||
|
|
||||||
func TestCopy(t *testing.T) {
|
func TestCopy(t *testing.T) {
|
||||||
l1 := &Line{
|
l1 := &Line{
|
||||||
Nick: "nick",
|
Nick: "nick",
|
||||||
Ident: "ident",
|
Ident: "ident",
|
||||||
Host: "host",
|
Host: "host",
|
||||||
Src: "src",
|
Src: "src",
|
||||||
Cmd: "cmd",
|
Cmd: "cmd",
|
||||||
Raw: "raw",
|
Raw: "raw",
|
||||||
Args: []string{"arg", "text"},
|
Args: []string{"arg", "text"},
|
||||||
Time: time.Now(),
|
Time: time.Now(),
|
||||||
}
|
}
|
||||||
|
|
||||||
l2 := l1.Copy()
|
l2 := l1.Copy()
|
||||||
|
|
|
@ -9,19 +9,19 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
var stHandlers = map[string]HandlerFunc{
|
var stHandlers = map[string]HandlerFunc{
|
||||||
"JOIN": (*Conn).h_JOIN,
|
"JOIN": (*Conn).h_JOIN,
|
||||||
"KICK": (*Conn).h_KICK,
|
"KICK": (*Conn).h_KICK,
|
||||||
"MODE": (*Conn).h_MODE,
|
"MODE": (*Conn).h_MODE,
|
||||||
"NICK": (*Conn).h_STNICK,
|
"NICK": (*Conn).h_STNICK,
|
||||||
"PART": (*Conn).h_PART,
|
"PART": (*Conn).h_PART,
|
||||||
"QUIT": (*Conn).h_QUIT,
|
"QUIT": (*Conn).h_QUIT,
|
||||||
"TOPIC": (*Conn).h_TOPIC,
|
"TOPIC": (*Conn).h_TOPIC,
|
||||||
"311": (*Conn).h_311,
|
"311": (*Conn).h_311,
|
||||||
"324": (*Conn).h_324,
|
"324": (*Conn).h_324,
|
||||||
"332": (*Conn).h_332,
|
"332": (*Conn).h_332,
|
||||||
"352": (*Conn).h_352,
|
"352": (*Conn).h_352,
|
||||||
"353": (*Conn).h_353,
|
"353": (*Conn).h_353,
|
||||||
"671": (*Conn).h_671,
|
"671": (*Conn).h_671,
|
||||||
}
|
}
|
||||||
|
|
||||||
func (conn *Conn) addSTHandlers() {
|
func (conn *Conn) addSTHandlers() {
|
||||||
|
|
Loading…
Reference in New Issue