qbit/mcchunkie
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Allow setting key-value pairs from matrix

Browse Source
This commit is contained in:
Aaron Bieber 2020-05-07 19:35:36 -06:00
parent 09ba1cade3
commit 48a68c48fa
1 changed files with 35 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
main.go
View File

@ -22,7 +22,7 @@ const header = `
` `
func main() { func main() {
var username, password, userID, accessToken, server, db, avatar, botOwner string var username, shortname, password, userID, accessToken, server, db, avatar, botOwner string
var key, value, get string var key, value, get string
var setup, doc, verbose bool var setup, doc, verbose bool
@ -48,6 +48,7 @@ func main() {
var help = `^help: (\w+)$` var help = `^help: (\w+)$`
var helpRE = regexp.MustCompile(help) var helpRE = regexp.MustCompile(help)
var kvRE = regexp.MustCompile(`^(.+)\s->\s(.+)$`)
var store, err = NewStore(db) var store, err = NewStore(db)
if err != nil { if err != nil {
log.Fatalf("%s\n", err) log.Fatalf("%s\n", err)
@ -136,6 +137,8 @@ func main() {
botOwner, _ = store.Get("bot_owner") botOwner, _ = store.Get("bot_owner")
} }
shortname = plugins.NameRE.ReplaceAllString(username, "$1")
cli.SetCredentials(userID, accessToken) cli.SetCredentials(userID, accessToken)
cli.Store = store cli.Store = store
syncer := gomatrix.NewDefaultSyncer(username, store) syncer := gomatrix.NewDefaultSyncer(username, store)
@ -146,12 +149,16 @@ func main() {
if ev.Sender == username { if ev.Sender == username {
return return
} }
switch ev.Sender {
if ev.Sender == botOwner && ev.Content["membership"] == "invite" { case botOwner:
log.Println("message from owner")
if ev.Content["membership"] == "invite" {
log.Printf("Joining %s (invite from %s)\n", ev.RoomID, ev.Sender) log.Printf("Joining %s (invite from %s)\n", ev.RoomID, ev.Sender)
if _, err := cli.JoinRoom(ev.RoomID, "", nil); err != nil { if _, err := cli.JoinRoom(ev.RoomID, "", nil); err != nil {
log.Fatalln(err) log.Fatalln(err)
} }
return
}
} }
}) })
@ -199,6 +206,28 @@ func main() {
return return
} }
switch ev.Sender {
case botOwner:
var post string
var ok bool
if post, ok = ev.Body(); !ok {
return
}
if plugins.ToMe(username, post) {
mp := plugins.RemoveName(shortname, post)
if kvRE.MatchString(mp) {
key := kvRE.ReplaceAllString(post, "$1")
val := kvRE.ReplaceAllString(post, "$2")
store.Set(key, val)
log.Printf("Setting %q to %q", key, val)
plugins.SendMD(cli, ev.RoomID, fmt.Sprintf("Set **%q** = *%q*", key, val))
return
}
}
}
// Sending a response per plugin hits issues, so save them and // Sending a response per plugin hits issues, so save them and
// send as one message. // send as one message.
helps := []string{} helps := []string{}