bots/jitsibot
3
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

set command updated to handleSetCommand

Browse Source
This commit is contained in:
David Haukeness
2020-03-31 16:55:50 +00:00
parent 1890ee0585
commit 150e0b63ab
2 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
commands.go
View File

@ -76,30 +76,33 @@ func (b *bot) handleFeedback(m chat1.MsgSummary) {
}
}
func (b *bot) setKValue(convid chat1.ConvIDStr, msgID chat1.MessageID, args []string) {
if args[0] != "set" {
func (b *bot) handleSetCommand(m chat1.MsgSummary) {
// first normalize the text and extract the arguments
args := strings.Fields(strings.ToLower(m.Content.Text.Body))
if args[1] != "set" {
return
}
switch len(args) {
case 3:
if args[1] == "url" {
if args[2] == "url" {
// first validate the URL
u, err := url.ParseRequestURI(args[2])
if err != nil {
b.k.ReplyByConvID(convid, msgID, "ERROR - `%s`", err)
eid := b.logError(err)
b.k.ReactByConvID(m.ConvID, m.Id, "Error ID %s", eid)
return
}
// then make sure its HTTPS
if u.Scheme != "https" {
b.k.ReplyByConvID(convid, msgID, "ERROR - HTTPS Required")
b.k.ReactByConvID(m.ConvID, m.Id, "ERROR: HTTPS Required")
return
}
// then get the current options
var opts ConvOptions
err = b.KVStoreGetStruct(convid, &opts)
err = b.KVStoreGetStruct(m.ConvID, &opts)
if err != nil {
eid := b.logError(err)
b.k.ReactByConvID(convid, msgID, "Error %s", eid)
b.k.ReactByConvID(m.ConvID, m.Id, "Error ID %s", eid)
return
}
// then update the struct using only the scheme and hostname:port
@ -109,13 +112,13 @@ func (b *bot) setKValue(convid chat1.ConvIDStr, msgID chat1.MessageID, args []st
opts.CustomURL = fmt.Sprintf("%s://%s/", u.Scheme, u.Hostname())
}
// then write that back to kvstore, with revision
err = b.KVStorePutStruct(convid, opts)
err = b.KVStorePutStruct(m.ConvID, opts)
if err != nil {
eid := b.logError(err)
b.k.ReactByConvID(convid, msgID, "ERROR %s", eid)
b.k.ReactByConvID(m.ConvID, m.Id, "Error ID %s", eid)
return
}
b.k.ReactByConvID(convid, msgID, "OK!")
b.k.ReactByConvID(m.ConvID, m.Id, "OK!")
return
}
default:

2
handlers.go
View File

@ -51,7 +51,7 @@ func (b *bot) chatHandler(m chat1.MsgSummary) {
if nargs > 0 {
switch args[0] {
case "set":
b.setKValue(m.ConvID, m.Id, args)
b.handleSetCommand(m)
case "list":
b.listKValue(m.ConvID, m.Id, args)
}