61 lines
1.2 KiB
Go
61 lines
1.2 KiB
Go
/*
|
|
* @Author: Bartuccio Antoine
|
|
* @Date: 2018-07-23 15:24:22
|
|
* @Last Modified by: klmp200
|
|
* @Last Modified time: 2018-07-24 02:33:57
|
|
*/
|
|
|
|
package main
|
|
|
|
import (
|
|
"./settings"
|
|
"./shared"
|
|
tb "gopkg.in/tucnak/telebot.v2"
|
|
"log"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func main() {
|
|
if err := settings.LoadSettings("settings.json", "settings_custom.json"); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
|
|
log.Println("Initialize history")
|
|
shared.InitHistory(int(settings.Settings["history size"].(float64)))
|
|
|
|
log.Println("Bot initialisation")
|
|
b, err := tb.NewBot(tb.Settings{
|
|
Token: settings.Settings["token"].(string),
|
|
Poller: &tb.LongPoller{Timeout: 10 * time.Second},
|
|
})
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
return
|
|
}
|
|
|
|
b.Handle("/hello", func(m *tb.Message) {
|
|
b.Send(m.Chat, "Bonjour "+m.Sender.Username)
|
|
})
|
|
|
|
b.Handle(tb.OnText, func(m *tb.Message) {
|
|
shared.History.AddMessage(m.Chat.ID, m.Text)
|
|
})
|
|
|
|
b.Handle("/sponge", func(m *tb.Message) {
|
|
message := ""
|
|
for i, char := range shared.History.LastMessage(m.Chat.ID) {
|
|
if i%2 == 0 {
|
|
message += strings.ToLower(string(char))
|
|
} else {
|
|
message += strings.ToUpper(string(char))
|
|
}
|
|
}
|
|
b.Send(m.Chat, message)
|
|
|
|
})
|
|
|
|
log.Println("Starting bot")
|
|
b.Start()
|
|
}
|