Avatar gm -> matrix

mute system messages

ROADMAP.md

@@ -31,7 +31,7 @@
   * [ ] Chat types
     * [ ] Private chat
     * [x] Group chat
-  * [ ] Avatars
+  * [x] Avatars
   * [ ] Presence
   * [ ] Typing notifications
   * [ ] Read receipts
@@ -56,7 +56,7 @@
   * [ ] Automatic portal creation
     * [x] At startup
     * [ ] When receiving invite
-    * [ ] When receiving message
+    * [x] When receiving message
   * [ ] Private chat creation by inviting Matrix puppet of WhatsApp user to new room
   * [ ] Option to use own Matrix account for messages sent from WhatsApp mobile/other web clients
   * [ ] Shared group chat portals

portal.go

@@ -339,6 +339,8 @@ func (portal *Portal) startHandling(source *User, info *groupme.Message) *appser
 		portal.log.Debugfln("Not handling %s: message was recently handled", info.ID)
 	} else if portal.isDuplicate(info.ID) {
 		portal.log.Debugfln("Not handling %s: message is duplicate", info.ID)
+	} else if info.System {
+		portal.log.Debugfln("Not handling %s: message is from system: %s", info.ID, info.Text)
 	} else {
 		portal.lastMessageTs = uint64(info.CreatedAt.ToTime().Unix())
 		intent := portal.getMessageIntent(source, info)

puppet.go

@@ -18,6 +18,8 @@ package main
 
 import (
 	"fmt"
+	"io/ioutil"
+	"net/http"
 	"regexp"
 	"strings"
 
@@ -180,55 +182,53 @@ func (puppet *Puppet) DefaultIntent() *appservice.IntentAPI {
 	return puppet.bridge.AS.Intent(puppet.MXID)
 }
 
-func (puppet *Puppet) UpdateAvatar(source *User, avatar *whatsappExt.ProfilePicInfo) bool {
-	// if avatar == nil {
-	// 	var err error
-	// 	avatar, err = source.Conn.GetProfilePicThumb(puppet.JID)
-	// 	if err != nil {
-	// 		puppet.log.Warnln("Failed to get avatar:", err)
-	// 		return false
-	// 	}
-	// }
+func (puppet *Puppet) UpdateAvatar(source *User, avatar string) bool {
+	if len(avatar) == 0 {
+		err := puppet.DefaultIntent().SetAvatarURL(id.ContentURI{})
+		if err != nil {
+			puppet.log.Warnln("Failed to remove avatar:", err)
+		}
+		puppet.AvatarURL = id.ContentURI{}
+		puppet.Avatar = avatar
+		go puppet.updatePortalAvatar()
+		return true
+	}
 
-	// if avatar.Status != 0 {
-	// 	return false
-	// }
+	if puppet.Avatar == avatar {
+		return false // up to date
+	}
 
-	// if avatar.Tag == puppet.Avatar {
-	// 	return false
-	// }
+	//TODO check its actually groupme?
+	response, err := http.Get(avatar + ".large")
+	if err != nil {
+		puppet.log.Warnln("Failed to download avatar:", err)
+		return false
+	}
+	defer response.Body.Close()
 
-	// if len(avatar.URL) == 0 {
-	// 	err := puppet.DefaultIntent().SetAvatarURL(id.ContentURI{})
-	// 	if err != nil {
-	// 		puppet.log.Warnln("Failed to remove avatar:", err)
-	// 	}
-	// 	puppet.AvatarURL = id.ContentURI{}
-	// 	puppet.Avatar = avatar.Tag
-	// 	go puppet.updatePortalAvatar()
-	// 	return true
-	// }
+	image, err := ioutil.ReadAll(response.Body)
+	if err != nil {
+		puppet.log.Warnln("Failed to read downloaded avatar:", err)
+		return false
+	}
 
-	// data, err := avatar.DownloadBytes()
-	// if err != nil {
-	// 	puppet.log.Warnln("Failed to download avatar:", err)
-	// 	return false
-	// }
+	mime := response.Header.Get("Content-Type")
+	if len(mime) == 0 {
+		mime = http.DetectContentType(image)
+	}
+	resp, err := puppet.DefaultIntent().UploadBytes(image, mime)
+	if err != nil {
+		puppet.log.Warnln("Failed to upload avatar:", err)
+		return false
+	}
 
-	// mime := http.DetectContentType(data)
-	// resp, err := puppet.DefaultIntent().UploadBytes(data, mime)
-	// if err != nil {
-	// 	puppet.log.Warnln("Failed to upload avatar:", err)
-	// 	return false
-	// }
-
-	// puppet.AvatarURL = resp.ContentURI
-	// err = puppet.DefaultIntent().SetAvatarURL(puppet.AvatarURL)
-	// if err != nil {
-	// 	puppet.log.Warnln("Failed to set avatar:", err)
-	// }
-	// puppet.Avatar = avatar.Tag
-	// go puppet.updatePortalAvatar()
+	puppet.AvatarURL = resp.ContentURI
+	err = puppet.DefaultIntent().SetAvatarURL(puppet.AvatarURL)
+	if err != nil {
+		puppet.log.Warnln("Failed to set avatar:", err)
+	}
+	puppet.Avatar = avatar
+	go puppet.updatePortalAvatar()
 	return true
 }
 
@@ -297,7 +297,7 @@ func (puppet *Puppet) Sync(source *User, contact groupme.User) {
 
 	update := false
 	update = puppet.UpdateName(source, contact) || update
-	update = puppet.UpdateAvatar(source, nil) || update
+	update = puppet.UpdateAvatar(source, contact.AvatarURL) || update
 	if update {
 		puppet.Update()
 	}

user.go

@@ -802,6 +802,7 @@ func (user *User) handleMessageLoop() {
 				puppet.Sync(user, groupme.User{
 					ID:        msg.data.ID,
 					Name:      msg.data.Name,
+					AvatarURL: msg.data.AvatarURL,
 				}) //TODO: add params or docs?
 			}
 			user.GetPortalByJID(msg.chat).messages <- msg