71 lines
1.2 KiB
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)
|
|
}
|
|
}
|