Pull IRC client out

irc/client.go

@@ -0,0 +1,129 @@
+package irc
+
+import (
+	"bufio"
+	"crypto/tls"
+	"net"
+	"strings"
+	"sync"
+)
+
+type Client struct {
+	Server    string
+	Host      string
+	TLS       bool
+	TLSConfig *tls.Config
+	Password  string
+	Username  string
+	Realname  string
+	Messages  chan *Message
+
+	nick string
+
+	conn      net.Conn
+	connected bool
+	dialer    *net.Dialer
+	reader    *bufio.Reader
+	out       chan string
+
+	quit      chan struct{}
+	reconnect chan struct{}
+	ready     sync.WaitGroup
+	once      sync.Once
+	lock      sync.Mutex
+}
+
+func NewClient(nick, username string) *Client {
+	return &Client{
+		nick:      nick,
+		Username:  username,
+		Realname:  nick,
+		Messages:  make(chan *Message, 32),
+		out:       make(chan string, 32),
+		quit:      make(chan struct{}),
+		reconnect: make(chan struct{}),
+	}
+}
+
+func (c *Client) GetNick() string {
+	c.lock.Lock()
+	defer c.lock.Unlock()
+
+	return c.nick
+}
+
+func (c *Client) Connected() bool {
+	c.lock.Lock()
+	defer c.lock.Unlock()
+
+	return c.connected
+}
+
+func (c *Client) Pass(password string) {
+	c.write("PASS " + password)
+}
+
+func (c *Client) Nick(nick string) {
+	c.Write("NICK " + nick)
+
+	c.lock.Lock()
+	c.nick = nick
+	c.lock.Unlock()
+}
+
+func (c *Client) User(username, realname string) {
+	c.writef("USER %s 0 * :%s", username, realname)
+}
+
+func (c *Client) Oper(name, password string) {
+	c.Write("OPER " + name + " " + password)
+}
+
+func (c *Client) Mode(target, modes, params string) {
+	c.Write(strings.TrimRight("MODE "+target+" "+modes+" "+params, " "))
+}
+
+func (c *Client) Quit() {
+	go func() {
+		if c.Connected() {
+			c.write("QUIT")
+		}
+		close(c.quit)
+	}()
+}
+
+func (c *Client) Join(channels ...string) {
+	c.Write("JOIN " + strings.Join(channels, ","))
+}
+
+func (c *Client) Part(channels ...string) {
+	c.Write("PART " + strings.Join(channels, ","))
+}
+
+func (c *Client) Topic(channel string) {
+	c.Write("TOPIC " + channel)
+}
+
+func (c *Client) Invite(nick, channel string) {
+	c.Write("INVITE " + nick + " " + channel)
+}
+
+func (c *Client) Kick(channel string, users ...string) {
+	c.Write("KICK " + channel + " " + strings.Join(users, ","))
+}
+
+func (c *Client) Privmsg(target, msg string) {
+	c.Writef("PRIVMSG %s :%s", target, msg)
+}
+
+func (c *Client) Notice(target, msg string) {
+	c.Writef("NOTICE %s :%s", target, msg)
+}
+
+func (c *Client) Whois(nick string) {
+	c.Write("WHOIS " + nick)
+}
+
+func (c *Client) Away(message string) {
+	c.Write("AWAY :" + message)
+}