forked from MirrorHub/mautrix-whatsapp
122 lines
4.5 KiB
Go
122 lines
4.5 KiB
Go
// mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
|
|
// Copyright (C) 2022 Tulir Asokan
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Affero General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Affero General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Affero General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package main
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"go.mau.fi/whatsmeow/types"
|
|
|
|
"maunium.net/go/mautrix"
|
|
"maunium.net/go/mautrix/bridge"
|
|
"maunium.net/go/mautrix/event"
|
|
"maunium.net/go/mautrix/format"
|
|
"maunium.net/go/mautrix/id"
|
|
|
|
"maunium.net/go/mautrix-whatsapp/database"
|
|
)
|
|
|
|
func (br *WABridge) CreatePrivatePortal(roomID id.RoomID, brInviter bridge.User, brGhost bridge.Ghost) {
|
|
inviter := brInviter.(*User)
|
|
puppet := brGhost.(*Puppet)
|
|
key := database.NewPortalKey(puppet.JID, inviter.JID)
|
|
portal := br.GetPortalByJID(key)
|
|
|
|
if len(portal.MXID) == 0 {
|
|
br.createPrivatePortalFromInvite(roomID, inviter, puppet, portal)
|
|
return
|
|
}
|
|
|
|
err := portal.MainIntent().EnsureInvited(portal.MXID, inviter.MXID)
|
|
if err != nil {
|
|
br.Log.Warnfln("Failed to invite %s to existing private chat portal %s with %s: %v. Redirecting portal to new room...", inviter.MXID, portal.MXID, puppet.JID, err)
|
|
br.createPrivatePortalFromInvite(roomID, inviter, puppet, portal)
|
|
return
|
|
}
|
|
intent := puppet.DefaultIntent()
|
|
errorMessage := fmt.Sprintf("You already have a private chat portal with me at [%[1]s](https://matrix.to/#/%[1]s)", portal.MXID)
|
|
errorContent := format.RenderMarkdown(errorMessage, true, false)
|
|
_, _ = intent.SendMessageEvent(roomID, event.EventMessage, errorContent)
|
|
br.Log.Debugfln("Leaving private chat room %s as %s after accepting invite from %s as we already have chat with the user", roomID, puppet.MXID, inviter.MXID)
|
|
_, _ = intent.LeaveRoom(roomID)
|
|
}
|
|
|
|
func (br *WABridge) createPrivatePortalFromInvite(roomID id.RoomID, inviter *User, puppet *Puppet, portal *Portal) {
|
|
portal.MXID = roomID
|
|
portal.Topic = PrivateChatTopic
|
|
_, _ = portal.MainIntent().SetRoomTopic(portal.MXID, portal.Topic)
|
|
if portal.bridge.Config.Bridge.PrivateChatPortalMeta {
|
|
portal.Name = puppet.Displayname
|
|
portal.AvatarURL = puppet.AvatarURL
|
|
portal.Avatar = puppet.Avatar
|
|
_, _ = portal.MainIntent().SetRoomName(portal.MXID, portal.Name)
|
|
_, _ = portal.MainIntent().SetRoomAvatar(portal.MXID, portal.AvatarURL)
|
|
} else {
|
|
portal.Name = ""
|
|
}
|
|
portal.log.Infofln("Created private chat portal in %s after invite from %s", roomID, inviter.MXID)
|
|
intent := puppet.DefaultIntent()
|
|
|
|
if br.Config.Bridge.Encryption.Default {
|
|
_, err := intent.InviteUser(roomID, &mautrix.ReqInviteUser{UserID: br.Bot.UserID})
|
|
if err != nil {
|
|
portal.log.Warnln("Failed to invite bridge bot to enable e2be:", err)
|
|
}
|
|
err = br.Bot.EnsureJoined(roomID)
|
|
if err != nil {
|
|
portal.log.Warnln("Failed to join as bridge bot to enable e2be:", err)
|
|
}
|
|
_, err = intent.SendStateEvent(roomID, event.StateEncryption, "", portal.GetEncryptionEventContent())
|
|
if err != nil {
|
|
portal.log.Warnln("Failed to enable e2be:", err)
|
|
}
|
|
br.AS.StateStore.SetMembership(roomID, inviter.MXID, event.MembershipJoin)
|
|
br.AS.StateStore.SetMembership(roomID, puppet.MXID, event.MembershipJoin)
|
|
br.AS.StateStore.SetMembership(roomID, br.Bot.UserID, event.MembershipJoin)
|
|
portal.Encrypted = true
|
|
}
|
|
portal.Update(nil)
|
|
portal.UpdateBridgeInfo()
|
|
_, _ = intent.SendNotice(roomID, "Private chat portal created")
|
|
}
|
|
|
|
func (br *WABridge) HandlePresence(evt *event.Event) {
|
|
user := br.GetUserByMXIDIfExists(evt.Sender)
|
|
if user == nil || !user.IsLoggedIn() {
|
|
return
|
|
}
|
|
customPuppet := br.GetPuppetByCustomMXID(user.MXID)
|
|
// TODO move this flag to the user and/or portal data
|
|
if customPuppet != nil && !customPuppet.EnablePresence {
|
|
return
|
|
}
|
|
|
|
presence := types.PresenceAvailable
|
|
if evt.Content.AsPresence().Presence != event.PresenceOnline {
|
|
presence = types.PresenceUnavailable
|
|
user.log.Debugln("Marking offline")
|
|
} else {
|
|
user.log.Debugln("Marking online")
|
|
}
|
|
user.lastPresence = presence
|
|
if user.Client.Store.PushName != "" {
|
|
err := user.Client.SendPresence(presence)
|
|
if err != nil {
|
|
user.log.Warnln("Failed to set presence:", err)
|
|
}
|
|
}
|
|
}
|