mirror of https://github.com/velour/catbase.git
Compare commits
No commits in common. "6707902caff66dc1e7a2686548e942c199b6afeb" and "4626d0270c36f0e65288ba1aae4b467644f1e5de" have entirely different histories.
6707902caf
...
4626d0270c
|
@ -151,27 +151,7 @@ func (d *Discord) sendMessage(channel, message string, meMessage bool, args ...a
|
||||||
Interface("data", data).
|
Interface("data", data).
|
||||||
Msg("sending message")
|
Msg("sending message")
|
||||||
|
|
||||||
maxLen := 2000
|
st, err := d.client.ChannelMessageSendComplex(channel, data)
|
||||||
chunkSize := maxLen - 100
|
|
||||||
var st *discordgo.Message
|
|
||||||
var err error
|
|
||||||
if len(data.Content) > maxLen {
|
|
||||||
tmp := data.Content
|
|
||||||
data.Content = tmp[:chunkSize]
|
|
||||||
st, err = d.client.ChannelMessageSendComplex(channel, data)
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
for i := chunkSize; i < len(data.Content); i += chunkSize {
|
|
||||||
data := &discordgo.MessageSend{Content: tmp[i : i+chunkSize]}
|
|
||||||
st, err = d.client.ChannelMessageSendComplex(channel, data)
|
|
||||||
if err != nil {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
st, err = d.client.ChannelMessageSendComplex(channel, data)
|
|
||||||
}
|
|
||||||
|
|
||||||
//st, err := d.client.ChannelMessageSend(channel, message)
|
//st, err := d.client.ChannelMessageSend(channel, message)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
|
4
main.go
4
main.go
|
@ -135,6 +135,7 @@ func main() {
|
||||||
b.AddPlugin(roles.New(b))
|
b.AddPlugin(roles.New(b))
|
||||||
b.AddPlugin(twitch.New(b))
|
b.AddPlugin(twitch.New(b))
|
||||||
b.AddPlugin(pagecomment.New(b))
|
b.AddPlugin(pagecomment.New(b))
|
||||||
|
b.AddPlugin(gpt.New(b))
|
||||||
b.AddPlugin(secrets.New(b))
|
b.AddPlugin(secrets.New(b))
|
||||||
b.AddPlugin(mayi.New(b))
|
b.AddPlugin(mayi.New(b))
|
||||||
b.AddPlugin(giphy.New(b))
|
b.AddPlugin(giphy.New(b))
|
||||||
|
@ -178,9 +179,8 @@ func main() {
|
||||||
b.AddPlugin(cowboy.New(b))
|
b.AddPlugin(cowboy.New(b))
|
||||||
b.AddPlugin(topic.New(b))
|
b.AddPlugin(topic.New(b))
|
||||||
b.AddPlugin(talker.New(b))
|
b.AddPlugin(talker.New(b))
|
||||||
b.AddPlugin(fact.New(b))
|
|
||||||
// catches anything left, will always return true
|
// catches anything left, will always return true
|
||||||
b.AddPlugin(gpt.New(b))
|
b.AddPlugin(fact.New(b))
|
||||||
|
|
||||||
if err := client.Serve(); err != nil {
|
if err := client.Serve(); err != nil {
|
||||||
log.Fatal().Err(err)
|
log.Fatal().Err(err)
|
||||||
|
|
|
@ -90,8 +90,21 @@ func New(botInst bot.Bot) *FactoidPlugin {
|
||||||
|
|
||||||
// findAction simply regexes a string for the action verb
|
// findAction simply regexes a string for the action verb
|
||||||
func findAction(message string) string {
|
func findAction(message string) string {
|
||||||
r := regexp.MustCompile("<.+?>")
|
r, err := regexp.Compile("<.+?>")
|
||||||
return r.FindString(message)
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
action := r.FindString(message)
|
||||||
|
|
||||||
|
if action == "" {
|
||||||
|
if strings.Contains(message, " is ") {
|
||||||
|
return "is"
|
||||||
|
} else if strings.Contains(message, " are ") {
|
||||||
|
return "are"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return action
|
||||||
}
|
}
|
||||||
|
|
||||||
// learnFact assumes we have a learning situation and inserts a new fact
|
// learnFact assumes we have a learning situation and inserts a new fact
|
||||||
|
@ -472,7 +485,18 @@ func (p *FactoidPlugin) register() {
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
return false
|
notFound := p.c.GetArray("fact.notfound", []string{
|
||||||
|
"I don't know.",
|
||||||
|
"NONONONO",
|
||||||
|
"((",
|
||||||
|
"*pukes*",
|
||||||
|
"NOPE! NOPE! NOPE!",
|
||||||
|
"One time, I learned how to jump rope.",
|
||||||
|
})
|
||||||
|
|
||||||
|
// We didn't find anything, panic!
|
||||||
|
p.b.Send(c, bot.Message, message.Channel, notFound[rand.Intn(len(notFound))])
|
||||||
|
return true
|
||||||
}},
|
}},
|
||||||
}
|
}
|
||||||
p.b.RegisterTable(p, p.handlers)
|
p.b.RegisterTable(p, p.handlers)
|
||||||
|
|
|
@ -54,11 +54,6 @@ func (p *GPTPlugin) register() {
|
||||||
HelpText: "set the ChatGPT prompt",
|
HelpText: "set the ChatGPT prompt",
|
||||||
Handler: p.setPromptMessage,
|
Handler: p.setPromptMessage,
|
||||||
},
|
},
|
||||||
{
|
|
||||||
Kind: bot.Message, IsCmd: true,
|
|
||||||
Regex: regexp.MustCompile(`(?P<text>.*)`),
|
|
||||||
Handler: p.chatMessage,
|
|
||||||
},
|
|
||||||
}
|
}
|
||||||
log.Debug().Msg("Registering GPT3 handlers")
|
log.Debug().Msg("Registering GPT3 handlers")
|
||||||
p.b.RegisterTable(p, p.h)
|
p.b.RegisterTable(p, p.h)
|
||||||
|
|
Loading…
Reference in New Issue