Pull IRC client out

irc/conn.go

@@ -0,0 +1,177 @@
+package irc
+
+import (
+	"bufio"
+	"crypto/tls"
+	"fmt"
+	"net"
+	"strings"
+	"sync"
+	"time"
+)
+
+func (c *Client) Connect(address string) {
+	if idx := strings.Index(address, ":"); idx < 0 {
+		c.Host = address
+
+		if c.TLS {
+			address += ":6697"
+		} else {
+			address += ":6667"
+		}
+	} else {
+		c.Host = address[:idx]
+	}
+	c.Server = address
+	c.dialer = &net.Dialer{Timeout: 10 * time.Second}
+
+	go c.run()
+}
+
+func (c *Client) Write(data string) {
+	c.out <- data + "\r\n"
+}
+
+func (c *Client) Writef(format string, a ...interface{}) {
+	c.out <- fmt.Sprintf(format+"\r\n", a...)
+}
+
+func (c *Client) write(data string) {
+	c.conn.Write([]byte(data + "\r\n"))
+}
+
+func (c *Client) writef(format string, a ...interface{}) {
+	fmt.Fprintf(c.conn, format+"\r\n", a...)
+}
+
+func (c *Client) connect() error {
+	if c.TLS {
+		if c.TLSConfig == nil {
+			c.TLSConfig = &tls.Config{InsecureSkipVerify: true}
+		}
+
+		if conn, err := tls.DialWithDialer(c.dialer, "tcp", c.Server, c.TLSConfig); err != nil {
+			return err
+		} else {
+			c.conn = conn
+		}
+	} else {
+		if conn, err := c.dialer.Dial("tcp", c.Server); err != nil {
+			return err
+		} else {
+			c.conn = conn
+		}
+	}
+
+	c.lock.Lock()
+	c.connected = true
+	c.lock.Unlock()
+
+	c.reader = bufio.NewReader(c.conn)
+
+	if c.Password != "" {
+		c.Pass(c.Password)
+	}
+	c.write("NICK " + c.nick)
+	c.User(c.Username, c.Realname)
+
+	c.ready.Add(1)
+	go c.send()
+	go c.recv()
+
+	return nil
+}
+
+func (c *Client) tryConnect() {
+	// TODO: backoff
+	for {
+		select {
+		case <-c.quit:
+			return
+		default:
+		}
+
+		err := c.connect()
+		if err == nil {
+			return
+		}
+	}
+}
+
+func (c *Client) run() {
+	c.tryConnect()
+	for {
+		select {
+		case <-c.quit:
+			c.close()
+			c.lock.Lock()
+			c.connected = false
+			c.lock.Unlock()
+			return
+
+		case <-c.reconnect:
+			c.reconnect = make(chan struct{})
+			c.once = sync.Once{}
+			c.tryConnect()
+		}
+	}
+}
+
+func (c *Client) send() {
+	c.ready.Wait()
+	for {
+		select {
+		case <-c.quit:
+			return
+
+		case <-c.reconnect:
+			return
+
+		case msg := <-c.out:
+			_, err := c.conn.Write([]byte(msg))
+			if err != nil {
+				return
+			}
+		}
+	}
+}
+
+func (c *Client) recv() {
+	defer c.conn.Close()
+	for {
+		line, err := c.reader.ReadString('\n')
+		if err != nil {
+			c.lock.Lock()
+			c.connected = false
+			c.lock.Unlock()
+			c.once.Do(c.ready.Done)
+			close(c.reconnect)
+			return
+		}
+
+		select {
+		case <-c.quit:
+			return
+		default:
+		}
+
+		msg := parseMessage(line)
+		c.Messages <- msg
+
+		switch msg.Command {
+		case Ping:
+			go c.write("PONG :" + msg.Trailing)
+
+		case ReplyWelcome:
+			c.once.Do(c.ready.Done)
+		}
+	}
+}
+
+func (c *Client) close() {
+	if c.Connected() {
+		c.once.Do(c.ready.Done)
+	}
+	close(c.out)
+	close(c.Messages)
+}