user.go 37 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2022 Tulir Asokan
  3. //
  4. // This program is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Affero General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // This program is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Affero General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Affero General Public License
  15. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  16. package main
  17. import (
  18. "context"
  19. "encoding/json"
  20. "errors"
  21. "fmt"
  22. "math"
  23. "net/http"
  24. "strconv"
  25. "strings"
  26. "sync"
  27. "time"
  28. log "maunium.net/go/maulogger/v2"
  29. "maunium.net/go/mautrix"
  30. "maunium.net/go/mautrix/appservice"
  31. "maunium.net/go/mautrix/event"
  32. "maunium.net/go/mautrix/format"
  33. "maunium.net/go/mautrix/id"
  34. "maunium.net/go/mautrix/pushrules"
  35. "go.mau.fi/whatsmeow"
  36. "go.mau.fi/whatsmeow/appstate"
  37. waProto "go.mau.fi/whatsmeow/binary/proto"
  38. "go.mau.fi/whatsmeow/store"
  39. "go.mau.fi/whatsmeow/types"
  40. "go.mau.fi/whatsmeow/types/events"
  41. waLog "go.mau.fi/whatsmeow/util/log"
  42. "maunium.net/go/mautrix-whatsapp/database"
  43. )
  44. type User struct {
  45. *database.User
  46. Client *whatsmeow.Client
  47. Session *store.Device
  48. bridge *Bridge
  49. log log.Logger
  50. Admin bool
  51. Whitelisted bool
  52. RelayWhitelisted bool
  53. mgmtCreateLock sync.Mutex
  54. spaceCreateLock sync.Mutex
  55. connLock sync.Mutex
  56. historySyncs chan *events.HistorySync
  57. prevBridgeStatus *BridgeState
  58. lastPresence types.Presence
  59. historySyncLoopsStarted bool
  60. spaceMembershipChecked bool
  61. lastPhoneOfflineWarning time.Time
  62. groupListCache []*types.GroupInfo
  63. groupListCacheLock sync.Mutex
  64. groupListCacheTime time.Time
  65. bridgeStateQueue chan BridgeState
  66. BackfillQueue *BackfillQueue
  67. }
  68. func (bridge *Bridge) getUserByMXID(userID id.UserID, onlyIfExists bool) *User {
  69. _, isPuppet := bridge.ParsePuppetMXID(userID)
  70. if isPuppet || userID == bridge.Bot.UserID {
  71. return nil
  72. }
  73. bridge.usersLock.Lock()
  74. defer bridge.usersLock.Unlock()
  75. user, ok := bridge.usersByMXID[userID]
  76. if !ok {
  77. userIDPtr := &userID
  78. if onlyIfExists {
  79. userIDPtr = nil
  80. }
  81. return bridge.loadDBUser(bridge.DB.User.GetByMXID(userID), userIDPtr)
  82. }
  83. return user
  84. }
  85. func (bridge *Bridge) GetUserByMXID(userID id.UserID) *User {
  86. return bridge.getUserByMXID(userID, false)
  87. }
  88. func (bridge *Bridge) GetUserByMXIDIfExists(userID id.UserID) *User {
  89. return bridge.getUserByMXID(userID, true)
  90. }
  91. func (bridge *Bridge) GetUserByJID(jid types.JID) *User {
  92. bridge.usersLock.Lock()
  93. defer bridge.usersLock.Unlock()
  94. user, ok := bridge.usersByUsername[jid.User]
  95. if !ok {
  96. return bridge.loadDBUser(bridge.DB.User.GetByUsername(jid.User), nil)
  97. }
  98. return user
  99. }
  100. func (user *User) addToJIDMap() {
  101. user.bridge.usersLock.Lock()
  102. user.bridge.usersByUsername[user.JID.User] = user
  103. user.bridge.usersLock.Unlock()
  104. }
  105. func (user *User) removeFromJIDMap(state BridgeState) {
  106. user.bridge.usersLock.Lock()
  107. jidUser, ok := user.bridge.usersByUsername[user.JID.User]
  108. if ok && user == jidUser {
  109. delete(user.bridge.usersByUsername, user.JID.User)
  110. }
  111. user.bridge.usersLock.Unlock()
  112. user.bridge.Metrics.TrackLoginState(user.JID, false)
  113. user.sendBridgeState(state)
  114. }
  115. func (bridge *Bridge) GetAllUsers() []*User {
  116. bridge.usersLock.Lock()
  117. defer bridge.usersLock.Unlock()
  118. dbUsers := bridge.DB.User.GetAll()
  119. output := make([]*User, len(dbUsers))
  120. for index, dbUser := range dbUsers {
  121. user, ok := bridge.usersByMXID[dbUser.MXID]
  122. if !ok {
  123. user = bridge.loadDBUser(dbUser, nil)
  124. }
  125. output[index] = user
  126. }
  127. return output
  128. }
  129. func (bridge *Bridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User {
  130. if dbUser == nil {
  131. if mxid == nil {
  132. return nil
  133. }
  134. dbUser = bridge.DB.User.New()
  135. dbUser.MXID = *mxid
  136. dbUser.Insert()
  137. }
  138. user := bridge.NewUser(dbUser)
  139. bridge.usersByMXID[user.MXID] = user
  140. if !user.JID.IsEmpty() {
  141. var err error
  142. user.Session, err = bridge.WAContainer.GetDevice(user.JID)
  143. if err != nil {
  144. user.log.Errorfln("Failed to load user's whatsapp session: %v", err)
  145. } else if user.Session == nil {
  146. user.log.Warnfln("Didn't find session data for %s, treating user as logged out", user.JID)
  147. user.JID = types.EmptyJID
  148. user.Update()
  149. } else {
  150. bridge.usersByUsername[user.JID.User] = user
  151. }
  152. }
  153. if len(user.ManagementRoom) > 0 {
  154. bridge.managementRooms[user.ManagementRoom] = user
  155. }
  156. return user
  157. }
  158. func (bridge *Bridge) NewUser(dbUser *database.User) *User {
  159. user := &User{
  160. User: dbUser,
  161. bridge: bridge,
  162. log: bridge.Log.Sub("User").Sub(string(dbUser.MXID)),
  163. historySyncs: make(chan *events.HistorySync, 32),
  164. lastPresence: types.PresenceUnavailable,
  165. }
  166. user.RelayWhitelisted = user.bridge.Config.Bridge.Permissions.IsRelayWhitelisted(user.MXID)
  167. user.Whitelisted = user.bridge.Config.Bridge.Permissions.IsWhitelisted(user.MXID)
  168. user.Admin = user.bridge.Config.Bridge.Permissions.IsAdmin(user.MXID)
  169. if len(user.bridge.Config.Homeserver.StatusEndpoint) > 0 {
  170. user.bridgeStateQueue = make(chan BridgeState, 10)
  171. go user.bridgeStateLoop()
  172. }
  173. return user
  174. }
  175. func (user *User) ensureInvited(intent *appservice.IntentAPI, roomID id.RoomID, isDirect bool) (ok bool) {
  176. inviteContent := event.Content{
  177. Parsed: &event.MemberEventContent{
  178. Membership: event.MembershipInvite,
  179. IsDirect: isDirect,
  180. },
  181. Raw: map[string]interface{}{},
  182. }
  183. customPuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  184. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  185. inviteContent.Raw["fi.mau.will_auto_accept"] = true
  186. }
  187. _, err := intent.SendStateEvent(roomID, event.StateMember, user.MXID.String(), &inviteContent)
  188. var httpErr mautrix.HTTPError
  189. if err != nil && errors.As(err, &httpErr) && httpErr.RespError != nil && strings.Contains(httpErr.RespError.Err, "is already in the room") {
  190. user.bridge.StateStore.SetMembership(roomID, user.MXID, event.MembershipJoin)
  191. ok = true
  192. return
  193. } else if err != nil {
  194. user.log.Warnfln("Failed to invite user to %s: %v", roomID, err)
  195. } else {
  196. ok = true
  197. }
  198. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  199. err = customPuppet.CustomIntent().EnsureJoined(roomID, appservice.EnsureJoinedParams{IgnoreCache: true})
  200. if err != nil {
  201. user.log.Warnfln("Failed to auto-join %s: %v", roomID, err)
  202. ok = false
  203. } else {
  204. ok = true
  205. }
  206. }
  207. return
  208. }
  209. func (user *User) GetSpaceRoom() id.RoomID {
  210. if !user.bridge.Config.Bridge.PersonalFilteringSpaces {
  211. return ""
  212. }
  213. if len(user.SpaceRoom) == 0 {
  214. user.spaceCreateLock.Lock()
  215. defer user.spaceCreateLock.Unlock()
  216. if len(user.SpaceRoom) > 0 {
  217. return user.SpaceRoom
  218. }
  219. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  220. Visibility: "private",
  221. Name: "WhatsApp",
  222. Topic: "Your WhatsApp bridged chats",
  223. InitialState: []*event.Event{{
  224. Type: event.StateRoomAvatar,
  225. Content: event.Content{
  226. Parsed: &event.RoomAvatarEventContent{
  227. URL: user.bridge.Config.AppService.Bot.ParsedAvatar,
  228. },
  229. },
  230. }},
  231. CreationContent: map[string]interface{}{
  232. "type": event.RoomTypeSpace,
  233. },
  234. PowerLevelOverride: &event.PowerLevelsEventContent{
  235. Users: map[id.UserID]int{
  236. user.bridge.Bot.UserID: 9001,
  237. user.MXID: 50,
  238. },
  239. },
  240. })
  241. if err != nil {
  242. user.log.Errorln("Failed to auto-create space room:", err)
  243. } else {
  244. user.SpaceRoom = resp.RoomID
  245. user.Update()
  246. user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false)
  247. }
  248. } else if !user.spaceMembershipChecked && !user.bridge.StateStore.IsInRoom(user.SpaceRoom, user.MXID) {
  249. user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false)
  250. }
  251. user.spaceMembershipChecked = true
  252. return user.SpaceRoom
  253. }
  254. func (user *User) GetManagementRoom() id.RoomID {
  255. if len(user.ManagementRoom) == 0 {
  256. user.mgmtCreateLock.Lock()
  257. defer user.mgmtCreateLock.Unlock()
  258. if len(user.ManagementRoom) > 0 {
  259. return user.ManagementRoom
  260. }
  261. creationContent := make(map[string]interface{})
  262. if !user.bridge.Config.Bridge.FederateRooms {
  263. creationContent["m.federate"] = false
  264. }
  265. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  266. Topic: "WhatsApp bridge notices",
  267. IsDirect: true,
  268. CreationContent: creationContent,
  269. })
  270. if err != nil {
  271. user.log.Errorln("Failed to auto-create management room:", err)
  272. } else {
  273. user.SetManagementRoom(resp.RoomID)
  274. }
  275. }
  276. return user.ManagementRoom
  277. }
  278. func (user *User) SetManagementRoom(roomID id.RoomID) {
  279. existingUser, ok := user.bridge.managementRooms[roomID]
  280. if ok {
  281. existingUser.ManagementRoom = ""
  282. existingUser.Update()
  283. }
  284. user.ManagementRoom = roomID
  285. user.bridge.managementRooms[user.ManagementRoom] = user
  286. user.Update()
  287. }
  288. type waLogger struct{ l log.Logger }
  289. func (w *waLogger) Debugf(msg string, args ...interface{}) { w.l.Debugfln(msg, args...) }
  290. func (w *waLogger) Infof(msg string, args ...interface{}) { w.l.Infofln(msg, args...) }
  291. func (w *waLogger) Warnf(msg string, args ...interface{}) { w.l.Warnfln(msg, args...) }
  292. func (w *waLogger) Errorf(msg string, args ...interface{}) { w.l.Errorfln(msg, args...) }
  293. func (w *waLogger) Sub(module string) waLog.Logger { return &waLogger{l: w.l.Sub(module)} }
  294. var ErrAlreadyLoggedIn = errors.New("already logged in")
  295. func (user *User) createClient(sess *store.Device) {
  296. user.Client = whatsmeow.NewClient(sess, &waLogger{user.log.Sub("Client")})
  297. user.Client.DebugDecodeBeforeSend = user.bridge.Config.WhatsApp.DebugDecodeBeforeSend
  298. user.Client.AddEventHandler(user.HandleEvent)
  299. user.Client.SetForceActiveDeliveryReceipts(user.bridge.Config.Bridge.ForceActiveDeliveryReceipts)
  300. user.Client.GetMessageForRetry = func(to types.JID, id types.MessageID) *waProto.Message {
  301. user.bridge.Metrics.TrackRetryReceipt(0, false)
  302. return nil
  303. }
  304. user.Client.PreRetryCallback = func(receipt *events.Receipt, retryCount int, msg *waProto.Message) bool {
  305. user.bridge.Metrics.TrackRetryReceipt(retryCount, true)
  306. return true
  307. }
  308. }
  309. func (user *User) Login(ctx context.Context) (<-chan whatsmeow.QRChannelItem, error) {
  310. user.connLock.Lock()
  311. defer user.connLock.Unlock()
  312. if user.Session != nil {
  313. return nil, ErrAlreadyLoggedIn
  314. } else if user.Client != nil {
  315. user.unlockedDeleteConnection()
  316. }
  317. newSession := user.bridge.WAContainer.NewDevice()
  318. newSession.Log = &waLogger{user.log.Sub("Session")}
  319. user.createClient(newSession)
  320. qrChan, err := user.Client.GetQRChannel(ctx)
  321. if err != nil {
  322. return nil, fmt.Errorf("failed to get QR channel: %w", err)
  323. }
  324. err = user.Client.Connect()
  325. if err != nil {
  326. return nil, fmt.Errorf("failed to connect to WhatsApp: %w", err)
  327. }
  328. return qrChan, nil
  329. }
  330. func (user *User) Connect() bool {
  331. user.connLock.Lock()
  332. defer user.connLock.Unlock()
  333. if user.Client != nil {
  334. return user.Client.IsConnected()
  335. } else if user.Session == nil {
  336. return false
  337. }
  338. user.log.Debugln("Connecting to WhatsApp")
  339. user.sendBridgeState(BridgeState{StateEvent: StateConnecting, Error: WAConnecting})
  340. user.createClient(user.Session)
  341. err := user.Client.Connect()
  342. if err != nil {
  343. user.log.Warnln("Error connecting to WhatsApp:", err)
  344. user.sendBridgeState(BridgeState{
  345. StateEvent: StateUnknownError,
  346. Error: WAConnectionFailed,
  347. Info: map[string]interface{}{
  348. "go_error": err.Error(),
  349. },
  350. })
  351. return false
  352. }
  353. return true
  354. }
  355. func (user *User) unlockedDeleteConnection() {
  356. if user.Client == nil {
  357. return
  358. }
  359. user.Client.Disconnect()
  360. user.Client.RemoveEventHandlers()
  361. user.Client = nil
  362. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  363. }
  364. func (user *User) DeleteConnection() {
  365. user.connLock.Lock()
  366. defer user.connLock.Unlock()
  367. user.unlockedDeleteConnection()
  368. }
  369. func (user *User) HasSession() bool {
  370. return user.Session != nil
  371. }
  372. func (user *User) DeleteSession() {
  373. if user.Session != nil {
  374. err := user.Session.Delete()
  375. if err != nil {
  376. user.log.Warnln("Failed to delete session:", err)
  377. }
  378. user.Session = nil
  379. }
  380. if !user.JID.IsEmpty() {
  381. user.JID = types.EmptyJID
  382. user.Update()
  383. }
  384. // Delete all of the backfill and history sync data.
  385. user.bridge.DB.BackfillQuery.DeleteAll(user.MXID)
  386. user.bridge.DB.HistorySyncQuery.DeleteAllConversations(user.MXID)
  387. user.bridge.DB.HistorySyncQuery.DeleteAllMessages(user.MXID)
  388. }
  389. func (user *User) IsConnected() bool {
  390. return user.Client != nil && user.Client.IsConnected()
  391. }
  392. func (user *User) IsLoggedIn() bool {
  393. return user.IsConnected() && user.Client.IsLoggedIn()
  394. }
  395. func (user *User) tryAutomaticDoublePuppeting() {
  396. if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) {
  397. return
  398. }
  399. user.log.Debugln("Checking if double puppeting needs to be enabled")
  400. puppet := user.bridge.GetPuppetByJID(user.JID)
  401. if len(puppet.CustomMXID) > 0 {
  402. user.log.Debugln("User already has double-puppeting enabled")
  403. // Custom puppet already enabled
  404. return
  405. }
  406. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  407. if err != nil {
  408. user.log.Warnln("Failed to login with shared secret:", err)
  409. return
  410. }
  411. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  412. if err != nil {
  413. puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
  414. return
  415. }
  416. user.log.Infoln("Successfully automatically enabled custom puppet")
  417. }
  418. func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) {
  419. if user.bridge.Config.Bridge.DisableBridgeAlerts {
  420. return
  421. }
  422. notice := fmt.Sprintf(formatString, args...)
  423. content := format.RenderMarkdown(notice, true, false)
  424. _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content)
  425. if err != nil {
  426. user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err)
  427. }
  428. }
  429. const callEventMaxAge = 15 * time.Minute
  430. func (user *User) handleCallStart(sender types.JID, id, callType string, ts time.Time) {
  431. if !user.bridge.Config.Bridge.CallStartNotices || ts.Add(callEventMaxAge).Before(time.Now()) {
  432. return
  433. }
  434. portal := user.GetPortalByJID(sender)
  435. text := "Incoming call"
  436. if callType != "" {
  437. text = fmt.Sprintf("Incoming %s call", callType)
  438. }
  439. portal.messages <- PortalMessage{
  440. fake: &fakeMessage{
  441. Sender: sender,
  442. Text: text,
  443. ID: id,
  444. Time: ts,
  445. Important: true,
  446. },
  447. source: user,
  448. }
  449. }
  450. const PhoneDisconnectWarningTime = 12 * 24 * time.Hour // 12 days
  451. const PhoneDisconnectPingTime = 10 * 24 * time.Hour
  452. const PhoneMinPingInterval = 24 * time.Hour
  453. func (user *User) sendHackyPhonePing() {
  454. msgID := whatsmeow.GenerateMessageID()
  455. user.PhoneLastPinged = time.Now()
  456. ts, err := user.Client.SendMessage(user.JID.ToNonAD(), msgID, &waProto.Message{
  457. ProtocolMessage: &waProto.ProtocolMessage{},
  458. })
  459. if err != nil {
  460. user.log.Warnfln("Failed to send hacky phone ping: %v", err)
  461. } else {
  462. user.log.Debugfln("Sent hacky phone ping %s/%s because phone has been offline for >10 days", msgID, ts)
  463. user.PhoneLastPinged = ts
  464. user.Update()
  465. }
  466. }
  467. func (user *User) PhoneRecentlySeen(doPing bool) bool {
  468. if doPing && !user.PhoneLastSeen.IsZero() && user.PhoneLastSeen.Add(PhoneDisconnectPingTime).Before(time.Now()) && user.PhoneLastPinged.Add(PhoneMinPingInterval).Before(time.Now()) {
  469. // Over 10 days since the phone was seen and over a day since the last somewhat hacky ping, send a new ping.
  470. go user.sendHackyPhonePing()
  471. }
  472. return user.PhoneLastSeen.IsZero() || user.PhoneLastSeen.Add(PhoneDisconnectWarningTime).After(time.Now())
  473. }
  474. // phoneSeen records a timestamp when the user's main device was seen online.
  475. // The stored timestamp can later be used to warn the user if the main device is offline for too long.
  476. func (user *User) phoneSeen(ts time.Time) {
  477. if user.PhoneLastSeen.Add(1 * time.Hour).After(ts) {
  478. // The last seen timestamp isn't going to be perfectly accurate in any case,
  479. // so don't spam the database with an update every time there's an event.
  480. return
  481. } else if !user.PhoneRecentlySeen(false) && user.GetPrevBridgeState().Error == WAPhoneOffline && user.IsConnected() {
  482. user.log.Debugfln("Saw phone after current bridge state said it has been offline, switching state back to connected")
  483. go user.sendBridgeState(BridgeState{StateEvent: StateConnected})
  484. }
  485. user.PhoneLastSeen = ts
  486. go user.Update()
  487. }
  488. func formatDisconnectTime(dur time.Duration) string {
  489. days := int(math.Floor(dur.Hours() / 24))
  490. hours := int(dur.Hours()) % 24
  491. if hours == 0 {
  492. return fmt.Sprintf("%d days", days)
  493. } else if hours == 1 {
  494. return fmt.Sprintf("%d days and 1 hour", days)
  495. } else {
  496. return fmt.Sprintf("%d days and %d hours", days, hours)
  497. }
  498. }
  499. func (user *User) sendPhoneOfflineWarning() {
  500. if user.lastPhoneOfflineWarning.Add(12 * time.Hour).After(time.Now()) {
  501. // Don't spam the warning too much
  502. return
  503. }
  504. user.lastPhoneOfflineWarning = time.Now()
  505. timeSinceSeen := time.Now().Sub(user.PhoneLastSeen)
  506. user.sendMarkdownBridgeAlert("Your phone hasn't been seen in %s. The server will force the bridge to log out if the phone is not active at least every 2 weeks.", formatDisconnectTime(timeSinceSeen))
  507. }
  508. func (user *User) HandleEvent(event interface{}) {
  509. switch v := event.(type) {
  510. case *events.LoggedOut:
  511. go user.handleLoggedOut(v.OnConnect, v.Reason)
  512. case *events.Connected:
  513. user.bridge.Metrics.TrackConnectionState(user.JID, true)
  514. user.bridge.Metrics.TrackLoginState(user.JID, true)
  515. if len(user.Client.Store.PushName) > 0 {
  516. go func() {
  517. err := user.Client.SendPresence(user.lastPresence)
  518. if err != nil {
  519. user.log.Warnln("Failed to send initial presence:", err)
  520. }
  521. }()
  522. }
  523. go user.tryAutomaticDoublePuppeting()
  524. if user.bridge.Config.Bridge.HistorySync.Backfill && !user.historySyncLoopsStarted {
  525. go user.handleHistorySyncsLoop()
  526. user.historySyncLoopsStarted = true
  527. if user.bridge.Config.Bridge.HistorySync.BackfillMedia && user.bridge.Config.Bridge.HistorySync.EnqueueBackfillMediaNextStart {
  528. user.EnqueueMediaBackfills(user.bridge.GetAllPortalsForUser(user.MXID))
  529. }
  530. }
  531. case *events.OfflineSyncPreview:
  532. user.log.Infofln("Server says it's going to send %d messages and %d receipts that were missed during downtime", v.Messages, v.Receipts)
  533. go user.sendBridgeState(BridgeState{
  534. StateEvent: StateBackfilling,
  535. Message: fmt.Sprintf("backfilling %d messages and %d receipts", v.Messages, v.Receipts),
  536. })
  537. case *events.OfflineSyncCompleted:
  538. if !user.PhoneRecentlySeen(true) {
  539. user.log.Infofln("Offline sync completed, but phone last seen date is still %s - sending phone offline bridge status", user.PhoneLastSeen)
  540. go user.sendBridgeState(BridgeState{StateEvent: StateTransientDisconnect, Error: WAPhoneOffline})
  541. } else {
  542. if user.GetPrevBridgeState().StateEvent == StateBackfilling {
  543. user.log.Infoln("Offline sync completed")
  544. }
  545. go user.sendBridgeState(BridgeState{StateEvent: StateConnected})
  546. }
  547. case *events.AppStateSyncComplete:
  548. if len(user.Client.Store.PushName) > 0 && v.Name == appstate.WAPatchCriticalBlock {
  549. err := user.Client.SendPresence(user.lastPresence)
  550. if err != nil {
  551. user.log.Warnln("Failed to send presence after app state sync:", err)
  552. }
  553. } else if v.Name == appstate.WAPatchCriticalUnblockLow {
  554. go func() {
  555. err := user.ResyncContacts()
  556. if err != nil {
  557. user.log.Errorln("Failed to resync puppets: %v", err)
  558. }
  559. }()
  560. }
  561. case *events.PushNameSetting:
  562. // Send presence available when connecting and when the pushname is changed.
  563. // This makes sure that outgoing messages always have the right pushname.
  564. err := user.Client.SendPresence(user.lastPresence)
  565. if err != nil {
  566. user.log.Warnln("Failed to send presence after push name update:", err)
  567. }
  568. case *events.PairSuccess:
  569. user.PhoneLastSeen = time.Now()
  570. user.Session = user.Client.Store
  571. user.JID = v.ID
  572. user.addToJIDMap()
  573. user.Update()
  574. case *events.StreamError:
  575. var message string
  576. if v.Code != "" {
  577. message = fmt.Sprintf("Unknown stream error with code %s", v.Code)
  578. } else if children := v.Raw.GetChildren(); len(children) > 0 {
  579. message = fmt.Sprintf("Unknown stream error (contains %s node)", children[0].Tag)
  580. } else {
  581. message = "Unknown stream error"
  582. }
  583. go user.sendBridgeState(BridgeState{StateEvent: StateUnknownError, Message: message})
  584. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  585. case *events.ConnectFailure:
  586. go user.sendBridgeState(BridgeState{StateEvent: StateUnknownError, Message: fmt.Sprintf("Unknown connection failure: %s", v.Reason)})
  587. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  588. case *events.ClientOutdated:
  589. user.log.Errorfln("Got a client outdated connect failure. The bridge is likely out of date, please update immediately.")
  590. go user.sendBridgeState(BridgeState{StateEvent: StateUnknownError, Message: "Connect failure: 405 client outdated"})
  591. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  592. case *events.TemporaryBan:
  593. go user.sendBridgeState(BridgeState{StateEvent: StateBadCredentials, Message: v.String()})
  594. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  595. case *events.Disconnected:
  596. // Don't send the normal transient disconnect state if we're already in a different transient disconnect state.
  597. // TODO remove this if/when the phone offline state is moved to a sub-state of CONNECTED
  598. if user.GetPrevBridgeState().Error != WAPhoneOffline && user.PhoneRecentlySeen(false) {
  599. go user.sendBridgeState(BridgeState{StateEvent: StateTransientDisconnect, Message: "Disconnected from WhatsApp. Trying to reconnect."})
  600. }
  601. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  602. case *events.Contact:
  603. go user.syncPuppet(v.JID, "contact event")
  604. case *events.PushName:
  605. go user.syncPuppet(v.JID, "push name event")
  606. case *events.GroupInfo:
  607. user.groupListCache = nil
  608. go user.handleGroupUpdate(v)
  609. case *events.JoinedGroup:
  610. user.groupListCache = nil
  611. go user.handleGroupCreate(v)
  612. case *events.Picture:
  613. go user.handlePictureUpdate(v)
  614. case *events.Receipt:
  615. if v.IsFromMe && v.Sender.Device == 0 {
  616. user.phoneSeen(v.Timestamp)
  617. }
  618. go user.handleReceipt(v)
  619. case *events.ChatPresence:
  620. go user.handleChatPresence(v)
  621. case *events.Message:
  622. portal := user.GetPortalByMessageSource(v.Info.MessageSource)
  623. portal.messages <- PortalMessage{evt: v, source: user}
  624. case *events.MediaRetry:
  625. user.phoneSeen(v.Timestamp)
  626. portal := user.GetPortalByJID(v.ChatID)
  627. portal.mediaRetries <- PortalMediaRetry{evt: v, source: user}
  628. case *events.CallOffer:
  629. user.handleCallStart(v.CallCreator, v.CallID, "", v.Timestamp)
  630. case *events.CallOfferNotice:
  631. user.handleCallStart(v.CallCreator, v.CallID, v.Type, v.Timestamp)
  632. case *events.IdentityChange:
  633. puppet := user.bridge.GetPuppetByJID(v.JID)
  634. portal := user.GetPortalByJID(v.JID)
  635. if len(portal.MXID) > 0 && user.bridge.Config.Bridge.IdentityChangeNotices {
  636. text := fmt.Sprintf("Your security code with %s changed.", puppet.Displayname)
  637. if v.Implicit {
  638. text = fmt.Sprintf("Your security code with %s (device #%d) changed.", puppet.Displayname, v.JID.Device)
  639. }
  640. portal.messages <- PortalMessage{
  641. fake: &fakeMessage{
  642. Sender: v.JID,
  643. Text: text,
  644. ID: strconv.FormatInt(v.Timestamp.Unix(), 10),
  645. Time: v.Timestamp,
  646. Important: false,
  647. },
  648. source: user,
  649. }
  650. }
  651. case *events.CallTerminate, *events.CallRelayLatency, *events.CallAccept, *events.UnknownCallEvent:
  652. // ignore
  653. case *events.UndecryptableMessage:
  654. portal := user.GetPortalByMessageSource(v.Info.MessageSource)
  655. portal.messages <- PortalMessage{undecryptable: v, source: user}
  656. case *events.HistorySync:
  657. if user.bridge.Config.Bridge.HistorySync.Backfill {
  658. user.historySyncs <- v
  659. }
  660. case *events.Mute:
  661. portal := user.GetPortalByJID(v.JID)
  662. if portal != nil {
  663. var mutedUntil time.Time
  664. if v.Action.GetMuted() {
  665. mutedUntil = time.Unix(v.Action.GetMuteEndTimestamp(), 0)
  666. }
  667. go user.updateChatMute(nil, portal, mutedUntil)
  668. }
  669. case *events.Archive:
  670. portal := user.GetPortalByJID(v.JID)
  671. if portal != nil {
  672. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.ArchiveTag, v.Action.GetArchived())
  673. }
  674. case *events.Pin:
  675. portal := user.GetPortalByJID(v.JID)
  676. if portal != nil {
  677. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.PinnedTag, v.Action.GetPinned())
  678. }
  679. case *events.AppState:
  680. // Ignore
  681. default:
  682. user.log.Debugfln("Unknown type of event in HandleEvent: %T", v)
  683. }
  684. }
  685. func (user *User) updateChatMute(intent *appservice.IntentAPI, portal *Portal, mutedUntil time.Time) {
  686. if len(portal.MXID) == 0 || !user.bridge.Config.Bridge.MuteBridging {
  687. return
  688. } else if intent == nil {
  689. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  690. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  691. return
  692. }
  693. intent = doublePuppet.CustomIntent()
  694. }
  695. var err error
  696. if mutedUntil.IsZero() && mutedUntil.Before(time.Now()) {
  697. user.log.Debugfln("Portal %s is muted until %s, unmuting...", portal.MXID, mutedUntil)
  698. err = intent.DeletePushRule("global", pushrules.RoomRule, string(portal.MXID))
  699. } else {
  700. user.log.Debugfln("Portal %s is muted until %s, muting...", portal.MXID, mutedUntil)
  701. err = intent.PutPushRule("global", pushrules.RoomRule, string(portal.MXID), &mautrix.ReqPutPushRule{
  702. Actions: []pushrules.PushActionType{pushrules.ActionDontNotify},
  703. })
  704. }
  705. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  706. user.log.Warnfln("Failed to update push rule for %s through double puppet: %v", portal.MXID, err)
  707. }
  708. }
  709. type CustomTagData struct {
  710. Order json.Number `json:"order"`
  711. DoublePuppet string `json:"fi.mau.double_puppet_source"`
  712. }
  713. type CustomTagEventContent struct {
  714. Tags map[string]CustomTagData `json:"tags"`
  715. }
  716. func (user *User) updateChatTag(intent *appservice.IntentAPI, portal *Portal, tag string, active bool) {
  717. if len(portal.MXID) == 0 || len(tag) == 0 {
  718. return
  719. } else if intent == nil {
  720. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  721. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  722. return
  723. }
  724. intent = doublePuppet.CustomIntent()
  725. }
  726. var existingTags CustomTagEventContent
  727. err := intent.GetTagsWithCustomData(portal.MXID, &existingTags)
  728. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  729. user.log.Warnfln("Failed to get tags of %s: %v", portal.MXID, err)
  730. }
  731. currentTag, ok := existingTags.Tags[tag]
  732. if active && !ok {
  733. user.log.Debugln("Adding tag", tag, "to", portal.MXID)
  734. data := CustomTagData{"0.5", doublePuppetValue}
  735. err = intent.AddTagWithCustomData(portal.MXID, tag, &data)
  736. } else if !active && ok && currentTag.DoublePuppet == doublePuppetValue {
  737. user.log.Debugln("Removing tag", tag, "from", portal.MXID)
  738. err = intent.RemoveTag(portal.MXID, tag)
  739. } else {
  740. err = nil
  741. }
  742. if err != nil {
  743. user.log.Warnfln("Failed to update tag %s for %s through double puppet: %v", tag, portal.MXID, err)
  744. }
  745. }
  746. type CustomReadReceipt struct {
  747. Timestamp int64 `json:"ts,omitempty"`
  748. DoublePuppetSource string `json:"fi.mau.double_puppet_source,omitempty"`
  749. }
  750. type CustomReadMarkers struct {
  751. mautrix.ReqSetReadMarkers
  752. ReadExtra CustomReadReceipt `json:"com.beeper.read.extra"`
  753. FullyReadExtra CustomReadReceipt `json:"com.beeper.fully_read.extra"`
  754. }
  755. func (user *User) syncChatDoublePuppetDetails(portal *Portal, justCreated bool) {
  756. doublePuppet := portal.bridge.GetPuppetByCustomMXID(user.MXID)
  757. if doublePuppet == nil {
  758. return
  759. }
  760. if doublePuppet == nil || doublePuppet.CustomIntent() == nil || len(portal.MXID) == 0 {
  761. return
  762. }
  763. if justCreated || !user.bridge.Config.Bridge.TagOnlyOnCreate {
  764. chat, err := user.Client.Store.ChatSettings.GetChatSettings(portal.Key.JID)
  765. if err != nil {
  766. user.log.Warnfln("Failed to get settings of %s: %v", portal.Key.JID, err)
  767. return
  768. }
  769. intent := doublePuppet.CustomIntent()
  770. if portal.Key.JID == types.StatusBroadcastJID && justCreated && user.bridge.Config.Bridge.MuteStatusBroadcast {
  771. user.updateChatMute(intent, portal, time.Now().Add(365*24*time.Hour))
  772. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, true)
  773. return
  774. } else if !chat.Found {
  775. return
  776. }
  777. user.updateChatMute(intent, portal, chat.MutedUntil)
  778. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, chat.Archived)
  779. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.PinnedTag, chat.Pinned)
  780. }
  781. }
  782. func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
  783. res := make(map[id.UserID][]id.RoomID)
  784. privateChats := user.bridge.DB.Portal.FindPrivateChats(user.JID.ToNonAD())
  785. for _, portal := range privateChats {
  786. if len(portal.MXID) > 0 {
  787. res[user.bridge.FormatPuppetMXID(portal.Key.JID)] = []id.RoomID{portal.MXID}
  788. }
  789. }
  790. return res
  791. }
  792. func (user *User) UpdateDirectChats(chats map[id.UserID][]id.RoomID) {
  793. if !user.bridge.Config.Bridge.SyncDirectChatList {
  794. return
  795. }
  796. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  797. if puppet == nil || puppet.CustomIntent() == nil {
  798. return
  799. }
  800. intent := puppet.CustomIntent()
  801. method := http.MethodPatch
  802. if chats == nil {
  803. chats = user.getDirectChats()
  804. method = http.MethodPut
  805. }
  806. user.log.Debugln("Updating m.direct list on homeserver")
  807. var err error
  808. if user.bridge.Config.Homeserver.Asmux {
  809. urlPath := intent.BuildClientURL("unstable", "com.beeper.asmux", "dms")
  810. _, err = intent.MakeFullRequest(mautrix.FullRequest{
  811. Method: method,
  812. URL: urlPath,
  813. Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
  814. RequestJSON: chats,
  815. })
  816. } else {
  817. existingChats := make(map[id.UserID][]id.RoomID)
  818. err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
  819. if err != nil {
  820. user.log.Warnln("Failed to get m.direct list to update it:", err)
  821. return
  822. }
  823. for userID, rooms := range existingChats {
  824. if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
  825. // This is not a ghost user, include it in the new list
  826. chats[userID] = rooms
  827. } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
  828. // This is a ghost user, but we're not replacing the whole list, so include it too
  829. chats[userID] = rooms
  830. }
  831. }
  832. err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
  833. }
  834. if err != nil {
  835. user.log.Warnln("Failed to update m.direct list:", err)
  836. }
  837. }
  838. func (user *User) handleLoggedOut(onConnect bool, reason events.ConnectFailureReason) {
  839. errorCode := WAUnknownLogout
  840. if reason == events.ConnectFailureLoggedOut {
  841. errorCode = WALoggedOut
  842. } else if reason == events.ConnectFailureBanned {
  843. errorCode = WAAccountBanned
  844. }
  845. user.removeFromJIDMap(BridgeState{StateEvent: StateBadCredentials, Error: errorCode})
  846. user.DeleteConnection()
  847. user.Session = nil
  848. user.JID = types.EmptyJID
  849. user.Update()
  850. if onConnect {
  851. user.sendMarkdownBridgeAlert("Connecting to WhatsApp failed as the device was unlinked (error %s). Please link the bridge to your phone again.", reason)
  852. } else {
  853. user.sendMarkdownBridgeAlert("You were logged out from another device. Please link the bridge to your phone again.")
  854. }
  855. }
  856. func (user *User) GetPortalByMessageSource(ms types.MessageSource) *Portal {
  857. jid := ms.Chat
  858. if ms.IsIncomingBroadcast() {
  859. if ms.IsFromMe {
  860. jid = ms.BroadcastListOwner.ToNonAD()
  861. } else {
  862. jid = ms.Sender.ToNonAD()
  863. }
  864. if jid.IsEmpty() {
  865. return nil
  866. }
  867. }
  868. return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
  869. }
  870. func (user *User) GetPortalByJID(jid types.JID) *Portal {
  871. return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
  872. }
  873. func (user *User) syncPuppet(jid types.JID, reason string) {
  874. user.bridge.GetPuppetByJID(jid).SyncContact(user, false, reason)
  875. }
  876. func (user *User) ResyncContacts() error {
  877. contacts, err := user.Client.Store.Contacts.GetAllContacts()
  878. if err != nil {
  879. return fmt.Errorf("failed to get cached contacts: %w", err)
  880. }
  881. user.log.Infofln("Resyncing displaynames with %d contacts", len(contacts))
  882. for jid, contact := range contacts {
  883. puppet := user.bridge.GetPuppetByJID(jid)
  884. if puppet != nil {
  885. puppet.Sync(user, contact)
  886. } else {
  887. user.log.Warnfln("Got a nil puppet for %s while syncing contacts", jid)
  888. }
  889. }
  890. return nil
  891. }
  892. func (user *User) ResyncGroups(createPortals bool) error {
  893. groups, err := user.Client.GetJoinedGroups()
  894. if err != nil {
  895. return fmt.Errorf("failed to get group list from server: %w", err)
  896. }
  897. for _, group := range groups {
  898. portal := user.GetPortalByJID(group.JID)
  899. if len(portal.MXID) == 0 {
  900. if createPortals {
  901. err = portal.CreateMatrixRoom(user, group, true, true)
  902. if err != nil {
  903. return fmt.Errorf("failed to create room for %s: %w", group.JID, err)
  904. }
  905. }
  906. } else {
  907. portal.UpdateMatrixRoom(user, group)
  908. }
  909. }
  910. return nil
  911. }
  912. const WATypingTimeout = 15 * time.Second
  913. func (user *User) handleChatPresence(presence *events.ChatPresence) {
  914. puppet := user.bridge.GetPuppetByJID(presence.Sender)
  915. portal := user.GetPortalByJID(presence.Chat)
  916. if puppet == nil || portal == nil || len(portal.MXID) == 0 {
  917. return
  918. }
  919. if presence.State == types.ChatPresenceComposing {
  920. if puppet.typingIn != "" && puppet.typingAt.Add(WATypingTimeout).Before(time.Now()) {
  921. if puppet.typingIn == portal.MXID {
  922. return
  923. }
  924. _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  925. }
  926. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, WATypingTimeout.Milliseconds())
  927. puppet.typingIn = portal.MXID
  928. puppet.typingAt = time.Now()
  929. } else {
  930. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, false, 0)
  931. puppet.typingIn = ""
  932. }
  933. }
  934. func (user *User) handleReceipt(receipt *events.Receipt) {
  935. if receipt.Type != events.ReceiptTypeRead && receipt.Type != events.ReceiptTypeReadSelf {
  936. return
  937. }
  938. portal := user.GetPortalByMessageSource(receipt.MessageSource)
  939. if portal == nil || len(portal.MXID) == 0 {
  940. return
  941. }
  942. portal.receipts <- PortalReceipt{evt: receipt, source: user}
  943. }
  944. func makeReadMarkerContent(eventID id.EventID, doublePuppet bool) CustomReadMarkers {
  945. var extra CustomReadReceipt
  946. if doublePuppet {
  947. extra.DoublePuppetSource = doublePuppetValue
  948. }
  949. return CustomReadMarkers{
  950. ReqSetReadMarkers: mautrix.ReqSetReadMarkers{
  951. Read: eventID,
  952. FullyRead: eventID,
  953. },
  954. ReadExtra: extra,
  955. FullyReadExtra: extra,
  956. }
  957. }
  958. func (user *User) markSelfReadFull(portal *Portal) {
  959. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  960. if puppet == nil || puppet.CustomIntent() == nil {
  961. return
  962. }
  963. lastMessage := user.bridge.DB.Message.GetLastInChat(portal.Key)
  964. if lastMessage == nil {
  965. return
  966. }
  967. user.SetLastReadTS(portal.Key, lastMessage.Timestamp)
  968. err := puppet.CustomIntent().SetReadMarkers(portal.MXID, makeReadMarkerContent(lastMessage.MXID, true))
  969. if err != nil {
  970. user.log.Warnfln("Failed to mark %s (last message) in %s as read: %v", lastMessage.MXID, portal.MXID, err)
  971. } else {
  972. user.log.Debugfln("Marked %s (last message) in %s as read", lastMessage.MXID, portal.MXID)
  973. }
  974. }
  975. func (user *User) handleGroupCreate(evt *events.JoinedGroup) {
  976. portal := user.GetPortalByJID(evt.JID)
  977. if len(portal.MXID) == 0 {
  978. err := portal.CreateMatrixRoom(user, &evt.GroupInfo, true, true)
  979. if err != nil {
  980. user.log.Errorln("Failed to create Matrix room after join notification: %v", err)
  981. }
  982. } else {
  983. portal.UpdateMatrixRoom(user, &evt.GroupInfo)
  984. }
  985. }
  986. func (user *User) handleGroupUpdate(evt *events.GroupInfo) {
  987. portal := user.GetPortalByJID(evt.JID)
  988. if portal == nil || len(portal.MXID) == 0 {
  989. user.log.Debugfln("Ignoring group info update in chat with no portal: %+v", evt)
  990. return
  991. }
  992. switch {
  993. case evt.Announce != nil:
  994. portal.RestrictMessageSending(evt.Announce.IsAnnounce)
  995. case evt.Locked != nil:
  996. portal.RestrictMetadataChanges(evt.Locked.IsLocked)
  997. case evt.Name != nil:
  998. portal.UpdateName(evt.Name.Name, evt.Name.NameSetBy, true)
  999. case evt.Topic != nil:
  1000. portal.UpdateTopic(evt.Topic.Topic, evt.Topic.TopicSetBy, true)
  1001. case evt.Leave != nil:
  1002. if evt.Sender != nil && !evt.Sender.IsEmpty() {
  1003. portal.HandleWhatsAppKick(user, *evt.Sender, evt.Leave)
  1004. }
  1005. case evt.Join != nil:
  1006. portal.HandleWhatsAppInvite(user, evt.Sender, evt.Join)
  1007. case evt.Promote != nil:
  1008. portal.ChangeAdminStatus(evt.Promote, true)
  1009. case evt.Demote != nil:
  1010. portal.ChangeAdminStatus(evt.Demote, false)
  1011. case evt.Ephemeral != nil:
  1012. portal.UpdateGroupDisappearingMessages(evt.Sender, evt.Timestamp, evt.Ephemeral.DisappearingTimer)
  1013. }
  1014. }
  1015. func (user *User) handlePictureUpdate(evt *events.Picture) {
  1016. if evt.JID.Server == types.DefaultUserServer {
  1017. puppet := user.bridge.GetPuppetByJID(evt.JID)
  1018. user.log.Debugfln("Received picture update for puppet %s (current: %s, new: %s)", evt.JID, puppet.Avatar, evt.PictureID)
  1019. if puppet.Avatar != evt.PictureID {
  1020. puppet.UpdateAvatar(user)
  1021. }
  1022. } else if portal := user.GetPortalByJID(evt.JID); portal != nil {
  1023. user.log.Debugfln("Received picture update for portal %s (current: %s, new: %s)", evt.JID, portal.Avatar, evt.PictureID)
  1024. if portal.Avatar != evt.PictureID {
  1025. portal.UpdateAvatar(user, evt.Author, true)
  1026. }
  1027. }
  1028. }
  1029. func (user *User) StartPM(jid types.JID, reason string) (*Portal, *Puppet, bool, error) {
  1030. user.log.Debugln("Starting PM with", jid, "from", reason)
  1031. puppet := user.bridge.GetPuppetByJID(jid)
  1032. puppet.SyncContact(user, true, reason)
  1033. portal := user.GetPortalByJID(puppet.JID)
  1034. if len(portal.MXID) > 0 {
  1035. ok := portal.ensureUserInvited(user)
  1036. if !ok {
  1037. portal.log.Warnfln("ensureUserInvited(%s) returned false, creating new portal", user.MXID)
  1038. portal.MXID = ""
  1039. } else {
  1040. return portal, puppet, false, nil
  1041. }
  1042. }
  1043. err := portal.CreateMatrixRoom(user, nil, false, true)
  1044. return portal, puppet, true, err
  1045. }
  1046. const groupListCacheMaxAge = 24 * time.Hour
  1047. func (user *User) getCachedGroupList() ([]*types.GroupInfo, error) {
  1048. user.groupListCacheLock.Lock()
  1049. defer user.groupListCacheLock.Unlock()
  1050. if user.groupListCache != nil && user.groupListCacheTime.Add(groupListCacheMaxAge).After(time.Now()) {
  1051. return user.groupListCache, nil
  1052. }
  1053. var err error
  1054. user.groupListCache, err = user.Client.GetJoinedGroups()
  1055. user.groupListCacheTime = time.Now()
  1056. return user.groupListCache, err
  1057. }