Initial working TCP version

irccat.json

@@ -0,0 +1,12 @@
+{
+  "tcp_listen": ":12345",
+  "http_listen": ":8045",
+  "irc": {
+    "server": "irc.irccloud.com:6667",
+    "tls": false,
+    "tls_skip_verify": true,
+    "nick": "gocat",
+    "realname": "IRCCat",
+    "channels": ["#russtest"]
+  }
+}

main.go

@@ -0,0 +1,70 @@
+package main
+
+import (
+	"crypto/tls"
+	"github.com/irccloud/irccat/tcplistener"
+	"github.com/juju/loggo"
+	"github.com/spf13/viper"
+	"github.com/thoj/go-ircevent"
+)
+
+var log = loggo.GetLogger("main")
+
+type IRCCat struct {
+	irc *irc.Connection
+	tcp *tcplistener.TCPListener
+}
+
+func main() {
+	loggo.ConfigureLoggers("<root>=DEBUG")
+	viper.SetConfigName("irccat")
+	viper.AddConfigPath("/etc")
+	viper.AddConfigPath(".")
+	var err error
+
+	err = viper.ReadInConfig()
+	if err != nil {
+		return
+	}
+
+	irccat := IRCCat{}
+
+	irccat.tcp, err = tcplistener.Bind()
+	if err != nil {
+		return
+	}
+
+	err = irccat.connectIRC()
+
+	if err != nil {
+		log.Criticalf("Error connecting to IRC server: %s", err)
+		return
+	}
+
+	irccat.tcp.Run(irccat.irc)
+	irccat.irc.Loop()
+}
+
+func (i *IRCCat) connectIRC() error {
+	irccon := irc.IRC(viper.GetString("irc.nick"), viper.GetString("irc.realname"))
+	irccon.UseTLS = viper.GetBool("irc.tls")
+	if viper.GetBool("irc.tls_skip_verify") {
+		irccon.TLSConfig = &tls.Config{InsecureSkipVerify: true}
+	}
+
+	err := irccon.Connect(viper.GetString("irc.server"))
+	if err != nil {
+		return err
+	}
+
+	irccon.AddCallback("001", i.handleWelcome)
+
+	i.irc = irccon
+	return nil
+}
+
+func (i *IRCCat) handleWelcome(e *irc.Event) {
+	for _, channel := range viper.GetStringSlice("irc.channels") {
+		i.irc.Join(channel)
+	}
+}

tcplistener/colours.go

@@ -0,0 +1,34 @@
+package tcplistener
+
+import "strings"
+
+var colourMap map[string]string = map[string]string{
+	"NORMAL":    "\x0f",
+	"BOLD":      "\x02",
+	"UNDERLINE": "\x1f",
+	"REVERSE":   "\x16",
+	"WHITE":     "\x0300",
+	"BLACK":     "\x0301",
+	"DBLUE":     "\x0302",
+	"DGREEN":    "\x0303",
+	"RED":       "\x0304",
+	"BROWN":     "\x0305",
+	"PURPLE":    "\x0306",
+	"ORANGE":    "\x0307",
+	"YELLOW":    "\x0308",
+	"GREEN":     "\x0309",
+	"TEAL":      "\x0310",
+	"CYAN":      "\x0311",
+	"BLUE":      "\x0312",
+	"PINK":      "\x0313",
+	"DGRAY":     "\x0314",
+	"GRAY":      "\x0315",
+}
+
+func replaceFormatting(msg string) string {
+	for colour, code := range colourMap {
+		msg = strings.Replace(msg, "%"+colour, code, -1)
+		msg = strings.Replace(msg, "#"+colour, code, -1)
+	}
+	return msg
+}

tcplistener/tcplistener.go

@@ -0,0 +1,87 @@
+package tcplistener
+
+import (
+	"bufio"
+	"github.com/juju/loggo"
+	"github.com/spf13/viper"
+	"github.com/thoj/go-ircevent"
+	"net"
+	"strings"
+)
+
+var log = loggo.GetLogger("TCPListener")
+
+type TCPListener struct {
+	socket net.Listener
+	irc    *irc.Connection
+}
+
+func Bind() (*TCPListener, error) {
+	var err error
+
+	listener := TCPListener{}
+	listener.socket, err = net.Listen("tcp", viper.GetString("tcp_listen"))
+	if err != nil {
+		return nil, err
+	}
+
+	return &listener, nil
+}
+
+func (l *TCPListener) Run(irccon *irc.Connection) {
+	log.Infof("Listening for TCP requests on %s", viper.GetString("tcp_listen"))
+	l.irc = irccon
+	go l.acceptConnections()
+}
+
+func (l *TCPListener) acceptConnections() {
+	for {
+		conn, err := l.socket.Accept()
+		if err != nil {
+			break
+		}
+		go l.handleConnection(conn)
+	}
+	l.socket.Close()
+}
+
+func (l *TCPListener) handleConnection(conn net.Conn) {
+	reader := bufio.NewReader(conn)
+	for {
+		msg, err := reader.ReadString('\n')
+		if err != nil {
+			break
+		}
+		msg = strings.Trim(msg, "\r\n")
+		log.Debugf("Message: %s", msg)
+		l.parseMessage(msg)
+	}
+}
+
+func (l *TCPListener) parseMessage(msg string) {
+	channels := viper.GetStringSlice("irc.channels")
+
+	if msg[0] == '#' || msg[0] == '@' {
+		parts := strings.SplitN(msg, " ", 2)
+		if parts[0] == "#*" {
+			for _, channel := range channels {
+				l.irc.Privmsg(channel, replaceFormatting(parts[1]))
+			}
+		} else {
+			targets := strings.Split(parts[0], ",")
+			for _, target := range targets {
+				if target[0] == '@' {
+					target = target[1:]
+				}
+				l.irc.Privmsg(target, replaceFormatting(parts[1]))
+			}
+		}
+	} else if msg[0:6] == "%TOPIC" {
+		parts := strings.SplitN(msg, " ", 3)
+		l.irc.SendRawf("TOPIC %s :%s", parts[1], replaceFormatting(parts[2]))
+	} else {
+		if len(channels) > 0 {
+			l.irc.Privmsg(channels[0], replaceFormatting(msg))
+		}
+	}
+}