123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960 |
- // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
- // Copyright (C) 2021 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 (
- "context"
- "encoding/json"
- "errors"
- "fmt"
- "net/http"
- "strconv"
- "strings"
- "sync"
- "time"
- log "maunium.net/go/maulogger/v2"
- "maunium.net/go/mautrix"
- "maunium.net/go/mautrix/appservice"
- "maunium.net/go/mautrix/event"
- "maunium.net/go/mautrix/format"
- "maunium.net/go/mautrix/id"
- "maunium.net/go/mautrix/pushrules"
- "go.mau.fi/whatsmeow"
- "go.mau.fi/whatsmeow/appstate"
- "go.mau.fi/whatsmeow/store"
- "go.mau.fi/whatsmeow/types"
- "go.mau.fi/whatsmeow/types/events"
- waLog "go.mau.fi/whatsmeow/util/log"
- "maunium.net/go/mautrix-whatsapp/database"
- )
- type User struct {
- *database.User
- Client *whatsmeow.Client
- Session *store.Device
- bridge *Bridge
- log log.Logger
- Admin bool
- Whitelisted bool
- RelayWhitelisted bool
- mgmtCreateLock sync.Mutex
- spaceCreateLock sync.Mutex
- connLock sync.Mutex
- historySyncs chan *events.HistorySync
- prevBridgeStatus *BridgeState
- lastPresence types.Presence
- spaceMembershipChecked bool
- }
- func (bridge *Bridge) getUserByMXID(userID id.UserID, onlyIfExists bool) *User {
- _, isPuppet := bridge.ParsePuppetMXID(userID)
- if isPuppet || userID == bridge.Bot.UserID {
- return nil
- }
- bridge.usersLock.Lock()
- defer bridge.usersLock.Unlock()
- user, ok := bridge.usersByMXID[userID]
- if !ok {
- userIDPtr := &userID
- if onlyIfExists {
- userIDPtr = nil
- }
- return bridge.loadDBUser(bridge.DB.User.GetByMXID(userID), userIDPtr)
- }
- return user
- }
- func (bridge *Bridge) GetUserByMXID(userID id.UserID) *User {
- return bridge.getUserByMXID(userID, false)
- }
- func (bridge *Bridge) GetUserByMXIDIfExists(userID id.UserID) *User {
- return bridge.getUserByMXID(userID, true)
- }
- func (bridge *Bridge) GetUserByJID(jid types.JID) *User {
- bridge.usersLock.Lock()
- defer bridge.usersLock.Unlock()
- user, ok := bridge.usersByUsername[jid.User]
- if !ok {
- return bridge.loadDBUser(bridge.DB.User.GetByUsername(jid.User), nil)
- }
- return user
- }
- func (user *User) addToJIDMap() {
- user.bridge.usersLock.Lock()
- user.bridge.usersByUsername[user.JID.User] = user
- user.bridge.usersLock.Unlock()
- }
- func (user *User) removeFromJIDMap(state BridgeStateEvent) {
- user.bridge.usersLock.Lock()
- jidUser, ok := user.bridge.usersByUsername[user.JID.User]
- if ok && user == jidUser {
- delete(user.bridge.usersByUsername, user.JID.User)
- }
- user.bridge.usersLock.Unlock()
- user.bridge.Metrics.TrackLoginState(user.JID, false)
- user.sendBridgeState(BridgeState{StateEvent: state, Error: WANotLoggedIn})
- }
- func (bridge *Bridge) GetAllUsers() []*User {
- bridge.usersLock.Lock()
- defer bridge.usersLock.Unlock()
- dbUsers := bridge.DB.User.GetAll()
- output := make([]*User, len(dbUsers))
- for index, dbUser := range dbUsers {
- user, ok := bridge.usersByMXID[dbUser.MXID]
- if !ok {
- user = bridge.loadDBUser(dbUser, nil)
- }
- output[index] = user
- }
- return output
- }
- func (bridge *Bridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User {
- if dbUser == nil {
- if mxid == nil {
- return nil
- }
- dbUser = bridge.DB.User.New()
- dbUser.MXID = *mxid
- dbUser.Insert()
- }
- user := bridge.NewUser(dbUser)
- bridge.usersByMXID[user.MXID] = user
- if !user.JID.IsEmpty() {
- var err error
- user.Session, err = bridge.WAContainer.GetDevice(user.JID)
- if err != nil {
- user.log.Errorfln("Failed to load user's whatsapp session: %v", err)
- } else if user.Session == nil {
- user.log.Warnfln("Didn't find session data for %s, treating user as logged out", user.JID)
- user.JID = types.EmptyJID
- user.Update()
- } else {
- bridge.usersByUsername[user.JID.User] = user
- }
- }
- if len(user.ManagementRoom) > 0 {
- bridge.managementRooms[user.ManagementRoom] = user
- }
- return user
- }
- func (bridge *Bridge) NewUser(dbUser *database.User) *User {
- user := &User{
- User: dbUser,
- bridge: bridge,
- log: bridge.Log.Sub("User").Sub(string(dbUser.MXID)),
- historySyncs: make(chan *events.HistorySync, 32),
- lastPresence: types.PresenceUnavailable,
- }
- user.RelayWhitelisted = user.bridge.Config.Bridge.Permissions.IsRelayWhitelisted(user.MXID)
- user.Whitelisted = user.bridge.Config.Bridge.Permissions.IsWhitelisted(user.MXID)
- user.Admin = user.bridge.Config.Bridge.Permissions.IsAdmin(user.MXID)
- go user.handleHistorySyncsLoop()
- return user
- }
- func (user *User) ensureInvited(intent *appservice.IntentAPI, roomID id.RoomID, isDirect bool) (ok bool) {
- inviteContent := event.Content{
- Parsed: &event.MemberEventContent{
- Membership: event.MembershipInvite,
- IsDirect: isDirect,
- },
- Raw: map[string]interface{}{},
- }
- customPuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
- if customPuppet != nil && customPuppet.CustomIntent() != nil {
- inviteContent.Raw["fi.mau.will_auto_accept"] = true
- }
- _, err := intent.SendStateEvent(roomID, event.StateMember, user.MXID.String(), &inviteContent)
- var httpErr mautrix.HTTPError
- if err != nil && errors.As(err, &httpErr) && httpErr.RespError != nil && strings.Contains(httpErr.RespError.Err, "is already in the room") {
- user.bridge.StateStore.SetMembership(roomID, user.MXID, event.MembershipJoin)
- ok = true
- } else if err != nil {
- user.log.Warnfln("Failed to invite user to %s: %v", roomID, err)
- } else {
- ok = true
- }
- if customPuppet != nil && customPuppet.CustomIntent() != nil {
- err = customPuppet.CustomIntent().EnsureJoined(roomID)
- if err != nil {
- user.log.Warnfln("Failed to auto-join %s: %v", roomID, err)
- ok = false
- } else {
- ok = true
- }
- }
- return
- }
- func (user *User) GetSpaceRoom() id.RoomID {
- if !user.bridge.Config.Bridge.PersonalFilteringSpaces {
- return ""
- }
- if len(user.SpaceRoom) == 0 {
- user.spaceCreateLock.Lock()
- defer user.spaceCreateLock.Unlock()
- if len(user.SpaceRoom) > 0 {
- return user.SpaceRoom
- }
- resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
- Visibility: "private",
- Name: "WhatsApp",
- Topic: "Your WhatsApp bridged chats",
- InitialState: []*event.Event{{
- Type: event.StateRoomAvatar,
- Content: event.Content{
- Parsed: &event.RoomAvatarEventContent{
- URL: user.bridge.Config.AppService.Bot.ParsedAvatar,
- },
- },
- }},
- CreationContent: map[string]interface{}{
- "type": event.RoomTypeSpace,
- },
- PowerLevelOverride: &event.PowerLevelsEventContent{
- Users: map[id.UserID]int{
- user.bridge.Bot.UserID: 9001,
- user.MXID: 50,
- },
- },
- })
- if err != nil {
- user.log.Errorln("Failed to auto-create space room:", err)
- } else {
- user.SpaceRoom = resp.RoomID
- user.Update()
- user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false)
- }
- } else if !user.spaceMembershipChecked && !user.bridge.StateStore.IsInRoom(user.SpaceRoom, user.MXID) {
- user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false)
- }
- user.spaceMembershipChecked = true
- return user.SpaceRoom
- }
- func (user *User) GetManagementRoom() id.RoomID {
- if len(user.ManagementRoom) == 0 {
- user.mgmtCreateLock.Lock()
- defer user.mgmtCreateLock.Unlock()
- if len(user.ManagementRoom) > 0 {
- return user.ManagementRoom
- }
- creationContent := make(map[string]interface{})
- if !user.bridge.Config.Bridge.FederateRooms {
- creationContent["m.federate"] = false
- }
- resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
- Topic: "WhatsApp bridge notices",
- IsDirect: true,
- CreationContent: creationContent,
- })
- if err != nil {
- user.log.Errorln("Failed to auto-create management room:", err)
- } else {
- user.SetManagementRoom(resp.RoomID)
- }
- }
- return user.ManagementRoom
- }
- func (user *User) SetManagementRoom(roomID id.RoomID) {
- existingUser, ok := user.bridge.managementRooms[roomID]
- if ok {
- existingUser.ManagementRoom = ""
- existingUser.Update()
- }
- user.ManagementRoom = roomID
- user.bridge.managementRooms[user.ManagementRoom] = user
- user.Update()
- }
- type waLogger struct{ l log.Logger }
- func (w *waLogger) Debugf(msg string, args ...interface{}) { w.l.Debugfln(msg, args...) }
- func (w *waLogger) Infof(msg string, args ...interface{}) { w.l.Infofln(msg, args...) }
- func (w *waLogger) Warnf(msg string, args ...interface{}) { w.l.Warnfln(msg, args...) }
- func (w *waLogger) Errorf(msg string, args ...interface{}) { w.l.Errorfln(msg, args...) }
- func (w *waLogger) Sub(module string) waLog.Logger { return &waLogger{l: w.l.Sub(module)} }
- var ErrAlreadyLoggedIn = errors.New("already logged in")
- func (user *User) Login(ctx context.Context) (<-chan whatsmeow.QRChannelItem, error) {
- user.connLock.Lock()
- defer user.connLock.Unlock()
- if user.Session != nil {
- return nil, ErrAlreadyLoggedIn
- } else if user.Client != nil {
- user.unlockedDeleteConnection()
- }
- newSession := user.bridge.WAContainer.NewDevice()
- newSession.Log = &waLogger{user.log.Sub("Session")}
- user.Client = whatsmeow.NewClient(newSession, &waLogger{user.log.Sub("Client")})
- user.Client.AddEventHandler(user.HandleEvent)
- qrChan, err := user.Client.GetQRChannel(ctx)
- if err != nil {
- return nil, fmt.Errorf("failed to get QR channel: %w", err)
- }
- err = user.Client.Connect()
- if err != nil {
- return nil, fmt.Errorf("failed to connect to WhatsApp: %w", err)
- }
- return qrChan, nil
- }
- func (user *User) Connect() bool {
- user.connLock.Lock()
- defer user.connLock.Unlock()
- if user.Client != nil {
- return user.Client.IsConnected()
- } else if user.Session == nil {
- return false
- }
- user.log.Debugln("Connecting to WhatsApp")
- user.sendBridgeState(BridgeState{StateEvent: StateConnecting, Error: WAConnecting})
- user.Client = whatsmeow.NewClient(user.Session, &waLogger{user.log.Sub("Client")})
- user.Client.AddEventHandler(user.HandleEvent)
- err := user.Client.Connect()
- if err != nil {
- user.log.Warnln("Error connecting to WhatsApp:", err)
- return false
- }
- return true
- }
- func (user *User) unlockedDeleteConnection() {
- if user.Client == nil {
- return
- }
- user.Client.Disconnect()
- user.Client.RemoveEventHandlers()
- user.Client = nil
- user.bridge.Metrics.TrackConnectionState(user.JID, false)
- }
- func (user *User) DeleteConnection() {
- user.connLock.Lock()
- defer user.connLock.Unlock()
- user.unlockedDeleteConnection()
- }
- func (user *User) HasSession() bool {
- return user.Session != nil
- }
- func (user *User) DeleteSession() {
- if user.Session != nil {
- err := user.Session.Delete()
- if err != nil {
- user.log.Warnln("Failed to delete session:", err)
- }
- user.Session = nil
- }
- if !user.JID.IsEmpty() {
- user.JID = types.EmptyJID
- user.Update()
- }
- }
- func (user *User) IsConnected() bool {
- return user.Client != nil && user.Client.IsConnected()
- }
- func (user *User) IsLoggedIn() bool {
- return user.IsConnected() && user.Client.IsLoggedIn()
- }
- func (user *User) tryAutomaticDoublePuppeting() {
- if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) {
- return
- }
- user.log.Debugln("Checking if double puppeting needs to be enabled")
- puppet := user.bridge.GetPuppetByJID(user.JID)
- if len(puppet.CustomMXID) > 0 {
- user.log.Debugln("User already has double-puppeting enabled")
- // Custom puppet already enabled
- return
- }
- accessToken, err := puppet.loginWithSharedSecret(user.MXID)
- if err != nil {
- user.log.Warnln("Failed to login with shared secret:", err)
- return
- }
- err = puppet.SwitchCustomMXID(accessToken, user.MXID)
- if err != nil {
- puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
- return
- }
- user.log.Infoln("Successfully automatically enabled custom puppet")
- }
- func (user *User) sendBridgeNotice(formatString string, args ...interface{}) {
- notice := fmt.Sprintf(formatString, args...)
- _, err := user.bridge.Bot.SendNotice(user.GetManagementRoom(), notice)
- if err != nil {
- user.log.Warnf("Failed to send bridge notice \"%s\": %v", notice, err)
- }
- }
- func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) {
- notice := fmt.Sprintf(formatString, args...)
- content := format.RenderMarkdown(notice, true, false)
- _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content)
- if err != nil {
- user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err)
- }
- }
- const callEventMaxAge = 15 * time.Minute
- func (user *User) handleCallStart(sender types.JID, id, callType string, ts time.Time) {
- if !user.bridge.Config.Bridge.CallStartNotices || ts.Add(callEventMaxAge).Before(time.Now()) {
- return
- }
- portal := user.GetPortalByJID(sender)
- text := "Incoming call"
- if callType != "" {
- text = fmt.Sprintf("Incoming %s call", callType)
- }
- portal.messages <- PortalMessage{
- fake: &fakeMessage{
- Sender: sender,
- Text: text,
- ID: id,
- Time: ts,
- Important: true,
- },
- source: user,
- }
- }
- func (user *User) HandleEvent(event interface{}) {
- switch v := event.(type) {
- case *events.LoggedOut:
- go user.handleLoggedOut(v.OnConnect)
- user.bridge.Metrics.TrackConnectionState(user.JID, false)
- user.bridge.Metrics.TrackLoginState(user.JID, false)
- case *events.Connected:
- go user.sendBridgeState(BridgeState{StateEvent: StateConnected})
- user.bridge.Metrics.TrackConnectionState(user.JID, true)
- user.bridge.Metrics.TrackLoginState(user.JID, true)
- if len(user.Client.Store.PushName) > 0 {
- go func() {
- err := user.Client.SendPresence(user.lastPresence)
- if err != nil {
- user.log.Warnln("Failed to send initial presence:", err)
- }
- }()
- }
- go user.tryAutomaticDoublePuppeting()
- case *events.AppStateSyncComplete:
- if len(user.Client.Store.PushName) > 0 && v.Name == appstate.WAPatchCriticalBlock {
- err := user.Client.SendPresence(user.lastPresence)
- if err != nil {
- user.log.Warnln("Failed to send presence after app state sync:", err)
- }
- } else if v.Name == appstate.WAPatchCriticalUnblockLow {
- go func() {
- err := user.ResyncContacts()
- if err != nil {
- user.log.Errorln("Failed to resync puppets: %v", err)
- }
- }()
- }
- case *events.PushNameSetting:
- // Send presence available when connecting and when the pushname is changed.
- // This makes sure that outgoing messages always have the right pushname.
- err := user.Client.SendPresence(user.lastPresence)
- if err != nil {
- user.log.Warnln("Failed to send presence after push name update:", err)
- }
- case *events.PairSuccess:
- user.Session = user.Client.Store
- user.JID = v.ID
- user.addToJIDMap()
- user.Update()
- case *events.ConnectFailure, *events.StreamError:
- go user.sendBridgeState(BridgeState{StateEvent: StateUnknownError})
- user.bridge.Metrics.TrackConnectionState(user.JID, false)
- case *events.Disconnected:
- go user.sendBridgeState(BridgeState{StateEvent: StateTransientDisconnect})
- user.bridge.Metrics.TrackConnectionState(user.JID, false)
- case *events.Contact:
- go user.syncPuppet(v.JID, "contact event")
- case *events.PushName:
- go user.syncPuppet(v.JID, "push name event")
- case *events.GroupInfo:
- go user.handleGroupUpdate(v)
- case *events.JoinedGroup:
- go user.handleGroupCreate(v)
- case *events.Picture:
- go user.handlePictureUpdate(v)
- case *events.Receipt:
- go user.handleReceipt(v)
- case *events.ChatPresence:
- go user.handleChatPresence(v)
- case *events.Message:
- portal := user.GetPortalByMessageSource(v.Info.MessageSource)
- portal.messages <- PortalMessage{evt: v, source: user}
- case *events.CallOffer:
- user.handleCallStart(v.CallCreator, v.CallID, "", v.Timestamp)
- case *events.CallOfferNotice:
- user.handleCallStart(v.CallCreator, v.CallID, v.Type, v.Timestamp)
- case *events.IdentityChange:
- puppet := user.bridge.GetPuppetByJID(v.JID)
- portal := user.GetPortalByJID(v.JID)
- if len(portal.MXID) > 0 && user.bridge.Config.Bridge.IdentityChangeNotices {
- text := fmt.Sprintf("Your security code with %s changed.", puppet.Displayname)
- if v.Implicit {
- text = fmt.Sprintf("Your security code with %s (device #%d) changed.", puppet.Displayname, v.JID.Device)
- }
- portal.messages <- PortalMessage{
- fake: &fakeMessage{
- Sender: v.JID,
- Text: text,
- ID: strconv.FormatInt(v.Timestamp.Unix(), 10),
- Time: v.Timestamp,
- Important: false,
- },
- source: user,
- }
- }
- case *events.CallTerminate, *events.CallRelayLatency, *events.CallAccept, *events.UnknownCallEvent:
- // ignore
- case *events.UndecryptableMessage:
- portal := user.GetPortalByMessageSource(v.Info.MessageSource)
- portal.messages <- PortalMessage{undecryptable: v, source: user}
- case *events.HistorySync:
- user.historySyncs <- v
- case *events.Mute:
- portal := user.GetPortalByJID(v.JID)
- if portal != nil {
- var mutedUntil time.Time
- if v.Action.GetMuted() {
- mutedUntil = time.Unix(v.Action.GetMuteEndTimestamp(), 0)
- }
- go user.updateChatMute(nil, portal, mutedUntil)
- }
- case *events.Archive:
- portal := user.GetPortalByJID(v.JID)
- if portal != nil {
- go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.ArchiveTag, v.Action.GetArchived())
- }
- case *events.Pin:
- portal := user.GetPortalByJID(v.JID)
- if portal != nil {
- go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.PinnedTag, v.Action.GetPinned())
- }
- case *events.AppState:
- // Ignore
- default:
- user.log.Debugfln("Unknown type of event in HandleEvent: %T", v)
- }
- }
- func (user *User) updateChatMute(intent *appservice.IntentAPI, portal *Portal, mutedUntil time.Time) {
- if len(portal.MXID) == 0 || !user.bridge.Config.Bridge.MuteBridging {
- return
- } else if intent == nil {
- doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
- if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
- return
- }
- intent = doublePuppet.CustomIntent()
- }
- var err error
- if mutedUntil.IsZero() && mutedUntil.Before(time.Now()) {
- user.log.Debugfln("Portal %s is muted until %s, unmuting...", portal.MXID, mutedUntil)
- err = intent.DeletePushRule("global", pushrules.RoomRule, string(portal.MXID))
- } else {
- user.log.Debugfln("Portal %s is muted until %s, muting...", portal.MXID, mutedUntil)
- err = intent.PutPushRule("global", pushrules.RoomRule, string(portal.MXID), &mautrix.ReqPutPushRule{
- Actions: []pushrules.PushActionType{pushrules.ActionDontNotify},
- })
- }
- if err != nil && !errors.Is(err, mautrix.MNotFound) {
- user.log.Warnfln("Failed to update push rule for %s through double puppet: %v", portal.MXID, err)
- }
- }
- type CustomTagData struct {
- Order json.Number `json:"order"`
- DoublePuppet string `json:"fi.mau.double_puppet_source"`
- }
- type CustomTagEventContent struct {
- Tags map[string]CustomTagData `json:"tags"`
- }
- func (user *User) updateChatTag(intent *appservice.IntentAPI, portal *Portal, tag string, active bool) {
- if len(portal.MXID) == 0 || len(tag) == 0 {
- return
- } else if intent == nil {
- doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
- if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
- return
- }
- intent = doublePuppet.CustomIntent()
- }
- var existingTags CustomTagEventContent
- err := intent.GetTagsWithCustomData(portal.MXID, &existingTags)
- if err != nil && !errors.Is(err, mautrix.MNotFound) {
- user.log.Warnfln("Failed to get tags of %s: %v", portal.MXID, err)
- }
- currentTag, ok := existingTags.Tags[tag]
- if active && !ok {
- user.log.Debugln("Adding tag", tag, "to", portal.MXID)
- data := CustomTagData{"0.5", doublePuppetValue}
- err = intent.AddTagWithCustomData(portal.MXID, tag, &data)
- } else if !active && ok && currentTag.DoublePuppet == doublePuppetValue {
- user.log.Debugln("Removing tag", tag, "from", portal.MXID)
- err = intent.RemoveTag(portal.MXID, tag)
- } else {
- err = nil
- }
- if err != nil {
- user.log.Warnfln("Failed to update tag %s for %s through double puppet: %v", tag, portal.MXID, err)
- }
- }
- type CustomReadReceipt struct {
- Timestamp int64 `json:"ts,omitempty"`
- DoublePuppetSource string `json:"fi.mau.double_puppet_source,omitempty"`
- }
- func (user *User) syncChatDoublePuppetDetails(portal *Portal, justCreated bool) {
- doublePuppet := portal.bridge.GetPuppetByCustomMXID(user.MXID)
- if doublePuppet == nil {
- return
- }
- if doublePuppet == nil || doublePuppet.CustomIntent() == nil || len(portal.MXID) == 0 {
- return
- }
- if justCreated || !user.bridge.Config.Bridge.TagOnlyOnCreate {
- chat, err := user.Client.Store.ChatSettings.GetChatSettings(portal.Key.JID)
- if err != nil {
- user.log.Warnfln("Failed to get settings of %s: %v", portal.Key.JID, err)
- return
- }
- intent := doublePuppet.CustomIntent()
- if portal.Key.JID == types.StatusBroadcastJID && justCreated && user.bridge.Config.Bridge.MuteStatusBroadcast {
- user.updateChatMute(intent, portal, time.Now().Add(365*24*time.Hour))
- user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, true)
- return
- } else if !chat.Found {
- return
- }
- user.updateChatMute(intent, portal, chat.MutedUntil)
- user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, chat.Archived)
- user.updateChatTag(intent, portal, user.bridge.Config.Bridge.PinnedTag, chat.Pinned)
- }
- }
- func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
- res := make(map[id.UserID][]id.RoomID)
- privateChats := user.bridge.DB.Portal.FindPrivateChats(user.JID.ToNonAD())
- for _, portal := range privateChats {
- if len(portal.MXID) > 0 {
- res[user.bridge.FormatPuppetMXID(portal.Key.JID)] = []id.RoomID{portal.MXID}
- }
- }
- return res
- }
- func (user *User) UpdateDirectChats(chats map[id.UserID][]id.RoomID) {
- if !user.bridge.Config.Bridge.SyncDirectChatList {
- return
- }
- puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
- if puppet == nil || puppet.CustomIntent() == nil {
- return
- }
- intent := puppet.CustomIntent()
- method := http.MethodPatch
- if chats == nil {
- chats = user.getDirectChats()
- method = http.MethodPut
- }
- user.log.Debugln("Updating m.direct list on homeserver")
- var err error
- if user.bridge.Config.Homeserver.Asmux {
- urlPath := intent.BuildBaseURL("_matrix", "client", "unstable", "com.beeper.asmux", "dms")
- _, err = intent.MakeFullRequest(mautrix.FullRequest{
- Method: method,
- URL: urlPath,
- Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
- RequestJSON: chats,
- })
- } else {
- existingChats := make(map[id.UserID][]id.RoomID)
- err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
- if err != nil {
- user.log.Warnln("Failed to get m.direct list to update it:", err)
- return
- }
- for userID, rooms := range existingChats {
- if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
- // This is not a ghost user, include it in the new list
- chats[userID] = rooms
- } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
- // This is a ghost user, but we're not replacing the whole list, so include it too
- chats[userID] = rooms
- }
- }
- err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
- }
- if err != nil {
- user.log.Warnln("Failed to update m.direct list:", err)
- }
- }
- func (user *User) handleLoggedOut(onConnect bool) {
- user.JID = types.EmptyJID
- user.Update()
- if onConnect {
- user.sendMarkdownBridgeAlert("Connecting to WhatsApp failed as the device was logged out. Please link the bridge to your phone again.")
- } else {
- user.sendMarkdownBridgeAlert("You were logged out from another device. Please link the bridge to your phone again.")
- }
- user.sendBridgeState(BridgeState{StateEvent: StateBadCredentials, Error: WANotLoggedIn})
- }
- func (user *User) GetPortalByMessageSource(ms types.MessageSource) *Portal {
- jid := ms.Chat
- if ms.IsIncomingBroadcast() {
- if ms.IsFromMe {
- jid = ms.BroadcastListOwner.ToNonAD()
- } else {
- jid = ms.Sender.ToNonAD()
- }
- if jid.IsEmpty() {
- return nil
- }
- }
- return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
- }
- func (user *User) GetPortalByJID(jid types.JID) *Portal {
- return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
- }
- func (user *User) syncPuppet(jid types.JID, reason string) {
- user.bridge.GetPuppetByJID(jid).SyncContact(user, false, reason)
- }
- func (user *User) ResyncContacts() error {
- contacts, err := user.Client.Store.Contacts.GetAllContacts()
- if err != nil {
- return fmt.Errorf("failed to get cached contacts: %w", err)
- }
- user.log.Infofln("Resyncing displaynames with %d contacts", len(contacts))
- for jid, contact := range contacts {
- puppet := user.bridge.GetPuppetByJID(jid)
- puppet.Sync(user, contact)
- }
- return nil
- }
- func (user *User) ResyncGroups(createPortals bool) error {
- groups, err := user.Client.GetJoinedGroups()
- if err != nil {
- return fmt.Errorf("failed to get group list from server: %w", err)
- }
- for _, group := range groups {
- portal := user.GetPortalByJID(group.JID)
- if len(portal.MXID) == 0 {
- if createPortals {
- err = portal.CreateMatrixRoom(user, group, true)
- if err != nil {
- return fmt.Errorf("failed to create room for %s: %w", group.JID, err)
- }
- }
- } else {
- portal.UpdateMatrixRoom(user, group)
- }
- }
- return nil
- }
- const WATypingTimeout = 15 * time.Second
- func (user *User) handleChatPresence(presence *events.ChatPresence) {
- puppet := user.bridge.GetPuppetByJID(presence.Sender)
- portal := user.GetPortalByJID(presence.Chat)
- if puppet == nil || portal == nil || len(portal.MXID) == 0 {
- return
- }
- if presence.State == types.ChatPresenceComposing {
- if puppet.typingIn != "" && puppet.typingAt.Add(WATypingTimeout).Before(time.Now()) {
- if puppet.typingIn == portal.MXID {
- return
- }
- _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
- }
- _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, WATypingTimeout.Milliseconds())
- puppet.typingIn = portal.MXID
- puppet.typingAt = time.Now()
- } else {
- _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, false, 0)
- puppet.typingIn = ""
- }
- }
- func (user *User) handleReceipt(receipt *events.Receipt) {
- if receipt.Type != events.ReceiptTypeRead && receipt.Type != events.ReceiptTypeReadSelf {
- return
- }
- portal := user.GetPortalByMessageSource(receipt.MessageSource)
- if portal == nil || len(portal.MXID) == 0 {
- return
- }
- // The order of the message ID array depends on the sender's platform, so we just have to find
- // the last message based on timestamp. Also, timestamps only have second precision, so if
- // there are many messages at the same second just mark them all as read, because we don't
- // know which one is last
- markAsRead := make([]*database.Message, 0, 1)
- var bestTimestamp time.Time
- for _, msgID := range receipt.MessageIDs {
- msg := user.bridge.DB.Message.GetByJID(portal.Key, msgID)
- if msg == nil || msg.IsFakeMXID() {
- continue
- }
- if msg.Timestamp.After(bestTimestamp) {
- bestTimestamp = msg.Timestamp
- markAsRead = append(markAsRead[:0], msg)
- } else if msg != nil && msg.Timestamp.Equal(bestTimestamp) {
- markAsRead = append(markAsRead, msg)
- }
- }
- if receipt.Sender.User == user.JID.User {
- if len(markAsRead) > 0 {
- user.SetLastReadTS(portal.Key, markAsRead[0].Timestamp)
- } else {
- user.SetLastReadTS(portal.Key, receipt.Timestamp)
- }
- }
- intent := user.bridge.GetPuppetByJID(receipt.Sender).IntentFor(portal)
- var rrContent CustomReadReceipt
- if intent.IsCustomPuppet {
- rrContent.DoublePuppetSource = doublePuppetValue
- }
- for _, msg := range markAsRead {
- err := intent.MarkReadWithContent(portal.MXID, msg.MXID, &rrContent)
- if err != nil {
- user.log.Warnfln("Failed to mark message %s as read by %s: %v", msg.MXID, intent.UserID, err)
- } else {
- user.log.Debugfln("Marked %s as read by %s", msg.MXID, intent.UserID)
- }
- }
- }
- func (user *User) markSelfReadFull(portal *Portal) {
- puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
- if puppet == nil || puppet.CustomIntent() == nil {
- return
- }
- lastMessage := user.bridge.DB.Message.GetLastInChat(portal.Key)
- if lastMessage == nil {
- return
- }
- user.SetLastReadTS(portal.Key, lastMessage.Timestamp)
- err := puppet.CustomIntent().MarkReadWithContent(portal.MXID, lastMessage.MXID, &CustomReadReceipt{DoublePuppetSource: doublePuppetValue})
- if err != nil {
- user.log.Warnfln("Failed to mark %s (last message) in %s as read: %v", lastMessage.MXID, portal.MXID, err)
- } else {
- user.log.Debugfln("Marked %s (last message) in %s as read", lastMessage.MXID, portal.MXID)
- }
- }
- func (user *User) handleGroupCreate(evt *events.JoinedGroup) {
- portal := user.GetPortalByJID(evt.JID)
- if len(portal.MXID) == 0 {
- err := portal.CreateMatrixRoom(user, &evt.GroupInfo, true)
- if err != nil {
- user.log.Errorln("Failed to create Matrix room after join notification: %v", err)
- }
- } else {
- portal.UpdateMatrixRoom(user, &evt.GroupInfo)
- }
- }
- func (user *User) handleGroupUpdate(evt *events.GroupInfo) {
- portal := user.GetPortalByJID(evt.JID)
- if portal == nil || len(portal.MXID) == 0 {
- user.log.Debugfln("Ignoring group info update in chat with no portal: %+v", evt)
- return
- }
- switch {
- case evt.Announce != nil:
- portal.RestrictMessageSending(evt.Announce.IsAnnounce)
- case evt.Locked != nil:
- portal.RestrictMetadataChanges(evt.Locked.IsLocked)
- case evt.Name != nil:
- portal.UpdateName(evt.Name.Name, evt.Name.NameSetBy, true)
- case evt.Topic != nil:
- portal.UpdateTopic(evt.Topic.Topic, evt.Topic.TopicSetBy, true)
- case evt.Leave != nil:
- if evt.Sender != nil && !evt.Sender.IsEmpty() {
- portal.HandleWhatsAppKick(user, *evt.Sender, evt.Leave)
- }
- case evt.Join != nil:
- portal.HandleWhatsAppInvite(user, evt.Sender, evt.Join)
- case evt.Promote != nil:
- portal.ChangeAdminStatus(evt.Promote, true)
- case evt.Demote != nil:
- portal.ChangeAdminStatus(evt.Demote, false)
- case evt.Ephemeral != nil:
- portal.UpdateGroupDisappearingMessages(evt.Sender, evt.Timestamp, evt.Ephemeral.DisappearingTimer)
- }
- }
- func (user *User) handlePictureUpdate(evt *events.Picture) {
- if evt.JID.Server == types.DefaultUserServer {
- puppet := user.bridge.GetPuppetByJID(evt.JID)
- user.log.Debugfln("Received picture update for puppet %s (current: %s, new: %s)", evt.JID, puppet.Avatar, evt.PictureID)
- if puppet.Avatar != evt.PictureID {
- puppet.UpdateAvatar(user)
- }
- } else if portal := user.GetPortalByJID(evt.JID); portal != nil {
- user.log.Debugfln("Received picture update for portal %s (current: %s, new: %s)", evt.JID, portal.Avatar, evt.PictureID)
- if portal.Avatar != evt.PictureID {
- portal.UpdateAvatar(user, evt.Author, true)
- }
- }
- }
|