mirror of https://github.com/velour/catbase.git
99 lines
2.3 KiB
Go
99 lines
2.3 KiB
Go
// © 2013 the CatBase Authors under the WTFPL. See AUTHORS for the list of authors.
|
|
|
|
package main
|
|
|
|
//go:generate templ generate
|
|
|
|
import (
|
|
"flag"
|
|
"github.com/velour/catbase/plugins"
|
|
"github.com/velour/catbase/plugins/talklikeapirate"
|
|
"io"
|
|
"math/rand"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/rs/zerolog"
|
|
"github.com/rs/zerolog/log"
|
|
"github.com/velour/catbase/bot/msg"
|
|
"github.com/velour/catbase/connectors/discord"
|
|
|
|
"github.com/velour/catbase/bot"
|
|
"github.com/velour/catbase/config"
|
|
"github.com/velour/catbase/connectors/irc"
|
|
"github.com/velour/catbase/connectors/slackapp"
|
|
)
|
|
|
|
var (
|
|
key = flag.String("set", "", "Configuration key to set")
|
|
val = flag.String("val", "", "Configuration value to set")
|
|
initDB = flag.Bool("init", false, "Initialize the configuration db")
|
|
prettyLog = flag.Bool("pretty", false, "Use pretty console logger")
|
|
debug = flag.Bool("debug", false, "Turn on debug logging")
|
|
)
|
|
|
|
func main() {
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
var dbpath = flag.String("db", "catbase.db",
|
|
"Database file to load. (Defaults to catbase.db)")
|
|
flag.Parse() // parses the logging flags.
|
|
|
|
var output io.Writer = os.Stdout
|
|
if *prettyLog {
|
|
output = zerolog.ConsoleWriter{Out: output}
|
|
}
|
|
log.Logger = log.Output(output).With().Caller().Stack().Logger()
|
|
|
|
zerolog.SetGlobalLevel(zerolog.InfoLevel)
|
|
if *debug {
|
|
zerolog.SetGlobalLevel(zerolog.DebugLevel)
|
|
}
|
|
|
|
c := config.ReadConfig(*dbpath)
|
|
|
|
if *key != "" && *val != "" {
|
|
c.Set(*key, *val)
|
|
log.Info().Msgf("Set config %s: %s", *key, *val)
|
|
return
|
|
}
|
|
if (*initDB && len(flag.Args()) != 2) || (!*initDB && c.GetInt("init", 0) != 1) {
|
|
log.Fatal().Msgf(`You must run "catbase -init <channel> <nick>"`)
|
|
} else if *initDB {
|
|
c.SetDefaults(flag.Arg(0), flag.Arg(1))
|
|
return
|
|
}
|
|
|
|
var client bot.Connector
|
|
|
|
switch c.Get("type", "slackapp") {
|
|
case "irc":
|
|
client = irc.New(c)
|
|
case "slackapp":
|
|
client = slackapp.New(c)
|
|
case "discord":
|
|
d := discord.New(c)
|
|
d.Pirate = talklikeapirate.New(c)
|
|
client = d
|
|
default:
|
|
log.Fatal().Msgf("Unknown connection type: %s", c.Get("type", "UNSET"))
|
|
}
|
|
|
|
b := bot.New(c, client)
|
|
|
|
if r, path := client.GetRouter(); r != nil {
|
|
b.GetWeb().RegisterWeb(r, path)
|
|
}
|
|
|
|
plugins.Register(b)
|
|
|
|
if err := client.Serve(); err != nil {
|
|
log.Fatal().Err(err)
|
|
}
|
|
|
|
log.Debug().Msgf("Sending bot.Startup message")
|
|
b.Receive(client, bot.Startup, msg.Message{})
|
|
|
|
b.ListenAndServe()
|
|
}
|