145 lines
3.4 KiB
Go
145 lines
3.4 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"math/rand"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/hamcha/clessy/tg"
|
|
)
|
|
|
|
type Mod struct {
|
|
OnInit func()
|
|
OnMessage func(*tg.Broker, tg.APIMessage)
|
|
}
|
|
|
|
var mods = map[string]Mod{
|
|
"metafora": {
|
|
OnMessage: metafora,
|
|
},
|
|
"viaggi": {
|
|
OnInit: initviaggi,
|
|
OnMessage: viaggi,
|
|
},
|
|
"meme": {
|
|
OnInit: initmeme,
|
|
OnMessage: memegen,
|
|
},
|
|
"unsplash": {
|
|
OnInit: initunsplash,
|
|
OnMessage: unsplash,
|
|
},
|
|
"macro": {
|
|
OnInit: initmacro,
|
|
OnMessage: macro,
|
|
},
|
|
"snapchat": {
|
|
OnInit: initsnapchat,
|
|
OnMessage: snapchat,
|
|
},
|
|
"proverbio": {
|
|
OnInit: initproverbio,
|
|
OnMessage: proverbio,
|
|
},
|
|
"talk": {
|
|
OnInit: inittalk,
|
|
OnMessage: talk,
|
|
},
|
|
"stt": {
|
|
OnInit: initstt,
|
|
OnMessage: stt,
|
|
},
|
|
"remind": {
|
|
OnInit: initremind,
|
|
OnMessage: remind,
|
|
},
|
|
}
|
|
|
|
func initmods() {
|
|
var enabledmods []string
|
|
for name, mod := range mods {
|
|
log.Printf("Initializing %s..", name)
|
|
if mod.OnInit != nil {
|
|
mod.OnInit()
|
|
}
|
|
enabledmods = append(enabledmods, name)
|
|
}
|
|
log.Println("===============================")
|
|
log.Println("All mods have been initialized!")
|
|
log.Println("===============================")
|
|
log.Println("Active modules: " + strings.Join(enabledmods, ", "))
|
|
}
|
|
|
|
func dispatch(broker *tg.Broker, update tg.APIMessage) {
|
|
for _, mod := range mods {
|
|
if mod.OnMessage != nil {
|
|
go mod.OnMessage(broker, update)
|
|
}
|
|
}
|
|
}
|
|
|
|
func isCommand(update tg.APIMessage, cmdname string) bool {
|
|
if update.Text == nil {
|
|
return false
|
|
}
|
|
|
|
text := *(update.Text)
|
|
return strings.HasPrefix(text, "/"+cmdname+"@"+*botname) || (strings.HasPrefix(text, "/"+cmdname) && !strings.Contains(text, "@"))
|
|
}
|
|
|
|
var broker *tg.Broker
|
|
var botname *string
|
|
var impact *string
|
|
var gillmt *string
|
|
var sourcesans *string
|
|
var proverbi *string
|
|
var talktoken *string
|
|
var gapifile *string
|
|
|
|
func main() {
|
|
brokerAddr := flag.String("broker", "localhost:7314", "Broker address:port")
|
|
botname = flag.String("botname", "maudbot", "Bot name for /targetet@commands")
|
|
impact = flag.String("impact", "impact.ttf", "Path to impact.ttf (Impact font)")
|
|
gillmt = flag.String("gillmt", "gill.ttf", "Path to gill.ttf (Gill Sans MT font)")
|
|
sourcesans = flag.String("sourcesans", "source.ttf", "Path to source.ttf (Source Sans Pro font)")
|
|
macropath = flag.String("macropath", "macros.json", "Path to macros db (JSON)")
|
|
remindpath = flag.String("remindpath", "reminders.json", "Path to reminder db (JSON)")
|
|
proverbi = flag.String("proverbi", "proverbi.txt", "Path to proverbi pairs (separated by /)")
|
|
talktoken = flag.String("apiai", "@apiai.token", "api.ai token")
|
|
gapifile = flag.String("gapi", "gapi.json", "Google API Service Credentials file")
|
|
disable := flag.String("disable", "", "Blacklist mods (separated by comma)")
|
|
enable := flag.String("enable", "", "Whitelist mods (separated by comma)")
|
|
flag.Parse()
|
|
|
|
if *disable != "" {
|
|
for _, modname := range strings.Split(*disable, ",") {
|
|
modname = strings.TrimSpace(modname)
|
|
delete(mods, modname)
|
|
}
|
|
} else if *enable != "" {
|
|
newmods := make(map[string]Mod)
|
|
for _, modname := range strings.Split(*enable, ",") {
|
|
newmods[modname] = mods[modname]
|
|
}
|
|
mods = newmods
|
|
}
|
|
|
|
rand.Seed(time.Now().Unix())
|
|
|
|
initmods()
|
|
|
|
var err error
|
|
broker, err = tg.ConnectToBroker(*brokerAddr)
|
|
assert(err)
|
|
defer broker.Close()
|
|
|
|
assert(tg.RunBrokerClient(broker, dispatch))
|
|
}
|
|
|
|
func assert(err error) {
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|