147 lines
3.0 KiB
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()
|
|
|
|
}
|