mirror of
https://github.com/lyx0/ollama-twitch-bot.git
synced 2024-11-06 18:52:03 +01:00
move command logic into main file
This commit is contained in:
parent
f31e75b73b
commit
779cc8650c
2 changed files with 33 additions and 50 deletions
44
commands.go
44
commands.go
|
@ -1,44 +0,0 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/gempir/go-twitch-irc/v4"
|
||||
)
|
||||
|
||||
// handleCommand is called each time a message starts with "()"
|
||||
func (app *application) handleCommand(message twitch.PrivateMessage) {
|
||||
var reply string
|
||||
|
||||
// msgLen is the amount of words in a message without the prefix.
|
||||
// Useful to check if enough cmdParams are provided.
|
||||
msgLen := len(strings.SplitN(message.Message, " ", -2))
|
||||
|
||||
// commandName is the actual name of the command without the prefix.
|
||||
// e.g. `()gpt` is `gpt`.
|
||||
commandName := strings.ToLower(strings.SplitN(message.Message, " ", 3)[0][2:])
|
||||
switch commandName {
|
||||
case "gpt":
|
||||
if msgLen < 2 {
|
||||
reply = "Not enough arguments provided. Usage: ()gpt <query>"
|
||||
} else {
|
||||
switch app.cfg.ollamaContext {
|
||||
case "none":
|
||||
app.generateNoContext(message.Channel, message.Message[6:len(message.Message)])
|
||||
return
|
||||
|
||||
case "general":
|
||||
app.chatGeneralContext(message.Channel, message.Message[6:len(message.Message)])
|
||||
return
|
||||
|
||||
case "user":
|
||||
app.chatUserContext(message.Channel, message.User.Name, message.Message[6:len(message.Message)])
|
||||
return
|
||||
}
|
||||
}
|
||||
if reply != "" {
|
||||
go app.send(message.Channel, reply)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
37
main.go
37
main.go
|
@ -67,12 +67,39 @@ func main() {
|
|||
}
|
||||
|
||||
// Message was shorter than our prefix is therefore it's irrelevant for us.
|
||||
if len(message.Message) >= 2 {
|
||||
// Check if the first 2 characters of the mesage were our prefix.
|
||||
// if they were forward the message to the command handler.
|
||||
if message.Message[:2] == "()" {
|
||||
go app.handleCommand(message)
|
||||
if len(message.Message) >= 2 && message.Message[:2] == "()" {
|
||||
var reply string
|
||||
|
||||
// msgLen is the amount of words in a message without the prefix.
|
||||
msgLen := len(strings.SplitN(message.Message, " ", -2))
|
||||
|
||||
// commandName is the actual name of the command without the prefix.
|
||||
// e.g. `()gpt` is `gpt`.
|
||||
commandName := strings.ToLower(strings.SplitN(message.Message, " ", 3)[0][2:])
|
||||
switch commandName {
|
||||
case "gpt":
|
||||
if msgLen < 2 {
|
||||
reply = "Not enough arguments provided. Usage: ()gpt <query>"
|
||||
return
|
||||
} else {
|
||||
switch app.cfg.ollamaContext {
|
||||
case "none":
|
||||
app.generateNoContext(message.Channel, message.Message[6:len(message.Message)])
|
||||
return
|
||||
|
||||
case "general":
|
||||
app.chatGeneralContext(message.Channel, message.Message[6:len(message.Message)])
|
||||
return
|
||||
|
||||
case "user":
|
||||
app.chatUserContext(message.Channel, message.User.Name, message.Message[6:len(message.Message)])
|
||||
return
|
||||
}
|
||||
}
|
||||
if reply != "" {
|
||||
go app.send(message.Channel, reply)
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
})
|
||||
|
|
Loading…
Reference in a new issue