mirror of
https://github.com/Luzifer/twitch-bot.git
synced 2024-09-07 20:39:07 +00:00
96 lines
2.2 KiB
Go
96 lines
2.2 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
_ "embed"
|
|
"runtime/debug"
|
|
"sort"
|
|
"text/template"
|
|
"time"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/sirupsen/logrus"
|
|
"gopkg.in/irc.v4"
|
|
|
|
"github.com/Luzifer/go_helpers/v2/fieldcollection"
|
|
"github.com/Luzifer/twitch-bot/v3/plugins"
|
|
)
|
|
|
|
//go:embed tplDocs.tpl
|
|
var tplDocsTemplate string
|
|
|
|
func generateTplDocs() ([]byte, error) {
|
|
tpl, err := template.New("tplDocs").Funcs(map[string]any{
|
|
"renderExample": generateTplDocsRender,
|
|
}).Parse(tplDocsTemplate)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "parsing tplDocs template")
|
|
}
|
|
|
|
sort.Slice(tplFuncs.docs, func(i, j int) bool { return tplFuncs.docs[i].Name < tplFuncs.docs[j].Name })
|
|
|
|
buf := new(bytes.Buffer)
|
|
if err := tpl.Execute(buf, struct {
|
|
Funcs []plugins.TemplateFuncDocumentation
|
|
}{
|
|
Funcs: tplFuncs.docs,
|
|
}); err != nil {
|
|
return nil, errors.Wrap(err, "rendering tplDocs template")
|
|
}
|
|
|
|
return buf.Bytes(), nil
|
|
}
|
|
|
|
func generateTplDocsRender(e *plugins.TemplateFuncDocumentationExample) (string, error) {
|
|
defer func() {
|
|
if err := recover(); err != nil {
|
|
logrus.WithError(err.(error)).Fatalf("%s", debug.Stack())
|
|
}
|
|
}()
|
|
|
|
content := e.MessageContent
|
|
if content == "" {
|
|
content = "Hello World"
|
|
}
|
|
|
|
msg := &irc.Message{
|
|
Command: "PRIVMSG",
|
|
Params: []string{
|
|
"#example",
|
|
content,
|
|
},
|
|
Prefix: &irc.Prefix{
|
|
Name: "exampleuser",
|
|
User: "exampleuser",
|
|
Host: "exampleuser.tmi.twitch.tv",
|
|
},
|
|
Tags: map[string]string{
|
|
"badge-info": "subscriber/26",
|
|
"badges": "moderator/1,subscriber/24",
|
|
"color": "#8A2BE2",
|
|
"display-name": "ExampleUser",
|
|
"emotes": "",
|
|
"first-msg": "0",
|
|
"flags": "",
|
|
"id": "d3167f1f-5a0c-4d78-ba68-1a6c0018d284",
|
|
"mod": "1",
|
|
"returning-chatter": "0",
|
|
"room-id": "123456",
|
|
"subscriber": "1",
|
|
"tmi-sent-ts": "1679582970403",
|
|
"turbo": "0",
|
|
"user-id": "987654",
|
|
"user-type": "mod",
|
|
},
|
|
}
|
|
|
|
rule := &plugins.Rule{}
|
|
if e.MatchMessage != "" {
|
|
rule.MatchMessage = &e.MatchMessage
|
|
}
|
|
|
|
return formatMessage(e.Template, msg, rule, fieldcollection.FieldCollectionFromData(map[string]any{
|
|
"testDuration": 5*time.Hour + 33*time.Minute + 12*time.Second,
|
|
}))
|
|
}
|