flokati/irc.go

135 lines
3.2 KiB
Go

// vi:ts=4:sts=4:sw=4:noet:tw=72
//
// flokatirc
//
// Copyright (c) 2015,2016 Andreas Neue <an@dnix.de>
package main
import (
"crypto/tls"
"flag"
"fmt"
"strings"
"time"
modules "git.dnix.de/an/flokati/modules"
"git.dnix.de/an/xlog"
irc "github.com/fluffle/goirc/client"
)
var (
usetls = flag.Bool("tls", true, "Use TLS")
port = flag.String("port", "6697", "Server port")
nsname = flag.String("nsname", "NickServ", "NickServ name")
nspass = flag.String("nspass", "", "NickServ password")
automsgTo = flag.String("automsg_to", "", "Autosend IRC msg recipient")
automsgText = flag.String("automsg_text", "", "Autosend IRC msg text")
)
func Irc(say chan string) {
//bot := ircx.Classic(*server, *name)
cfg := irc.NewConfig(*name)
cfg.SSL = *usetls
cfg.SSLConfig = &tls.Config{ServerName: *server}
cfg.Server = *server + ":" + *port
cfg.NewNick = func(n string) string { return n + "^" }
cfg.Me.Ident = *name
cfg.Me.Name = *name
bot := irc.Client(cfg)
SetupHandlers(bot, say)
go CheckConnection(bot)
go Ping(bot)
for {
var targets string
for {
if bot.Connected() {
break
}
time.Sleep(1 * time.Second)
}
line := strings.Split(<-say, "\n")
xlog.Debug("Say: \"%s\"", line)
if len(line) < 2 {
continue
}
if line[0] != "*" {
targets = line[0]
} else {
targets = *channels
}
for _, tar := range strings.Split(targets, ",") {
bot.Privmsg(tar, line[1])
time.Sleep(1 * time.Second)
}
}
time.Sleep(1 * time.Millisecond)
}
func SetupHandlers(bot *irc.Conn, say chan string) {
xlog.Info("Registering handlers")
bot.HandleFunc("001", func(c *irc.Conn, l *irc.Line) {
go func(c *irc.Conn, l *irc.Line) {
if *nspass != "" {
xlog.Info("Authenticating with NickServ: %v, %v", *name, "xxxxxx")
bot.Privmsg(*nsname, "IDENTIFY "+*name+" "+*nspass)
}
if *automsgTo != "" && *automsgText != "" {
xlog.Info("Sending automsg ...")
bot.Privmsg(*automsgTo, *automsgText)
}
xlog.Info("Joining channels: %v", *channels)
for _, ch := range strings.Split(*channels, ",") {
bot.Join(ch)
}
time.Sleep(2 * time.Second)
msg := fmt.Sprintf("running on %s", SoftwareInfo())
say <- fmt.Sprintf("%s\n\001ACTION %s\001", "*", msg)
}(c, l)
})
bot.HandleFunc("PRIVMSG", func(c *irc.Conn, l *irc.Line) {
go func(c *irc.Conn, l *irc.Line) {
//TODO: implement message handler table
HandleMessage(l)
modules.HandleMessage(&modules.Message{From: l.Nick, Channel: l.Args[0], Text: l.Text()})
}(c, l)
})
}
func CheckConnection(bot *irc.Conn) {
for {
if !bot.Connected() {
xlog.Info("Not connected, connecting ...")
if err := bot.Connect(); err != nil {
xlog.Error("Unable to dial IRC Server: %v", err)
}
time.Sleep(10 * time.Second)
}
time.Sleep(1 * time.Millisecond)
}
}
func HandleMessage(l *irc.Line) {
tok := strings.Split(l.Text(), " ")
if len(tok) < 1 {
return
}
switch tok[0] {
case "!version":
//msg := ctcp.Action(fmt.Sprintf("running on %s", SoftwareInfo()))
//sayCh <- fmt.Sprintf("%s\n%s", "*", msg)
//sayCh <- fmt.Sprintf("%s\n%s", "*", SoftwareInfo())
default:
}
}
func Ping(bot *irc.Conn) {
for {
time.Sleep(1 * time.Minute)
bot.Raw("PING")
}
}