This repository has been archived on 2023-07-05. You can view files and clone it, but cannot push or open issues or pull requests.
clessy/mods/oroscopo.go

71 lines
1.2 KiB
Go

package main
import (
"log"
"os"
"strings"
"unicode"
"io/ioutil"
"path/filepath"
"github.com/hamcha/tg"
gomarkov "github.com/simon-weber/gomarkov"
)
var orogen *gomarkov.Chain
func oroscopo_init() {
counter := 0
orogen = gomarkov.NewChain()
filepath.Walk(*oropath, func(path string, info os.FileInfo, err error) error {
if info.IsDir() {
return nil
}
data, err := ioutil.ReadFile(path)
if err != nil {
return err
}
orogen.Update(string(data))
counter++
return nil
})
log.Printf("[oroscopo] Loaded corpus from %d files\n", counter)
}
func oroscopo_message(broker *tg.Broker, update tg.APIMessage) {
if isCommand(update, "oroscopo") {
var err error
var txt string
tries := 0
for tries < 10 {
txt, err = orogen.Respond("", 20, 100)
if err != nil {
txt = err.Error()
tries++
continue
}
// Find end
idx := strings.LastIndexByte(txt, '.')
if idx < 10 {
tries++
continue
}
txt = txt[:idx]
// Find beginning
idx = strings.IndexFunc(txt, unicode.IsUpper)
if idx < 0 {
tries++
continue
}
txt = txt[idx:]
break
}
broker.SendTextMessage(update.Chat, "<b>Ecco cosa dicono le stelle:</b>\n"+txt+".", &update.MessageID)
}
}