user.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2021 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. "net/http"
  23. "sync"
  24. "time"
  25. log "maunium.net/go/maulogger/v2"
  26. "go.mau.fi/whatsmeow/appstate"
  27. "maunium.net/go/mautrix/appservice"
  28. "maunium.net/go/mautrix/pushrules"
  29. "go.mau.fi/whatsmeow"
  30. "go.mau.fi/whatsmeow/store"
  31. "go.mau.fi/whatsmeow/types"
  32. "go.mau.fi/whatsmeow/types/events"
  33. waLog "go.mau.fi/whatsmeow/util/log"
  34. "maunium.net/go/mautrix"
  35. "maunium.net/go/mautrix/event"
  36. "maunium.net/go/mautrix/format"
  37. "maunium.net/go/mautrix/id"
  38. "maunium.net/go/mautrix-whatsapp/database"
  39. )
  40. type User struct {
  41. *database.User
  42. Client *whatsmeow.Client
  43. Session *store.Device
  44. bridge *Bridge
  45. log log.Logger
  46. Admin bool
  47. Whitelisted bool
  48. RelayWhitelisted bool
  49. mgmtCreateLock sync.Mutex
  50. connLock sync.Mutex
  51. historySyncs chan *events.HistorySync
  52. prevBridgeStatus *BridgeState
  53. lastPresence types.Presence
  54. }
  55. func (bridge *Bridge) GetUserByMXID(userID id.UserID) *User {
  56. _, isPuppet := bridge.ParsePuppetMXID(userID)
  57. if isPuppet || userID == bridge.Bot.UserID {
  58. return nil
  59. }
  60. bridge.usersLock.Lock()
  61. defer bridge.usersLock.Unlock()
  62. user, ok := bridge.usersByMXID[userID]
  63. if !ok {
  64. return bridge.loadDBUser(bridge.DB.User.GetByMXID(userID), &userID)
  65. }
  66. return user
  67. }
  68. func (bridge *Bridge) GetUserByJID(jid types.JID) *User {
  69. bridge.usersLock.Lock()
  70. defer bridge.usersLock.Unlock()
  71. user, ok := bridge.usersByUsername[jid.User]
  72. if !ok {
  73. return bridge.loadDBUser(bridge.DB.User.GetByUsername(jid.User), nil)
  74. }
  75. return user
  76. }
  77. func (user *User) addToJIDMap() {
  78. user.bridge.usersLock.Lock()
  79. user.bridge.usersByUsername[user.JID.User] = user
  80. user.bridge.usersLock.Unlock()
  81. }
  82. func (user *User) removeFromJIDMap(state BridgeStateEvent) {
  83. user.bridge.usersLock.Lock()
  84. jidUser, ok := user.bridge.usersByUsername[user.JID.User]
  85. if ok && user == jidUser {
  86. delete(user.bridge.usersByUsername, user.JID.User)
  87. }
  88. user.bridge.usersLock.Unlock()
  89. user.bridge.Metrics.TrackLoginState(user.JID, false)
  90. user.sendBridgeState(BridgeState{StateEvent: state, Error: WANotLoggedIn})
  91. }
  92. func (bridge *Bridge) GetAllUsers() []*User {
  93. bridge.usersLock.Lock()
  94. defer bridge.usersLock.Unlock()
  95. dbUsers := bridge.DB.User.GetAll()
  96. output := make([]*User, len(dbUsers))
  97. for index, dbUser := range dbUsers {
  98. user, ok := bridge.usersByMXID[dbUser.MXID]
  99. if !ok {
  100. user = bridge.loadDBUser(dbUser, nil)
  101. }
  102. output[index] = user
  103. }
  104. return output
  105. }
  106. func (bridge *Bridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User {
  107. if dbUser == nil {
  108. if mxid == nil {
  109. return nil
  110. }
  111. dbUser = bridge.DB.User.New()
  112. dbUser.MXID = *mxid
  113. dbUser.Insert()
  114. }
  115. user := bridge.NewUser(dbUser)
  116. bridge.usersByMXID[user.MXID] = user
  117. if !user.JID.IsEmpty() {
  118. var err error
  119. user.Session, err = bridge.WAContainer.GetDevice(user.JID)
  120. if err != nil {
  121. user.log.Errorfln("Failed to load user's whatsapp session: %v", err)
  122. } else if user.Session == nil {
  123. user.log.Warnfln("Didn't find session data for %s, treating user as logged out", user.JID)
  124. user.JID = types.EmptyJID
  125. user.Update()
  126. } else {
  127. bridge.usersByUsername[user.JID.User] = user
  128. }
  129. }
  130. if len(user.ManagementRoom) > 0 {
  131. bridge.managementRooms[user.ManagementRoom] = user
  132. }
  133. return user
  134. }
  135. func (bridge *Bridge) NewUser(dbUser *database.User) *User {
  136. user := &User{
  137. User: dbUser,
  138. bridge: bridge,
  139. log: bridge.Log.Sub("User").Sub(string(dbUser.MXID)),
  140. historySyncs: make(chan *events.HistorySync, 32),
  141. lastPresence: types.PresenceUnavailable,
  142. }
  143. user.RelayWhitelisted = user.bridge.Config.Bridge.Permissions.IsRelayWhitelisted(user.MXID)
  144. user.Whitelisted = user.bridge.Config.Bridge.Permissions.IsWhitelisted(user.MXID)
  145. user.Admin = user.bridge.Config.Bridge.Permissions.IsAdmin(user.MXID)
  146. go user.handleHistorySyncsLoop()
  147. return user
  148. }
  149. func (user *User) GetManagementRoom() id.RoomID {
  150. if len(user.ManagementRoom) == 0 {
  151. user.mgmtCreateLock.Lock()
  152. defer user.mgmtCreateLock.Unlock()
  153. if len(user.ManagementRoom) > 0 {
  154. return user.ManagementRoom
  155. }
  156. creationContent := make(map[string]interface{})
  157. if !user.bridge.Config.Bridge.FederateRooms {
  158. creationContent["m.federate"] = false
  159. }
  160. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  161. Topic: "WhatsApp bridge notices",
  162. IsDirect: true,
  163. CreationContent: creationContent,
  164. })
  165. if err != nil {
  166. user.log.Errorln("Failed to auto-create management room:", err)
  167. } else {
  168. user.SetManagementRoom(resp.RoomID)
  169. }
  170. }
  171. return user.ManagementRoom
  172. }
  173. func (user *User) SetManagementRoom(roomID id.RoomID) {
  174. existingUser, ok := user.bridge.managementRooms[roomID]
  175. if ok {
  176. existingUser.ManagementRoom = ""
  177. existingUser.Update()
  178. }
  179. user.ManagementRoom = roomID
  180. user.bridge.managementRooms[user.ManagementRoom] = user
  181. user.Update()
  182. }
  183. type waLogger struct{ l log.Logger }
  184. func (w *waLogger) Debugf(msg string, args ...interface{}) { w.l.Debugfln(msg, args...) }
  185. func (w *waLogger) Infof(msg string, args ...interface{}) { w.l.Infofln(msg, args...) }
  186. func (w *waLogger) Warnf(msg string, args ...interface{}) { w.l.Warnfln(msg, args...) }
  187. func (w *waLogger) Errorf(msg string, args ...interface{}) { w.l.Errorfln(msg, args...) }
  188. func (w *waLogger) Sub(module string) waLog.Logger { return &waLogger{l: w.l.Sub(module)} }
  189. var ErrAlreadyLoggedIn = errors.New("already logged in")
  190. func (user *User) Login(ctx context.Context) (<-chan whatsmeow.QRChannelItem, error) {
  191. user.connLock.Lock()
  192. defer user.connLock.Unlock()
  193. if user.Session != nil {
  194. return nil, ErrAlreadyLoggedIn
  195. } else if user.Client != nil {
  196. user.unlockedDeleteConnection()
  197. }
  198. newSession := user.bridge.WAContainer.NewDevice()
  199. newSession.Log = &waLogger{user.log.Sub("Session")}
  200. user.Client = whatsmeow.NewClient(newSession, &waLogger{user.log.Sub("Client")})
  201. user.Client.AddEventHandler(user.HandleEvent)
  202. qrChan, err := user.Client.GetQRChannel(ctx)
  203. if err != nil {
  204. return nil, fmt.Errorf("failed to get QR channel: %w", err)
  205. }
  206. err = user.Client.Connect()
  207. if err != nil {
  208. return nil, fmt.Errorf("failed to connect to WhatsApp: %w", err)
  209. }
  210. return qrChan, nil
  211. }
  212. func (user *User) Connect() bool {
  213. user.connLock.Lock()
  214. defer user.connLock.Unlock()
  215. if user.Client != nil {
  216. return user.Client.IsConnected()
  217. } else if user.Session == nil {
  218. return false
  219. }
  220. user.log.Debugln("Connecting to WhatsApp")
  221. user.sendBridgeState(BridgeState{StateEvent: StateConnecting, Error: WAConnecting})
  222. user.Client = whatsmeow.NewClient(user.Session, &waLogger{user.log.Sub("Client")})
  223. user.Client.AddEventHandler(user.HandleEvent)
  224. err := user.Client.Connect()
  225. if err != nil {
  226. user.log.Warnln("Error connecting to WhatsApp:", err)
  227. return false
  228. }
  229. return true
  230. }
  231. func (user *User) unlockedDeleteConnection() {
  232. if user.Client == nil {
  233. return
  234. }
  235. user.Client.Disconnect()
  236. user.Client.RemoveEventHandlers()
  237. user.Client = nil
  238. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  239. }
  240. func (user *User) DeleteConnection() {
  241. user.connLock.Lock()
  242. defer user.connLock.Unlock()
  243. user.unlockedDeleteConnection()
  244. }
  245. func (user *User) HasSession() bool {
  246. return user.Session != nil
  247. }
  248. func (user *User) DeleteSession() {
  249. if user.Session != nil {
  250. err := user.Session.Delete()
  251. if err != nil {
  252. user.log.Warnln("Failed to delete session:", err)
  253. }
  254. user.Session = nil
  255. }
  256. if !user.JID.IsEmpty() {
  257. user.JID = types.EmptyJID
  258. user.Update()
  259. }
  260. }
  261. func (user *User) IsConnected() bool {
  262. return user.Client != nil && user.Client.IsConnected()
  263. }
  264. func (user *User) IsLoggedIn() bool {
  265. return user.IsConnected() && user.Client.IsLoggedIn
  266. }
  267. func (user *User) tryAutomaticDoublePuppeting() {
  268. if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) {
  269. return
  270. }
  271. user.log.Debugln("Checking if double puppeting needs to be enabled")
  272. puppet := user.bridge.GetPuppetByJID(user.JID)
  273. if len(puppet.CustomMXID) > 0 {
  274. user.log.Debugln("User already has double-puppeting enabled")
  275. // Custom puppet already enabled
  276. return
  277. }
  278. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  279. if err != nil {
  280. user.log.Warnln("Failed to login with shared secret:", err)
  281. return
  282. }
  283. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  284. if err != nil {
  285. puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
  286. return
  287. }
  288. user.log.Infoln("Successfully automatically enabled custom puppet")
  289. }
  290. func (user *User) sendBridgeNotice(formatString string, args ...interface{}) {
  291. notice := fmt.Sprintf(formatString, args...)
  292. _, err := user.bridge.Bot.SendNotice(user.GetManagementRoom(), notice)
  293. if err != nil {
  294. user.log.Warnf("Failed to send bridge notice \"%s\": %v", notice, err)
  295. }
  296. }
  297. func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) {
  298. notice := fmt.Sprintf(formatString, args...)
  299. content := format.RenderMarkdown(notice, true, false)
  300. _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content)
  301. if err != nil {
  302. user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err)
  303. }
  304. }
  305. const callEventMaxAge = 15 * time.Minute
  306. func (user *User) handleCallStart(sender types.JID, id, callType string, ts time.Time) {
  307. if !user.bridge.Config.Bridge.CallStartNotices || ts.Add(callEventMaxAge).Before(time.Now()) {
  308. return
  309. }
  310. portal := user.GetPortalByJID(sender)
  311. text := "Incoming call"
  312. if callType != "" {
  313. text = fmt.Sprintf("Incoming %s call", callType)
  314. }
  315. portal.messages <- PortalMessage{
  316. fake: &fakeMessage{
  317. Sender: sender,
  318. Text: text,
  319. ID: id,
  320. },
  321. source: user,
  322. }
  323. }
  324. func (user *User) HandleEvent(event interface{}) {
  325. switch v := event.(type) {
  326. case *events.LoggedOut:
  327. go user.handleLoggedOut(v.OnConnect)
  328. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  329. user.bridge.Metrics.TrackLoginState(user.JID, false)
  330. case *events.Connected:
  331. go user.sendBridgeState(BridgeState{StateEvent: StateConnected})
  332. user.bridge.Metrics.TrackConnectionState(user.JID, true)
  333. user.bridge.Metrics.TrackLoginState(user.JID, true)
  334. if len(user.Client.Store.PushName) > 0 {
  335. go func() {
  336. err := user.Client.SendPresence(user.lastPresence)
  337. if err != nil {
  338. user.log.Warnln("Failed to send initial presence:", err)
  339. }
  340. }()
  341. }
  342. go user.tryAutomaticDoublePuppeting()
  343. case *events.AppStateSyncComplete:
  344. if len(user.Client.Store.PushName) > 0 && v.Name == appstate.WAPatchCriticalBlock {
  345. err := user.Client.SendPresence(user.lastPresence)
  346. if err != nil {
  347. user.log.Warnln("Failed to send presence after app state sync:", err)
  348. }
  349. } else if v.Name == appstate.WAPatchCriticalUnblockLow {
  350. go func() {
  351. err := user.ResyncContacts()
  352. if err != nil {
  353. user.log.Errorln("Failed to resync puppets: %v", err)
  354. }
  355. }()
  356. }
  357. case *events.PushNameSetting:
  358. // Send presence available when connecting and when the pushname is changed.
  359. // This makes sure that outgoing messages always have the right pushname.
  360. err := user.Client.SendPresence(user.lastPresence)
  361. if err != nil {
  362. user.log.Warnln("Failed to send presence after push name update:", err)
  363. }
  364. case *events.PairSuccess:
  365. user.Session = user.Client.Store
  366. user.JID = v.ID
  367. user.addToJIDMap()
  368. user.Update()
  369. case *events.ConnectFailure, *events.StreamError:
  370. go user.sendBridgeState(BridgeState{StateEvent: StateUnknownError})
  371. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  372. case *events.Disconnected:
  373. go user.sendBridgeState(BridgeState{StateEvent: StateTransientDisconnect})
  374. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  375. case *events.Contact:
  376. go user.syncPuppet(v.JID, "contact event")
  377. case *events.PushName:
  378. go user.syncPuppet(v.JID, "push name event")
  379. case *events.GroupInfo:
  380. go user.handleGroupUpdate(v)
  381. case *events.JoinedGroup:
  382. go user.handleGroupCreate(v)
  383. case *events.Picture:
  384. go user.handlePictureUpdate(v)
  385. case *events.Receipt:
  386. go user.handleReceipt(v)
  387. case *events.ChatPresence:
  388. go user.handleChatPresence(v)
  389. case *events.Message:
  390. portal := user.GetPortalByJID(v.Info.Chat)
  391. portal.messages <- PortalMessage{evt: v, source: user}
  392. case *events.CallOffer:
  393. user.handleCallStart(v.CallCreator, v.CallID, "", v.Timestamp)
  394. case *events.CallOfferNotice:
  395. user.handleCallStart(v.CallCreator, v.CallID, v.Type, v.Timestamp)
  396. case *events.CallTerminate, *events.CallRelayLatency, *events.CallAccept, *events.UnknownCallEvent:
  397. // ignore
  398. case *events.UndecryptableMessage:
  399. portal := user.GetPortalByJID(v.Info.Chat)
  400. portal.messages <- PortalMessage{undecryptable: v, source: user}
  401. case *events.HistorySync:
  402. user.historySyncs <- v
  403. case *events.Mute:
  404. portal := user.GetPortalByJID(v.JID)
  405. if portal != nil {
  406. var mutedUntil time.Time
  407. if v.Action.GetMuted() {
  408. mutedUntil = time.Unix(v.Action.GetMuteEndTimestamp(), 0)
  409. }
  410. go user.updateChatMute(nil, portal, mutedUntil)
  411. }
  412. case *events.Archive:
  413. portal := user.GetPortalByJID(v.JID)
  414. if portal != nil {
  415. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.ArchiveTag, v.Action.GetArchived())
  416. }
  417. case *events.Pin:
  418. portal := user.GetPortalByJID(v.JID)
  419. if portal != nil {
  420. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.PinnedTag, v.Action.GetPinned())
  421. }
  422. case *events.AppState:
  423. // Ignore
  424. default:
  425. user.log.Debugfln("Unknown type of event in HandleEvent: %T", v)
  426. }
  427. }
  428. func (user *User) updateChatMute(intent *appservice.IntentAPI, portal *Portal, mutedUntil time.Time) {
  429. if len(portal.MXID) == 0 || !user.bridge.Config.Bridge.MuteBridging {
  430. return
  431. } else if intent == nil {
  432. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  433. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  434. return
  435. }
  436. intent = doublePuppet.CustomIntent()
  437. }
  438. var err error
  439. if mutedUntil.IsZero() && mutedUntil.Before(time.Now()) {
  440. user.log.Debugfln("Portal %s is muted until %d, unmuting...", portal.MXID, mutedUntil)
  441. err = intent.DeletePushRule("global", pushrules.RoomRule, string(portal.MXID))
  442. } else {
  443. user.log.Debugfln("Portal %s is muted until %d, muting...", portal.MXID, mutedUntil)
  444. err = intent.PutPushRule("global", pushrules.RoomRule, string(portal.MXID), &mautrix.ReqPutPushRule{
  445. Actions: []pushrules.PushActionType{pushrules.ActionDontNotify},
  446. })
  447. }
  448. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  449. user.log.Warnfln("Failed to update push rule for %s through double puppet: %v", portal.MXID, err)
  450. }
  451. }
  452. type CustomTagData struct {
  453. Order json.Number `json:"order"`
  454. DoublePuppet bool `json:"net.maunium.whatsapp.puppet"`
  455. }
  456. type CustomTagEventContent struct {
  457. Tags map[string]CustomTagData `json:"tags"`
  458. }
  459. func (user *User) updateChatTag(intent *appservice.IntentAPI, portal *Portal, tag string, active bool) {
  460. if len(portal.MXID) == 0 || len(tag) == 0 {
  461. return
  462. } else if intent == nil {
  463. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  464. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  465. return
  466. }
  467. intent = doublePuppet.CustomIntent()
  468. }
  469. var existingTags CustomTagEventContent
  470. err := intent.GetTagsWithCustomData(portal.MXID, &existingTags)
  471. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  472. user.log.Warnfln("Failed to get tags of %s: %v", portal.MXID, err)
  473. }
  474. currentTag, ok := existingTags.Tags[tag]
  475. if active && !ok {
  476. user.log.Debugln("Adding tag", tag, "to", portal.MXID)
  477. data := CustomTagData{"0.5", true}
  478. err = intent.AddTagWithCustomData(portal.MXID, tag, &data)
  479. } else if !active && ok && currentTag.DoublePuppet {
  480. user.log.Debugln("Removing tag", tag, "from", portal.MXID)
  481. err = intent.RemoveTag(portal.MXID, tag)
  482. } else {
  483. err = nil
  484. }
  485. if err != nil {
  486. user.log.Warnfln("Failed to update tag %s for %s through double puppet: %v", tag, portal.MXID, err)
  487. }
  488. }
  489. type CustomReadReceipt struct {
  490. Timestamp int64 `json:"ts,omitempty"`
  491. DoublePuppet bool `json:"net.maunium.whatsapp.puppet,omitempty"`
  492. }
  493. func (user *User) syncChatDoublePuppetDetails(portal *Portal, justCreated bool) {
  494. doublePuppet := portal.bridge.GetPuppetByCustomMXID(user.MXID)
  495. if doublePuppet == nil {
  496. return
  497. }
  498. if doublePuppet == nil || doublePuppet.CustomIntent() == nil || len(portal.MXID) == 0 {
  499. return
  500. }
  501. if justCreated || !user.bridge.Config.Bridge.TagOnlyOnCreate {
  502. chat, err := user.Client.Store.ChatSettings.GetChatSettings(portal.Key.JID)
  503. if err != nil {
  504. user.log.Warnfln("Failed to get settings of %s: %v", portal.Key.JID, err)
  505. return
  506. }
  507. intent := doublePuppet.CustomIntent()
  508. user.updateChatMute(intent, portal, chat.MutedUntil)
  509. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, chat.Archived)
  510. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.PinnedTag, chat.Pinned)
  511. }
  512. }
  513. func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
  514. res := make(map[id.UserID][]id.RoomID)
  515. privateChats := user.bridge.DB.Portal.FindPrivateChats(user.JID.ToNonAD())
  516. for _, portal := range privateChats {
  517. if len(portal.MXID) > 0 {
  518. res[user.bridge.FormatPuppetMXID(portal.Key.JID)] = []id.RoomID{portal.MXID}
  519. }
  520. }
  521. return res
  522. }
  523. func (user *User) UpdateDirectChats(chats map[id.UserID][]id.RoomID) {
  524. if !user.bridge.Config.Bridge.SyncDirectChatList {
  525. return
  526. }
  527. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  528. if puppet == nil || puppet.CustomIntent() == nil {
  529. return
  530. }
  531. intent := puppet.CustomIntent()
  532. method := http.MethodPatch
  533. if chats == nil {
  534. chats = user.getDirectChats()
  535. method = http.MethodPut
  536. }
  537. user.log.Debugln("Updating m.direct list on homeserver")
  538. var err error
  539. if user.bridge.Config.Homeserver.Asmux {
  540. urlPath := intent.BuildBaseURL("_matrix", "client", "unstable", "com.beeper.asmux", "dms")
  541. _, err = intent.MakeFullRequest(mautrix.FullRequest{
  542. Method: method,
  543. URL: urlPath,
  544. Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
  545. RequestJSON: chats,
  546. })
  547. } else {
  548. existingChats := make(map[id.UserID][]id.RoomID)
  549. err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
  550. if err != nil {
  551. user.log.Warnln("Failed to get m.direct list to update it:", err)
  552. return
  553. }
  554. for userID, rooms := range existingChats {
  555. if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
  556. // This is not a ghost user, include it in the new list
  557. chats[userID] = rooms
  558. } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
  559. // This is a ghost user, but we're not replacing the whole list, so include it too
  560. chats[userID] = rooms
  561. }
  562. }
  563. err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
  564. }
  565. if err != nil {
  566. user.log.Warnln("Failed to update m.direct list:", err)
  567. }
  568. }
  569. func (user *User) handleLoggedOut(onConnect bool) {
  570. user.JID = types.EmptyJID
  571. user.Update()
  572. if onConnect {
  573. user.sendMarkdownBridgeAlert("Connecting to WhatsApp failed as the device was logged out. Please link the bridge to your phone again.")
  574. } else {
  575. user.sendMarkdownBridgeAlert("You were logged out from another device. Please link the bridge to your phone again.")
  576. }
  577. user.sendBridgeState(BridgeState{StateEvent: StateBadCredentials, Error: WANotLoggedIn})
  578. }
  579. func (user *User) GetPortalByJID(jid types.JID) *Portal {
  580. return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
  581. }
  582. func (user *User) syncPuppet(jid types.JID, reason string) {
  583. user.bridge.GetPuppetByJID(jid).SyncContact(user, false, reason)
  584. }
  585. func (user *User) ResyncContacts() error {
  586. contacts, err := user.Client.Store.Contacts.GetAllContacts()
  587. if err != nil {
  588. return fmt.Errorf("failed to get cached contacts: %w", err)
  589. }
  590. user.log.Infofln("Resyncing displaynames with %d contacts", len(contacts))
  591. for jid, contact := range contacts {
  592. puppet := user.bridge.GetPuppetByJID(jid)
  593. puppet.Sync(user, contact)
  594. }
  595. return nil
  596. }
  597. func (user *User) ResyncGroups(createPortals bool) error {
  598. groups, err := user.Client.GetJoinedGroups()
  599. if err != nil {
  600. return fmt.Errorf("failed to get group list from server: %w", err)
  601. }
  602. for _, group := range groups {
  603. portal := user.GetPortalByJID(group.JID)
  604. if len(portal.MXID) == 0 {
  605. if createPortals {
  606. err = portal.CreateMatrixRoom(user, group, true)
  607. if err != nil {
  608. return fmt.Errorf("failed to create room for %s: %w", group.JID, err)
  609. }
  610. }
  611. } else {
  612. portal.UpdateMatrixRoom(user, group)
  613. }
  614. }
  615. return nil
  616. }
  617. const WATypingTimeout = 15 * time.Second
  618. func (user *User) handleChatPresence(presence *events.ChatPresence) {
  619. puppet := user.bridge.GetPuppetByJID(presence.Sender)
  620. portal := user.GetPortalByJID(presence.Chat)
  621. if puppet == nil || portal == nil || len(portal.MXID) == 0 {
  622. return
  623. }
  624. if presence.State == types.ChatPresenceComposing {
  625. if puppet.typingIn != "" && puppet.typingAt.Add(WATypingTimeout).Before(time.Now()) {
  626. if puppet.typingIn == portal.MXID {
  627. return
  628. }
  629. _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  630. }
  631. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, WATypingTimeout.Milliseconds())
  632. puppet.typingIn = portal.MXID
  633. puppet.typingAt = time.Now()
  634. } else {
  635. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, false, 0)
  636. puppet.typingIn = ""
  637. }
  638. }
  639. func (user *User) handleReceipt(receipt *events.Receipt) {
  640. if receipt.Type != events.ReceiptTypeRead {
  641. return
  642. }
  643. portal := user.GetPortalByJID(receipt.Chat)
  644. if portal == nil || len(portal.MXID) == 0 {
  645. return
  646. }
  647. if receipt.IsFromMe {
  648. user.markSelfRead(portal, receipt.MessageID)
  649. } else {
  650. intent := user.bridge.GetPuppetByJID(receipt.Sender).IntentFor(portal)
  651. ok := user.markOtherRead(portal, intent, receipt.MessageID)
  652. if !ok {
  653. // Message not found, try any previous IDs
  654. for i := len(receipt.PreviousIDs) - 1; i >= 0; i-- {
  655. ok = user.markOtherRead(portal, intent, receipt.PreviousIDs[i])
  656. if ok {
  657. break
  658. }
  659. }
  660. }
  661. }
  662. }
  663. func (user *User) markOtherRead(portal *Portal, intent *appservice.IntentAPI, messageID types.MessageID) bool {
  664. msg := user.bridge.DB.Message.GetByJID(portal.Key, messageID)
  665. if msg == nil || msg.IsFakeMXID() {
  666. return false
  667. }
  668. err := intent.MarkReadWithContent(portal.MXID, msg.MXID, &CustomReadReceipt{DoublePuppet: intent.IsCustomPuppet})
  669. if err != nil {
  670. user.log.Warnfln("Failed to mark message %s as read by %s: %v", msg.MXID, intent.UserID, err)
  671. }
  672. return true
  673. }
  674. func (user *User) markSelfRead(portal *Portal, messageID types.MessageID) {
  675. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  676. if puppet == nil || puppet.CustomIntent() == nil {
  677. return
  678. }
  679. var message *database.Message
  680. if messageID == "" {
  681. message = user.bridge.DB.Message.GetLastInChat(portal.Key)
  682. if message == nil {
  683. return
  684. }
  685. user.log.Debugfln("User read chat %s/%s in WhatsApp mobile (last known event: %s/%s)", portal.Key.JID, portal.MXID, message.JID, message.MXID)
  686. } else {
  687. message = user.bridge.DB.Message.GetByJID(portal.Key, messageID)
  688. if message == nil || message.IsFakeMXID() {
  689. return
  690. }
  691. user.log.Debugfln("User read message %s/%s in %s/%s in WhatsApp mobile", message.JID, message.MXID, portal.Key.JID, portal.MXID)
  692. }
  693. err := puppet.CustomIntent().MarkReadWithContent(portal.MXID, message.MXID, &CustomReadReceipt{DoublePuppet: true})
  694. if err != nil {
  695. user.log.Warnfln("Failed to bridge own read receipt in %s: %v", portal.Key.JID, err)
  696. }
  697. }
  698. func (user *User) markSelfReadFull(portal *Portal) {
  699. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  700. if puppet == nil || puppet.CustomIntent() == nil {
  701. return
  702. }
  703. lastMessage := user.bridge.DB.Message.GetLastInChat(portal.Key)
  704. if lastMessage == nil {
  705. return
  706. }
  707. err := puppet.CustomIntent().MarkReadWithContent(portal.MXID, lastMessage.MXID, &CustomReadReceipt{DoublePuppet: true})
  708. if err != nil {
  709. user.log.Warnfln("Failed to mark %s in %s as read after backfill: %v", lastMessage.MXID, portal.MXID, err)
  710. }
  711. }
  712. func (user *User) handleGroupCreate(evt *events.JoinedGroup) {
  713. portal := user.GetPortalByJID(evt.JID)
  714. if len(portal.MXID) == 0 {
  715. err := portal.CreateMatrixRoom(user, &evt.GroupInfo, true)
  716. if err != nil {
  717. user.log.Errorln("Failed to create Matrix room after join notification: %v", err)
  718. }
  719. } else {
  720. portal.UpdateMatrixRoom(user, &evt.GroupInfo)
  721. }
  722. }
  723. func (user *User) handleGroupUpdate(evt *events.GroupInfo) {
  724. portal := user.GetPortalByJID(evt.JID)
  725. if portal == nil || len(portal.MXID) == 0 {
  726. user.log.Debugfln("Ignoring group info update in chat with no portal: %+v", evt)
  727. return
  728. }
  729. switch {
  730. case evt.Announce != nil:
  731. portal.RestrictMessageSending(evt.Announce.IsAnnounce)
  732. case evt.Locked != nil:
  733. portal.RestrictMetadataChanges(evt.Locked.IsLocked)
  734. case evt.Name != nil:
  735. portal.UpdateName(evt.Name.Name, evt.Name.NameSetBy, true)
  736. case evt.Topic != nil:
  737. portal.UpdateTopic(evt.Topic.Topic, evt.Topic.TopicSetBy, true)
  738. case evt.Leave != nil:
  739. if evt.Sender != nil && !evt.Sender.IsEmpty() {
  740. portal.HandleWhatsAppKick(user, *evt.Sender, evt.Leave)
  741. }
  742. case evt.Join != nil:
  743. portal.HandleWhatsAppInvite(user, evt.Sender, evt.Join)
  744. case evt.Promote != nil:
  745. portal.ChangeAdminStatus(evt.Promote, true)
  746. case evt.Demote != nil:
  747. portal.ChangeAdminStatus(evt.Demote, false)
  748. }
  749. }
  750. func (user *User) handlePictureUpdate(evt *events.Picture) {
  751. if evt.JID.Server == types.DefaultUserServer {
  752. puppet := user.bridge.GetPuppetByJID(evt.JID)
  753. user.log.Debugfln("Received picture update for puppet %s (current: %s, new: %s)", evt.JID, puppet.Avatar, evt.PictureID)
  754. if puppet.Avatar != evt.PictureID {
  755. puppet.UpdateAvatar(user)
  756. }
  757. } else if portal := user.GetPortalByJID(evt.JID); portal != nil {
  758. user.log.Debugfln("Received picture update for portal %s (current: %s, new: %s)", evt.JID, portal.Avatar, evt.PictureID)
  759. if portal.Avatar != evt.PictureID {
  760. portal.UpdateAvatar(user, evt.Author, true)
  761. }
  762. }
  763. }