handle*: add support for avatar changes (#81)

Co-authored-by: Tulir Asokan <[email protected]>

Author: rnons

pkg/connector/handlematrix.go

@@ -20,6 +20,7 @@ import (
 	"context"
 	"errors"
 	"fmt"
+	"net/http"
 
 	"github.com/google/uuid"
 	"github.com/rs/zerolog"
@@ -48,6 +49,7 @@ var (
 	_ bridgev2.TypingHandlingNetworkAPI      = (*TwitterClient)(nil)
 	_ bridgev2.ChatViewingNetworkAPI         = (*TwitterClient)(nil)
 	_ bridgev2.DeleteChatHandlingNetworkAPI  = (*TwitterClient)(nil)
+	_ bridgev2.RoomAvatarHandlingNetworkAPI  = (*TwitterClient)(nil)
 )
 
 var _ bridgev2.TransactionIDGeneratingNetwork = (*TwitterConnector)(nil)
@@ -250,3 +252,41 @@ func (tc *TwitterClient) HandleMatrixDeleteChat(ctx context.Context, chat *bridg
 	reqQuery := payload.DMRequestQuery{}.Default()
 	return tc.client.DeleteConversation(ctx, conversationID, &reqQuery)
 }
+
+func (tc *TwitterClient) HandleMatrixRoomAvatar(ctx context.Context, msg *bridgev2.MatrixRoomAvatar) (bool, error) {
+	if msg.Portal.RoomType == database.RoomTypeDM {
+		return false, errors.New("cannot set room avatar for DM")
+	}
+
+	if msg.Content.URL != "" || msg.Content.MSC3414File != nil {
+		data, err := msg.Portal.Bridge.Bot.DownloadMedia(ctx, msg.Content.URL, msg.Content.MSC3414File)
+		if err != nil {
+			return false, fmt.Errorf("failed to download avatar: %w", err)
+		}
+
+		var mediaType string
+		if msg.Content.Info != nil {
+			mediaType = msg.Content.Info.MimeType
+		} else {
+			mediaType = http.DetectContentType(data)
+		}
+
+		uploadMediaParams := &payload.UploadMediaQuery{
+			MediaType: mediaType,
+		}
+		uploadedMediaResponse, err := tc.client.UploadMedia(ctx, uploadMediaParams, data)
+		if err != nil {
+			return false, err
+		}
+
+		updateAvatarParams := &payload.DMRequestQuery{
+			AvatarID: uploadedMediaResponse.MediaIDString,
+		}
+		err = tc.client.UpdateConversationAvatar(ctx, string(msg.Portal.ID), updateAvatarParams)
+		if err != nil {
+			return false, err
+		}
+		return true, nil
+	}
+	return false, errors.New("avatar not found")
+}

pkg/connector/handletwit.go

@@ -181,6 +181,23 @@ func (tc *TwitterClient) HandleTwitterEvent(rawEvt types.TwitterEvent, inbox *re
 			},
 		}
 		return tc.userLogin.QueueRemoteEvent(portalUpdateRemoteEvent).Success
+	case *types.ConversationAvatarUpdate:
+		chatInfo := &bridgev2.ChatInfo{
+			Avatar: makeAvatar(tc.client, evt.ConversationAvatarImageHttps),
+		}
+		success := tc.userLogin.QueueRemoteEvent(&simplevent.ChatInfoChange{
+			EventMeta: simplevent.EventMeta{
+				Type:        bridgev2.RemoteEventChatInfoChange,
+				Sender:      tc.MakeEventSender(evt.ByUserID),
+				PortalKey:   tc.makePortalKeyFromInbox(evt.ConversationID, inbox),
+				StreamOrder: methods.ParseSnowflakeInt(evt.ID),
+				Timestamp:   methods.ParseSnowflake(evt.ID),
+			},
+			ChatInfoChange: &bridgev2.ChatInfoChange{
+				ChatInfo: chatInfo,
+			},
+		}).Success
+		return success
 	case *types.ConversationMetadataUpdate:
 		tc.client.Logger.Warn().Any("data", evt).Msg("Unhandled conversation metadata update event")
 		return true

pkg/twittermeow/data/endpoints/endpoints.go

@@ -29,6 +29,7 @@ const (
 	SEARCH_TYPEAHEAD_URL              = BASE_URL + "/i/api/1.1/search/typeahead.json"
 	DM_PERMISSIONS_URL                = BASE_URL + "/i/api/1.1/dm/permissions.json"
 	DELETE_CONVERSATION_URL           = BASE_URL + "/i/api/1.1/dm/conversation/%s/delete.json"
+	UPDATE_CONVERSATION_AVATAR_URL    = BASE_URL + "/i/api/1.1/dm/conversation/%s/update_avatar.json"
 	PIN_CONVERSATION_URL              = BASE_URL + "/i/api/graphql/o0aymgGiJY-53Y52YSUGVA/DMPinnedInboxAppend_Mutation"
 	UNPIN_CONVERSATION_URL            = BASE_URL + "/i/api/graphql/_TQxP2Rb0expwVP9ktGrTQ/DMPinnedInboxDelete_Mutation"
 	GET_PINNED_CONVERSATIONS_URL      = BASE_URL + "/i/api/graphql/_gBQBgClVuMQb8efxWkbbQ/DMPinnedInboxQuery"

pkg/twittermeow/data/payload/form.go

@@ -49,6 +49,7 @@ const (
 )
 
 type DMRequestQuery struct {
+	AvatarID                            string      `url:"avatar_id,omitempty"`
 	ActiveConversationID                string      `url:"active_conversation_id,omitempty"`
 	Cursor                              string      `url:"cursor,omitempty"`
 	Count                               int         `url:"count,omitempty"`

pkg/twittermeow/data/types/event.go

@@ -22,6 +22,7 @@ func (*ConversationDelete) isTwitterEvent()         {}
 func (*ConversationJoin) isTwitterEvent()           {}
 func (*ParticipantsJoin) isTwitterEvent()           {}
 func (*ParticipantsLeave) isTwitterEvent()          {}
+func (*ConversationAvatarUpdate) isTwitterEvent()   {}
 func (*ConversationMetadataUpdate) isTwitterEvent() {}
 func (*ConversationNameUpdate) isTwitterEvent()     {}
 func (*ConversationRead) isTwitterEvent()           {}
@@ -52,6 +53,7 @@ type twitterEventContainer struct {
 	ConversationJoin           *ConversationJoin           `json:"join_conversation,omitempty"`
 	ParticipantsJoin           *ParticipantsJoin           `json:"participants_join,omitempty"`
 	ParticipantsLeave          *ParticipantsLeave          `json:"participants_leave,omitempty"`
+	ConversationAvatarUpdate   *ConversationAvatarUpdate   `json:"conversation_avatar_update,omitempty"`
 	ConversationMetadataUpdate *ConversationMetadataUpdate `json:"conversation_metadata_update,omitempty"`
 	ConversationNameUpdate     *ConversationNameUpdate     `json:"conversation_name_update,omitempty"`
 	ConversationRead           *ConversationRead           `json:"conversation_read,omitempty"`
@@ -86,6 +88,8 @@ func (rte *RawTwitterEvent) Parse() (TwitterEvent, map[string]any, error) {
 		return tec.ParticipantsJoin, nil, nil
 	case tec.ParticipantsLeave != nil:
 		return tec.ParticipantsLeave, nil, nil
+	case tec.ConversationAvatarUpdate != nil:
+		return tec.ConversationAvatarUpdate, nil, nil
 	case tec.ConversationMetadataUpdate != nil:
 		return tec.ConversationMetadataUpdate, nil, nil
 	case tec.ConversationNameUpdate != nil:

pkg/twittermeow/data/types/messaging.go

@@ -54,6 +54,16 @@ type ConversationRead struct {
 	LastReadEventID string `json:"last_read_event_id,omitempty"`
 }
 
+type ConversationAvatarUpdate struct {
+	ID                           string `json:"id,omitempty"`
+	Time                         string `json:"time,omitempty"`
+	AffectsSort                  bool   `json:"affects_sort,omitempty"`
+	ConversationID               string `json:"conversation_id,omitempty"`
+	ConversationAvatarImageHttps string `json:"conversation_avatar_image_https,omitempty"`
+	Avatar                       Avatar `json:"avatar,omitempty"`
+	ByUserID                     string `json:"by_user_id,omitempty"`
+}
+
 type ConversationMetadataUpdate struct {
 	ID             string `json:"id,omitempty"`
 	Time           string `json:"time,omitempty"`

pkg/twittermeow/messaging.go

@@ -348,3 +348,29 @@ func (c *Client) React(ctx context.Context, reactionPayload *payload.ReactionAct
 	data := response.ReactionResponse{}
 	return &data, json.Unmarshal(respBody, &data)
 }
+
+func (c *Client) UpdateConversationAvatar(ctx context.Context, conversationID string, payload *payload.DMRequestQuery) error {
+	encodedQueryBody, err := payload.Encode()
+	if err != nil {
+		return err
+	}
+
+	resp, respBody, err := c.makeAPIRequest(ctx, apiRequestOpts{
+		URL:            fmt.Sprintf(endpoints.UPDATE_CONVERSATION_AVATAR_URL, conversationID),
+		Method:         http.MethodPost,
+		WithClientUUID: true,
+		Body:           encodedQueryBody,
+		Referer:        endpoints.BASE_MESSAGES_URL,
+		Origin:         endpoints.BASE_URL,
+		ContentType:    types.ContentTypeForm,
+	})
+	if err != nil {
+		return err
+	}
+
+	if resp.StatusCode > 204 {
+		return fmt.Errorf("failed to update conversation avatar id=%s (status_code=%d, response_body=%s)", conversationID, resp.StatusCode, string(respBody))
+	}
+
+	return nil
+}