watn3y/bloaterbot
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

refine shutup feature

Browse source
This commit is contained in:
Watn3y 2023-03-05 05:08:34 +01:00
parent ed2d94f810
commit e0f0f2d282
7 changed files with 70 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

48
bot.go
View file

@ -1,15 +1,10 @@
package main
import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"log"
"regexp"
"strings"
"time"
"watn3y/bloaterbotv3/botIO"
"watn3y/bloaterbotv3/commonlogic"
"watn3y/bloaterbotv3/config"
"watn3y/bloaterbotv3/text/nhentai"
"watn3y/bloaterbotv3/commands"
"watn3y/bloaterbotv3/text"
)
var shutuptime time.Time
@ -23,48 +18,13 @@ func bot() {
}
if update.Message.IsCommand() {
go commands(update, bot)
commands.Commands(update, bot)
} else {
if time.Since(shutuptime).Minutes() >= 60 {
go textMatcher(update, bot)
text.Matcher(update, bot)
}
}
}
}
func textMatcher(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
//nhentai
const nhentaiRegex string = "([ \\t\\n\\r]|^)\\d{2,}([ \\t\\n\\r]|$)"
isNhentai, err := regexp.MatchString(nhentaiRegex, update.Message.Text)
if err != nil {
log.Panicf("Failed to compile regex for nhentai", err)
}
if isNhentai && commonlogic.ContainsInt64(config.BotConfig.Nhentai.EnabledChats, update.Message.Chat.ID) {
go nhentai.Nhentai(nhentaiRegex, update, bot)
}
//balonlyl
var isBalonlyl bool = false
for _, word := range config.BotConfig.Balonlyl.TriggerWords {
if strings.Contains(strings.ToLower(update.Message.Text), word) && commonlogic.ContainsInt64(config.BotConfig.Balonlyl.EnabledChats, update.Message.Chat.ID) {
isBalonlyl = true
break
}
}
if isBalonlyl {
go balonlyl(update, bot)
}
}
func commands(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
if strings.ToLower(update.Message.Command()) == "shutup" || strings.ToLower(update.Message.Command()) == "shut" {
shutuptime = time.Now().UTC()
msg := tgbotapi.NewMessage(update.Message.Chat.ID, "Shutting up")
botIO.SendMessage(msg, bot)
}
}

4
build.sh Normal file
View file

@ -0,0 +1,4 @@
###!/usr/bin/env bash
HASH=$(git log -1 --pretty=%h)
docker build -t bloaterbot:latest -t bloaterbot:$HASH .

1
commands.go
View file

@ -1 +0,0 @@
package main

17
commands/commands.go Normal file
View file

@ -0,0 +1,17 @@
package commands
import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"strings"
"time"
"watn3y/bloaterbotv3/botIO"
"watn3y/bloaterbotv3/config"
)
func Commands(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
if strings.ToLower(update.Message.Command()) == "shutup" || strings.ToLower(update.Message.Command()) == "shut" {
config.BotConfig.Shutup[update.Message.Chat.ID] = time.Now().UTC()
msg := tgbotapi.NewMessage(update.Message.Chat.ID, "Shutting up")
botIO.SendMessage(msg, bot)
}
}

2
config/config.go
View file

@ -12,7 +12,7 @@ type config struct {
APIToken string
DebugMode bool
PrivilegedUsers []int64
ShutupTime time.Time
Shutup map[int64]time.Time
Nhentai struct {
EnabledChats []int64
APIEndpoint string

4
text.go → text/balonlyl/balonlyl.go
View file

@ -1,4 +1,4 @@
package main
package balonlyl
import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
@ -6,7 +6,7 @@ import (
"watn3y/bloaterbotv3/config"
)
func balonlyl(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
func Balonlyl(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
/*balonlylLines := [4]string{
balonlylAT + " they're talking about you",

42
text/matcher.go Normal file
View file

@ -0,0 +1,42 @@
package text
import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"log"
"regexp"
"strings"
"time"
"watn3y/bloaterbotv3/commonlogic"
"watn3y/bloaterbotv3/config"
"watn3y/bloaterbotv3/text/balonlyl"
"watn3y/bloaterbotv3/text/nhentai"
)
func Matcher(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
if time.Since(config.BotConfig.Shutup[update.Message.Chat.ID]) <= time.Minute {
return
}
//nhentai
const nhentaiRegex string = "([ \\t\\n\\r]|^)\\d{2,}([ \\t\\n\\r]|$)"
isNhentai, err := regexp.MatchString(nhentaiRegex, update.Message.Text)
if err != nil {
log.Panicf("Failed to compile regex for nhentai", err)
}
if isNhentai && commonlogic.ContainsInt64(config.BotConfig.Nhentai.EnabledChats, update.Message.Chat.ID) {
go nhentai.Nhentai(nhentaiRegex, update, bot)
}
//balonlyl
var isBalonlyl bool = false
for _, word := range config.BotConfig.Balonlyl.TriggerWords {
if strings.Contains(strings.ToLower(update.Message.Text), word) && commonlogic.ContainsInt64(config.BotConfig.Balonlyl.EnabledChats, update.Message.Chat.ID) {
isBalonlyl = true
break
}
}
if isBalonlyl {
go balonlyl.Balonlyl(update, bot)
}
}