mirror of
https://github.com/fluffle/goirc
synced 2025-07-10 23:39:23 +00:00
vastly updated bot framework, now with state tracking etc.
This commit is contained in:
Alex Bramley
parent
4d610d1718
commit
79822340b5
8 changed files with 977 additions and 175 deletions
|
|
@ -1,5 +1,3 @@
|
|||
// Some IRC testing code!
|
||||
|
||||
package irc
|
||||
|
||||
import (
|
||||
|
|
@ -11,96 +9,99 @@ import (
|
|||
)
|
||||
|
||||
// the IRC connection object
|
||||
type IRCConn struct {
|
||||
sock *bufio.ReadWriter;
|
||||
Host string;
|
||||
Me string;
|
||||
Ident string;
|
||||
Name string;
|
||||
con bool;
|
||||
reg bool;
|
||||
events map[string] []func (*IRCConn, *IRCLine);
|
||||
chans map[string] *IRCChan;
|
||||
nicks map[string] *IRCNick;
|
||||
type Conn struct {
|
||||
// Hostname, Nickname, etc.
|
||||
Host string;
|
||||
Me *Nick;
|
||||
|
||||
// I/O stuff to server
|
||||
sock *net.TCPConn;
|
||||
io *bufio.ReadWriter;
|
||||
in chan *Line;
|
||||
out chan string;
|
||||
connected bool;
|
||||
|
||||
// Error channel to transmit any fail back to the user
|
||||
Err chan os.Error;
|
||||
|
||||
// Event handler mapping
|
||||
events map[string] []func (*Conn, *Line);
|
||||
|
||||
// Map of channels we're on
|
||||
chans map[string] *Channel;
|
||||
|
||||
// Map of nicks we know about
|
||||
nicks map[string] *Nick;
|
||||
}
|
||||
|
||||
// We'll parse an incoming line into this struct
|
||||
// raw =~ ":nick!user@host cmd args[] :text"
|
||||
// src == "nick!user@host"
|
||||
type IRCLine struct {
|
||||
Nick string;
|
||||
User string;
|
||||
Host string;
|
||||
Src string;
|
||||
Cmd string;
|
||||
type Line struct {
|
||||
Nick, Ident, Host, Src string;
|
||||
Cmd, Text, Raw string;
|
||||
Args []string;
|
||||
Text string;
|
||||
Raw string;
|
||||
}
|
||||
|
||||
// A struct representing an IRC channel
|
||||
type IRCChan struct {
|
||||
Name string;
|
||||
Topic string;
|
||||
Modes map[string] string;
|
||||
Nicks map[string] *IRCNick;
|
||||
}
|
||||
|
||||
// A struct representing an IRC nick
|
||||
type IRCNick struct {
|
||||
Name string;
|
||||
Chans map[string] *IRCChan;
|
||||
}
|
||||
|
||||
// construct a new IRC Connection object
|
||||
func New(nick, user, name string) (conn *IRCConn) {
|
||||
conn = &IRCConn{Me: nick, Ident: user, Name: name};
|
||||
// allocate meh some memoraaaahh
|
||||
conn.nicks = make(map[string] *IRCNick);
|
||||
conn.chans = make(map[string] *IRCChan);
|
||||
conn.events = make(map[string] []func(*IRCConn, *IRCLine));
|
||||
func New(nick, user, name string) *Conn {
|
||||
conn := new(Conn);
|
||||
conn.initialise();
|
||||
conn.Me = conn.NewNick(nick, user, name, "");
|
||||
conn.setupEvents();
|
||||
return conn
|
||||
return conn;
|
||||
}
|
||||
|
||||
func (conn *Conn) initialise() {
|
||||
// allocate meh some memoraaaahh
|
||||
fmt.Println("irc.initialise(): initialising...");
|
||||
conn.nicks = make(map[string] *Nick);
|
||||
conn.chans = make(map[string] *Channel);
|
||||
conn.in = make(chan *Line, 32);
|
||||
conn.out = make(chan string, 32);
|
||||
conn.Err = make(chan os.Error, 4);
|
||||
conn.io = nil;
|
||||
conn.sock = nil;
|
||||
}
|
||||
|
||||
// connect the IRC connection object to a host
|
||||
func (conn *IRCConn) Connect(host, pass string) (err os.Error) {
|
||||
func (conn *Conn) Connect(host, pass string) os.Error {
|
||||
if conn.connected {
|
||||
return os.NewError(fmt.Sprintf("irc.Connect(): already connected to %s, cannot connect to %s", conn.Host, host));
|
||||
}
|
||||
if !hasPort(host) {
|
||||
host += ":6667";
|
||||
}
|
||||
sock, err := net.Dial("tcp", "", host);
|
||||
if err != nil {
|
||||
|
||||
if addr, err := net.ResolveTCPAddr(host); err != nil {
|
||||
return err
|
||||
} else if conn.sock, err = net.DialTCP("tcp", nil, addr); err != nil {
|
||||
return err
|
||||
}
|
||||
conn.sock = bufio.NewReadWriter(bufio.NewReader(sock), bufio.NewWriter(sock));
|
||||
conn.con = true;
|
||||
fmt.Println("irc.Connect(): connected happily...");
|
||||
conn.Host = host;
|
||||
|
||||
// initial connection set-up
|
||||
// verify valid nick/user/name here?
|
||||
conn.io = bufio.NewReadWriter(
|
||||
bufio.NewReader(conn.sock),
|
||||
bufio.NewWriter(conn.sock)
|
||||
);
|
||||
go conn.send();
|
||||
go conn.recv();
|
||||
|
||||
if pass != "" {
|
||||
conn.Pass(pass)
|
||||
}
|
||||
conn.Nick(conn.Me);
|
||||
conn.User(conn.Ident, conn.Name);
|
||||
conn.Nick(conn.Me.Nick);
|
||||
conn.User(conn.Me.Ident, conn.Me.Name);
|
||||
|
||||
for line, err := conn.recv(); err == nil; line, err = conn.recv() {
|
||||
// initial loop to get us to the point where we're connected
|
||||
conn.dispatchEvent(line);
|
||||
if line.Cmd == "001" {
|
||||
break;
|
||||
}
|
||||
}
|
||||
return err;
|
||||
go conn.runLoop();
|
||||
fmt.Println("irc.Connect(): launched runLoop() goroutine.");
|
||||
return nil;
|
||||
}
|
||||
|
||||
func (conn *IRCConn) RunLoop(c chan os.Error) {
|
||||
var err os.Error;
|
||||
for line, err := conn.recv(); err == nil; line, err = conn.recv() {
|
||||
conn.dispatchEvent(line);
|
||||
}
|
||||
c <- err;
|
||||
return;
|
||||
// dispatch a nicely formatted os.Error to the error channel
|
||||
func (conn *Conn) error(s string, a ...) {
|
||||
conn.Err <- os.NewError(fmt.Sprintf(s, a));
|
||||
}
|
||||
|
||||
// copied from http.client for great justice
|
||||
|
|
@ -108,56 +109,120 @@ func hasPort(s string) bool {
|
|||
return strings.LastIndex(s, ":") > strings.LastIndex(s, "]")
|
||||
}
|
||||
|
||||
// send \r\n terminated line to peer, propagate errors
|
||||
func (conn *IRCConn) send(line string) (err os.Error) {
|
||||
err = conn.sock.WriteString(line + "\r\n");
|
||||
conn.sock.Flush();
|
||||
fmt.Println("-> " + line);
|
||||
return err
|
||||
// dispatch input from channel as \r\n terminated line to peer
|
||||
func (conn *Conn) send() {
|
||||
for {
|
||||
line := <-conn.out;
|
||||
if closed(conn.out) {
|
||||
break;
|
||||
}
|
||||
if err := conn.io.WriteString(line + "\r\n"); err != nil {
|
||||
conn.error("irc.send(): %s", err.String());
|
||||
conn.shutdown();
|
||||
break;
|
||||
}
|
||||
conn.io.Flush();
|
||||
fmt.Println("-> " + line);
|
||||
}
|
||||
}
|
||||
|
||||
// receive one \r\n terminated line from peer and parse it, propagate errors
|
||||
func (conn *IRCConn) recv() (line *IRCLine, err os.Error) {
|
||||
s, err := conn.sock.ReadString('\n');
|
||||
if err != nil {
|
||||
return line, err
|
||||
}
|
||||
// chop off \r\n
|
||||
s = s[0:len(s)-2];
|
||||
fmt.Println("<- " + s);
|
||||
// receive one \r\n terminated line from peer, parse and dispatch it
|
||||
func (conn *Conn) recv() {
|
||||
for {
|
||||
s, err := conn.io.ReadString('\n');
|
||||
if err != nil {
|
||||
conn.error("irc.recv(): %s", err.String());
|
||||
conn.shutdown();
|
||||
break;
|
||||
}
|
||||
// chop off \r\n
|
||||
s = s[0:len(s)-2];
|
||||
fmt.Println("<- " + s);
|
||||
|
||||
line = &IRCLine{Raw: s};
|
||||
if s[0] == ':' {
|
||||
// remove a source and parse it
|
||||
if idx := strings.Index(s, " "); idx != -1 {
|
||||
line.Src, s = s[1:idx], s[idx+1:len(s)];
|
||||
} else {
|
||||
// pretty sure we shouldn't get here ...
|
||||
line.Src = s[1:len(s)];
|
||||
return line, nil;
|
||||
line := &Line{Raw: s};
|
||||
if s[0] == ':' {
|
||||
// remove a source and parse it
|
||||
if idx := strings.Index(s, " "); idx != -1 {
|
||||
line.Src, s = s[1:idx], s[idx+1:len(s)];
|
||||
} else {
|
||||
// pretty sure we shouldn't get here ...
|
||||
line.Src = s[1:len(s)];
|
||||
conn.in <- line;
|
||||
continue;
|
||||
}
|
||||
|
||||
// src can be the hostname of the irc server or a nick!user@host
|
||||
line.Host = line.Src;
|
||||
nidx, uidx := strings.Index(line.Src, "!"), strings.Index(line.Src, "@");
|
||||
if uidx != -1 && nidx != -1 {
|
||||
line.Nick = line.Src[0:nidx];
|
||||
line.Ident = line.Src[nidx+1:uidx];
|
||||
line.Host = line.Src[uidx+1:len(line.Src)];
|
||||
}
|
||||
}
|
||||
|
||||
// src can be the hostname of the irc server or a nick!user@host
|
||||
line.Host = line.Src;
|
||||
nidx, uidx := strings.Index(line.Src, "!"), strings.Index(line.Src, "@");
|
||||
if uidx != -1 && nidx != -1 {
|
||||
line.Nick = line.Src[0:nidx];
|
||||
line.User = line.Src[nidx+1:uidx];
|
||||
line.Host = line.Src[uidx+1:len(line.Src)];
|
||||
// now we're here, we've parsed a :nick!user@host or :server off
|
||||
// s should contain "cmd args[] :text"
|
||||
args := strings.Split(s, " :", 2);
|
||||
if len(args) > 1 {
|
||||
line.Text = args[1];
|
||||
}
|
||||
args = strings.Split(args[0], " ", 0);
|
||||
line.Cmd = strings.ToUpper(args[0]);
|
||||
if len(args) > 1 {
|
||||
line.Args = args[1:len(args)];
|
||||
}
|
||||
conn.in <- line
|
||||
}
|
||||
|
||||
// now we're here, we've parsed a :nick!user@host or :server off
|
||||
// s should contain "cmd args[] :text"
|
||||
args := strings.Split(s, " :", 2);
|
||||
if len(args) > 1 {
|
||||
line.Text = args[1];
|
||||
}
|
||||
args = strings.Split(args[0], " ", 0);
|
||||
line.Cmd = strings.ToUpper(args[0]);
|
||||
if len(args) > 1 {
|
||||
line.Args = args[1:len(args)];
|
||||
}
|
||||
return line, nil
|
||||
}
|
||||
|
||||
func (conn *Conn) runLoop() {
|
||||
for {
|
||||
if closed(conn.in) {
|
||||
break;
|
||||
}
|
||||
select {
|
||||
case line := <-conn.in:
|
||||
conn.dispatchEvent(line);
|
||||
}
|
||||
}
|
||||
fmt.Println("irc.runLoop(): Exited runloop...");
|
||||
// if we fall off the end here due to shutdown,
|
||||
// reinit everything once the runloop is done
|
||||
// so that Connect() can be called again.
|
||||
conn.initialise();
|
||||
}
|
||||
|
||||
func (conn *Conn) shutdown() {
|
||||
close(conn.in);
|
||||
close(conn.out);
|
||||
close(conn.Err);
|
||||
conn.connected = false;
|
||||
conn.sock.Close();
|
||||
fmt.Println("irc.shutdown(): shut down sockets and channels!");
|
||||
}
|
||||
|
||||
func (conn *Conn) String() string {
|
||||
str := "GoIRC Connection\n";
|
||||
str += "----------------\n\n";
|
||||
if conn.connected {
|
||||
str += "Connected to " + conn.Host + "\n\n"
|
||||
} else {
|
||||
str += "Not currently connected!\n\n";
|
||||
}
|
||||
str += conn.Me.String() + "\n";
|
||||
str += "GoIRC Channels\n";
|
||||
str += "--------------\n\n";
|
||||
for _, ch := range conn.chans {
|
||||
str += ch.String() + "\n"
|
||||
}
|
||||
str += "GoIRC NickNames\n";
|
||||
str += "---------------\n\n";
|
||||
for _, n := range conn.nicks {
|
||||
if n != conn.Me {
|
||||
str += n.String() + "\n"
|
||||
}
|
||||
}
|
||||
return str;
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue