2015-11-12 09:07:53 +00:00
|
|
|
// Copyright 2015 Robert S. Gerus. All rights reserved.
|
|
|
|
// Use of this source code is governed by a MIT-style
|
|
|
|
// license that can be found in the LICENSE file.
|
2015-11-16 19:25:18 +00:00
|
|
|
|
|
|
|
// Gorepost is an overengineered IRC bot that i use for learning Go.
|
2015-10-15 19:04:24 +00:00
|
|
|
package main
|
2015-10-15 14:50:32 +00:00
|
|
|
|
|
|
|
import (
|
2015-10-19 23:54:07 +00:00
|
|
|
"log"
|
|
|
|
"os"
|
2015-11-10 20:00:40 +00:00
|
|
|
"path"
|
2015-10-15 14:50:32 +00:00
|
|
|
|
2015-11-08 19:55:47 +00:00
|
|
|
"github.com/arachnist/gorepost/bot"
|
2015-11-11 20:38:33 +00:00
|
|
|
cfg "github.com/arachnist/gorepost/config"
|
2015-10-19 23:56:34 +00:00
|
|
|
"github.com/arachnist/gorepost/irc"
|
2015-10-15 14:50:32 +00:00
|
|
|
)
|
|
|
|
|
2015-11-11 20:38:33 +00:00
|
|
|
func fileListFuncBuilder(basedir, common string) func(map[string]string) []string {
|
2015-11-10 21:12:36 +00:00
|
|
|
return func(c map[string]string) []string {
|
|
|
|
var r []string
|
|
|
|
|
|
|
|
if c["Network"] != "" {
|
2015-11-11 00:11:50 +00:00
|
|
|
if c["Target"] != "" {
|
|
|
|
if c["Source"] != "" {
|
|
|
|
r = append(r, path.Join(basedir, c["Network"], c["Target"], c["Source"]+".json"))
|
|
|
|
r = append(r, path.Join(basedir, c["Network"], c["Source"]+".json"))
|
2015-11-10 21:12:36 +00:00
|
|
|
}
|
2015-11-11 00:11:50 +00:00
|
|
|
r = append(r, path.Join(basedir, c["Network"], c["Target"]+".json"))
|
2015-11-10 21:12:36 +00:00
|
|
|
}
|
|
|
|
r = append(r, path.Join(basedir, c["Network"]+".json"))
|
|
|
|
}
|
|
|
|
|
|
|
|
return append(r, path.Join(basedir, common))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-10-15 14:50:32 +00:00
|
|
|
func main() {
|
2015-10-20 00:59:05 +00:00
|
|
|
var exit chan struct{}
|
2015-11-10 20:00:40 +00:00
|
|
|
context := make(map[string]string)
|
2015-11-08 21:00:33 +00:00
|
|
|
|
2015-11-09 00:05:06 +00:00
|
|
|
if len(os.Args) < 2 {
|
2015-11-09 13:32:26 +00:00
|
|
|
log.Fatalln("Usage:", os.Args[0], "<configuration directory>")
|
2015-11-09 00:05:06 +00:00
|
|
|
}
|
2015-11-09 13:32:26 +00:00
|
|
|
|
|
|
|
d, err := os.Stat(os.Args[1])
|
2015-10-19 23:54:07 +00:00
|
|
|
if err != nil {
|
2015-11-09 00:05:06 +00:00
|
|
|
log.Fatalln("Error reading configuration from", os.Args[1], "error:", err.Error())
|
2015-10-19 23:54:07 +00:00
|
|
|
}
|
2015-11-09 13:32:26 +00:00
|
|
|
if !d.IsDir() {
|
|
|
|
log.Fatalln("Not a directory:", os.Args[1])
|
|
|
|
}
|
2015-10-15 19:04:24 +00:00
|
|
|
|
2015-11-11 20:38:33 +00:00
|
|
|
cfg.SetFileListBuilder(fileListFuncBuilder(os.Args[1], "common.json"))
|
2015-11-09 13:32:26 +00:00
|
|
|
|
2015-11-18 13:07:38 +00:00
|
|
|
logfile, err := os.OpenFile(cfg.LookupString(context, "Logpath"), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0644)
|
2015-10-19 23:54:07 +00:00
|
|
|
if err != nil {
|
2015-11-18 13:07:38 +00:00
|
|
|
log.Fatalln("Error opening", cfg.LookupString(context, "Logpath"), "for writing, error:", err.Error())
|
2015-10-19 23:54:07 +00:00
|
|
|
}
|
|
|
|
log.SetOutput(logfile)
|
2015-10-19 08:37:08 +00:00
|
|
|
|
2015-11-18 13:07:38 +00:00
|
|
|
networks := cfg.LookupStringSlice(context, "Networks")
|
2015-11-09 23:13:30 +00:00
|
|
|
|
|
|
|
log.Println("Configured networks:", len(networks), networks)
|
|
|
|
|
2015-11-11 10:50:29 +00:00
|
|
|
for i, conn := range make([]irc.Connection, len(networks)) {
|
|
|
|
conn := conn
|
2015-11-18 13:07:38 +00:00
|
|
|
log.Println("Setting up", networks[i], "connection")
|
|
|
|
conn.Setup(bot.Dispatcher, networks[i])
|
2015-10-20 00:59:05 +00:00
|
|
|
}
|
|
|
|
<-exit
|
2015-10-15 14:50:32 +00:00
|
|
|
}
|