clessy-ng/main.go

124 lines
3.7 KiB
Go
Raw Permalink Normal View History

2022-03-24 11:47:37 +00:00
package main
import (
"flag"
"log"
2022-03-25 19:06:13 +00:00
"net/url"
"os"
2022-03-24 11:47:37 +00:00
"strings"
"git.fromouter.space/crunchy-rocks/clessy-ng/modules"
2022-03-24 15:28:00 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/macro"
2022-03-27 20:40:10 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/meme"
2022-03-24 11:47:37 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/metafora"
2022-03-24 15:28:00 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/proverbio"
2022-03-25 14:29:14 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/remind"
2022-03-27 20:40:10 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/snapchat"
2022-03-27 01:21:07 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/modules/unsplash"
2022-03-25 19:06:13 +00:00
"git.fromouter.space/crunchy-rocks/clessy-ng/utils"
2022-03-24 11:47:37 +00:00
2022-03-27 01:21:07 +00:00
"git.fromouter.space/crunchy-rocks/emoji"
2022-03-24 11:47:37 +00:00
"git.fromouter.space/hamcha/tg"
2022-03-25 19:06:13 +00:00
"github.com/cockroachdb/pebble"
2022-03-24 11:47:37 +00:00
)
var mods = map[string]modules.Module{
2022-03-24 15:28:00 +00:00
"metafora": &metafora.Module{},
"proverbio": &proverbio.Module{},
"macro": &macro.Module{},
2022-03-25 14:29:14 +00:00
"remind": &remind.Module{},
2022-03-27 01:21:07 +00:00
"unsplash": &unsplash.Module{},
2022-03-27 20:40:10 +00:00
"snapchat": &snapchat.Module{},
"meme": &meme.Module{},
2022-03-24 11:47:37 +00:00
}
func checkErr(err error, message string, args ...interface{}) {
if err != nil {
args = append(args, err)
2022-03-24 15:28:00 +00:00
log.Fatalf("FATAL: "+message+":\n\t%s", args...)
2022-03-24 11:47:37 +00:00
}
}
func main() {
disable := flag.String("disable", "", "Blacklist mods (separated by comma)")
enable := flag.String("enable", "", "Whitelist mods (separated by comma)")
macrofile := flag.String("import-macros", "", "If specified, path to JSON file containing macros to import to DB")
remindfile := flag.String("import-reminders", "", "If specified, path to JSON file containing reminders to import to DB")
2022-03-24 11:47:37 +00:00
flag.Parse()
// Make Telegram API client
2022-03-25 19:06:13 +00:00
api := tg.MakeAPIClient(utils.RequireEnv("CLESSY_TOKEN"))
2022-03-24 11:47:37 +00:00
name, err := api.GetMe()
checkErr(err, "could not retrieve bot info")
2022-03-27 01:21:07 +00:00
// Load emojis
emojis, err := emoji.ScanEmojiDirectory(utils.RequireEnv("CLESSY_EMOJI_PATH"))
if err != nil {
log.Printf("[x-emoji] Error while loading emojis: %s\n", err.Error())
log.Println("[x-emoji] Emoji support will be disabled")
} else {
log.Printf("[x-emoji] Loaded %d emojis\n", len(emojis))
}
2022-03-25 19:06:13 +00:00
// Initialize your database
db, err := pebble.Open(utils.RequireEnv("CLESSY_DB_DIR"), &pebble.Options{})
checkErr(err, "could not open database")
defer db.Close()
// Perform imports
if *macrofile != "" {
byt, err := os.ReadFile(*macrofile)
checkErr(err, "could not load macro file")
db.Set([]byte("mod/macros/data"), byt, &pebble.WriteOptions{Sync: true})
log.Println("Imported macros")
}
if *remindfile != "" {
byt, err := os.ReadFile(*macrofile)
checkErr(err, "could not load macro file")
db.Set([]byte("mod/macros/data"), byt, &pebble.WriteOptions{Sync: true})
log.Println("Imported reminders")
}
// Pick modules
2022-03-24 11:47:37 +00:00
toActivate := make(map[string]modules.Module)
if *disable != "" {
for _, modname := range strings.Split(*disable, ",") {
modname = strings.TrimSpace(modname)
delete(mods, modname)
}
toActivate = mods
} else if *enable != "" {
for _, modname := range strings.Split(*enable, ",") {
toActivate[modname] = mods[modname]
}
2022-03-24 12:24:20 +00:00
} else {
toActivate = mods
2022-03-24 11:47:37 +00:00
}
// Initialize modules
2022-03-24 11:47:37 +00:00
for modname, mod := range toActivate {
log.Printf("Initializing %s", modname)
2022-03-25 19:06:13 +00:00
err := mod.Initialize(modules.ModuleOptions{
2022-03-27 01:21:07 +00:00
API: api,
Name: name.Username,
KV: db,
Emojis: emojis,
2022-03-25 19:06:13 +00:00
})
2022-03-24 11:47:37 +00:00
checkErr(err, "Starting module %s failed with error", modname)
}
// Set webhook and handle calls
2022-03-25 19:06:13 +00:00
webhook := utils.RequireEnv("CLESSY_WEBHOOK")
uri, err := url.Parse(webhook)
checkErr(err, "Specified webhook is not a valid URL")
api.SetWebhook(webhook)
log.Fatal(api.HandleWebhook(utils.EnvFallback("CLESSY_BIND", ":8080"), uri.Path, func(update tg.APIUpdate) {
2022-03-24 12:24:20 +00:00
for _, mod := range toActivate {
mod.OnUpdate(update)
}
}))
2022-03-24 11:47:37 +00:00
}