1
0
mirror of https://github.com/klmp200/kaamelott-soundboard-telegram-bot synced 2024-10-29 20:58:03 +00:00
kaamelott-soundboard-telegr.../bot.go

147 lines
3.0 KiB
Go

package main
import (
"encoding/json"
"fmt"
"io/ioutil"
"log"
"net/http"
"os"
"strconv"
"time"
tb "gopkg.in/tucnak/telebot.v2"
)
type settings struct {
ListeningAddress string `json:"listening_address"`
Domain string `json:"domain"`
TelegramKey string `json:"telegram_key"`
}
const (
soundDatabase = "./sounds/sounds.json"
soundFolderPrefix = "./sounds"
defaultlisteningAddress = "0.0.0.0:8080"
defaultDomain = "http://localhost:8080"
defaultTelegramKey = "nope"
)
var index *SoundIndex
var cfg *settings
func loadSettings(path string) (*settings, error) {
loaded := settings{
ListeningAddress: defaultlisteningAddress,
Domain: defaultDomain,
TelegramKey: defaultTelegramKey,
}
jsonFile, err := os.Open(path)
if err != nil {
log.Printf("non trouvé mais c'est pas grave on fait sans")
return &loaded, nil
}
byteData, err := ioutil.ReadAll(jsonFile)
if err != nil {
return nil, fmt.Errorf("Je peux pas le lire ton fichier %s > %w", path, err)
}
if json.Unmarshal(byteData, &loaded) != nil {
return nil, fmt.Errorf("Ton json %s il est tout pété > %w", path, err)
}
return &loaded, nil
}
func main() {
cfg, err := loadSettings("settings.json")
if err != nil {
log.Fatal(err)
return
}
index, err := loadAndIndexSounds(soundDatabase)
if err != nil {
log.Fatal(err)
return
}
b, err := tb.NewBot(tb.Settings{
Token: cfg.TelegramKey,
Poller: &tb.LongPoller{Timeout: 10 * time.Second},
})
if err != nil {
log.Fatal(err)
return
}
fs := http.FileServer(http.Dir(soundFolderPrefix))
http.Handle("/", fs)
log.Printf("Écoute sur %s...", cfg.ListeningAddress)
go http.ListenAndServe(cfg.ListeningAddress, nil)
b.Handle("/cite", func(m *tb.Message) {
sounds, err := index.Search(m.Text)
if err != nil {
b.Send(m.Chat, "Pas de fichier audio trouvé")
return
}
path := fmt.Sprintf("%s/%s", soundFolderPrefix, sounds[0].File)
_, err = b.Send(
m.Chat,
&tb.Audio{
File: tb.FromDisk(path),
Caption: sounds[0].Title,
Title: sounds[0].Episode,
Performer: sounds[0].Character,
},
)
if err != nil {
log.Panicf("Impossible d'envoyer le fichier %s: %s", path, err)
b.Send(m.Chat, "Erreur d'envoi du fichier")
}
})
b.Handle(tb.OnQuery, func(q *tb.Query) {
sounds, err := index.Search(q.Text)
if err != nil {
log.Println("Pas de fichier audio trouvé")
return
}
results := make(tb.Results, len(sounds))
for i, sound := range sounds {
url := fmt.Sprintf("%s/%s", cfg.Domain, sound.File)
log.Println(url)
results[i] = &tb.AudioResult{
URL: url,
Caption: fmt.Sprintf(
"<i>%s</i>\n\n%s",
sound.Episode,
sound.Title,
),
Title: sound.Title,
Performer: sound.Character,
ParseMode: tb.ModeHTML,
}
results[i].SetResultID(strconv.Itoa(i))
}
err = b.Answer(q, &tb.QueryResponse{
Results: results,
SwitchPMParameter: "Ajoute moi",
CacheTime: 60, // une minute
})
if err != nil {
log.Println(err)
}
})
b.Start()
}