diff --git a/alfred.go b/alfred.go index e84b9a3..39f3709 100644 --- a/alfred.go +++ b/alfred.go @@ -2,15 +2,17 @@ * @Author: Bartuccio Antoine * @Date: 2018-07-23 15:24:22 * @Last Modified by: klmp200 -* @Last Modified time: 2018-07-23 19:14:10 +* @Last Modified time: 2018-07-24 02:33:57 */ package main import ( "./settings" + "./shared" tb "gopkg.in/tucnak/telebot.v2" "log" + "strings" "time" ) @@ -19,6 +21,9 @@ func main() { 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), @@ -33,6 +38,23 @@ func main() { 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() } diff --git a/settings.json b/settings.json index d691ca0..f6f81c4 100644 --- a/settings.json +++ b/settings.json @@ -1,3 +1,4 @@ { - "token": "INSERT TOKEN HERE" + "token": "INSERT TOKEN HERE", + "history size": 10 } \ No newline at end of file diff --git a/shared/history.go b/shared/history.go new file mode 100644 index 0000000..0dc315e --- /dev/null +++ b/shared/history.go @@ -0,0 +1,85 @@ +/* +* @Author: Bartuccio Antoine +* @Date: 2018-07-24 01:27:11 +* @Last Modified by: klmp200 +* @Last Modified time: 2018-07-24 02:09:53 + */ + +package shared + +import ( + "sync" +) + +type history struct { + mutex sync.Mutex + size int + data map[int64][]string +} + +var History history + +func InitHistory(size int) { + History = history{} + History.mutex.Lock() + defer History.mutex.Unlock() + History.size = size + History.data = make(map[int64][]string) +} + +func (h history) Size() int { + h.mutex.Lock() + defer h.mutex.Unlock() + return h.size +} + +func (h history) Message(chatID int64, n int) string { + h.mutex.Lock() + defer h.mutex.Unlock() + array, exists := h.data[chatID] + if exists && n >= 0 && n < len(array) { + return array[n] + } + return "" +} + +func (h history) AddMessage(chatID int64, m string) { + h.mutex.Lock() + defer h.mutex.Unlock() + if _, exists := h.data[chatID]; !exists { + h.data[chatID] = make([]string, h.size, h.size) + } + h.append(chatID, m) +} + +func (h history) LastMessage(chatID int64) string { + h.mutex.Lock() + defer h.mutex.Unlock() + if _, exists := h.data[chatID]; exists { + return h.data[chatID][h.size-1] + } + return "" +} + +func (h history) ChatHistory(chatID int64) []string { + h.mutex.Lock() + defer h.mutex.Unlock() + array, exists := h.data[chatID] + if exists { + c := make([]string, h.size, h.size) + copy(c, array) + return c + } + return nil +} + +// Assert that the slice exists and mutex already locked +func (h history) append(chatID int64, m string) { + c := make([]string, h.size-1, h.size-1) + array, _ := h.data[chatID] + copy(c, array[1:]) + for i, val := range c { + array[i] = val + } + array[h.size-1] = m +}