Add DM support

MilesBreslin · Mar 2, 2020 · 1fdbec4 · 1fdbec4
1 parent 8897f5f
commit 1fdbec4
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/rocket/messages.go b/rocket/messages.go
@@ -156,6 +156,13 @@ func (msg *Message) Reply(text string) (Message, error) {
     return msg.rocketCon.SendMessage(msg.RoomId, text)
 }
 
+func (msg *Message) DM(text string) (Message, error) {
+    if msg.IsDirect {
+        return msg.Reply(text)
+    }
+    return msg.rocketCon.DM(msg.UserName, text)
+}
+
 func (msg *Message) KickUser() {
     msg.Reply("/kick @"+msg.UserName)
 }

diff --git a/rocket/rocket.go b/rocket/rocket.go
@@ -562,6 +562,22 @@ func (rock *RocketCon) SendMessage(rid string, text string) (Message, error) {
     return msg, nil
 }
 
+func (rock *RocketCon) DM(username string, text string) (Message, error) {
+    obj := map[string] interface{} {
+        "method": "createDirectMessage",
+        "params": []string {
+            username,
+        },
+    }
+
+    reply, err := rock.runMethod(obj)
+    if err != nil {
+        return Message{}, err
+    }
+    rid := reply["result"].(map[string] interface{})["rid"].(string)
+    return rock.SendMessage(rid, text)
+}
+
 func (rock *RocketCon) React(mid string, emoji string) error {
     reaction := map[string] interface{} {
         "method": "setReaction",