127 lines
2.4 KiB
Go
127 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"strings"
|
|
|
|
"git.fromouter.space/mcg/mlp-server-tools/draftbot"
|
|
|
|
"git.fromouter.space/hamcha/tg"
|
|
"github.com/go-kit/kit/log"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
func checkErr(err error, msg string, args ...interface{}) {
|
|
if err != nil {
|
|
fmt.Printf("FATAL ERROR\n"+msg+":\n ", args...)
|
|
fmt.Println(err.Error())
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
var api *tg.Telegram
|
|
var botname string
|
|
var tgapi *tgInterface
|
|
var logger log.Logger
|
|
|
|
func main() {
|
|
viper.AutomaticEnv()
|
|
|
|
bind := viper.GetString("bind")
|
|
if bind == "" {
|
|
panic("Missing BIND")
|
|
}
|
|
|
|
token := viper.GetString("token")
|
|
if token == "" {
|
|
panic("Missing TOKEN")
|
|
}
|
|
|
|
webhookURL := viper.GetString("webhookurl")
|
|
if webhookURL == "" {
|
|
panic("Missing WEBHOOKURL")
|
|
}
|
|
|
|
webhookPath := viper.GetString("webhookpath")
|
|
if webhookURL == "" {
|
|
panic("Missing WEBHOOKPATH")
|
|
}
|
|
|
|
botname = viper.GetString("botname")
|
|
if botname == "" {
|
|
panic("Missing BOTNAME")
|
|
}
|
|
|
|
initTemplates()
|
|
|
|
logger = log.NewLogfmtLogger(os.Stderr)
|
|
logger = log.With(logger, "ts", log.DefaultTimestampUTC)
|
|
logger = log.With(logger, "caller", log.DefaultCaller)
|
|
|
|
// Ignore CA errors
|
|
http.DefaultTransport.(*http.Transport).TLSClientConfig = &tls.Config{InsecureSkipVerify: true}
|
|
|
|
api = tg.MakeAPIClient(token)
|
|
api.SetWebhook(webhookURL)
|
|
tgapi = &tgInterface{
|
|
client: api,
|
|
usermap: make(map[string]int64),
|
|
}
|
|
bot := draftbot.NewDraftBot(tgapi, botname)
|
|
bot.Logger = logger
|
|
tgapi.Bind(bot)
|
|
logger.Log("err", api.HandleWebhook(bind, webhookPath, webhook))
|
|
}
|
|
|
|
func webhook(update tg.APIUpdate) {
|
|
if update.Message != nil {
|
|
msg := *update.Message
|
|
if msg.Chat == nil {
|
|
return
|
|
}
|
|
switch msg.Chat.Type {
|
|
case tg.ChatTypeGroup, tg.ChatTypeSupergroup:
|
|
if isCommand(msg, "newdraft") {
|
|
tgapi.NewDraft(msg.Chat, msg.User)
|
|
return
|
|
}
|
|
if isCommand(msg, "join") {
|
|
tgapi.JoinDraft(msg.Chat, msg.User)
|
|
return
|
|
}
|
|
if isCommand(msg, "startdraft") {
|
|
tgapi.StartDraft(msg.Chat, msg.User)
|
|
return
|
|
}
|
|
case tg.ChatTypePrivate:
|
|
//todo
|
|
}
|
|
}
|
|
}
|
|
|
|
func isCommand(update tg.APIMessage, cmdname string) bool {
|
|
if update.Text == nil {
|
|
return false
|
|
}
|
|
|
|
text := strings.TrimSpace(*(update.Text))
|
|
|
|
shortcmd := "/" + cmdname
|
|
fullcmd := shortcmd + "@" + botname
|
|
|
|
// Check short form
|
|
if text == shortcmd || strings.HasPrefix(text, shortcmd+" ") {
|
|
return true
|
|
}
|
|
|
|
// Check long form
|
|
if text == fullcmd || strings.HasPrefix(text, fullcmd+" ") {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|