user.go 27 KB

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