/
main.go
111 lines (93 loc) · 2.57 KB
/
main.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
package main
import (
"bytes"
"context"
"expvar"
_ "expvar"
"flag"
"fmt"
"log"
"net/http"
"strings"
"time"
"github.com/Xe/x/idp/idpmiddleware"
"github.com/Xe/x/internal"
_ "github.com/Xe/x/tokipona"
"github.com/Xe/x/web/tokiponatokens"
_ "github.com/joho/godotenv/autoload"
tb "gopkg.in/tucnak/telebot.v2"
"within.website/johaus/parser"
_ "within.website/johaus/parser/alldialects"
"within.website/johaus/pretty"
"within.website/ln"
"within.website/ln/opname"
)
const (
tpapiurl = `https://us-central1-golden-cove-408.cloudfunctions.net/toki-pona-verb-marker`
selfURL = `http://10.0.0.240:5009/`
)
var (
telegramToken = flag.String("telegram-token", "", "telegram bot token")
tokiPonaAPIURL = flag.String("toki-pona-tokenizer-api-url", tpapiurl, "toki pona tokenizer API URL")
port = flag.String("port", "5009", "HTTP port for statistics")
)
func main() {
internal.HandleStartup()
b, err := tb.NewBot(tb.Settings{
Token: *telegramToken,
Poller: &tb.LongPoller{Timeout: 10 * time.Second},
})
errCount := expvar.NewInt("errors")
if err != nil {
log.Fatal(err)
return
}
b.Handle("/camxes", parserCommandFor(b, "camxes"))
b.Handle("/ilmentufa", parserCommandFor(b, "ilmentufa"))
b.Handle("/maftufa", parserCommandFor(b, "maftufa"))
b.Handle("/zantufa", parserCommandFor(b, "zantufa"))
tokiCount := expvar.NewInt("toki")
b.Handle("/toki", func(m *tb.Message) {
defer func() {
if r := recover(); r != nil {
fmt.Println("Recovered in f", r)
errCount.Add(1)
}
}()
defer tokiCount.Add(1)
msg := m.Payload
parts, err := tokiponatokens.Tokenize(*tokiPonaAPIURL, msg)
if err != nil {
b.Send(m.Sender, err.Error())
errCount.Add(1)
return
}
var sb strings.Builder
for _, sentence := range parts {
bracesReply := tokiBraces(sentence)
sb.WriteString(bracesReply)
sb.WriteRune('\n')
}
b.Send(m.Sender, sb.String())
})
ln.Log(opname.With(context.Background(), "main"), ln.Info("starting HTTP server"), ln.F{"port": *port, "using": "idpmiddleware", "self_url": selfURL})
go http.ListenAndServe(":"+*port, idpmiddleware.XeProtect(selfURL)(http.DefaultServeMux))
b.Start()
}
func parserCommandFor(b *tb.Bot, dialect string) func(*tb.Message) {
return func(m *tb.Message) {
msg := m.Payload
tree, err := parser.Parse(dialect, msg)
if err != nil {
b.Send(m.Sender, err.Error())
return
}
parser.RemoveMorphology(tree)
parser.AddElidedTerminators(tree)
parser.RemoveSpace(tree)
parser.CollapseLists(tree)
buf := bytes.NewBuffer(nil)
pretty.Braces(buf, tree)
b.Send(m.Sender, buf.String())
}
}