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

various improvements

Browse source
This commit is contained in:
watn3y 2023-12-29 02:35:22 +01:00
parent d5272262da
commit 1a6953900c
14 changed files with 71 additions and 49 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
bot.go
View file

@ -6,7 +6,6 @@ import (
"watn3y/bloaterbot/commands"
"watn3y/bloaterbot/commands/notify"
"watn3y/bloaterbot/text"
"github.com/rs/zerolog/log"
)

13
botIO/authenticate.go
View file

@ -2,25 +2,24 @@ package botIO
import (
"watn3y/bloaterbot/config"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"github.com/rs/zerolog/log"
)
func Authenticate() (tgbotapi.UpdatesChannel, *tgbotapi.BotAPI) {
b, err := tgbotapi.NewBotAPI(config.BotConfig.APIToken)
bot, err := tgbotapi.NewBotAPI(config.BotConfig.APIToken)
if err != nil {
log.Panic().Err(err).Msg("Failed to authorize bot")
}
b.Debug = config.BotConfig.DebugMode
bot.Debug = config.BotConfig.DebugMode
u := tgbotapi.NewUpdate(0)
u.Timeout = 60
updates := tgbotapi.NewUpdate(0)
updates.Timeout = 60
log.Info().Int64("ID", b.Self.ID).Str("username", b.Self.UserName).Msg("Successfully authorized bot")
log.Info().Int64("ID", bot.Self.ID).Str("username", bot.Self.UserName).Msg("Successfully authorized bot")
return b.GetUpdatesChan(u), b
return bot.GetUpdatesChan(updates), bot
}

8
botIO/getinfo.go
View file

@ -1,12 +1,18 @@
package botIO
import tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
import (
"github.com/rs/zerolog/log"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
func GetUserByID(userToGet tgbotapi.ChatConfigWithUser, bot *tgbotapi.BotAPI) (error bool, member tgbotapi.ChatMember) {
member, err := bot.GetChatMember(tgbotapi.GetChatMemberConfig{ChatConfigWithUser: userToGet})
if err != nil {
log.Error().Err(err).Msg("Unable to get info for user")
return true, member
}
log.Debug().Interface("member",member).Msg("Got Info for user")
return false, member
}

2
commands/commands.go
View file

@ -15,6 +15,8 @@ import (
"github.com/rs/zerolog/log"
)
//TODO logging
func Commands(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
cmd := strings.ToLower(update.Message.Command())

11
commands/download/download.go
View file

@ -22,6 +22,8 @@ import (
"github.com/rs/zerolog/log"
)
//TODO logging
func Download(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
log.Debug().Int64("chat", update.Message.Chat.ID).Int64("user", update.Message.From.ID).Msg("starting download")
@ -30,18 +32,17 @@ func Download(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
msg := tgbotapi.MessageConfig{
BaseChat: tgbotapi.BaseChat{ChatID: update.Message.Chat.ID, ReplyToMessageID: update.Message.MessageID},
Text: "Downloading your media...",
Text: "Downloading your YouTube Video...",
}
workingMessage := botIO.SendMessage(msg, bot)
if len(commandArgs) >= 1 && matchURL(commandArgs[0]) != "" {
//TODO distinguish
//service := matchURL(commandArgs[0])
} else {
log.Error().Int64("chat", update.Message.Chat.ID).Int64("user", update.Message.From.ID).Str("args", update.Message.CommandArguments()).Msg("failed to download. empty arg")
log.Error().Int64("chat", update.Message.Chat.ID).Int64("user", update.Message.From.ID).Str("args", update.Message.CommandArguments()).Msg("Failed to download YouTube Video. Empty args")
message := tgbotapi.EditMessageTextConfig{
BaseEdit: tgbotapi.BaseEdit{ChatID: workingMessage.Chat.ID, MessageID: workingMessage.MessageID},
Text: "Please specify a valid YouTube URL to download something.",
Text: "Please specify a valid YouTube URL.",
}
botIO.EditMessage(message, bot)
return

2
commands/gaypoints/gaypoints.go
View file

@ -10,7 +10,7 @@ import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"github.com/rs/zerolog/log"
)
//TODO logging
func GetGP(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
log.Debug().Int64("chat", update.Message.Chat.ID).Int64("user", update.Message.From.ID).Msg("getting gaypoints")

3
commands/gaypoints/sql.go
View file

@ -11,7 +11,8 @@ var gpSelectUser *sql.Stmt
var gpSelectChat *sql.Stmt
var gpSet *sql.Stmt
//TODO logging
//TODO switch to SQL
func sqlGetAllGP(chatid int64) (gaypoints []gaypointShortDetails) {
rows, err := gpSelectChat.Query(chatid)
if err != nil {

1
commands/notify/notify.go
View file

@ -10,6 +10,7 @@ import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"github.com/rs/zerolog/log"
)
//TODO logging
func Reminder(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
log.Debug().Str("args", update.Message.CommandArguments()).Msg("parsing new reminder")

3
commands/notify/sql.go
View file

@ -9,7 +9,8 @@ var notifySetReminder *sql.Stmt
var notifyGetReminders *sql.Stmt
var notifyGetReminderDetails *sql.Stmt
var notifyDeleteReminder *sql.Stmt
//TODO logging
//TODO switch to SQL
func InitDB() {
const dbPath string = "./bloater.db"
log.Info().Str("dbpath", dbPath).Msg("init database")

58
config/config.go
View file

@ -2,6 +2,7 @@ package config
import (
"github.com/BurntSushi/toml"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
"os"
"time"
@ -11,11 +12,11 @@ type config struct {
APIToken string
DebugMode bool
PrivilegedUsers []int64
Shutup map[int64]time.Time //is always filled at startup, not present in config
Webserver struct {
Shutup map[int64]time.Time //! is always filled at startup, not present in config
Webserver struct {
Port string
}
Balonlyl struct {
Balonlyl struct {
EnabledChats []int64
BalonlylAT string
TriggerWords []string
@ -39,7 +40,7 @@ func LoadConfig() {
configFile, err := os.ReadFile("config.toml")
if err != nil {
log.Panic().Err(err).Msg("rror opening config file")
log.Panic().Err(err).Msg("error opening config file")
}
_, err = toml.Decode(string(configFile), &BotConfig)
@ -49,6 +50,10 @@ func LoadConfig() {
BotConfig.Shutup = make(map[int64]time.Time)
if !BotConfig.DebugMode {
zerolog.SetGlobalLevel(zerolog.InfoLevel)
}
log.Info().Msg("loaded config file")
log.Debug().Interface("config", BotConfig).Msg("")
@ -59,27 +64,27 @@ func EncodeToml() {
exampleConfig := config{
APIToken: "1234567:h8Y7ma7hMG93kxDx5o",
DebugMode: false,
PrivilegedUsers: []int64{1234567,9764382423},
Webserver: struct{ Port string }{
PrivilegedUsers: []int64{1234567, 9764382423},
Webserver: struct{ Port string }{
Port: "3000",
},
Balonlyl: struct {
Balonlyl: struct {
EnabledChats []int64
BalonlylAT string
TriggerWords []string
}{
EnabledChats: []int64{353454534,8743658},
EnabledChats: []int64{353454534, 8743658},
BalonlylAT: "@Balonlyl",
TriggerWords: []string{"French, France"},
},
GayPoints: struct {
GayPoints: struct {
EnabledChats []int64
ModifyUsers []int64
}{
EnabledChats: []int64{353454534,8743658},
ModifyUsers: []int64{3827468324,1736576},
EnabledChats: []int64{353454534, 8743658},
ModifyUsers: []int64{3827468324, 1736576},
},
Download: struct {
Download: struct {
Prefix string
Yourls struct {
APIPath string
@ -97,21 +102,18 @@ func EncodeToml() {
},
}
f, err := os.Create("config.toml.example")
if err != nil {
log.Panic().Err(err).Msg("error encoding config file. Cannot open config.toml.example")
if err != nil {
log.Panic().Err(err).Msg("error encoding config file. Cannot open config.toml.example")
}
if err := toml.NewEncoder(f).Encode(exampleConfig); err != nil {
log.Panic().Err(err).Msg("error encoding config file")
}
if err := f.Close(); err != nil {
log.Panic().Err(err).Msg("error encoding config file. Cannot close config.toml.example")
}
log.Info().Msg("Wrote example config to config.toml.example. Bye Bye")
}
if err := toml.NewEncoder(f).Encode(exampleConfig); err != nil {
log.Panic().Err(err).Msg("error encoding config file")
}
if err := f.Close(); err != nil {
log.Panic().Err(err).Msg("error encoding config file. Cannot close config.toml.example")
}
log.Info().Msg("Wrote example config to config.toml.example. Bye Bye")
}

13
main.go
View file

@ -26,16 +26,25 @@ func main() {
}
config.LoadConfig()
gaypoints.InitDB()
notify.InitDB()
go webserver.RunWeb()
go webserver.RunServer()
bot()
}
func configureLogger() {
output := zerolog.ConsoleWriter{Out: os.Stdout, TimeFormat: time.DateTime}
log.Logger = zerolog.New(output).With().Timestamp().Caller().Logger()
zerolog.SetGlobalLevel(zerolog.TraceLevel)
//! note that we overwrite the loglevel after loading the config in config/config.go:53. This is just the default
zerolog.SetGlobalLevel(zerolog.DebugLevel)
log.Info().Msg("Started zerolog logger")
}

1
text/balonlyl/balonlyl.go
View file

@ -6,6 +6,7 @@ import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
//TODO logging
func Balonlyl(update tgbotapi.Update, bot *tgbotapi.BotAPI) {

2
text/matcher.go
View file

@ -10,7 +10,7 @@ import (
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
"github.com/rs/zerolog/log"
)
//TODO logging
func Matcher(update tgbotapi.Update, bot *tgbotapi.BotAPI) {
log.Debug().Str("text", update.Message.Text).Int64("chat", update.Message.Chat.ID).Msg("Starting text matcher")

2
webserver/webserver.go
View file

@ -7,7 +7,7 @@ import (
"github.com/rs/zerolog/log"
)
func RunWeb() {
func RunServer() {
fs := http.FileServer(http.Dir("./videos"))
http.Handle("/", fs)