all-tg-bot/main.go

76 lines
1.7 KiB
Go
Raw Normal View History

2022-04-01 15:33:10 +00:00
package main
2022-04-01 16:17:55 +00:00
import (
"errors"
2022-04-03 10:19:32 +00:00
"git.user-penguin.space/vladimir/all_tg_bot/commands"
2022-04-01 16:17:55 +00:00
tgbot "github.com/go-telegram-bot-api/telegram-bot-api"
"github.com/umputun/go-flags"
"log"
"os"
"strings"
)
const (
tag = "@all"
allUsers = "@vo13crabs @disconecto @v9gel @KeyM_v2 @k_ukolov @iamembarrassed @freel_Fedorov_Kirill @butinmv"
)
type Opts struct {
Token string `short:"t" long:"token" description:"Telegram api token"`
2022-04-03 10:19:32 +00:00
Name string `short:"n" long:"name" description:"Telegram bot name" default:"@butler_tg_bot"`
2022-04-01 16:17:55 +00:00
}
var opts Opts
2022-04-01 15:33:10 +00:00
func main() {
2022-04-01 16:17:55 +00:00
run()
}
func run() {
p := flags.NewParser(&opts, flags.PrintErrors|flags.PassDoubleDash|flags.HelpFlag)
p.SubcommandsOptional = true
if _, err := p.Parse(); err != nil {
if err.(*flags.Error).Type != flags.ErrHelp {
log.Println(errors.New("[ERROR] cli error: " + err.Error()))
}
os.Exit(2)
}
bot, err := tgbot.NewBotAPI(opts.Token)
if err != nil {
panic(err)
}
u := tgbot.NewUpdate(0)
u.Timeout = 60
updates, err := bot.GetUpdatesChan(u)
if err != nil {
panic(err)
}
for update := range updates {
if update.Message == nil {
continue
}
text := update.Message.Text
chatID := update.Message.Chat.ID
if strings.Contains(strings.ToLower(text), tag) {
_, _ = bot.Send(tgbot.NewMessage(chatID, allUsers))
}
2022-04-03 10:19:32 +00:00
command := commands.Command(strings.Replace(text, opts.Name, "", 1))
switch command {
case commands.Start:
_, _ = bot.Send(tgbot.NewMessage(chatID, "Здравствуйте, я ваш дворецкий."))
2022-04-03 10:35:07 +00:00
case commands.Help:
_, _ = bot.Send(tgbot.NewMessage(chatID, "Вот что я умею:\n\n1) Позвать всех"))
2022-04-03 10:19:32 +00:00
case commands.Ping:
_, _ = bot.Send(tgbot.NewMessage(chatID, "pong"))
}
2022-04-01 16:17:55 +00:00
}
2022-04-01 15:33:10 +00:00
}