You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
91 lines
2.5 KiB
91 lines
2.5 KiB
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) |
|
}
|
|
|