user.go 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841
  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. "encoding/json"
  19. "errors"
  20. "fmt"
  21. "net/http"
  22. "sync"
  23. "time"
  24. log "maunium.net/go/maulogger/v2"
  25. "maunium.net/go/mautrix/appservice"
  26. "maunium.net/go/mautrix/pushrules"
  27. "go.mau.fi/whatsmeow"
  28. "go.mau.fi/whatsmeow/store"
  29. "go.mau.fi/whatsmeow/types"
  30. "go.mau.fi/whatsmeow/types/events"
  31. waLog "go.mau.fi/whatsmeow/util/log"
  32. "maunium.net/go/mautrix"
  33. "maunium.net/go/mautrix/event"
  34. "maunium.net/go/mautrix/format"
  35. "maunium.net/go/mautrix/id"
  36. "maunium.net/go/mautrix-whatsapp/database"
  37. )
  38. type User struct {
  39. *database.User
  40. Client *whatsmeow.Client
  41. Session *store.Device
  42. bridge *Bridge
  43. log log.Logger
  44. Admin bool
  45. Whitelisted bool
  46. RelaybotWhitelisted bool
  47. IsRelaybot bool
  48. mgmtCreateLock sync.Mutex
  49. connLock sync.Mutex
  50. qrListener chan<- *events.QR
  51. loginListener chan<- *events.PairSuccess
  52. prevBridgeStatus *BridgeState
  53. }
  54. func (bridge *Bridge) GetUserByMXID(userID id.UserID) *User {
  55. _, isPuppet := bridge.ParsePuppetMXID(userID)
  56. if isPuppet || userID == bridge.Bot.UserID {
  57. return nil
  58. }
  59. bridge.usersLock.Lock()
  60. defer bridge.usersLock.Unlock()
  61. user, ok := bridge.usersByMXID[userID]
  62. if !ok {
  63. return bridge.loadDBUser(bridge.DB.User.GetByMXID(userID), &userID)
  64. }
  65. return user
  66. }
  67. func (bridge *Bridge) GetUserByJID(jid types.JID) *User {
  68. bridge.usersLock.Lock()
  69. defer bridge.usersLock.Unlock()
  70. user, ok := bridge.usersByUsername[jid.User]
  71. if !ok {
  72. return bridge.loadDBUser(bridge.DB.User.GetByUsername(jid.User), nil)
  73. }
  74. return user
  75. }
  76. func (user *User) addToJIDMap() {
  77. user.bridge.usersLock.Lock()
  78. user.bridge.usersByUsername[user.JID.User] = user
  79. user.bridge.usersLock.Unlock()
  80. }
  81. func (user *User) removeFromJIDMap(state BridgeStateEvent) {
  82. user.bridge.usersLock.Lock()
  83. jidUser, ok := user.bridge.usersByUsername[user.JID.User]
  84. if ok && user == jidUser {
  85. delete(user.bridge.usersByUsername, user.JID.User)
  86. }
  87. user.bridge.usersLock.Unlock()
  88. user.bridge.Metrics.TrackLoginState(user.JID, false)
  89. user.sendBridgeState(BridgeState{StateEvent: state, Error: WANotLoggedIn})
  90. }
  91. func (bridge *Bridge) GetAllUsers() []*User {
  92. bridge.usersLock.Lock()
  93. defer bridge.usersLock.Unlock()
  94. dbUsers := bridge.DB.User.GetAll()
  95. output := make([]*User, len(dbUsers))
  96. for index, dbUser := range dbUsers {
  97. user, ok := bridge.usersByMXID[dbUser.MXID]
  98. if !ok {
  99. user = bridge.loadDBUser(dbUser, nil)
  100. }
  101. output[index] = user
  102. }
  103. return output
  104. }
  105. func (bridge *Bridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User {
  106. if dbUser == nil {
  107. if mxid == nil {
  108. return nil
  109. }
  110. dbUser = bridge.DB.User.New()
  111. dbUser.MXID = *mxid
  112. dbUser.Insert()
  113. }
  114. user := bridge.NewUser(dbUser)
  115. bridge.usersByMXID[user.MXID] = user
  116. if !user.JID.IsEmpty() {
  117. var err error
  118. user.Session, err = bridge.WAContainer.GetDevice(user.JID)
  119. if err != nil {
  120. user.log.Errorfln("Failed to scan user's whatsapp session: %v", err)
  121. } else if user.Session == nil {
  122. user.log.Warnfln("Didn't find session data for %s, treating user as logged out", user.JID)
  123. user.JID = types.EmptyJID
  124. user.Update()
  125. } else {
  126. bridge.usersByUsername[user.JID.User] = user
  127. }
  128. }
  129. if len(user.ManagementRoom) > 0 {
  130. bridge.managementRooms[user.ManagementRoom] = user
  131. }
  132. return user
  133. }
  134. func (user *User) GetPortals() []*Portal {
  135. // FIXME
  136. //keys := user.User.GetPortalKeys()
  137. //portals := make([]*Portal, len(keys))
  138. //
  139. //user.bridge.portalsLock.Lock()
  140. //for i, key := range keys {
  141. // portal, ok := user.bridge.portalsByJID[key]
  142. // if !ok {
  143. // portal = user.bridge.loadDBPortal(user.bridge.DB.Portal.GetByJID(key), &key)
  144. // }
  145. // portals[i] = portal
  146. //}
  147. //user.bridge.portalsLock.Unlock()
  148. //return portals
  149. return nil
  150. }
  151. func (bridge *Bridge) NewUser(dbUser *database.User) *User {
  152. user := &User{
  153. User: dbUser,
  154. bridge: bridge,
  155. log: bridge.Log.Sub("User").Sub(string(dbUser.MXID)),
  156. IsRelaybot: false,
  157. }
  158. user.RelaybotWhitelisted = user.bridge.Config.Bridge.Permissions.IsRelaybotWhitelisted(user.MXID)
  159. user.Whitelisted = user.bridge.Config.Bridge.Permissions.IsWhitelisted(user.MXID)
  160. user.Admin = user.bridge.Config.Bridge.Permissions.IsAdmin(user.MXID)
  161. return user
  162. }
  163. func (user *User) GetManagementRoom() id.RoomID {
  164. if len(user.ManagementRoom) == 0 {
  165. user.mgmtCreateLock.Lock()
  166. defer user.mgmtCreateLock.Unlock()
  167. if len(user.ManagementRoom) > 0 {
  168. return user.ManagementRoom
  169. }
  170. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  171. Topic: "WhatsApp bridge notices",
  172. IsDirect: true,
  173. })
  174. if err != nil {
  175. user.log.Errorln("Failed to auto-create management room:", err)
  176. } else {
  177. user.SetManagementRoom(resp.RoomID)
  178. }
  179. }
  180. return user.ManagementRoom
  181. }
  182. func (user *User) SetManagementRoom(roomID id.RoomID) {
  183. existingUser, ok := user.bridge.managementRooms[roomID]
  184. if ok {
  185. existingUser.ManagementRoom = ""
  186. existingUser.Update()
  187. }
  188. user.ManagementRoom = roomID
  189. user.bridge.managementRooms[user.ManagementRoom] = user
  190. user.Update()
  191. }
  192. type waLogger struct{ l log.Logger }
  193. func (w *waLogger) Debugf(msg string, args ...interface{}) { w.l.Debugfln(msg, args...) }
  194. func (w *waLogger) Infof(msg string, args ...interface{}) { w.l.Infofln(msg, args...) }
  195. func (w *waLogger) Warnf(msg string, args ...interface{}) { w.l.Warnfln(msg, args...) }
  196. func (w *waLogger) Errorf(msg string, args ...interface{}) { w.l.Errorfln(msg, args...) }
  197. func (w *waLogger) Sub(module string) waLog.Logger { return &waLogger{l: w.l.Sub(module)} }
  198. func (user *User) Connect(evenIfNoSession bool) bool {
  199. user.connLock.Lock()
  200. defer user.connLock.Unlock()
  201. if user.Client != nil {
  202. return user.Client.IsConnected()
  203. } else if !evenIfNoSession && user.Session == nil {
  204. return false
  205. }
  206. user.log.Debugln("Connecting to WhatsApp")
  207. if user.Session != nil {
  208. user.sendBridgeState(BridgeState{StateEvent: StateConnecting, Error: WAConnecting})
  209. }
  210. if user.Session == nil {
  211. newSession := user.bridge.WAContainer.NewDevice()
  212. newSession.Log = &waLogger{user.log.Sub("Session")}
  213. user.Client = whatsmeow.NewClient(newSession, &waLogger{user.log.Sub("Client")})
  214. } else {
  215. user.Client = whatsmeow.NewClient(user.Session, &waLogger{user.log.Sub("Client")})
  216. }
  217. user.Client.AddEventHandler(user.HandleEvent)
  218. err := user.Client.Connect()
  219. if err != nil {
  220. user.log.Warnln("Error connecting to WhatsApp:", err)
  221. return false
  222. }
  223. return true
  224. }
  225. func (user *User) DeleteConnection() {
  226. user.connLock.Lock()
  227. defer user.connLock.Unlock()
  228. if user.Client == nil {
  229. return
  230. }
  231. user.Client.Disconnect()
  232. user.Client.RemoveEventHandlers()
  233. user.Client = nil
  234. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  235. user.sendBridgeState(BridgeState{StateEvent: StateBadCredentials, Error: WANotConnected})
  236. }
  237. func (user *User) HasSession() bool {
  238. return user.Session != nil
  239. }
  240. func (user *User) DeleteSession() {
  241. if user.Session != nil {
  242. err := user.Session.Delete()
  243. if err != nil {
  244. user.log.Warnln("Failed to delete session:", err)
  245. }
  246. user.Session = nil
  247. }
  248. if !user.JID.IsEmpty() {
  249. user.JID = types.EmptyJID
  250. user.Update()
  251. }
  252. }
  253. func (user *User) IsLoggedIn() bool {
  254. return user.Client != nil && user.Client.IsConnected() && user.Client.IsLoggedIn
  255. }
  256. func (user *User) PostLogin() {
  257. user.sendBridgeState(BridgeState{StateEvent: StateConnected})
  258. user.bridge.Metrics.TrackConnectionState(user.JID, true)
  259. user.bridge.Metrics.TrackLoginState(user.JID, true)
  260. go user.tryAutomaticDoublePuppeting()
  261. }
  262. func (user *User) tryAutomaticDoublePuppeting() {
  263. if !user.bridge.Config.CanDoublePuppet(user.MXID) {
  264. return
  265. }
  266. user.log.Debugln("Checking if double puppeting needs to be enabled")
  267. puppet := user.bridge.GetPuppetByJID(user.JID)
  268. if len(puppet.CustomMXID) > 0 {
  269. user.log.Debugln("User already has double-puppeting enabled")
  270. // Custom puppet already enabled
  271. return
  272. }
  273. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  274. if err != nil {
  275. user.log.Warnln("Failed to login with shared secret:", err)
  276. return
  277. }
  278. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  279. if err != nil {
  280. puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
  281. return
  282. }
  283. user.log.Infoln("Successfully automatically enabled custom puppet")
  284. }
  285. func (user *User) sendBridgeNotice(formatString string, args ...interface{}) {
  286. notice := fmt.Sprintf(formatString, args...)
  287. _, err := user.bridge.Bot.SendNotice(user.GetManagementRoom(), notice)
  288. if err != nil {
  289. user.log.Warnf("Failed to send bridge notice \"%s\": %v", notice, err)
  290. }
  291. }
  292. func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) {
  293. notice := fmt.Sprintf(formatString, args...)
  294. content := format.RenderMarkdown(notice, true, false)
  295. _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content)
  296. if err != nil {
  297. user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err)
  298. }
  299. }
  300. func (user *User) HandleEvent(event interface{}) {
  301. switch v := event.(type) {
  302. case *events.LoggedOut:
  303. go user.handleLoggedOut()
  304. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  305. user.bridge.Metrics.TrackLoginState(user.JID, false)
  306. case *events.Connected:
  307. go user.sendBridgeState(BridgeState{StateEvent: StateConnected})
  308. user.bridge.Metrics.TrackConnectionState(user.JID, true)
  309. user.bridge.Metrics.TrackLoginState(user.JID, true)
  310. go func() {
  311. err := user.Client.SendPresence(types.PresenceUnavailable)
  312. if err != nil {
  313. user.log.Warnln("Failed to send initial presence:", err)
  314. }
  315. }()
  316. case *events.PairSuccess:
  317. user.JID = v.ID
  318. user.addToJIDMap()
  319. user.Update()
  320. user.Session = user.Client.Store
  321. if user.loginListener != nil {
  322. select {
  323. case user.loginListener <- v:
  324. return
  325. default:
  326. }
  327. }
  328. user.log.Warnln("Got pair success event, but nothing waiting for it")
  329. case *events.QR:
  330. if user.qrListener != nil {
  331. select {
  332. case user.qrListener <- v:
  333. return
  334. default:
  335. }
  336. }
  337. user.log.Warnln("Got QR code event, but nothing waiting for it")
  338. case *events.ConnectFailure, *events.StreamError:
  339. go user.sendBridgeState(BridgeState{StateEvent: StateUnknownError})
  340. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  341. case *events.Disconnected:
  342. go user.sendBridgeState(BridgeState{StateEvent: StateTransientDisconnect})
  343. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  344. case *events.Contact:
  345. go user.syncPuppet(v.JID)
  346. case *events.PushName:
  347. go user.syncPuppet(v.JID)
  348. case *events.Receipt:
  349. go user.handleReceipt(v)
  350. case *events.Message:
  351. portal := user.GetPortalByJID(v.Info.Chat)
  352. portal.messages <- PortalMessage{v, user}
  353. case *events.Mute:
  354. portal := user.bridge.GetPortalByJID(user.PortalKey(v.JID))
  355. if portal != nil {
  356. var mutedUntil time.Time
  357. if v.Action.GetMuted() {
  358. mutedUntil = time.Unix(v.Action.GetMuteEndTimestamp(), 0)
  359. }
  360. go user.updateChatMute(nil, portal, mutedUntil)
  361. }
  362. case *events.Archive:
  363. portal := user.bridge.GetPortalByJID(user.PortalKey(v.JID))
  364. if portal != nil {
  365. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.ArchiveTag, v.Action.GetArchived())
  366. }
  367. case *events.Pin:
  368. portal := user.bridge.GetPortalByJID(user.PortalKey(v.JID))
  369. if portal != nil {
  370. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.PinnedTag, v.Action.GetPinned())
  371. }
  372. default:
  373. user.log.Debugfln("Unknown type of event in HandleEvent: %T", v)
  374. }
  375. }
  376. func (user *User) updateChatMute(intent *appservice.IntentAPI, portal *Portal, mutedUntil time.Time) {
  377. if len(portal.MXID) == 0 || !user.bridge.Config.Bridge.MuteBridging {
  378. return
  379. } else if intent == nil {
  380. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  381. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  382. return
  383. }
  384. intent = doublePuppet.CustomIntent()
  385. }
  386. var err error
  387. if mutedUntil.IsZero() && mutedUntil.Before(time.Now()) {
  388. user.log.Debugfln("Portal %s is muted until %d, unmuting...", portal.MXID, mutedUntil)
  389. err = intent.DeletePushRule("global", pushrules.RoomRule, string(portal.MXID))
  390. } else {
  391. user.log.Debugfln("Portal %s is muted until %d, muting...", portal.MXID, mutedUntil)
  392. err = intent.PutPushRule("global", pushrules.RoomRule, string(portal.MXID), &mautrix.ReqPutPushRule{
  393. Actions: []pushrules.PushActionType{pushrules.ActionDontNotify},
  394. })
  395. }
  396. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  397. user.log.Warnfln("Failed to update push rule for %s through double puppet: %v", portal.MXID, err)
  398. }
  399. }
  400. type CustomTagData struct {
  401. Order json.Number `json:"order"`
  402. DoublePuppet bool `json:"net.maunium.whatsapp.puppet"`
  403. }
  404. type CustomTagEventContent struct {
  405. Tags map[string]CustomTagData `json:"tags"`
  406. }
  407. func (user *User) updateChatTag(intent *appservice.IntentAPI, portal *Portal, tag string, active bool) {
  408. if len(portal.MXID) == 0 || len(tag) == 0 {
  409. return
  410. } else if intent == nil {
  411. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  412. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  413. return
  414. }
  415. intent = doublePuppet.CustomIntent()
  416. }
  417. var existingTags CustomTagEventContent
  418. err := intent.GetTagsWithCustomData(portal.MXID, &existingTags)
  419. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  420. user.log.Warnfln("Failed to get tags of %s: %v", portal.MXID, err)
  421. }
  422. currentTag, ok := existingTags.Tags[tag]
  423. if active && !ok {
  424. user.log.Debugln("Adding tag", tag, "to", portal.MXID)
  425. data := CustomTagData{"0.5", true}
  426. err = intent.AddTagWithCustomData(portal.MXID, tag, &data)
  427. } else if !active && ok && currentTag.DoublePuppet {
  428. user.log.Debugln("Removing tag", tag, "from", portal.MXID)
  429. err = intent.RemoveTag(portal.MXID, tag)
  430. } else {
  431. err = nil
  432. }
  433. if err != nil {
  434. user.log.Warnfln("Failed to update tag %s for %s through double puppet: %v", tag, portal.MXID, err)
  435. }
  436. }
  437. type CustomReadReceipt struct {
  438. Timestamp int64 `json:"ts,omitempty"`
  439. DoublePuppet bool `json:"net.maunium.whatsapp.puppet,omitempty"`
  440. }
  441. func (user *User) syncChatDoublePuppetDetails(doublePuppet *Puppet, portal *Portal, justCreated bool) {
  442. if doublePuppet == nil || doublePuppet.CustomIntent() == nil || len(portal.MXID) == 0 {
  443. return
  444. }
  445. intent := doublePuppet.CustomIntent()
  446. // FIXME this might not be possible to do anymore
  447. //if chat.UnreadCount == 0 && (justCreated || !user.bridge.Config.Bridge.MarkReadOnlyOnCreate) {
  448. // lastMessage := user.bridge.DB.Message.GetLastInChatBefore(chat.Portal.Key, chat.ReceivedAt.Unix())
  449. // if lastMessage != nil {
  450. // err := intent.MarkReadWithContent(chat.Portal.MXID, lastMessage.MXID, &CustomReadReceipt{DoublePuppet: true})
  451. // if err != nil {
  452. // user.log.Warnfln("Failed to mark %s in %s as read after backfill: %v", lastMessage.MXID, chat.Portal.MXID, err)
  453. // }
  454. // }
  455. //} else if chat.UnreadCount == -1 {
  456. // user.log.Debugfln("Invalid unread count (missing field?) in chat info %+v", chat.Source)
  457. //}
  458. if justCreated || !user.bridge.Config.Bridge.TagOnlyOnCreate {
  459. chat, err := user.Client.Store.ChatSettings.GetChatSettings(portal.Key.JID)
  460. if err != nil {
  461. user.log.Warnfln("Failed to get settings of %s: %v", portal.Key.JID, err)
  462. return
  463. }
  464. user.updateChatMute(intent, portal, chat.MutedUntil)
  465. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, chat.Archived)
  466. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.PinnedTag, chat.Pinned)
  467. }
  468. }
  469. //func (user *User) syncPortal(chat Chat) {
  470. // // Don't sync unless chat meta sync is enabled or portal doesn't exist
  471. // if user.bridge.Config.Bridge.ChatMetaSync || len(chat.Portal.MXID) == 0 {
  472. // failedToCreate := chat.Portal.Sync(user, chat.Contact)
  473. // if failedToCreate {
  474. // return
  475. // }
  476. // }
  477. // err := chat.Portal.BackfillHistory(user, chat.LastMessageTime)
  478. // if err != nil {
  479. // chat.Portal.log.Errorln("Error backfilling history:", err)
  480. // }
  481. //}
  482. //
  483. //func (user *User) syncPortals(chatMap map[string]whatsapp.Chat, createAll bool) {
  484. // // TODO use contexts instead of checking if user.Conn is the same?
  485. // connAtStart := user.Conn
  486. //
  487. // chats := user.collectChatList(chatMap)
  488. //
  489. // limit := user.bridge.Config.Bridge.InitialChatSync
  490. // if limit < 0 {
  491. // limit = len(chats)
  492. // }
  493. // if user.Conn != connAtStart {
  494. // user.log.Debugln("Connection seems to have changed before sync, cancelling")
  495. // return
  496. // }
  497. // now := time.Now().Unix()
  498. // user.log.Infoln("Syncing portals")
  499. // doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  500. // for i, chat := range chats {
  501. // if chat.LastMessageTime+user.bridge.Config.Bridge.SyncChatMaxAge < now {
  502. // break
  503. // }
  504. // create := (chat.LastMessageTime >= user.LastConnection && user.LastConnection > 0) || i < limit
  505. // if len(chat.Portal.MXID) > 0 || create || createAll {
  506. // user.log.Debugfln("Syncing chat %+v", chat.Chat.Source)
  507. // justCreated := len(chat.Portal.MXID) == 0
  508. // user.syncPortal(chat)
  509. // user.syncChatDoublePuppetDetails(doublePuppet, chat, justCreated)
  510. // }
  511. // }
  512. // if user.Conn != connAtStart {
  513. // user.log.Debugln("Connection seems to have changed during sync, cancelling")
  514. // return
  515. // }
  516. // user.UpdateDirectChats(nil)
  517. //
  518. // user.log.Infoln("Finished syncing portals")
  519. // select {
  520. // case user.syncPortalsDone <- struct{}{}:
  521. // default:
  522. // }
  523. //}
  524. func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
  525. res := make(map[id.UserID][]id.RoomID)
  526. privateChats := user.bridge.DB.Portal.FindPrivateChats(user.JID.ToNonAD())
  527. for _, portal := range privateChats {
  528. if len(portal.MXID) > 0 {
  529. res[user.bridge.FormatPuppetMXID(portal.Key.JID)] = []id.RoomID{portal.MXID}
  530. }
  531. }
  532. return res
  533. }
  534. func (user *User) UpdateDirectChats(chats map[id.UserID][]id.RoomID) {
  535. if !user.bridge.Config.Bridge.SyncDirectChatList {
  536. return
  537. }
  538. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  539. if puppet == nil || puppet.CustomIntent() == nil {
  540. return
  541. }
  542. intent := puppet.CustomIntent()
  543. method := http.MethodPatch
  544. if chats == nil {
  545. chats = user.getDirectChats()
  546. method = http.MethodPut
  547. }
  548. user.log.Debugln("Updating m.direct list on homeserver")
  549. var err error
  550. if user.bridge.Config.Homeserver.Asmux {
  551. urlPath := intent.BuildBaseURL("_matrix", "client", "unstable", "com.beeper.asmux", "dms")
  552. _, err = intent.MakeFullRequest(mautrix.FullRequest{
  553. Method: method,
  554. URL: urlPath,
  555. Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
  556. RequestJSON: chats,
  557. })
  558. } else {
  559. existingChats := make(map[id.UserID][]id.RoomID)
  560. err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
  561. if err != nil {
  562. user.log.Warnln("Failed to get m.direct list to update it:", err)
  563. return
  564. }
  565. for userID, rooms := range existingChats {
  566. if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
  567. // This is not a ghost user, include it in the new list
  568. chats[userID] = rooms
  569. } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
  570. // This is a ghost user, but we're not replacing the whole list, so include it too
  571. chats[userID] = rooms
  572. }
  573. }
  574. err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
  575. }
  576. if err != nil {
  577. user.log.Warnln("Failed to update m.direct list:", err)
  578. }
  579. }
  580. func (user *User) handleLoggedOut() {
  581. user.JID = types.EmptyJID
  582. user.Update()
  583. user.sendMarkdownBridgeAlert("Connecting to WhatsApp failed as the device was logged out. Please link the bridge to your phone again.")
  584. user.sendBridgeState(BridgeState{StateEvent: StateBadCredentials, Error: WANotLoggedIn})
  585. }
  586. func (user *User) PortalKey(jid types.JID) database.PortalKey {
  587. return database.NewPortalKey(jid, user.JID)
  588. }
  589. func (user *User) GetPortalByJID(jid types.JID) *Portal {
  590. return user.bridge.GetPortalByJID(user.PortalKey(jid))
  591. }
  592. func (user *User) syncPuppet(jid types.JID) {
  593. user.bridge.GetPuppetByJID(jid).SyncContact(user, false)
  594. }
  595. //func (user *User) HandlePresence(info whatsapp.PresenceEvent) {
  596. // puppet := user.bridge.GetPuppetByJID(info.SenderJID)
  597. // switch info.Status {
  598. // case whatsapp.PresenceUnavailable:
  599. // _ = puppet.DefaultIntent().SetPresence("offline")
  600. // case whatsapp.PresenceAvailable:
  601. // if len(puppet.typingIn) > 0 && puppet.typingAt+15 > time.Now().Unix() {
  602. // portal := user.bridge.GetPortalByMXID(puppet.typingIn)
  603. // _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  604. // puppet.typingIn = ""
  605. // puppet.typingAt = 0
  606. // } else {
  607. // _ = puppet.DefaultIntent().SetPresence("online")
  608. // }
  609. // case whatsapp.PresenceComposing:
  610. // portal := user.GetPortalByJID(info.JID)
  611. // if len(puppet.typingIn) > 0 && puppet.typingAt+15 > time.Now().Unix() {
  612. // if puppet.typingIn == portal.MXID {
  613. // return
  614. // }
  615. // _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  616. // }
  617. // if len(portal.MXID) > 0 {
  618. // puppet.typingIn = portal.MXID
  619. // puppet.typingAt = time.Now().Unix()
  620. // _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, 15*1000)
  621. // }
  622. // }
  623. //}
  624. func (user *User) handleReceipt(receipt *events.Receipt) {
  625. if receipt.Type != events.ReceiptTypeRead {
  626. return
  627. }
  628. portal := user.GetPortalByJID(receipt.Chat)
  629. if portal == nil || len(portal.MXID) == 0 {
  630. return
  631. }
  632. if receipt.IsFromMe {
  633. user.markSelfRead(portal, receipt.MessageID)
  634. } else {
  635. intent := user.bridge.GetPuppetByJID(receipt.Sender).IntentFor(portal)
  636. ok := user.markOtherRead(portal, intent, receipt.MessageID)
  637. if !ok {
  638. // Message not found, try any previous IDs
  639. for i := len(receipt.PreviousIDs) - 1; i >= 0; i-- {
  640. ok = user.markOtherRead(portal, intent, receipt.PreviousIDs[i])
  641. if ok {
  642. break
  643. }
  644. }
  645. }
  646. }
  647. }
  648. func (user *User) markOtherRead(portal *Portal, intent *appservice.IntentAPI, messageID types.MessageID) bool {
  649. msg := user.bridge.DB.Message.GetByJID(portal.Key, messageID)
  650. if msg == nil || msg.IsFakeMXID() {
  651. return false
  652. }
  653. err := intent.MarkReadWithContent(portal.MXID, msg.MXID, &CustomReadReceipt{DoublePuppet: intent.IsCustomPuppet})
  654. if err != nil {
  655. user.log.Warnfln("Failed to mark message %s as read by %s: %v", msg.MXID, intent.UserID, err)
  656. }
  657. return true
  658. }
  659. func (user *User) markSelfRead(portal *Portal, messageID types.MessageID) {
  660. puppet := user.bridge.GetPuppetByJID(user.JID)
  661. if puppet == nil {
  662. return
  663. }
  664. intent := puppet.CustomIntent()
  665. if intent == nil {
  666. return
  667. }
  668. var message *database.Message
  669. if messageID == "" {
  670. message = user.bridge.DB.Message.GetLastInChat(portal.Key)
  671. if message == nil {
  672. return
  673. }
  674. 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)
  675. } else {
  676. message = user.bridge.DB.Message.GetByJID(portal.Key, messageID)
  677. if message == nil || message.IsFakeMXID() {
  678. return
  679. }
  680. user.log.Debugfln("User read message %s/%s in %s/%s in WhatsApp mobile", message.JID, message.MXID, portal.Key.JID, portal.MXID)
  681. }
  682. err := intent.MarkReadWithContent(portal.MXID, message.MXID, &CustomReadReceipt{DoublePuppet: true})
  683. if err != nil {
  684. user.log.Warnfln("Failed to bridge own read receipt in %s: %v", portal.Key.JID, err)
  685. }
  686. }
  687. //func (user *User) HandleCommand(cmd whatsapp.JSONCommand) {
  688. // switch cmd.Type {
  689. // case whatsapp.CommandPicture:
  690. // if strings.HasSuffix(cmd.JID, whatsapp.NewUserSuffix) {
  691. // puppet := user.bridge.GetPuppetByJID(cmd.JID)
  692. // go puppet.UpdateAvatar(user, cmd.ProfilePicInfo)
  693. // } else if user.bridge.Config.Bridge.ChatMetaSync {
  694. // portal := user.GetPortalByJID(cmd.JID)
  695. // go portal.UpdateAvatar(user, cmd.ProfilePicInfo, true)
  696. // }
  697. // case whatsapp.CommandDisconnect:
  698. // if cmd.Kind == "replaced" {
  699. // user.cleanDisconnection = true
  700. // go user.sendMarkdownBridgeAlert("\u26a0 Your WhatsApp connection was closed by the server because you opened another WhatsApp Web client.\n\n" +
  701. // "Use the `reconnect` command to disconnect the other client and resume bridging.")
  702. // } else {
  703. // user.log.Warnln("Unknown kind of disconnect:", string(cmd.Raw))
  704. // go user.sendMarkdownBridgeAlert("\u26a0 Your WhatsApp connection was closed by the server (reason code: %s).\n\n"+
  705. // "Use the `reconnect` command to reconnect.", cmd.Kind)
  706. // }
  707. // }
  708. //}
  709. //func (user *User) HandleChatUpdate(cmd whatsapp.ChatUpdate) {
  710. // if cmd.Command != whatsapp.ChatUpdateCommandAction {
  711. // return
  712. // }
  713. //
  714. // portal := user.GetPortalByJID(cmd.JID)
  715. // if len(portal.MXID) == 0 {
  716. // if cmd.Data.Action == whatsapp.ChatActionIntroduce || cmd.Data.Action == whatsapp.ChatActionCreate {
  717. // go func() {
  718. // err := portal.CreateMatrixRoom(user)
  719. // if err != nil {
  720. // user.log.Errorln("Failed to create portal room after receiving join event:", err)
  721. // }
  722. // }()
  723. // }
  724. // return
  725. // }
  726. //
  727. // // These don't come down the message history :(
  728. // switch cmd.Data.Action {
  729. // case whatsapp.ChatActionAddTopic:
  730. // go portal.UpdateTopic(cmd.Data.AddTopic.Topic, cmd.Data.SenderJID, nil, true)
  731. // case whatsapp.ChatActionRemoveTopic:
  732. // go portal.UpdateTopic("", cmd.Data.SenderJID, nil, true)
  733. // case whatsapp.ChatActionRemove:
  734. // // We ignore leaving groups in the message history to avoid accidentally leaving rejoined groups,
  735. // // but if we get a real-time command that says we left, it should be safe to bridge it.
  736. // if !user.bridge.Config.Bridge.ChatMetaSync {
  737. // for _, jid := range cmd.Data.UserChange.JIDs {
  738. // if jid == user.JID {
  739. // go portal.HandleWhatsAppKick(nil, cmd.Data.SenderJID, cmd.Data.UserChange.JIDs)
  740. // break
  741. // }
  742. // }
  743. // }
  744. // }
  745. //
  746. // if !user.bridge.Config.Bridge.ChatMetaSync {
  747. // // Ignore chat update commands, we're relying on the message history.
  748. // return
  749. // }
  750. //
  751. // switch cmd.Data.Action {
  752. // case whatsapp.ChatActionNameChange:
  753. // go portal.UpdateName(cmd.Data.NameChange.Name, cmd.Data.SenderJID, nil, true)
  754. // case whatsapp.ChatActionPromote:
  755. // go portal.ChangeAdminStatus(cmd.Data.UserChange.JIDs, true)
  756. // case whatsapp.ChatActionDemote:
  757. // go portal.ChangeAdminStatus(cmd.Data.UserChange.JIDs, false)
  758. // case whatsapp.ChatActionAnnounce:
  759. // go portal.RestrictMessageSending(cmd.Data.Announce)
  760. // case whatsapp.ChatActionRestrict:
  761. // go portal.RestrictMetadataChanges(cmd.Data.Restrict)
  762. // case whatsapp.ChatActionRemove:
  763. // go portal.HandleWhatsAppKick(nil, cmd.Data.SenderJID, cmd.Data.UserChange.JIDs)
  764. // case whatsapp.ChatActionAdd:
  765. // go portal.HandleWhatsAppInvite(user, cmd.Data.SenderJID, nil, cmd.Data.UserChange.JIDs)
  766. // case whatsapp.ChatActionIntroduce:
  767. // if cmd.Data.SenderJID != "unknown" {
  768. // go portal.Sync(user, whatsapp.Contact{JID: portal.Key.JID})
  769. // }
  770. // }
  771. //}
  772. //
  773. //func (user *User) HandleJSONMessage(evt whatsapp.RawJSONMessage) {
  774. // if !json.Valid(evt.RawMessage) {
  775. // return
  776. // }
  777. // user.log.Debugfln("JSON message with tag %s: %s", evt.Tag, evt.RawMessage)
  778. // user.updateLastConnectionIfNecessary()
  779. //}
  780. func (user *User) NeedsRelaybot(portal *Portal) bool {
  781. return !user.HasSession() // || !user.IsInPortal(portal.Key)
  782. }