1 files changed, 419 insertions, 0 deletions
diff --git a/vendor/maunium.net/go/mautrix/appservice/intent.go b/vendor/maunium.net/go/mautrix/appservice/intent.go
new file mode 100644
index 00000000..af6fea37
--- /dev/null
+++ b/vendor/maunium.net/go/mautrix/appservice/intent.go
@@ -0,0 +1,419 @@
+// Copyright (c) 2020 Tulir Asokan
+//
+// This Source Code Form is subject to the terms of the Mozilla Public
+// License, v. 2.0. If a copy of the MPL was not distributed with this
+// file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+package appservice
+
+import (
+	"errors"
+	"fmt"
+	"strings"
+
+	"maunium.net/go/mautrix"
+	"maunium.net/go/mautrix/event"
+	"maunium.net/go/mautrix/id"
+)
+
+type IntentAPI struct {
+	*mautrix.Client
+	bot       *mautrix.Client
+	as        *AppService
+	Localpart string
+	UserID    id.UserID
+
+	IsCustomPuppet bool
+}
+
+func (as *AppService) NewIntentAPI(localpart string) *IntentAPI {
+	userID := id.NewUserID(localpart, as.HomeserverDomain)
+	bot := as.BotClient()
+	if userID == bot.UserID {
+		bot = nil
+	}
+	return &IntentAPI{
+		Client:    as.Client(userID),
+		bot:       bot,
+		as:        as,
+		Localpart: localpart,
+		UserID:    userID,
+
+		IsCustomPuppet: false,
+	}
+}
+
+func (intent *IntentAPI) Register() error {
+	_, _, err := intent.Client.Register(&mautrix.ReqRegister{
+		Username:     intent.Localpart,
+		Type:         mautrix.AuthTypeAppservice,
+		InhibitLogin: true,
+	})
+	return err
+}
+
+func (intent *IntentAPI) EnsureRegistered() error {
+	if intent.IsCustomPuppet || intent.as.StateStore.IsRegistered(intent.UserID) {
+		return nil
+	}
+
+	err := intent.Register()
+	if err != nil && !errors.Is(err, mautrix.MUserInUse) {
+		return fmt.Errorf("failed to ensure registered: %w", err)
+	}
+	intent.as.StateStore.MarkRegistered(intent.UserID)
+	return nil
+}
+
+type EnsureJoinedParams struct {
+	IgnoreCache bool
+	BotOverride *mautrix.Client
+}
+
+func (intent *IntentAPI) EnsureJoined(roomID id.RoomID, extra ...EnsureJoinedParams) error {
+	var params EnsureJoinedParams
+	if len(extra) > 1 {
+		panic("invalid number of extra parameters")
+	} else if len(extra) == 1 {
+		params = extra[0]
+	}
+	if intent.as.StateStore.IsInRoom(roomID, intent.UserID) && !params.IgnoreCache {
+		return nil
+	}
+
+	if err := intent.EnsureRegistered(); err != nil {
+		return fmt.Errorf("failed to ensure joined: %w", err)
+	}
+
+	resp, err := intent.JoinRoomByID(roomID)
+	if err != nil {
+		bot := intent.bot
+		if params.BotOverride != nil {
+			bot = params.BotOverride
+		}
+		if !errors.Is(err, mautrix.MForbidden) || bot == nil {
+			return fmt.Errorf("failed to ensure joined: %w", err)
+		}
+		_, inviteErr := bot.InviteUser(roomID, &mautrix.ReqInviteUser{
+			UserID: intent.UserID,
+		})
+		if inviteErr != nil {
+			return fmt.Errorf("failed to invite in ensure joined: %w", inviteErr)
+		}
+		resp, err = intent.JoinRoomByID(roomID)
+		if err != nil {
+			return fmt.Errorf("failed to ensure joined after invite: %w", err)
+		}
+	}
+	intent.as.StateStore.SetMembership(resp.RoomID, intent.UserID, event.MembershipJoin)
+	return nil
+}
+
+func (intent *IntentAPI) AddDoublePuppetValue(into interface{}) interface{} {
+	if !intent.IsCustomPuppet || intent.as.DoublePuppetValue == "" {
+		return into
+	}
+	switch val := into.(type) {
+	case *map[string]interface{}:
+		if *val == nil {
+			valNonPtr := make(map[string]interface{})
+			*val = valNonPtr
+		}
+		(*val)[DoublePuppetKey] = intent.as.DoublePuppetValue
+		return val
+	case map[string]interface{}:
+		val[DoublePuppetKey] = intent.as.DoublePuppetValue
+		return val
+	case *event.Content:
+		if val.Raw == nil {
+			val.Raw = make(map[string]interface{})
+		}
+		val.Raw[DoublePuppetKey] = intent.as.DoublePuppetValue
+		return val
+	case event.Content:
+		if val.Raw == nil {
+			val.Raw = make(map[string]interface{})
+		}
+		val.Raw[DoublePuppetKey] = intent.as.DoublePuppetValue
+		return val
+	default:
+		return &event.Content{
+			Raw: map[string]interface{}{
+				DoublePuppetKey: intent.as.DoublePuppetValue,
+			},
+			Parsed: val,
+		}
+	}
+}
+
+func (intent *IntentAPI) SendMessageEvent(roomID id.RoomID, eventType event.Type, contentJSON interface{}) (*mautrix.RespSendEvent, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	contentJSON = intent.AddDoublePuppetValue(contentJSON)
+	return intent.Client.SendMessageEvent(roomID, eventType, contentJSON)
+}
+
+func (intent *IntentAPI) SendMassagedMessageEvent(roomID id.RoomID, eventType event.Type, contentJSON interface{}, ts int64) (*mautrix.RespSendEvent, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	contentJSON = intent.AddDoublePuppetValue(contentJSON)
+	return intent.Client.SendMessageEvent(roomID, eventType, contentJSON, mautrix.ReqSendEvent{Timestamp: ts})
+}
+
+func (intent *IntentAPI) SendStateEvent(roomID id.RoomID, eventType event.Type, stateKey string, contentJSON interface{}) (*mautrix.RespSendEvent, error) {
+	if eventType != event.StateMember || stateKey != string(intent.UserID) {
+		if err := intent.EnsureJoined(roomID); err != nil {
+			return nil, err
+		}
+	}
+	contentJSON = intent.AddDoublePuppetValue(contentJSON)
+	return intent.Client.SendStateEvent(roomID, eventType, stateKey, contentJSON)
+}
+
+func (intent *IntentAPI) SendMassagedStateEvent(roomID id.RoomID, eventType event.Type, stateKey string, contentJSON interface{}, ts int64) (*mautrix.RespSendEvent, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	contentJSON = intent.AddDoublePuppetValue(contentJSON)
+	return intent.Client.SendMassagedStateEvent(roomID, eventType, stateKey, contentJSON, ts)
+}
+
+func (intent *IntentAPI) StateEvent(roomID id.RoomID, eventType event.Type, stateKey string, outContent interface{}) error {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return err
+	}
+	return intent.Client.StateEvent(roomID, eventType, stateKey, outContent)
+}
+
+func (intent *IntentAPI) State(roomID id.RoomID) (mautrix.RoomStateMap, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	return intent.Client.State(roomID)
+}
+
+func (intent *IntentAPI) SendCustomMembershipEvent(roomID id.RoomID, target id.UserID, membership event.Membership, reason string, extraContent ...map[string]interface{}) (*mautrix.RespSendEvent, error) {
+	content := &event.MemberEventContent{
+		Membership: membership,
+		Reason:     reason,
+	}
+	memberContent, ok := intent.as.StateStore.TryGetMember(roomID, target)
+	if !ok {
+		if intent.as.GetProfile != nil {
+			memberContent = intent.as.GetProfile(target, roomID)
+			ok = memberContent != nil
+		}
+		if !ok {
+			profile, err := intent.GetProfile(target)
+			if err != nil {
+				intent.Log.Debug().Err(err).
+					Str("target_user_id", target.String()).
+					Str("membership", string(membership)).
+					Msg("Failed to get profile to fill new membership event")
+			} else {
+				content.Displayname = profile.DisplayName
+				content.AvatarURL = profile.AvatarURL.CUString()
+			}
+		}
+	}
+	if ok && memberContent != nil {
+		content.Displayname = memberContent.Displayname
+		content.AvatarURL = memberContent.AvatarURL
+	}
+	var extra map[string]interface{}
+	if len(extraContent) > 0 {
+		extra = extraContent[0]
+	}
+	return intent.SendStateEvent(roomID, event.StateMember, target.String(), &event.Content{
+		Parsed: content,
+		Raw:    extra,
+	})
+}
+
+func (intent *IntentAPI) JoinRoomByID(roomID id.RoomID, extraContent ...map[string]interface{}) (resp *mautrix.RespJoinRoom, err error) {
+	if intent.IsCustomPuppet || len(extraContent) > 0 {
+		_, err = intent.SendCustomMembershipEvent(roomID, intent.UserID, event.MembershipJoin, "", extraContent...)
+		return &mautrix.RespJoinRoom{}, err
+	}
+	return intent.Client.JoinRoomByID(roomID)
+}
+
+func (intent *IntentAPI) LeaveRoom(roomID id.RoomID, extra ...interface{}) (resp *mautrix.RespLeaveRoom, err error) {
+	var extraContent map[string]interface{}
+	leaveReq := &mautrix.ReqLeave{}
+	for _, item := range extra {
+		switch val := item.(type) {
+		case map[string]interface{}:
+			extraContent = val
+		case *mautrix.ReqLeave:
+			leaveReq = val
+		}
+	}
+	if intent.IsCustomPuppet || extraContent != nil {
+		_, err = intent.SendCustomMembershipEvent(roomID, intent.UserID, event.MembershipLeave, leaveReq.Reason, extraContent)
+		return &mautrix.RespLeaveRoom{}, err
+	}
+	return intent.Client.LeaveRoom(roomID, leaveReq)
+}
+
+func (intent *IntentAPI) InviteUser(roomID id.RoomID, req *mautrix.ReqInviteUser, extraContent ...map[string]interface{}) (resp *mautrix.RespInviteUser, err error) {
+	if intent.IsCustomPuppet || len(extraContent) > 0 {
+		_, err = intent.SendCustomMembershipEvent(roomID, req.UserID, event.MembershipInvite, req.Reason, extraContent...)
+		return &mautrix.RespInviteUser{}, err
+	}
+	return intent.Client.InviteUser(roomID, req)
+}
+
+func (intent *IntentAPI) KickUser(roomID id.RoomID, req *mautrix.ReqKickUser, extraContent ...map[string]interface{}) (resp *mautrix.RespKickUser, err error) {
+	if intent.IsCustomPuppet || len(extraContent) > 0 {
+		_, err = intent.SendCustomMembershipEvent(roomID, req.UserID, event.MembershipLeave, req.Reason, extraContent...)
+		return &mautrix.RespKickUser{}, err
+	}
+	return intent.Client.KickUser(roomID, req)
+}
+
+func (intent *IntentAPI) BanUser(roomID id.RoomID, req *mautrix.ReqBanUser, extraContent ...map[string]interface{}) (resp *mautrix.RespBanUser, err error) {
+	if intent.IsCustomPuppet || len(extraContent) > 0 {
+		_, err = intent.SendCustomMembershipEvent(roomID, req.UserID, event.MembershipBan, req.Reason, extraContent...)
+		return &mautrix.RespBanUser{}, err
+	}
+	return intent.Client.BanUser(roomID, req)
+}
+
+func (intent *IntentAPI) UnbanUser(roomID id.RoomID, req *mautrix.ReqUnbanUser, extraContent ...map[string]interface{}) (resp *mautrix.RespUnbanUser, err error) {
+	if intent.IsCustomPuppet || len(extraContent) > 0 {
+		_, err = intent.SendCustomMembershipEvent(roomID, req.UserID, event.MembershipLeave, req.Reason, extraContent...)
+		return &mautrix.RespUnbanUser{}, err
+	}
+	return intent.Client.UnbanUser(roomID, req)
+}
+
+func (intent *IntentAPI) Member(roomID id.RoomID, userID id.UserID) *event.MemberEventContent {
+	member, ok := intent.as.StateStore.TryGetMember(roomID, userID)
+	if !ok {
+		_ = intent.StateEvent(roomID, event.StateMember, string(userID), &member)
+	}
+	return member
+}
+
+func (intent *IntentAPI) PowerLevels(roomID id.RoomID) (pl *event.PowerLevelsEventContent, err error) {
+	pl = intent.as.StateStore.GetPowerLevels(roomID)
+	if pl == nil {
+		pl = &event.PowerLevelsEventContent{}
+		err = intent.StateEvent(roomID, event.StatePowerLevels, "", pl)
+	}
+	return
+}
+
+func (intent *IntentAPI) SetPowerLevels(roomID id.RoomID, levels *event.PowerLevelsEventContent) (resp *mautrix.RespSendEvent, err error) {
+	return intent.SendStateEvent(roomID, event.StatePowerLevels, "", &levels)
+}
+
+func (intent *IntentAPI) SetPowerLevel(roomID id.RoomID, userID id.UserID, level int) (*mautrix.RespSendEvent, error) {
+	pl, err := intent.PowerLevels(roomID)
+	if err != nil {
+		return nil, err
+	}
+
+	if pl.GetUserLevel(userID) != level {
+		pl.SetUserLevel(userID, level)
+		return intent.SendStateEvent(roomID, event.StatePowerLevels, "", &pl)
+	}
+	return nil, nil
+}
+
+func (intent *IntentAPI) SendText(roomID id.RoomID, text string) (*mautrix.RespSendEvent, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	return intent.Client.SendText(roomID, text)
+}
+
+func (intent *IntentAPI) SendNotice(roomID id.RoomID, text string) (*mautrix.RespSendEvent, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	return intent.Client.SendNotice(roomID, text)
+}
+
+func (intent *IntentAPI) RedactEvent(roomID id.RoomID, eventID id.EventID, extra ...mautrix.ReqRedact) (*mautrix.RespSendEvent, error) {
+	if err := intent.EnsureJoined(roomID); err != nil {
+		return nil, err
+	}
+	var req mautrix.ReqRedact
+	if len(extra) > 0 {
+		req = extra[0]
+	}
+	intent.AddDoublePuppetValue(&req.Extra)
+	return intent.Client.RedactEvent(roomID, eventID, req)
+}
+
+func (intent *IntentAPI) SetRoomName(roomID id.RoomID, roomName string) (*mautrix.RespSendEvent, error) {
+	return intent.SendStateEvent(roomID, event.StateRoomName, "", map[string]interface{}{
+		"name": roomName,
+	})
+}
+
+func (intent *IntentAPI) SetRoomAvatar(roomID id.RoomID, avatarURL id.ContentURI) (*mautrix.RespSendEvent, error) {
+	return intent.SendStateEvent(roomID, event.StateRoomAvatar, "", map[string]interface{}{
+		"url": avatarURL.String(),
+	})
+}
+
+func (intent *IntentAPI) SetRoomTopic(roomID id.RoomID, topic string) (*mautrix.RespSendEvent, error) {
+	return intent.SendStateEvent(roomID, event.StateTopic, "", map[string]interface{}{
+		"topic": topic,
+	})
+}
+
+func (intent *IntentAPI) SetDisplayName(displayName string) error {
+	if err := intent.EnsureRegistered(); err != nil {
+		return err
+	}
+	resp, err := intent.Client.GetOwnDisplayName()
+	if err != nil {
+		return fmt.Errorf("failed to check current displayname: %w", err)
+	} else if resp.DisplayName == displayName {
+		// No need to update
+		return nil
+	}
+	return intent.Client.SetDisplayName(displayName)
+}
+
+func (intent *IntentAPI) SetAvatarURL(avatarURL id.ContentURI) error {
+	if err := intent.EnsureRegistered(); err != nil {
+		return err
+	}
+	resp, err := intent.Client.GetOwnAvatarURL()
+	if err != nil {
+		return fmt.Errorf("failed to check current avatar URL: %w", err)
+	} else if resp.FileID == avatarURL.FileID && resp.Homeserver == avatarURL.Homeserver {
+		// No need to update
+		return nil
+	}
+	return intent.Client.SetAvatarURL(avatarURL)
+}
+
+func (intent *IntentAPI) Whoami() (*mautrix.RespWhoami, error) {
+	if err := intent.EnsureRegistered(); err != nil {
+		return nil, err
+	}
+	return intent.Client.Whoami()
+}
+
+func (intent *IntentAPI) EnsureInvited(roomID id.RoomID, userID id.UserID) error {
+	if !intent.as.StateStore.IsInvited(roomID, userID) {
+		_, err := intent.InviteUser(roomID, &mautrix.ReqInviteUser{
+			UserID: userID,
+		})
+		if httpErr, ok := err.(mautrix.HTTPError); ok &&
+			httpErr.RespError != nil &&
+			(strings.Contains(httpErr.RespError.Err, "is already in the room") || strings.Contains(httpErr.RespError.Err, "is already joined to room")) {
+			return nil
+		}
+		return err
+	}
+	return nil
+}