92 lines
2.5 KiB
Go
92 lines
2.5 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"flag"
|
|
"log"
|
|
"math/rand"
|
|
"net"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/fluffle/goirc/client"
|
|
"github.com/tcolgate/hugot"
|
|
"github.com/tcolgate/hugot/adapters/irc"
|
|
|
|
"ur.gs/lysenko/adapters"
|
|
"ur.gs/lysenko/handlers/deadline"
|
|
"ur.gs/lysenko/handlers/ignore"
|
|
"ur.gs/lysenko/handlers/quotedb"
|
|
)
|
|
|
|
var (
|
|
channels = flag.String("channels", "##testing,##test", "Channels to join (separated by comma)")
|
|
ignores = flag.String("ignore", "", "Nicks to ignore (separated by comma)")
|
|
cooldown = flag.Int("cooldown", 120, "Command rate-limit, in seconds")
|
|
host = flag.String("host", "chat.freenode.net", "Server host[:port]")
|
|
ident = flag.String("ident", "lysenko", "Lysenko Bot")
|
|
nick = flag.String("nick", "lysenko", "Lysenko Bot")
|
|
nickserv = flag.String("nickserv", "", "NickServ password")
|
|
quotes = flag.String("quotedb", ":memory:", "sqlite3 quote database")
|
|
ssl = flag.Bool("ssl", true, "Enable SSL")
|
|
testmode = flag.Bool("testmode", false, "Use STDIN/STDOUT instead of IRC for messages")
|
|
)
|
|
|
|
func init() {
|
|
// TODO: better rand.Seed
|
|
rand.Seed(time.Now().Unix())
|
|
}
|
|
|
|
func ircAdapter() hugot.Adapter {
|
|
channelList := strings.Split(*channels, ",")
|
|
|
|
config := client.NewConfig(*nick, "lysenko", "Comrade Trofim Lysenko")
|
|
config.Server = *host
|
|
config.Pass = *nickserv
|
|
config.SSL = *ssl
|
|
|
|
if config.SSL {
|
|
hostname, _, err := net.SplitHostPort(config.Server)
|
|
if err != nil {
|
|
hostname = config.Server
|
|
}
|
|
config.SSLConfig = &tls.Config{
|
|
ServerName: hostname,
|
|
}
|
|
}
|
|
|
|
return irc.New(config, channelList...)
|
|
}
|
|
|
|
func main() {
|
|
flag.Parse()
|
|
var ignoresList []string
|
|
if *ignores != "" {
|
|
ignoresList = strings.Split(*ignores, ",")
|
|
}
|
|
|
|
adapter := adapters.NewPrefixAdapter(ircAdapter(), "!")
|
|
adapter.PrivateHelpOnly = true
|
|
|
|
ctx := context.Background()
|
|
go adapter.Run(ctx)
|
|
|
|
db, err := quotedb.New(*quotes)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer db.DB.Close()
|
|
|
|
deadline.Cooldown = time.Second * time.Duration(*cooldown)
|
|
|
|
mux := hugot.DefaultMux
|
|
mux.HandleHears(ignore.NewHears("edb.AddQuoteHandler{QuoteDB: db}, ignoresList...))
|
|
mux.HandleHears(ignore.NewHears(deadline.NewHears("edb.LastQuoteHandler{QuoteDB: db}), ignoresList...))
|
|
mux.HandleHears(ignore.NewHears(deadline.NewHears("edb.FindQuoteHandler{QuoteDB: db}), ignoresList...))
|
|
mux.HandleHears(ignore.NewHears(deadline.NewHears("edb.RandQuoteHandler{QuoteDB: db}), ignoresList...))
|
|
mux.HandleHears(ignore.NewHears(deadline.NewHears("edb.QuoteHandler{QuoteDB: db}), ignoresList...))
|
|
|
|
hugot.ListenAndServe(ctx, mux, adapter)
|
|
}
|