101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import (
|
||
|
|
"vultras.su/core/tg"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
// The component to get incoming messages and
|
||
|
|
// send back mutated version.
|
||
|
|
type MutateMessageCompo struct {
|
||
|
|
Mutate func(string) string
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewMutateMessageCompo(fn func(string) string) *MutateMessageCompo {
|
||
|
|
ret := &MutateMessageCompo{}
|
||
|
|
ret.Mutate = fn
|
||
|
|
return ret
|
||
|
|
}
|
||
|
|
|
||
|
|
func (w *MutateMessageCompo) Serve(c tg.Context) {
|
||
|
|
args, ok := c.Arg().([]any)
|
||
|
|
if ok {
|
||
|
|
for _, arg := range args {
|
||
|
|
c.Sendf("%v", arg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
for u := range c.Input() {
|
||
|
|
text := u.Message.Text
|
||
|
|
_, err := c.Sendf2("%s", w.Mutate(text))
|
||
|
|
if err != nil {
|
||
|
|
c.Sendf("debug: %q", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Implementing the Filter interface making
|
||
|
|
// possible to give the updates away for
|
||
|
|
// the underlying components.
|
||
|
|
func (w *MutateMessageCompo) Filter(u tg.Update) bool {
|
||
|
|
if u.Message == nil {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
const (
|
||
|
|
UpperCasePath tg.Path = "upper-case"
|
||
|
|
LowerCasePath = "lower-case"
|
||
|
|
EscapePath = "escape"
|
||
|
|
MutateMessagesPath = "mutate-messages"
|
||
|
|
)
|
||
|
|
|
||
|
|
var MutateMessagesWidget = tg.RenderFunc(func(c tg.Context) tg.UI {
|
||
|
|
return tg.UI{
|
||
|
|
tg.Messagef(
|
||
|
|
"Choose widget to mutate strings",
|
||
|
|
).Reply(
|
||
|
|
tg.NewKeyboard().Row(
|
||
|
|
tg.Buttonf("Upper case").Go(UpperCasePath),
|
||
|
|
tg.Buttonf("Lower case").Go(LowerCasePath),
|
||
|
|
tg.Buttonf("Escape chars").Go(EscapePath),
|
||
|
|
).Row(
|
||
|
|
BackButton,
|
||
|
|
).Reply(),
|
||
|
|
),
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
var MutateMessagesToLowerCaseWidget = tg.RenderFunc(func(c tg.Context) tg.UI {
|
||
|
|
return tg.UI{
|
||
|
|
tg.Messagef(
|
||
|
|
"Type a string and the bot will convert it to lower case",
|
||
|
|
).Reply(
|
||
|
|
BackKeyboard.Reply(),
|
||
|
|
),
|
||
|
|
NewMutateMessageCompo(strings.ToLower),
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|
||
|
|
var MutateMessagesToUpperCaseWidget = tg.RenderFunc(func(c tg.Context) tg.UI {
|
||
|
|
return tg.UI{
|
||
|
|
tg.Messagef(
|
||
|
|
"Type a string and the bot will convert it to upper case",
|
||
|
|
).Reply(
|
||
|
|
BackKeyboard.Reply(),
|
||
|
|
),
|
||
|
|
NewMutateMessageCompo(strings.ToUpper),
|
||
|
|
}
|
||
|
|
})
|
||
|
|
var MutateMessagesEscapeWidget = tg.RenderFunc(func(c tg.Context) tg.UI {
|
||
|
|
return tg.UI{
|
||
|
|
tg.Messagef(
|
||
|
|
"Type a string and the bot will escape characters in it",
|
||
|
|
).Reply(
|
||
|
|
BackKeyboard.Reply(),
|
||
|
|
),
|
||
|
|
NewMutateMessageCompo(tg.Escape2),
|
||
|
|
}
|
||
|
|
})
|
||
|
|
|