user.go 48 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2022 Tulir Asokan
  3. //
  4. // This program is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Affero General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // This program is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Affero General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Affero General Public License
  15. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  16. package main
  17. import (
  18. "context"
  19. "crypto/hmac"
  20. "crypto/sha256"
  21. "encoding/binary"
  22. "encoding/json"
  23. "errors"
  24. "fmt"
  25. "math"
  26. "math/rand"
  27. "net/http"
  28. "strconv"
  29. "strings"
  30. "sync"
  31. "time"
  32. "github.com/rs/zerolog"
  33. "maunium.net/go/maulogger/v2"
  34. "maunium.net/go/maulogger/v2/maulogadapt"
  35. "maunium.net/go/mautrix"
  36. "maunium.net/go/mautrix/appservice"
  37. "maunium.net/go/mautrix/bridge"
  38. "maunium.net/go/mautrix/bridge/bridgeconfig"
  39. "maunium.net/go/mautrix/bridge/status"
  40. "maunium.net/go/mautrix/event"
  41. "maunium.net/go/mautrix/format"
  42. "maunium.net/go/mautrix/id"
  43. "maunium.net/go/mautrix/pushrules"
  44. "go.mau.fi/whatsmeow"
  45. "go.mau.fi/whatsmeow/appstate"
  46. waProto "go.mau.fi/whatsmeow/binary/proto"
  47. "go.mau.fi/whatsmeow/store"
  48. "go.mau.fi/whatsmeow/types"
  49. "go.mau.fi/whatsmeow/types/events"
  50. waLog "go.mau.fi/whatsmeow/util/log"
  51. "maunium.net/go/mautrix-whatsapp/database"
  52. )
  53. type User struct {
  54. *database.User
  55. Client *whatsmeow.Client
  56. Session *store.Device
  57. bridge *WABridge
  58. zlog zerolog.Logger
  59. // Deprecated
  60. log maulogger.Logger
  61. Admin bool
  62. Whitelisted bool
  63. RelayWhitelisted bool
  64. PermissionLevel bridgeconfig.PermissionLevel
  65. mgmtCreateLock sync.Mutex
  66. spaceCreateLock sync.Mutex
  67. connLock sync.Mutex
  68. historySyncs chan *events.HistorySync
  69. lastPresence types.Presence
  70. historySyncLoopsStarted bool
  71. enqueueBackfillsTimer *time.Timer
  72. spaceMembershipChecked bool
  73. lastPhoneOfflineWarning time.Time
  74. groupListCache []*types.GroupInfo
  75. groupListCacheLock sync.Mutex
  76. groupListCacheTime time.Time
  77. BackfillQueue *BackfillQueue
  78. BridgeState *bridge.BridgeStateQueue
  79. resyncQueue map[types.JID]resyncQueueItem
  80. resyncQueueLock sync.Mutex
  81. nextResync time.Time
  82. createKeyDedup string
  83. skipGroupCreateDelay types.JID
  84. groupJoinLock sync.Mutex
  85. }
  86. type resyncQueueItem struct {
  87. portal *Portal
  88. puppet *Puppet
  89. }
  90. func (br *WABridge) getUserByMXID(userID id.UserID, onlyIfExists bool) *User {
  91. _, isPuppet := br.ParsePuppetMXID(userID)
  92. if isPuppet || userID == br.Bot.UserID {
  93. return nil
  94. }
  95. br.usersLock.Lock()
  96. defer br.usersLock.Unlock()
  97. user, ok := br.usersByMXID[userID]
  98. if !ok {
  99. userIDPtr := &userID
  100. if onlyIfExists {
  101. userIDPtr = nil
  102. }
  103. return br.loadDBUser(br.DB.User.GetByMXID(userID), userIDPtr)
  104. }
  105. return user
  106. }
  107. func (br *WABridge) GetUserByMXID(userID id.UserID) *User {
  108. return br.getUserByMXID(userID, false)
  109. }
  110. func (br *WABridge) GetIUser(userID id.UserID, create bool) bridge.User {
  111. u := br.getUserByMXID(userID, !create)
  112. if u == nil {
  113. return nil
  114. }
  115. return u
  116. }
  117. func (user *User) GetPermissionLevel() bridgeconfig.PermissionLevel {
  118. return user.PermissionLevel
  119. }
  120. func (user *User) GetManagementRoomID() id.RoomID {
  121. return user.ManagementRoom
  122. }
  123. func (user *User) GetMXID() id.UserID {
  124. return user.MXID
  125. }
  126. func (user *User) GetCommandState() map[string]interface{} {
  127. return nil
  128. }
  129. func (br *WABridge) GetUserByMXIDIfExists(userID id.UserID) *User {
  130. return br.getUserByMXID(userID, true)
  131. }
  132. func (br *WABridge) GetUserByJID(jid types.JID) *User {
  133. br.usersLock.Lock()
  134. defer br.usersLock.Unlock()
  135. user, ok := br.usersByUsername[jid.User]
  136. if !ok {
  137. return br.loadDBUser(br.DB.User.GetByUsername(jid.User), nil)
  138. }
  139. return user
  140. }
  141. func (user *User) addToJIDMap() {
  142. user.bridge.usersLock.Lock()
  143. user.bridge.usersByUsername[user.JID.User] = user
  144. user.bridge.usersLock.Unlock()
  145. }
  146. func (user *User) removeFromJIDMap(state status.BridgeState) {
  147. user.bridge.usersLock.Lock()
  148. jidUser, ok := user.bridge.usersByUsername[user.JID.User]
  149. if ok && user == jidUser {
  150. delete(user.bridge.usersByUsername, user.JID.User)
  151. }
  152. user.bridge.usersLock.Unlock()
  153. user.bridge.Metrics.TrackLoginState(user.JID, false)
  154. user.BridgeState.Send(state)
  155. }
  156. func (br *WABridge) GetAllUsers() []*User {
  157. br.usersLock.Lock()
  158. defer br.usersLock.Unlock()
  159. dbUsers := br.DB.User.GetAll()
  160. output := make([]*User, len(dbUsers))
  161. for index, dbUser := range dbUsers {
  162. user, ok := br.usersByMXID[dbUser.MXID]
  163. if !ok {
  164. user = br.loadDBUser(dbUser, nil)
  165. }
  166. output[index] = user
  167. }
  168. return output
  169. }
  170. func (br *WABridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User {
  171. if dbUser == nil {
  172. if mxid == nil {
  173. return nil
  174. }
  175. dbUser = br.DB.User.New()
  176. dbUser.MXID = *mxid
  177. dbUser.Insert()
  178. }
  179. user := br.NewUser(dbUser)
  180. br.usersByMXID[user.MXID] = user
  181. if !user.JID.IsEmpty() {
  182. var err error
  183. user.Session, err = br.WAContainer.GetDevice(user.JID)
  184. if err != nil {
  185. user.log.Errorfln("Failed to load user's whatsapp session: %v", err)
  186. } else if user.Session == nil {
  187. user.log.Warnfln("Didn't find session data for %s, treating user as logged out", user.JID)
  188. user.JID = types.EmptyJID
  189. user.Update()
  190. } else {
  191. user.Session.Log = &waLogger{user.log.Sub("Session")}
  192. br.usersByUsername[user.JID.User] = user
  193. }
  194. }
  195. if len(user.ManagementRoom) > 0 {
  196. br.managementRooms[user.ManagementRoom] = user
  197. }
  198. return user
  199. }
  200. func (br *WABridge) NewUser(dbUser *database.User) *User {
  201. user := &User{
  202. User: dbUser,
  203. bridge: br,
  204. zlog: br.ZLog.With().Str("user_id", dbUser.MXID.String()).Logger(),
  205. historySyncs: make(chan *events.HistorySync, 32),
  206. lastPresence: types.PresenceUnavailable,
  207. resyncQueue: make(map[types.JID]resyncQueueItem),
  208. }
  209. user.log = maulogadapt.ZeroAsMau(&user.zlog)
  210. user.PermissionLevel = user.bridge.Config.Bridge.Permissions.Get(user.MXID)
  211. user.RelayWhitelisted = user.PermissionLevel >= bridgeconfig.PermissionLevelRelay
  212. user.Whitelisted = user.PermissionLevel >= bridgeconfig.PermissionLevelUser
  213. user.Admin = user.PermissionLevel >= bridgeconfig.PermissionLevelAdmin
  214. user.BridgeState = br.NewBridgeStateQueue(user)
  215. user.enqueueBackfillsTimer = time.NewTimer(5 * time.Second)
  216. user.enqueueBackfillsTimer.Stop()
  217. go user.puppetResyncLoop()
  218. return user
  219. }
  220. const resyncMinInterval = 7 * 24 * time.Hour
  221. const resyncLoopInterval = 4 * time.Hour
  222. func (user *User) puppetResyncLoop() {
  223. user.nextResync = time.Now().Add(resyncLoopInterval).Add(-time.Duration(rand.Intn(3600)) * time.Second)
  224. for {
  225. time.Sleep(user.nextResync.Sub(time.Now()))
  226. user.nextResync = time.Now().Add(resyncLoopInterval)
  227. user.doPuppetResync()
  228. }
  229. }
  230. func (user *User) EnqueuePuppetResync(puppet *Puppet) {
  231. if puppet.LastSync.Add(resyncMinInterval).After(time.Now()) {
  232. return
  233. }
  234. user.resyncQueueLock.Lock()
  235. if _, exists := user.resyncQueue[puppet.JID]; !exists {
  236. user.resyncQueue[puppet.JID] = resyncQueueItem{puppet: puppet}
  237. user.log.Debugfln("Enqueued resync for %s (next sync in %s)", puppet.JID, user.nextResync.Sub(time.Now()))
  238. }
  239. user.resyncQueueLock.Unlock()
  240. }
  241. func (user *User) EnqueuePortalResync(portal *Portal) {
  242. if !portal.IsGroupChat() || portal.LastSync.Add(resyncMinInterval).After(time.Now()) {
  243. return
  244. }
  245. user.resyncQueueLock.Lock()
  246. if _, exists := user.resyncQueue[portal.Key.JID]; !exists {
  247. user.resyncQueue[portal.Key.JID] = resyncQueueItem{portal: portal}
  248. user.log.Debugfln("Enqueued resync for %s (next sync in %s)", portal.Key.JID, user.nextResync.Sub(time.Now()))
  249. }
  250. user.resyncQueueLock.Unlock()
  251. }
  252. func (user *User) doPuppetResync() {
  253. if !user.IsLoggedIn() {
  254. return
  255. }
  256. user.resyncQueueLock.Lock()
  257. if len(user.resyncQueue) == 0 {
  258. user.resyncQueueLock.Unlock()
  259. return
  260. }
  261. queue := user.resyncQueue
  262. user.resyncQueue = make(map[types.JID]resyncQueueItem)
  263. user.resyncQueueLock.Unlock()
  264. var puppetJIDs []types.JID
  265. var puppets []*Puppet
  266. var portals []*Portal
  267. for jid, item := range queue {
  268. var lastSync time.Time
  269. if item.puppet != nil {
  270. lastSync = item.puppet.LastSync
  271. } else if item.portal != nil {
  272. lastSync = item.portal.LastSync
  273. }
  274. if lastSync.Add(resyncMinInterval).After(time.Now()) {
  275. user.log.Debugfln("Not resyncing %s, last sync was %s ago", jid, time.Now().Sub(lastSync))
  276. continue
  277. }
  278. if item.puppet != nil {
  279. puppets = append(puppets, item.puppet)
  280. puppetJIDs = append(puppetJIDs, jid)
  281. } else if item.portal != nil {
  282. portals = append(portals, item.portal)
  283. }
  284. }
  285. for _, portal := range portals {
  286. groupInfo, err := user.Client.GetGroupInfo(portal.Key.JID)
  287. if err != nil {
  288. user.log.Warnfln("Failed to get group info for %s to do background sync: %v", portal.Key.JID, err)
  289. } else {
  290. user.log.Debugfln("Doing background sync for %s", portal.Key.JID)
  291. portal.UpdateMatrixRoom(user, groupInfo)
  292. }
  293. }
  294. if len(puppetJIDs) == 0 {
  295. return
  296. }
  297. user.log.Debugfln("Doing background sync for users: %+v", puppetJIDs)
  298. infos, err := user.Client.GetUserInfo(puppetJIDs)
  299. if err != nil {
  300. user.log.Errorfln("Error getting user info for background sync: %v", err)
  301. return
  302. }
  303. for _, puppet := range puppets {
  304. info, ok := infos[puppet.JID]
  305. if !ok {
  306. user.log.Warnfln("Didn't get info for %s in background sync", puppet.JID)
  307. continue
  308. }
  309. var contactPtr *types.ContactInfo
  310. contact, err := user.Session.Contacts.GetContact(puppet.JID)
  311. if err != nil {
  312. user.log.Warnfln("Failed to get contact info for %s in background sync: %v", puppet.JID, err)
  313. } else if contact.Found {
  314. contactPtr = &contact
  315. }
  316. puppet.Sync(user, contactPtr, info.PictureID != "" && info.PictureID != puppet.Avatar, true)
  317. }
  318. }
  319. func (user *User) ensureInvited(intent *appservice.IntentAPI, roomID id.RoomID, isDirect bool) (ok bool) {
  320. extraContent := make(map[string]interface{})
  321. if isDirect {
  322. extraContent["is_direct"] = true
  323. }
  324. customPuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  325. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  326. extraContent["fi.mau.will_auto_accept"] = true
  327. }
  328. _, err := intent.InviteUser(roomID, &mautrix.ReqInviteUser{UserID: user.MXID}, extraContent)
  329. var httpErr mautrix.HTTPError
  330. if err != nil && errors.As(err, &httpErr) && httpErr.RespError != nil && strings.Contains(httpErr.RespError.Err, "is already in the room") {
  331. user.bridge.StateStore.SetMembership(roomID, user.MXID, event.MembershipJoin)
  332. ok = true
  333. return
  334. } else if err != nil {
  335. user.log.Warnfln("Failed to invite user to %s: %v", roomID, err)
  336. } else {
  337. ok = true
  338. }
  339. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  340. err = customPuppet.CustomIntent().EnsureJoined(roomID, appservice.EnsureJoinedParams{IgnoreCache: true})
  341. if err != nil {
  342. user.log.Warnfln("Failed to auto-join %s: %v", roomID, err)
  343. ok = false
  344. } else {
  345. ok = true
  346. }
  347. }
  348. return
  349. }
  350. func (user *User) GetSpaceRoom() id.RoomID {
  351. if !user.bridge.Config.Bridge.PersonalFilteringSpaces {
  352. return ""
  353. }
  354. if len(user.SpaceRoom) == 0 {
  355. user.spaceCreateLock.Lock()
  356. defer user.spaceCreateLock.Unlock()
  357. if len(user.SpaceRoom) > 0 {
  358. return user.SpaceRoom
  359. }
  360. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  361. Visibility: "private",
  362. Name: "WhatsApp",
  363. Topic: "Your WhatsApp bridged chats",
  364. InitialState: []*event.Event{{
  365. Type: event.StateRoomAvatar,
  366. Content: event.Content{
  367. Parsed: &event.RoomAvatarEventContent{
  368. URL: user.bridge.Config.AppService.Bot.ParsedAvatar,
  369. },
  370. },
  371. }},
  372. CreationContent: map[string]interface{}{
  373. "type": event.RoomTypeSpace,
  374. },
  375. PowerLevelOverride: &event.PowerLevelsEventContent{
  376. Users: map[id.UserID]int{
  377. user.bridge.Bot.UserID: 9001,
  378. user.MXID: 50,
  379. },
  380. },
  381. })
  382. if err != nil {
  383. user.log.Errorln("Failed to auto-create space room:", err)
  384. } else {
  385. user.SpaceRoom = resp.RoomID
  386. user.Update()
  387. user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false)
  388. }
  389. } else if !user.spaceMembershipChecked && !user.bridge.StateStore.IsInRoom(user.SpaceRoom, user.MXID) {
  390. user.ensureInvited(user.bridge.Bot, user.SpaceRoom, false)
  391. }
  392. user.spaceMembershipChecked = true
  393. return user.SpaceRoom
  394. }
  395. func (user *User) GetManagementRoom() id.RoomID {
  396. if len(user.ManagementRoom) == 0 {
  397. user.mgmtCreateLock.Lock()
  398. defer user.mgmtCreateLock.Unlock()
  399. if len(user.ManagementRoom) > 0 {
  400. return user.ManagementRoom
  401. }
  402. creationContent := make(map[string]interface{})
  403. if !user.bridge.Config.Bridge.FederateRooms {
  404. creationContent["m.federate"] = false
  405. }
  406. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  407. Topic: "WhatsApp bridge notices",
  408. IsDirect: true,
  409. CreationContent: creationContent,
  410. })
  411. if err != nil {
  412. user.log.Errorln("Failed to auto-create management room:", err)
  413. } else {
  414. user.SetManagementRoom(resp.RoomID)
  415. }
  416. }
  417. return user.ManagementRoom
  418. }
  419. func (user *User) SetManagementRoom(roomID id.RoomID) {
  420. existingUser, ok := user.bridge.managementRooms[roomID]
  421. if ok {
  422. existingUser.ManagementRoom = ""
  423. existingUser.Update()
  424. }
  425. user.ManagementRoom = roomID
  426. user.bridge.managementRooms[user.ManagementRoom] = user
  427. user.Update()
  428. }
  429. type waLogger struct{ l maulogger.Logger }
  430. func (w *waLogger) Debugf(msg string, args ...interface{}) { w.l.Debugfln(msg, args...) }
  431. func (w *waLogger) Infof(msg string, args ...interface{}) { w.l.Infofln(msg, args...) }
  432. func (w *waLogger) Warnf(msg string, args ...interface{}) { w.l.Warnfln(msg, args...) }
  433. func (w *waLogger) Errorf(msg string, args ...interface{}) { w.l.Errorfln(msg, args...) }
  434. func (w *waLogger) Sub(module string) waLog.Logger { return &waLogger{l: w.l.Sub(module)} }
  435. var ErrAlreadyLoggedIn = errors.New("already logged in")
  436. func (user *User) obfuscateJID(jid types.JID) string {
  437. // Turn the first 4 bytes of HMAC-SHA256(hs_token, phone) into a number and replace the middle of the actual phone with that deterministic random number.
  438. randomNumber := binary.BigEndian.Uint32(hmac.New(sha256.New, []byte(user.bridge.Config.AppService.HSToken)).Sum([]byte(jid.User))[:4])
  439. return fmt.Sprintf("+%s-%d-%s:%d", jid.User[:1], randomNumber, jid.User[len(jid.User)-2:], jid.Device)
  440. }
  441. func (user *User) createClient(sess *store.Device) {
  442. user.Client = whatsmeow.NewClient(sess, &waLogger{user.log.Sub("Client")})
  443. user.Client.AddEventHandler(user.HandleEvent)
  444. user.Client.SetForceActiveDeliveryReceipts(user.bridge.Config.Bridge.ForceActiveDeliveryReceipts)
  445. user.Client.AutomaticMessageRerequestFromPhone = true
  446. user.Client.GetMessageForRetry = func(requester, to types.JID, id types.MessageID) *waProto.Message {
  447. Segment.Track(user.MXID, "WhatsApp incoming retry (message not found)", map[string]interface{}{
  448. "requester": user.obfuscateJID(requester),
  449. "messageID": id,
  450. })
  451. user.bridge.Metrics.TrackRetryReceipt(0, false)
  452. return nil
  453. }
  454. user.Client.PreRetryCallback = func(receipt *events.Receipt, messageID types.MessageID, retryCount int, msg *waProto.Message) bool {
  455. Segment.Track(user.MXID, "WhatsApp incoming retry (accepted)", map[string]interface{}{
  456. "requester": user.obfuscateJID(receipt.Sender),
  457. "messageID": messageID,
  458. "retryCount": retryCount,
  459. })
  460. user.bridge.Metrics.TrackRetryReceipt(retryCount, true)
  461. return true
  462. }
  463. }
  464. func (user *User) Login(ctx context.Context) (<-chan whatsmeow.QRChannelItem, error) {
  465. user.connLock.Lock()
  466. defer user.connLock.Unlock()
  467. if user.Session != nil {
  468. return nil, ErrAlreadyLoggedIn
  469. } else if user.Client != nil {
  470. user.unlockedDeleteConnection()
  471. }
  472. newSession := user.bridge.WAContainer.NewDevice()
  473. newSession.Log = &waLogger{user.log.Sub("Session")}
  474. user.createClient(newSession)
  475. qrChan, err := user.Client.GetQRChannel(ctx)
  476. if err != nil {
  477. return nil, fmt.Errorf("failed to get QR channel: %w", err)
  478. }
  479. err = user.Client.Connect()
  480. if err != nil {
  481. return nil, fmt.Errorf("failed to connect to WhatsApp: %w", err)
  482. }
  483. return qrChan, nil
  484. }
  485. func (user *User) Connect() bool {
  486. user.connLock.Lock()
  487. defer user.connLock.Unlock()
  488. if user.Client != nil {
  489. return user.Client.IsConnected()
  490. } else if user.Session == nil {
  491. return false
  492. }
  493. user.log.Debugln("Connecting to WhatsApp")
  494. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnecting, Error: WAConnecting})
  495. user.createClient(user.Session)
  496. err := user.Client.Connect()
  497. if err != nil {
  498. user.log.Warnln("Error connecting to WhatsApp:", err)
  499. user.BridgeState.Send(status.BridgeState{
  500. StateEvent: status.StateUnknownError,
  501. Error: WAConnectionFailed,
  502. Info: map[string]interface{}{
  503. "go_error": err.Error(),
  504. },
  505. })
  506. return false
  507. }
  508. return true
  509. }
  510. func (user *User) unlockedDeleteConnection() {
  511. if user.Client == nil {
  512. return
  513. }
  514. user.Client.Disconnect()
  515. user.Client.RemoveEventHandlers()
  516. user.Client = nil
  517. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  518. }
  519. func (user *User) DeleteConnection() {
  520. user.connLock.Lock()
  521. defer user.connLock.Unlock()
  522. user.unlockedDeleteConnection()
  523. }
  524. func (user *User) HasSession() bool {
  525. return user.Session != nil
  526. }
  527. func (user *User) DeleteSession() {
  528. if user.Session != nil {
  529. err := user.Session.Delete()
  530. if err != nil {
  531. user.log.Warnln("Failed to delete session:", err)
  532. }
  533. user.Session = nil
  534. }
  535. if !user.JID.IsEmpty() {
  536. user.JID = types.EmptyJID
  537. user.Update()
  538. }
  539. // Delete all of the backfill and history sync data.
  540. user.bridge.DB.Backfill.DeleteAll(user.MXID)
  541. user.bridge.DB.HistorySync.DeleteAllConversations(user.MXID)
  542. user.bridge.DB.HistorySync.DeleteAllMessages(user.MXID)
  543. user.bridge.DB.MediaBackfillRequest.DeleteAllMediaBackfillRequests(user.MXID)
  544. }
  545. func (user *User) IsConnected() bool {
  546. return user.Client != nil && user.Client.IsConnected()
  547. }
  548. func (user *User) IsLoggedIn() bool {
  549. return user.IsConnected() && user.Client.IsLoggedIn()
  550. }
  551. func (user *User) tryAutomaticDoublePuppeting() {
  552. if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) {
  553. return
  554. }
  555. user.log.Debugln("Checking if double puppeting needs to be enabled")
  556. puppet := user.bridge.GetPuppetByJID(user.JID)
  557. if len(puppet.CustomMXID) > 0 {
  558. user.log.Debugln("User already has double-puppeting enabled")
  559. // Custom puppet already enabled
  560. return
  561. }
  562. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  563. if err != nil {
  564. user.log.Warnln("Failed to login with shared secret:", err)
  565. return
  566. }
  567. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  568. if err != nil {
  569. puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
  570. return
  571. }
  572. user.log.Infoln("Successfully automatically enabled custom puppet")
  573. }
  574. func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) {
  575. if user.bridge.Config.Bridge.DisableBridgeAlerts {
  576. return
  577. }
  578. notice := fmt.Sprintf(formatString, args...)
  579. content := format.RenderMarkdown(notice, true, false)
  580. _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content)
  581. if err != nil {
  582. user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err)
  583. }
  584. }
  585. const callEventMaxAge = 15 * time.Minute
  586. func (user *User) handleCallStart(sender types.JID, id, callType string, ts time.Time) {
  587. if !user.bridge.Config.Bridge.CallStartNotices || ts.Add(callEventMaxAge).Before(time.Now()) {
  588. return
  589. }
  590. portal := user.GetPortalByJID(sender)
  591. text := "Incoming call"
  592. if callType != "" {
  593. text = fmt.Sprintf("Incoming %s call", callType)
  594. }
  595. portal.messages <- PortalMessage{
  596. fake: &fakeMessage{
  597. Sender: sender,
  598. Text: text,
  599. ID: id,
  600. Time: ts,
  601. Important: true,
  602. },
  603. source: user,
  604. }
  605. }
  606. const PhoneDisconnectWarningTime = 12 * 24 * time.Hour // 12 days
  607. const PhoneDisconnectPingTime = 10 * 24 * time.Hour
  608. const PhoneMinPingInterval = 24 * time.Hour
  609. func (user *User) sendHackyPhonePing() {
  610. user.PhoneLastPinged = time.Now()
  611. msgID := user.Client.GenerateMessageID()
  612. keyIDs := make([]*waProto.AppStateSyncKeyId, 0, 1)
  613. lastKeyID, err := user.GetLastAppStateKeyID()
  614. if lastKeyID != nil {
  615. keyIDs = append(keyIDs, &waProto.AppStateSyncKeyId{
  616. KeyId: lastKeyID,
  617. })
  618. } else {
  619. user.log.Warnfln("Failed to get last app state key ID to send hacky phone ping: %v - sending empty request", err)
  620. }
  621. resp, err := user.Client.SendMessage(context.Background(), user.JID.ToNonAD(), &waProto.Message{
  622. ProtocolMessage: &waProto.ProtocolMessage{
  623. Type: waProto.ProtocolMessage_APP_STATE_SYNC_KEY_REQUEST.Enum(),
  624. AppStateSyncKeyRequest: &waProto.AppStateSyncKeyRequest{
  625. KeyIds: keyIDs,
  626. },
  627. },
  628. }, whatsmeow.SendRequestExtra{Peer: true, ID: msgID})
  629. if err != nil {
  630. user.log.Warnfln("Failed to send hacky phone ping: %v", err)
  631. } else {
  632. user.log.Debugfln("Sent hacky phone ping %s/%s because phone has been offline for >10 days", msgID, resp.Timestamp.Unix())
  633. user.PhoneLastPinged = resp.Timestamp
  634. user.Update()
  635. }
  636. }
  637. func (user *User) PhoneRecentlySeen(doPing bool) bool {
  638. if doPing && !user.PhoneLastSeen.IsZero() && user.PhoneLastSeen.Add(PhoneDisconnectPingTime).Before(time.Now()) && user.PhoneLastPinged.Add(PhoneMinPingInterval).Before(time.Now()) {
  639. // Over 10 days since the phone was seen and over a day since the last somewhat hacky ping, send a new ping.
  640. go user.sendHackyPhonePing()
  641. }
  642. return user.PhoneLastSeen.IsZero() || user.PhoneLastSeen.Add(PhoneDisconnectWarningTime).After(time.Now())
  643. }
  644. // phoneSeen records a timestamp when the user's main device was seen online.
  645. // The stored timestamp can later be used to warn the user if the main device is offline for too long.
  646. func (user *User) phoneSeen(ts time.Time) {
  647. if user.PhoneLastSeen.Add(1 * time.Hour).After(ts) {
  648. // The last seen timestamp isn't going to be perfectly accurate in any case,
  649. // so don't spam the database with an update every time there's an event.
  650. return
  651. } else if !user.PhoneRecentlySeen(false) {
  652. if user.BridgeState.GetPrev().Error == WAPhoneOffline && user.IsConnected() {
  653. user.log.Debugfln("Saw phone after current bridge state said it has been offline, switching state back to connected")
  654. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnected})
  655. } else {
  656. user.log.Debugfln("Saw phone after current bridge state said it has been offline, not sending new bridge state (prev: %s, connected: %t)", user.BridgeState.GetPrev().Error, user.IsConnected())
  657. }
  658. }
  659. user.PhoneLastSeen = ts
  660. go user.Update()
  661. }
  662. func formatDisconnectTime(dur time.Duration) string {
  663. days := int(math.Floor(dur.Hours() / 24))
  664. hours := int(dur.Hours()) % 24
  665. if hours == 0 {
  666. return fmt.Sprintf("%d days", days)
  667. } else if hours == 1 {
  668. return fmt.Sprintf("%d days and 1 hour", days)
  669. } else {
  670. return fmt.Sprintf("%d days and %d hours", days, hours)
  671. }
  672. }
  673. func (user *User) sendPhoneOfflineWarning() {
  674. if user.lastPhoneOfflineWarning.Add(12 * time.Hour).After(time.Now()) {
  675. // Don't spam the warning too much
  676. return
  677. }
  678. user.lastPhoneOfflineWarning = time.Now()
  679. timeSinceSeen := time.Now().Sub(user.PhoneLastSeen)
  680. user.sendMarkdownBridgeAlert("Your phone hasn't been seen in %s. The server will force the bridge to log out if the phone is not active at least every 2 weeks.", formatDisconnectTime(timeSinceSeen))
  681. }
  682. func (user *User) HandleEvent(event interface{}) {
  683. switch v := event.(type) {
  684. case *events.LoggedOut:
  685. go user.handleLoggedOut(v.OnConnect, v.Reason)
  686. case *events.Connected:
  687. user.bridge.Metrics.TrackConnectionState(user.JID, true)
  688. user.bridge.Metrics.TrackLoginState(user.JID, true)
  689. if len(user.Client.Store.PushName) > 0 {
  690. go func() {
  691. err := user.Client.SendPresence(user.lastPresence)
  692. if err != nil {
  693. user.log.Warnln("Failed to send initial presence:", err)
  694. }
  695. }()
  696. }
  697. go user.tryAutomaticDoublePuppeting()
  698. if user.bridge.Config.Bridge.HistorySync.Backfill && !user.historySyncLoopsStarted {
  699. go user.handleHistorySyncsLoop()
  700. user.historySyncLoopsStarted = true
  701. }
  702. case *events.OfflineSyncPreview:
  703. user.log.Infofln("Server says it's going to send %d messages and %d receipts that were missed during downtime", v.Messages, v.Receipts)
  704. user.BridgeState.Send(status.BridgeState{
  705. StateEvent: status.StateBackfilling,
  706. Message: fmt.Sprintf("backfilling %d messages and %d receipts", v.Messages, v.Receipts),
  707. })
  708. case *events.OfflineSyncCompleted:
  709. if !user.PhoneRecentlySeen(true) {
  710. user.log.Infofln("Offline sync completed, but phone last seen date is still %s - sending phone offline bridge status", user.PhoneLastSeen)
  711. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateTransientDisconnect, Error: WAPhoneOffline})
  712. } else {
  713. if user.BridgeState.GetPrev().StateEvent == status.StateBackfilling {
  714. user.log.Infoln("Offline sync completed")
  715. }
  716. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnected})
  717. }
  718. case *events.AppStateSyncComplete:
  719. if len(user.Client.Store.PushName) > 0 && v.Name == appstate.WAPatchCriticalBlock {
  720. err := user.Client.SendPresence(user.lastPresence)
  721. if err != nil {
  722. user.log.Warnln("Failed to send presence after app state sync:", err)
  723. }
  724. } else if v.Name == appstate.WAPatchCriticalUnblockLow {
  725. go func() {
  726. err := user.ResyncContacts(false)
  727. if err != nil {
  728. user.log.Errorln("Failed to resync puppets: %v", err)
  729. }
  730. }()
  731. }
  732. case *events.PushNameSetting:
  733. // Send presence available when connecting and when the pushname is changed.
  734. // This makes sure that outgoing messages always have the right pushname.
  735. err := user.Client.SendPresence(user.lastPresence)
  736. if err != nil {
  737. user.log.Warnln("Failed to send presence after push name update:", err)
  738. }
  739. _, _, err = user.Client.Store.Contacts.PutPushName(user.JID.ToNonAD(), v.Action.GetName())
  740. if err != nil {
  741. user.log.Warnln("Failed to update push name in store:", err)
  742. }
  743. go user.syncPuppet(user.JID.ToNonAD(), "push name setting")
  744. case *events.PairSuccess:
  745. user.PhoneLastSeen = time.Now()
  746. user.Session = user.Client.Store
  747. user.JID = v.ID
  748. user.addToJIDMap()
  749. user.Update()
  750. case *events.StreamError:
  751. var message string
  752. if v.Code != "" {
  753. message = fmt.Sprintf("Unknown stream error with code %s", v.Code)
  754. } else if children := v.Raw.GetChildren(); len(children) > 0 {
  755. message = fmt.Sprintf("Unknown stream error (contains %s node)", children[0].Tag)
  756. } else {
  757. message = "Unknown stream error"
  758. }
  759. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateUnknownError, Message: message})
  760. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  761. case *events.StreamReplaced:
  762. if user.bridge.Config.Bridge.CrashOnStreamReplaced {
  763. user.log.Infofln("Stopping bridge due to StreamReplaced event")
  764. user.bridge.ManualStop(60)
  765. } else {
  766. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateUnknownError, Message: "Stream replaced"})
  767. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  768. user.sendMarkdownBridgeAlert("The bridge was started in another location. Use `reconnect` to reconnect this one.")
  769. }
  770. case *events.ConnectFailure:
  771. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateUnknownError, Message: fmt.Sprintf("Unknown connection failure: %s (%s)", v.Reason, v.Message)})
  772. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  773. user.bridge.Metrics.TrackConnectionFailure(fmt.Sprintf("status-%d", v.Reason))
  774. case *events.ClientOutdated:
  775. user.log.Errorfln("Got a client outdated connect failure. The bridge is likely out of date, please update immediately.")
  776. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateUnknownError, Message: "Connect failure: 405 client outdated"})
  777. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  778. user.bridge.Metrics.TrackConnectionFailure("client-outdated")
  779. case *events.TemporaryBan:
  780. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateBadCredentials, Message: v.String()})
  781. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  782. user.bridge.Metrics.TrackConnectionFailure("temporary-ban")
  783. case *events.Disconnected:
  784. // Don't send the normal transient disconnect state if we're already in a different transient disconnect state.
  785. // TODO remove this if/when the phone offline state is moved to a sub-state of CONNECTED
  786. if user.BridgeState.GetPrev().Error != WAPhoneOffline && user.PhoneRecentlySeen(false) {
  787. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateTransientDisconnect, Error: WADisconnected})
  788. }
  789. user.bridge.Metrics.TrackConnectionState(user.JID, false)
  790. case *events.Contact:
  791. go user.syncPuppet(v.JID, "contact event")
  792. case *events.PushName:
  793. go user.syncPuppet(v.JID, "push name event")
  794. case *events.BusinessName:
  795. go user.syncPuppet(v.JID, "business name event")
  796. case *events.GroupInfo:
  797. user.groupListCache = nil
  798. go user.handleGroupUpdate(v)
  799. case *events.JoinedGroup:
  800. user.groupListCache = nil
  801. go user.handleGroupCreate(v)
  802. case *events.Picture:
  803. go user.handlePictureUpdate(v)
  804. case *events.Receipt:
  805. if v.IsFromMe && v.Sender.Device == 0 {
  806. user.phoneSeen(v.Timestamp)
  807. }
  808. go user.handleReceipt(v)
  809. case *events.ChatPresence:
  810. go user.handleChatPresence(v)
  811. case *events.Message:
  812. portal := user.GetPortalByMessageSource(v.Info.MessageSource)
  813. portal.messages <- PortalMessage{evt: v, source: user}
  814. case *events.MediaRetry:
  815. user.phoneSeen(v.Timestamp)
  816. portal := user.GetPortalByJID(v.ChatID)
  817. portal.mediaRetries <- PortalMediaRetry{evt: v, source: user}
  818. case *events.CallOffer:
  819. user.handleCallStart(v.CallCreator, v.CallID, "", v.Timestamp)
  820. case *events.CallOfferNotice:
  821. user.handleCallStart(v.CallCreator, v.CallID, v.Type, v.Timestamp)
  822. case *events.IdentityChange:
  823. puppet := user.bridge.GetPuppetByJID(v.JID)
  824. portal := user.GetPortalByJID(v.JID)
  825. if len(portal.MXID) > 0 && user.bridge.Config.Bridge.IdentityChangeNotices {
  826. text := fmt.Sprintf("Your security code with %s changed.", puppet.Displayname)
  827. if v.Implicit {
  828. text = fmt.Sprintf("Your security code with %s (device #%d) changed.", puppet.Displayname, v.JID.Device)
  829. }
  830. portal.messages <- PortalMessage{
  831. fake: &fakeMessage{
  832. Sender: v.JID,
  833. Text: text,
  834. ID: strconv.FormatInt(v.Timestamp.Unix(), 10),
  835. Time: v.Timestamp,
  836. Important: false,
  837. },
  838. source: user,
  839. }
  840. }
  841. case *events.CallTerminate, *events.CallRelayLatency, *events.CallAccept, *events.UnknownCallEvent:
  842. // ignore
  843. case *events.UndecryptableMessage:
  844. portal := user.GetPortalByMessageSource(v.Info.MessageSource)
  845. portal.messages <- PortalMessage{undecryptable: v, source: user}
  846. case *events.HistorySync:
  847. if user.bridge.Config.Bridge.HistorySync.Backfill {
  848. user.historySyncs <- v
  849. }
  850. case *events.Mute:
  851. portal := user.GetPortalByJID(v.JID)
  852. if portal != nil {
  853. var mutedUntil time.Time
  854. if v.Action.GetMuted() {
  855. mutedUntil = time.Unix(v.Action.GetMuteEndTimestamp(), 0)
  856. }
  857. go user.updateChatMute(nil, portal, mutedUntil)
  858. }
  859. case *events.Archive:
  860. portal := user.GetPortalByJID(v.JID)
  861. if portal != nil {
  862. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.ArchiveTag, v.Action.GetArchived())
  863. }
  864. case *events.Pin:
  865. portal := user.GetPortalByJID(v.JID)
  866. if portal != nil {
  867. go user.updateChatTag(nil, portal, user.bridge.Config.Bridge.PinnedTag, v.Action.GetPinned())
  868. }
  869. case *events.AppState:
  870. // Ignore
  871. case *events.KeepAliveTimeout:
  872. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateTransientDisconnect, Error: WAKeepaliveTimeout})
  873. case *events.KeepAliveRestored:
  874. user.log.Infof("Keepalive restored after timeouts, sending connected event")
  875. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnected})
  876. case *events.MarkChatAsRead:
  877. if user.bridge.Config.Bridge.SyncManualMarkedUnread {
  878. user.markUnread(user.GetPortalByJID(v.JID), !v.Action.GetRead())
  879. }
  880. case *events.DeleteForMe:
  881. portal := user.GetPortalByJID(v.ChatJID)
  882. if portal != nil {
  883. portal.deleteForMe(user, v)
  884. }
  885. case *events.DeleteChat:
  886. portal := user.GetPortalByJID(v.JID)
  887. if portal != nil {
  888. portal.HandleWhatsAppDeleteChat(user)
  889. }
  890. default:
  891. user.log.Debugfln("Unknown type of event in HandleEvent: %T", v)
  892. }
  893. }
  894. func (user *User) updateChatMute(intent *appservice.IntentAPI, portal *Portal, mutedUntil time.Time) {
  895. if len(portal.MXID) == 0 || !user.bridge.Config.Bridge.MuteBridging {
  896. return
  897. } else if intent == nil {
  898. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  899. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  900. return
  901. }
  902. intent = doublePuppet.CustomIntent()
  903. }
  904. var err error
  905. if mutedUntil.IsZero() && mutedUntil.Before(time.Now()) {
  906. user.log.Debugfln("Portal %s is muted until %s, unmuting...", portal.MXID, mutedUntil)
  907. err = intent.DeletePushRule("global", pushrules.RoomRule, string(portal.MXID))
  908. } else {
  909. user.log.Debugfln("Portal %s is muted until %s, muting...", portal.MXID, mutedUntil)
  910. err = intent.PutPushRule("global", pushrules.RoomRule, string(portal.MXID), &mautrix.ReqPutPushRule{
  911. Actions: []pushrules.PushActionType{pushrules.ActionDontNotify},
  912. })
  913. }
  914. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  915. user.log.Warnfln("Failed to update push rule for %s through double puppet: %v", portal.MXID, err)
  916. }
  917. }
  918. type CustomTagData struct {
  919. Order json.Number `json:"order"`
  920. DoublePuppet string `json:"fi.mau.double_puppet_source"`
  921. }
  922. type CustomTagEventContent struct {
  923. Tags map[string]CustomTagData `json:"tags"`
  924. }
  925. func (user *User) updateChatTag(intent *appservice.IntentAPI, portal *Portal, tag string, active bool) {
  926. if len(portal.MXID) == 0 || len(tag) == 0 {
  927. return
  928. } else if intent == nil {
  929. doublePuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  930. if doublePuppet == nil || doublePuppet.CustomIntent() == nil {
  931. return
  932. }
  933. intent = doublePuppet.CustomIntent()
  934. }
  935. var existingTags CustomTagEventContent
  936. err := intent.GetTagsWithCustomData(portal.MXID, &existingTags)
  937. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  938. user.log.Warnfln("Failed to get tags of %s: %v", portal.MXID, err)
  939. }
  940. currentTag, ok := existingTags.Tags[tag]
  941. if active && !ok {
  942. user.log.Debugln("Adding tag", tag, "to", portal.MXID)
  943. data := CustomTagData{Order: "0.5", DoublePuppet: user.bridge.Name}
  944. err = intent.AddTagWithCustomData(portal.MXID, tag, &data)
  945. } else if !active && ok && currentTag.DoublePuppet == user.bridge.Name {
  946. user.log.Debugln("Removing tag", tag, "from", portal.MXID)
  947. err = intent.RemoveTag(portal.MXID, tag)
  948. } else {
  949. err = nil
  950. }
  951. if err != nil {
  952. user.log.Warnfln("Failed to update tag %s for %s through double puppet: %v", tag, portal.MXID, err)
  953. }
  954. }
  955. type CustomReadReceipt struct {
  956. Timestamp int64 `json:"ts,omitempty"`
  957. DoublePuppetSource string `json:"fi.mau.double_puppet_source,omitempty"`
  958. }
  959. type CustomReadMarkers struct {
  960. mautrix.ReqSetReadMarkers
  961. ReadExtra CustomReadReceipt `json:"com.beeper.read.extra"`
  962. FullyReadExtra CustomReadReceipt `json:"com.beeper.fully_read.extra"`
  963. }
  964. func (user *User) syncChatDoublePuppetDetails(portal *Portal, justCreated bool) {
  965. doublePuppet := portal.bridge.GetPuppetByCustomMXID(user.MXID)
  966. if doublePuppet == nil {
  967. return
  968. }
  969. if doublePuppet == nil || doublePuppet.CustomIntent() == nil || len(portal.MXID) == 0 {
  970. return
  971. }
  972. if justCreated || !user.bridge.Config.Bridge.TagOnlyOnCreate {
  973. chat, err := user.Client.Store.ChatSettings.GetChatSettings(portal.Key.JID)
  974. if err != nil {
  975. user.log.Warnfln("Failed to get settings of %s: %v", portal.Key.JID, err)
  976. return
  977. }
  978. intent := doublePuppet.CustomIntent()
  979. if portal.Key.JID == types.StatusBroadcastJID && justCreated {
  980. if user.bridge.Config.Bridge.MuteStatusBroadcast {
  981. user.updateChatMute(intent, portal, time.Now().Add(365*24*time.Hour))
  982. }
  983. if len(user.bridge.Config.Bridge.StatusBroadcastTag) > 0 {
  984. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.StatusBroadcastTag, true)
  985. }
  986. return
  987. } else if !chat.Found {
  988. return
  989. }
  990. user.updateChatMute(intent, portal, chat.MutedUntil)
  991. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.ArchiveTag, chat.Archived)
  992. user.updateChatTag(intent, portal, user.bridge.Config.Bridge.PinnedTag, chat.Pinned)
  993. }
  994. }
  995. func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
  996. res := make(map[id.UserID][]id.RoomID)
  997. privateChats := user.bridge.DB.Portal.FindPrivateChats(user.JID.ToNonAD())
  998. for _, portal := range privateChats {
  999. if len(portal.MXID) > 0 {
  1000. res[user.bridge.FormatPuppetMXID(portal.Key.JID)] = []id.RoomID{portal.MXID}
  1001. }
  1002. }
  1003. return res
  1004. }
  1005. func (user *User) UpdateDirectChats(chats map[id.UserID][]id.RoomID) {
  1006. if !user.bridge.Config.Bridge.SyncDirectChatList {
  1007. return
  1008. }
  1009. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  1010. if puppet == nil || puppet.CustomIntent() == nil {
  1011. return
  1012. }
  1013. intent := puppet.CustomIntent()
  1014. method := http.MethodPatch
  1015. if chats == nil {
  1016. chats = user.getDirectChats()
  1017. method = http.MethodPut
  1018. }
  1019. user.log.Debugln("Updating m.direct list on homeserver")
  1020. var err error
  1021. if user.bridge.Config.Homeserver.Software == bridgeconfig.SoftwareAsmux {
  1022. urlPath := intent.BuildClientURL("unstable", "com.beeper.asmux", "dms")
  1023. _, err = intent.MakeFullRequest(mautrix.FullRequest{
  1024. Method: method,
  1025. URL: urlPath,
  1026. Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
  1027. RequestJSON: chats,
  1028. })
  1029. } else {
  1030. existingChats := make(map[id.UserID][]id.RoomID)
  1031. err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
  1032. if err != nil {
  1033. user.log.Warnln("Failed to get m.direct list to update it:", err)
  1034. return
  1035. }
  1036. for userID, rooms := range existingChats {
  1037. if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
  1038. // This is not a ghost user, include it in the new list
  1039. chats[userID] = rooms
  1040. } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
  1041. // This is a ghost user, but we're not replacing the whole list, so include it too
  1042. chats[userID] = rooms
  1043. }
  1044. }
  1045. err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
  1046. }
  1047. if err != nil {
  1048. user.log.Warnln("Failed to update m.direct list:", err)
  1049. }
  1050. }
  1051. func (user *User) handleLoggedOut(onConnect bool, reason events.ConnectFailureReason) {
  1052. errorCode := WAUnknownLogout
  1053. if reason == events.ConnectFailureLoggedOut {
  1054. errorCode = WALoggedOut
  1055. } else if reason == events.ConnectFailureMainDeviceGone {
  1056. errorCode = WAMainDeviceGone
  1057. }
  1058. user.removeFromJIDMap(status.BridgeState{StateEvent: status.StateBadCredentials, Error: errorCode})
  1059. user.DeleteConnection()
  1060. user.Session = nil
  1061. user.JID = types.EmptyJID
  1062. user.Update()
  1063. if onConnect {
  1064. user.sendMarkdownBridgeAlert("Connecting to WhatsApp failed as the device was unlinked (error %s). Please link the bridge to your phone again.", reason)
  1065. } else {
  1066. user.sendMarkdownBridgeAlert("You were logged out from another device. Please link the bridge to your phone again.")
  1067. }
  1068. }
  1069. func (user *User) GetPortalByMessageSource(ms types.MessageSource) *Portal {
  1070. jid := ms.Chat
  1071. if ms.IsIncomingBroadcast() {
  1072. if ms.IsFromMe {
  1073. jid = ms.BroadcastListOwner.ToNonAD()
  1074. } else {
  1075. jid = ms.Sender.ToNonAD()
  1076. }
  1077. if jid.IsEmpty() {
  1078. return nil
  1079. }
  1080. }
  1081. return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
  1082. }
  1083. func (user *User) GetPortalByJID(jid types.JID) *Portal {
  1084. return user.bridge.GetPortalByJID(database.NewPortalKey(jid, user.JID))
  1085. }
  1086. func (user *User) syncPuppet(jid types.JID, reason string) {
  1087. user.bridge.GetPuppetByJID(jid).SyncContact(user, false, false, reason)
  1088. }
  1089. func (user *User) ResyncContacts(forceAvatarSync bool) error {
  1090. contacts, err := user.Client.Store.Contacts.GetAllContacts()
  1091. if err != nil {
  1092. return fmt.Errorf("failed to get cached contacts: %w", err)
  1093. }
  1094. user.log.Infofln("Resyncing displaynames with %d contacts", len(contacts))
  1095. for jid, contact := range contacts {
  1096. puppet := user.bridge.GetPuppetByJID(jid)
  1097. if puppet != nil {
  1098. puppet.Sync(user, &contact, forceAvatarSync, true)
  1099. } else {
  1100. user.log.Warnfln("Got a nil puppet for %s while syncing contacts", jid)
  1101. }
  1102. }
  1103. return nil
  1104. }
  1105. func (user *User) ResyncGroups(createPortals bool) error {
  1106. groups, err := user.Client.GetJoinedGroups()
  1107. if err != nil {
  1108. return fmt.Errorf("failed to get group list from server: %w", err)
  1109. }
  1110. user.groupListCacheLock.Lock()
  1111. user.groupListCache = groups
  1112. user.groupListCacheTime = time.Now()
  1113. user.groupListCacheLock.Unlock()
  1114. for _, group := range groups {
  1115. portal := user.GetPortalByJID(group.JID)
  1116. if len(portal.MXID) == 0 {
  1117. if createPortals {
  1118. err = portal.CreateMatrixRoom(user, group, true, true)
  1119. if err != nil {
  1120. return fmt.Errorf("failed to create room for %s: %w", group.JID, err)
  1121. }
  1122. }
  1123. } else {
  1124. portal.UpdateMatrixRoom(user, group)
  1125. }
  1126. }
  1127. return nil
  1128. }
  1129. const WATypingTimeout = 15 * time.Second
  1130. func (user *User) handleChatPresence(presence *events.ChatPresence) {
  1131. puppet := user.bridge.GetPuppetByJID(presence.Sender)
  1132. portal := user.GetPortalByJID(presence.Chat)
  1133. if puppet == nil || portal == nil || len(portal.MXID) == 0 {
  1134. return
  1135. }
  1136. if presence.State == types.ChatPresenceComposing {
  1137. if puppet.typingIn != "" && puppet.typingAt.Add(WATypingTimeout).Before(time.Now()) {
  1138. if puppet.typingIn == portal.MXID {
  1139. return
  1140. }
  1141. _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  1142. }
  1143. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, WATypingTimeout)
  1144. puppet.typingIn = portal.MXID
  1145. puppet.typingAt = time.Now()
  1146. } else {
  1147. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, false, 0)
  1148. puppet.typingIn = ""
  1149. }
  1150. }
  1151. func (user *User) handleReceipt(receipt *events.Receipt) {
  1152. if receipt.Type != events.ReceiptTypeRead && receipt.Type != events.ReceiptTypeReadSelf {
  1153. return
  1154. }
  1155. portal := user.GetPortalByMessageSource(receipt.MessageSource)
  1156. if portal == nil || len(portal.MXID) == 0 {
  1157. return
  1158. }
  1159. portal.messages <- PortalMessage{receipt: receipt, source: user}
  1160. }
  1161. func (user *User) makeReadMarkerContent(eventID id.EventID, doublePuppet bool) CustomReadMarkers {
  1162. var extra CustomReadReceipt
  1163. if doublePuppet {
  1164. extra.DoublePuppetSource = user.bridge.Name
  1165. }
  1166. return CustomReadMarkers{
  1167. ReqSetReadMarkers: mautrix.ReqSetReadMarkers{
  1168. Read: eventID,
  1169. FullyRead: eventID,
  1170. },
  1171. ReadExtra: extra,
  1172. FullyReadExtra: extra,
  1173. }
  1174. }
  1175. func (user *User) markSelfReadFull(portal *Portal) {
  1176. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  1177. if puppet == nil || puppet.CustomIntent() == nil {
  1178. return
  1179. }
  1180. lastMessage := user.bridge.DB.Message.GetLastInChat(portal.Key)
  1181. if lastMessage == nil {
  1182. return
  1183. }
  1184. user.SetLastReadTS(portal.Key, lastMessage.Timestamp)
  1185. err := puppet.CustomIntent().SetReadMarkers(portal.MXID, user.makeReadMarkerContent(lastMessage.MXID, true))
  1186. if err != nil {
  1187. user.log.Warnfln("Failed to mark %s (last message) in %s as read: %v", lastMessage.MXID, portal.MXID, err)
  1188. } else {
  1189. user.log.Debugfln("Marked %s (last message) in %s as read", lastMessage.MXID, portal.MXID)
  1190. }
  1191. }
  1192. func (user *User) markUnread(portal *Portal, unread bool) {
  1193. puppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  1194. if puppet == nil || puppet.CustomIntent() == nil {
  1195. return
  1196. }
  1197. err := puppet.CustomIntent().SetRoomAccountData(portal.MXID, "m.marked_unread",
  1198. map[string]bool{"unread": unread})
  1199. if err != nil {
  1200. user.log.Warnfln("Failed to mark %s as unread via m.marked_unread: %v", portal.MXID, err)
  1201. } else {
  1202. user.log.Debugfln("Marked %s as unread via m.marked_unread: %v", portal.MXID, err)
  1203. }
  1204. err = puppet.CustomIntent().SetRoomAccountData(portal.MXID, "com.famedly.marked_unread",
  1205. map[string]bool{"unread": unread})
  1206. if err != nil {
  1207. user.log.Warnfln("Failed to mark %s as unread via com.famedly.marked_unread: %v", portal.MXID, err)
  1208. } else {
  1209. user.log.Debugfln("Marked %s as unread via com.famedly.marked_unread: %v", portal.MXID, err)
  1210. }
  1211. }
  1212. func (user *User) handleGroupCreate(evt *events.JoinedGroup) {
  1213. portal := user.GetPortalByJID(evt.JID)
  1214. if evt.CreateKey == "" && len(portal.MXID) == 0 && portal.Key.JID != user.skipGroupCreateDelay {
  1215. user.log.Debugfln("Delaying handling group create with empty key to avoid race conditions")
  1216. time.Sleep(5 * time.Second)
  1217. }
  1218. if len(portal.MXID) == 0 {
  1219. if user.createKeyDedup != "" && evt.CreateKey == user.createKeyDedup {
  1220. user.log.Debugfln("Ignoring group create event with key %s", evt.CreateKey)
  1221. return
  1222. }
  1223. err := portal.CreateMatrixRoom(user, &evt.GroupInfo, true, true)
  1224. if err != nil {
  1225. user.log.Errorln("Failed to create Matrix room after join notification: %v", err)
  1226. }
  1227. } else {
  1228. portal.UpdateMatrixRoom(user, &evt.GroupInfo)
  1229. }
  1230. }
  1231. func (user *User) handleGroupUpdate(evt *events.GroupInfo) {
  1232. portal := user.GetPortalByJID(evt.JID)
  1233. with := user.zlog.With().
  1234. Str("chat_jid", evt.JID.String()).
  1235. Interface("group_event", evt)
  1236. if portal != nil {
  1237. with = with.Str("portal_mxid", portal.MXID.String())
  1238. }
  1239. log := with.Logger()
  1240. if portal == nil || len(portal.MXID) == 0 {
  1241. log.Debug().Msg("Ignoring group info update in chat with no portal")
  1242. return
  1243. }
  1244. switch {
  1245. case evt.Announce != nil:
  1246. log.Debug().Msg("Group announcement mode (message send permission) changed")
  1247. portal.RestrictMessageSending(evt.Announce.IsAnnounce)
  1248. case evt.Locked != nil:
  1249. log.Debug().Msg("Group locked mode (metadata change permission) changed")
  1250. portal.RestrictMetadataChanges(evt.Locked.IsLocked)
  1251. case evt.Name != nil:
  1252. log.Debug().Msg("Group name changed")
  1253. portal.UpdateName(evt.Name.Name, evt.Name.NameSetBy, true)
  1254. case evt.Topic != nil:
  1255. log.Debug().Msg("Group topic changed")
  1256. portal.UpdateTopic(evt.Topic.Topic, evt.Topic.TopicSetBy, true)
  1257. case evt.Leave != nil:
  1258. log.Debug().Msg("Someone left the group")
  1259. if evt.Sender != nil && !evt.Sender.IsEmpty() {
  1260. portal.HandleWhatsAppKick(user, *evt.Sender, evt.Leave)
  1261. }
  1262. case evt.Join != nil:
  1263. log.Debug().Msg("Someone joined the group")
  1264. portal.HandleWhatsAppInvite(user, evt.Sender, evt.Join)
  1265. case evt.Promote != nil:
  1266. log.Debug().Msg("Someone was promoted to admin")
  1267. portal.ChangeAdminStatus(evt.Promote, true)
  1268. case evt.Demote != nil:
  1269. log.Debug().Msg("Someone was demoted from admin")
  1270. portal.ChangeAdminStatus(evt.Demote, false)
  1271. case evt.Ephemeral != nil:
  1272. log.Debug().Msg("Group ephemeral mode (disappearing message timer) changed")
  1273. portal.UpdateGroupDisappearingMessages(evt.Sender, evt.Timestamp, evt.Ephemeral.DisappearingTimer)
  1274. case evt.Link != nil:
  1275. log.Debug().Msg("Group parent changed")
  1276. if evt.Link.Type == types.GroupLinkChangeTypeParent {
  1277. portal.UpdateParentGroup(user, evt.Link.Group.JID, true)
  1278. }
  1279. case evt.Unlink != nil:
  1280. log.Debug().Msg("Group parent removed")
  1281. if evt.Unlink.Type == types.GroupLinkChangeTypeParent && portal.ParentGroup == evt.Unlink.Group.JID {
  1282. portal.UpdateParentGroup(user, types.EmptyJID, true)
  1283. }
  1284. case evt.Delete != nil:
  1285. log.Debug().Msg("Group deleted")
  1286. portal.Delete()
  1287. portal.Cleanup(false)
  1288. default:
  1289. log.Warn().Msg("Unhandled group info update")
  1290. }
  1291. }
  1292. func (user *User) handlePictureUpdate(evt *events.Picture) {
  1293. if evt.JID.Server == types.DefaultUserServer {
  1294. puppet := user.bridge.GetPuppetByJID(evt.JID)
  1295. user.log.Debugfln("Received picture update for puppet %s (current: %s, new: %s)", evt.JID, puppet.Avatar, evt.PictureID)
  1296. if puppet.Avatar != evt.PictureID {
  1297. puppet.Sync(user, nil, true, false)
  1298. }
  1299. } else if portal := user.GetPortalByJID(evt.JID); portal != nil {
  1300. user.log.Debugfln("Received picture update for portal %s (current: %s, new: %s)", evt.JID, portal.Avatar, evt.PictureID)
  1301. if portal.Avatar != evt.PictureID {
  1302. portal.UpdateAvatar(user, evt.Author, true)
  1303. }
  1304. }
  1305. }
  1306. func (user *User) StartPM(jid types.JID, reason string) (*Portal, *Puppet, bool, error) {
  1307. user.log.Debugln("Starting PM with", jid, "from", reason)
  1308. puppet := user.bridge.GetPuppetByJID(jid)
  1309. puppet.SyncContact(user, true, false, reason)
  1310. portal := user.GetPortalByJID(puppet.JID)
  1311. if len(portal.MXID) > 0 {
  1312. ok := portal.ensureUserInvited(user)
  1313. if !ok {
  1314. portal.log.Warnfln("ensureUserInvited(%s) returned false, creating new portal", user.MXID)
  1315. portal.MXID = ""
  1316. } else {
  1317. return portal, puppet, false, nil
  1318. }
  1319. }
  1320. err := portal.CreateMatrixRoom(user, nil, false, true)
  1321. return portal, puppet, true, err
  1322. }
  1323. const groupListCacheMaxAge = 24 * time.Hour
  1324. func (user *User) getCachedGroupList() ([]*types.GroupInfo, error) {
  1325. user.groupListCacheLock.Lock()
  1326. defer user.groupListCacheLock.Unlock()
  1327. if user.groupListCache != nil && user.groupListCacheTime.Add(groupListCacheMaxAge).After(time.Now()) {
  1328. return user.groupListCache, nil
  1329. }
  1330. var err error
  1331. user.groupListCache, err = user.Client.GetJoinedGroups()
  1332. user.groupListCacheTime = time.Now()
  1333. return user.groupListCache, err
  1334. }