user.go 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2020 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. "fmt"
  20. "sort"
  21. "strconv"
  22. "strings"
  23. "sync"
  24. "time"
  25. "github.com/pkg/errors"
  26. "github.com/skip2/go-qrcode"
  27. log "maunium.net/go/maulogger/v2"
  28. "maunium.net/go/mautrix"
  29. "github.com/Rhymen/go-whatsapp"
  30. waProto "github.com/Rhymen/go-whatsapp/binary/proto"
  31. "maunium.net/go/mautrix/event"
  32. "maunium.net/go/mautrix/format"
  33. "maunium.net/go/mautrix/id"
  34. "maunium.net/go/mautrix-whatsapp/database"
  35. "maunium.net/go/mautrix-whatsapp/types"
  36. "maunium.net/go/mautrix-whatsapp/whatsapp-ext"
  37. )
  38. type User struct {
  39. *database.User
  40. Conn *whatsappExt.ExtendedConn
  41. bridge *Bridge
  42. log log.Logger
  43. Admin bool
  44. Whitelisted bool
  45. RelaybotWhitelisted bool
  46. IsRelaybot bool
  47. ConnectionErrors int
  48. CommunityID string
  49. cleanDisconnection bool
  50. batteryWarningsSent int
  51. chatListReceived chan struct{}
  52. syncPortalsDone chan struct{}
  53. messages chan PortalMessage
  54. syncStart chan struct{}
  55. syncWait sync.WaitGroup
  56. mgmtCreateLock sync.Mutex
  57. }
  58. func (bridge *Bridge) GetUserByMXID(userID id.UserID) *User {
  59. _, isPuppet := bridge.ParsePuppetMXID(userID)
  60. if isPuppet || userID == bridge.Bot.UserID {
  61. return nil
  62. }
  63. bridge.usersLock.Lock()
  64. defer bridge.usersLock.Unlock()
  65. user, ok := bridge.usersByMXID[userID]
  66. if !ok {
  67. return bridge.loadDBUser(bridge.DB.User.GetByMXID(userID), &userID)
  68. }
  69. return user
  70. }
  71. func (bridge *Bridge) GetUserByJID(userID types.WhatsAppID) *User {
  72. bridge.usersLock.Lock()
  73. defer bridge.usersLock.Unlock()
  74. user, ok := bridge.usersByJID[userID]
  75. if !ok {
  76. return bridge.loadDBUser(bridge.DB.User.GetByJID(userID), nil)
  77. }
  78. return user
  79. }
  80. func (user *User) addToJIDMap() {
  81. user.bridge.usersLock.Lock()
  82. user.bridge.usersByJID[user.JID] = user
  83. user.bridge.usersLock.Unlock()
  84. }
  85. func (user *User) removeFromJIDMap() {
  86. user.bridge.usersLock.Lock()
  87. delete(user.bridge.usersByJID, user.JID)
  88. user.bridge.usersLock.Unlock()
  89. }
  90. func (bridge *Bridge) GetAllUsers() []*User {
  91. bridge.usersLock.Lock()
  92. defer bridge.usersLock.Unlock()
  93. dbUsers := bridge.DB.User.GetAll()
  94. output := make([]*User, len(dbUsers))
  95. for index, dbUser := range dbUsers {
  96. user, ok := bridge.usersByMXID[dbUser.MXID]
  97. if !ok {
  98. user = bridge.loadDBUser(dbUser, nil)
  99. }
  100. output[index] = user
  101. }
  102. return output
  103. }
  104. func (bridge *Bridge) loadDBUser(dbUser *database.User, mxid *id.UserID) *User {
  105. if dbUser == nil {
  106. if mxid == nil {
  107. return nil
  108. }
  109. dbUser = bridge.DB.User.New()
  110. dbUser.MXID = *mxid
  111. dbUser.Insert()
  112. }
  113. user := bridge.NewUser(dbUser)
  114. bridge.usersByMXID[user.MXID] = user
  115. if len(user.JID) > 0 {
  116. bridge.usersByJID[user.JID] = user
  117. }
  118. if len(user.ManagementRoom) > 0 {
  119. bridge.managementRooms[user.ManagementRoom] = user
  120. }
  121. return user
  122. }
  123. func (user *User) GetPortals() []*Portal {
  124. keys := user.User.GetPortalKeys()
  125. portals := make([]*Portal, len(keys))
  126. user.bridge.portalsLock.Lock()
  127. for i, key := range keys {
  128. portal, ok := user.bridge.portalsByJID[key]
  129. if !ok {
  130. portal = user.bridge.loadDBPortal(user.bridge.DB.Portal.GetByJID(key), &key)
  131. }
  132. portals[i] = portal
  133. }
  134. user.bridge.portalsLock.Unlock()
  135. return portals
  136. }
  137. func (bridge *Bridge) NewUser(dbUser *database.User) *User {
  138. user := &User{
  139. User: dbUser,
  140. bridge: bridge,
  141. log: bridge.Log.Sub("User").Sub(string(dbUser.MXID)),
  142. IsRelaybot: false,
  143. chatListReceived: make(chan struct{}, 1),
  144. syncPortalsDone: make(chan struct{}, 1),
  145. syncStart: make(chan struct{}, 1),
  146. messages: make(chan PortalMessage, bridge.Config.Bridge.UserMessageBuffer),
  147. }
  148. user.RelaybotWhitelisted = user.bridge.Config.Bridge.Permissions.IsRelaybotWhitelisted(user.MXID)
  149. user.Whitelisted = user.bridge.Config.Bridge.Permissions.IsWhitelisted(user.MXID)
  150. user.Admin = user.bridge.Config.Bridge.Permissions.IsAdmin(user.MXID)
  151. go user.handleMessageLoop()
  152. return user
  153. }
  154. func (user *User) GetManagementRoom() id.RoomID {
  155. if len(user.ManagementRoom) == 0 {
  156. user.mgmtCreateLock.Lock()
  157. defer user.mgmtCreateLock.Unlock()
  158. if len(user.ManagementRoom) > 0 {
  159. return user.ManagementRoom
  160. }
  161. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  162. Topic: "WhatsApp bridge notices",
  163. IsDirect: true,
  164. })
  165. if err != nil {
  166. user.log.Errorln("Failed to auto-create management room:", err)
  167. } else {
  168. user.SetManagementRoom(resp.RoomID)
  169. }
  170. }
  171. return user.ManagementRoom
  172. }
  173. func (user *User) SetManagementRoom(roomID id.RoomID) {
  174. existingUser, ok := user.bridge.managementRooms[roomID]
  175. if ok {
  176. existingUser.ManagementRoom = ""
  177. existingUser.Update()
  178. }
  179. user.ManagementRoom = roomID
  180. user.bridge.managementRooms[user.ManagementRoom] = user
  181. user.Update()
  182. }
  183. func (user *User) SetSession(session *whatsapp.Session) {
  184. user.Session = session
  185. if session == nil {
  186. user.LastConnection = 0
  187. }
  188. user.Update()
  189. }
  190. func (user *User) Connect(evenIfNoSession bool) bool {
  191. if user.Conn != nil {
  192. return true
  193. } else if !evenIfNoSession && user.Session == nil {
  194. return false
  195. }
  196. user.log.Debugln("Connecting to WhatsApp")
  197. timeout := time.Duration(user.bridge.Config.Bridge.ConnectionTimeout)
  198. if timeout == 0 {
  199. timeout = 20
  200. }
  201. conn, err := whatsapp.NewConn(timeout * time.Second)
  202. if err != nil {
  203. user.log.Errorln("Failed to connect to WhatsApp:", err)
  204. user.sendMarkdownBridgeAlert("\u26a0 Failed to connect to WhatsApp server. " +
  205. "This indicates a network problem on the bridge server. See bridge logs for more info.")
  206. return false
  207. }
  208. user.Conn = whatsappExt.ExtendConn(conn)
  209. _ = user.Conn.SetClientName(user.bridge.Config.WhatsApp.DeviceName, user.bridge.Config.WhatsApp.ShortName, WAVersion)
  210. user.log.Debugln("WhatsApp connection successful")
  211. user.Conn.AddHandler(user)
  212. return user.RestoreSession()
  213. }
  214. func (user *User) RestoreSession() bool {
  215. if user.Session != nil {
  216. sess, err := user.Conn.RestoreWithSession(*user.Session)
  217. if err == whatsapp.ErrAlreadyLoggedIn {
  218. return true
  219. } else if err != nil {
  220. user.log.Errorln("Failed to restore session:", err)
  221. user.sendMarkdownBridgeAlert("\u26a0 Failed to connect to WhatsApp. Make sure WhatsApp " +
  222. "on your phone is reachable and use `reconnect` to try connecting again.")
  223. user.log.Debugln("Disconnecting due to failed session restore...")
  224. _, err := user.Conn.Disconnect()
  225. if err != nil {
  226. user.log.Errorln("Failed to disconnect after failed session restore:", err)
  227. }
  228. return false
  229. }
  230. user.ConnectionErrors = 0
  231. user.SetSession(&sess)
  232. user.log.Debugln("Session restored successfully")
  233. user.PostLogin()
  234. }
  235. return true
  236. }
  237. func (user *User) HasSession() bool {
  238. return user.Session != nil
  239. }
  240. func (user *User) IsConnected() bool {
  241. return user.Conn != nil && user.Conn.IsConnected() && user.Conn.IsLoggedIn()
  242. }
  243. func (user *User) IsLoginInProgress() bool {
  244. return user.Conn != nil && user.Conn.IsLoginInProgress()
  245. }
  246. func (user *User) loginQrChannel(ce *CommandEvent, qrChan <-chan string, eventIDChan chan<- id.EventID) {
  247. var qrEventID id.EventID
  248. for code := range qrChan {
  249. if code == "stop" {
  250. return
  251. }
  252. qrCode, err := qrcode.Encode(code, qrcode.Low, 256)
  253. if err != nil {
  254. user.log.Errorln("Failed to encode QR code:", err)
  255. ce.Reply("Failed to encode QR code: %v", err)
  256. return
  257. }
  258. bot := user.bridge.AS.BotClient()
  259. resp, err := bot.UploadBytes(qrCode, "image/png")
  260. if err != nil {
  261. user.log.Errorln("Failed to upload QR code:", err)
  262. ce.Reply("Failed to upload QR code: %v", err)
  263. return
  264. }
  265. if qrEventID == "" {
  266. sendResp, err := bot.SendImage(ce.RoomID, code, resp.ContentURI)
  267. if err != nil {
  268. user.log.Errorln("Failed to send QR code to user:", err)
  269. return
  270. }
  271. qrEventID = sendResp.EventID
  272. eventIDChan <- qrEventID
  273. } else {
  274. _, err = bot.SendMessageEvent(ce.RoomID, event.EventMessage, &event.MessageEventContent{
  275. MsgType: event.MsgImage,
  276. Body: code,
  277. URL: resp.ContentURI.CUString(),
  278. NewContent: &event.MessageEventContent{
  279. MsgType: event.MsgImage,
  280. Body: code,
  281. URL: resp.ContentURI.CUString(),
  282. },
  283. RelatesTo: &event.RelatesTo{
  284. Type: event.RelReplace,
  285. EventID: qrEventID,
  286. },
  287. })
  288. if err != nil {
  289. user.log.Errorln("Failed to send edited QR code to user:", err)
  290. }
  291. }
  292. }
  293. }
  294. func (user *User) Login(ce *CommandEvent) {
  295. qrChan := make(chan string, 3)
  296. eventIDChan := make(chan id.EventID, 1)
  297. go user.loginQrChannel(ce, qrChan, eventIDChan)
  298. session, err := user.Conn.LoginWithRetry(qrChan, user.bridge.Config.Bridge.LoginQRRegenCount)
  299. qrChan <- "stop"
  300. if err != nil {
  301. var eventID id.EventID
  302. select {
  303. case eventID = <-eventIDChan:
  304. default:
  305. }
  306. reply := event.MessageEventContent{
  307. MsgType: event.MsgText,
  308. }
  309. if err == whatsapp.ErrAlreadyLoggedIn {
  310. reply.Body = "You're already logged in"
  311. } else if err == whatsapp.ErrLoginInProgress {
  312. reply.Body = "You have a login in progress already."
  313. } else if err == whatsapp.ErrLoginTimedOut {
  314. reply.Body = "QR code scan timed out. Please try again."
  315. } else {
  316. user.log.Warnln("Failed to log in:", err)
  317. reply.Body = fmt.Sprintf("Unknown error while logging in: %v", err)
  318. }
  319. msg := reply
  320. if eventID != "" {
  321. msg.NewContent = &reply
  322. msg.RelatesTo = &event.RelatesTo{
  323. Type: event.RelReplace,
  324. EventID: eventID,
  325. }
  326. }
  327. _, _ = ce.Bot.SendMessageEvent(ce.RoomID, event.EventMessage, &msg)
  328. return
  329. }
  330. // TODO there's a bit of duplication between this and the provisioning API login method
  331. // Also between the two logout methods (commands.go and provisioning.go)
  332. user.ConnectionErrors = 0
  333. user.JID = strings.Replace(user.Conn.Info.Wid, whatsappExt.OldUserSuffix, whatsappExt.NewUserSuffix, 1)
  334. user.addToJIDMap()
  335. user.SetSession(&session)
  336. ce.Reply("Successfully logged in, synchronizing chats...")
  337. user.PostLogin()
  338. }
  339. type Chat struct {
  340. Portal *Portal
  341. LastMessageTime uint64
  342. Contact whatsapp.Contact
  343. }
  344. type ChatList []Chat
  345. func (cl ChatList) Len() int {
  346. return len(cl)
  347. }
  348. func (cl ChatList) Less(i, j int) bool {
  349. return cl[i].LastMessageTime > cl[j].LastMessageTime
  350. }
  351. func (cl ChatList) Swap(i, j int) {
  352. cl[i], cl[j] = cl[j], cl[i]
  353. }
  354. func (user *User) PostLogin() {
  355. user.log.Debugln("Locking processing of incoming messages and starting post-login sync")
  356. user.syncWait.Add(1)
  357. user.syncStart <- struct{}{}
  358. go user.intPostLogin()
  359. }
  360. func (user *User) tryAutomaticDoublePuppeting() {
  361. if len(user.bridge.Config.Bridge.LoginSharedSecret) == 0 {
  362. // Automatic login not enabled
  363. return
  364. } else if _, homeserver, _ := user.MXID.Parse(); homeserver != user.bridge.Config.Homeserver.Domain {
  365. // user is on another homeserver
  366. return
  367. }
  368. puppet := user.bridge.GetPuppetByJID(user.JID)
  369. if len(puppet.CustomMXID) > 0 {
  370. // Custom puppet already enabled
  371. return
  372. }
  373. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  374. if err != nil {
  375. user.log.Warnln("Failed to login with shared secret:", err)
  376. return
  377. }
  378. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  379. if err != nil {
  380. puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
  381. return
  382. }
  383. user.log.Infoln("Successfully automatically enabled custom puppet")
  384. }
  385. func (user *User) sendBridgeNotice(formatString string, args ...interface{}) {
  386. notice := fmt.Sprintf(formatString, args...)
  387. _, err := user.bridge.Bot.SendNotice(user.GetManagementRoom(), notice)
  388. if err != nil {
  389. user.log.Warnf("Failed to send bridge notice \"%s\": %v", notice, err)
  390. }
  391. }
  392. func (user *User) sendMarkdownBridgeAlert(formatString string, args ...interface{}) {
  393. notice := fmt.Sprintf(formatString, args...)
  394. content := format.RenderMarkdown(notice, true, false)
  395. _, err := user.bridge.Bot.SendMessageEvent(user.GetManagementRoom(), event.EventMessage, content)
  396. if err != nil {
  397. user.log.Warnf("Failed to send bridge alert \"%s\": %v", notice, err)
  398. }
  399. }
  400. func (user *User) intPostLogin() {
  401. defer user.syncWait.Done()
  402. user.createCommunity()
  403. user.tryAutomaticDoublePuppeting()
  404. select {
  405. case <-user.chatListReceived:
  406. user.log.Debugln("Chat list receive confirmation received in PostLogin")
  407. case <-time.After(time.Duration(user.bridge.Config.Bridge.ChatListWait) * time.Second):
  408. user.log.Warnln("Timed out waiting for chat list to arrive!")
  409. return
  410. }
  411. select {
  412. case <-user.syncPortalsDone:
  413. user.log.Debugln("Post-login portal sync complete, unlocking processing of incoming messages.")
  414. case <-time.After(time.Duration(user.bridge.Config.Bridge.PortalSyncWait) * time.Second):
  415. user.log.Warnln("Timed out waiting for portal sync to complete! Unlocking processing of incoming messages.")
  416. }
  417. }
  418. func (user *User) HandleChatList(chats []whatsapp.Chat) {
  419. user.log.Infoln("Chat list received")
  420. chatMap := make(map[string]whatsapp.Chat)
  421. for _, chat := range user.Conn.Store.Chats {
  422. chatMap[chat.Jid] = chat
  423. }
  424. for _, chat := range chats {
  425. chatMap[chat.Jid] = chat
  426. }
  427. select {
  428. case user.chatListReceived <- struct{}{}:
  429. default:
  430. }
  431. go user.syncPortals(chatMap, false)
  432. }
  433. func (user *User) syncPortals(chatMap map[string]whatsapp.Chat, createAll bool) {
  434. if chatMap == nil {
  435. chatMap = user.Conn.Store.Chats
  436. }
  437. user.log.Infoln("Reading chat list")
  438. chats := make(ChatList, 0, len(chatMap))
  439. existingKeys := user.GetInCommunityMap()
  440. portalKeys := make([]database.PortalKeyWithMeta, 0, len(chatMap))
  441. for _, chat := range chatMap {
  442. ts, err := strconv.ParseUint(chat.LastMessageTime, 10, 64)
  443. if err != nil {
  444. user.log.Warnfln("Non-integer last message time in %s: %s", chat.Jid, chat.LastMessageTime)
  445. continue
  446. }
  447. portal := user.GetPortalByJID(chat.Jid)
  448. chats = append(chats, Chat{
  449. Portal: portal,
  450. Contact: user.Conn.Store.Contacts[chat.Jid],
  451. LastMessageTime: ts,
  452. })
  453. var inCommunity, ok bool
  454. if inCommunity, ok = existingKeys[portal.Key]; !ok || !inCommunity {
  455. inCommunity = user.addPortalToCommunity(portal)
  456. if portal.IsPrivateChat() {
  457. puppet := user.bridge.GetPuppetByJID(portal.Key.JID)
  458. user.addPuppetToCommunity(puppet)
  459. }
  460. }
  461. portalKeys = append(portalKeys, database.PortalKeyWithMeta{PortalKey: portal.Key, InCommunity: inCommunity})
  462. }
  463. user.log.Infoln("Read chat list, updating user-portal mapping")
  464. err := user.SetPortalKeys(portalKeys)
  465. if err != nil {
  466. user.log.Warnln("Failed to update user-portal mapping:", err)
  467. }
  468. sort.Sort(chats)
  469. limit := user.bridge.Config.Bridge.InitialChatSync
  470. if limit < 0 {
  471. limit = len(chats)
  472. }
  473. now := uint64(time.Now().Unix())
  474. user.log.Infoln("Syncing portals")
  475. for i, chat := range chats {
  476. if chat.LastMessageTime+user.bridge.Config.Bridge.SyncChatMaxAge < now {
  477. break
  478. }
  479. create := (chat.LastMessageTime >= user.LastConnection && user.LastConnection > 0) || i < limit
  480. if len(chat.Portal.MXID) > 0 || create || createAll {
  481. chat.Portal.Sync(user, chat.Contact)
  482. err := chat.Portal.BackfillHistory(user, chat.LastMessageTime)
  483. if err != nil {
  484. chat.Portal.log.Errorln("Error backfilling history:", err)
  485. }
  486. }
  487. }
  488. user.log.Infoln("Finished syncing portals")
  489. select {
  490. case user.syncPortalsDone <- struct{}{}:
  491. default:
  492. }
  493. }
  494. func (user *User) HandleContactList(contacts []whatsapp.Contact) {
  495. contactMap := make(map[string]whatsapp.Contact)
  496. for _, contact := range contacts {
  497. contactMap[contact.Jid] = contact
  498. }
  499. go user.syncPuppets(contactMap)
  500. }
  501. func (user *User) syncPuppets(contacts map[string]whatsapp.Contact) {
  502. if contacts == nil {
  503. contacts = user.Conn.Store.Contacts
  504. }
  505. user.log.Infoln("Syncing puppet info from contacts")
  506. for jid, contact := range contacts {
  507. if strings.HasSuffix(jid, whatsappExt.NewUserSuffix) {
  508. puppet := user.bridge.GetPuppetByJID(contact.Jid)
  509. puppet.Sync(user, contact)
  510. }
  511. }
  512. user.log.Infoln("Finished syncing puppet info from contacts")
  513. }
  514. func (user *User) updateLastConnectionIfNecessary() {
  515. if user.LastConnection+60 < uint64(time.Now().Unix()) {
  516. user.UpdateLastConnection()
  517. }
  518. }
  519. func (user *User) HandleError(err error) {
  520. if errors.Cause(err) != whatsapp.ErrInvalidWsData {
  521. user.log.Errorfln("WhatsApp error: %v", err)
  522. }
  523. if closed, ok := err.(*whatsapp.ErrConnectionClosed); ok {
  524. user.bridge.Metrics.TrackDisconnection(user.MXID)
  525. if closed.Code == 1000 && user.cleanDisconnection {
  526. user.cleanDisconnection = false
  527. user.log.Infoln("Clean disconnection by server")
  528. return
  529. }
  530. go user.tryReconnect(fmt.Sprintf("Your WhatsApp connection was closed with websocket status code %d", closed.Code))
  531. } else if failed, ok := err.(*whatsapp.ErrConnectionFailed); ok {
  532. user.bridge.Metrics.TrackDisconnection(user.MXID)
  533. user.ConnectionErrors++
  534. go user.tryReconnect(fmt.Sprintf("Your WhatsApp connection failed: %v", failed.Err))
  535. }
  536. // Otherwise unknown error, probably mostly harmless
  537. }
  538. func (user *User) tryReconnect(msg string) {
  539. if user.ConnectionErrors > user.bridge.Config.Bridge.MaxConnectionAttempts {
  540. user.sendMarkdownBridgeAlert("%s. Use the `reconnect` command to reconnect.", msg)
  541. return
  542. }
  543. if user.bridge.Config.Bridge.ReportConnectionRetry {
  544. user.sendBridgeNotice("%s. Reconnecting...", msg)
  545. // Don't want the same error to be repeated
  546. msg = ""
  547. }
  548. var tries uint
  549. var exponentialBackoff bool
  550. baseDelay := time.Duration(user.bridge.Config.Bridge.ConnectionRetryDelay)
  551. if baseDelay < 0 {
  552. exponentialBackoff = true
  553. baseDelay = -baseDelay + 1
  554. }
  555. delay := baseDelay
  556. for user.ConnectionErrors <= user.bridge.Config.Bridge.MaxConnectionAttempts {
  557. err := user.Conn.Restore()
  558. if err == nil {
  559. user.ConnectionErrors = 0
  560. if user.bridge.Config.Bridge.ReportConnectionRetry {
  561. user.sendBridgeNotice("Reconnected successfully")
  562. }
  563. user.PostLogin()
  564. return
  565. } else if err.Error() == "init responded with 400" {
  566. user.log.Infoln("Got init 400 error when trying to reconnect, resetting connection...")
  567. sess, err := user.Conn.Disconnect()
  568. if err != nil {
  569. user.log.Debugln("Error while disconnecting for connection reset:", err)
  570. }
  571. if len(sess.Wid) > 0 {
  572. user.SetSession(&sess)
  573. }
  574. }
  575. user.log.Errorln("Error while trying to reconnect after disconnection:", err)
  576. tries++
  577. user.ConnectionErrors++
  578. if user.ConnectionErrors <= user.bridge.Config.Bridge.MaxConnectionAttempts {
  579. if exponentialBackoff {
  580. delay = (1 << tries) + baseDelay
  581. }
  582. if user.bridge.Config.Bridge.ReportConnectionRetry {
  583. user.sendBridgeNotice("Reconnection attempt failed: %v. Retrying in %d seconds...", err, delay)
  584. }
  585. time.Sleep(delay * time.Second)
  586. }
  587. }
  588. if user.bridge.Config.Bridge.ReportConnectionRetry {
  589. user.sendMarkdownBridgeAlert("%d reconnection attempts failed. Use the `reconnect` command to try to reconnect manually.", tries)
  590. } else {
  591. user.sendMarkdownBridgeAlert("\u26a0 %s. Additionally, %d reconnection attempts failed. Use the `reconnect` command to try to reconnect.", msg, tries)
  592. }
  593. }
  594. func (user *User) ShouldCallSynchronously() bool {
  595. return true
  596. }
  597. func (user *User) HandleJSONParseError(err error) {
  598. user.log.Errorln("WhatsApp JSON parse error:", err)
  599. }
  600. func (user *User) PortalKey(jid types.WhatsAppID) database.PortalKey {
  601. return database.NewPortalKey(jid, user.JID)
  602. }
  603. func (user *User) GetPortalByJID(jid types.WhatsAppID) *Portal {
  604. return user.bridge.GetPortalByJID(user.PortalKey(jid))
  605. }
  606. func (user *User) handleMessageLoop() {
  607. for {
  608. select {
  609. case msg := <-user.messages:
  610. user.GetPortalByJID(msg.chat).messages <- msg
  611. case <-user.syncStart:
  612. user.syncWait.Wait()
  613. }
  614. }
  615. }
  616. func (user *User) putMessage(message PortalMessage) {
  617. select {
  618. case user.messages <- message:
  619. default:
  620. user.log.Warnln("Buffer is full, dropping message in", message.chat)
  621. }
  622. }
  623. func (user *User) HandleNewContact(contact whatsapp.Contact) {
  624. user.log.Debugfln("Contact message: %+v", contact)
  625. go func() {
  626. if strings.HasSuffix(contact.Jid, whatsappExt.OldUserSuffix) {
  627. contact.Jid = strings.Replace(contact.Jid, whatsappExt.OldUserSuffix, whatsappExt.NewUserSuffix, -1)
  628. }
  629. puppet := user.bridge.GetPuppetByJID(contact.Jid)
  630. puppet.UpdateName(user, contact)
  631. }()
  632. }
  633. func (user *User) HandleBatteryMessage(battery whatsapp.BatteryMessage) {
  634. user.log.Debugfln("Battery message: %+v", battery)
  635. var notice string
  636. if !battery.Plugged && battery.Percentage < 15 && user.batteryWarningsSent < 1 {
  637. notice = fmt.Sprintf("Phone battery low (%d %% remaining)", battery.Percentage)
  638. user.batteryWarningsSent = 1
  639. } else if !battery.Plugged && battery.Percentage < 5 && user.batteryWarningsSent < 2 {
  640. notice = fmt.Sprintf("Phone battery very low (%d %% remaining)", battery.Percentage)
  641. user.batteryWarningsSent = 2
  642. } else if battery.Percentage > 15 || battery.Plugged {
  643. user.batteryWarningsSent = 0
  644. }
  645. if notice != "" {
  646. go user.sendBridgeNotice("%s", notice)
  647. }
  648. }
  649. func (user *User) HandleTextMessage(message whatsapp.TextMessage) {
  650. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  651. }
  652. func (user *User) HandleImageMessage(message whatsapp.ImageMessage) {
  653. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  654. }
  655. func (user *User) HandleStickerMessage(message whatsapp.StickerMessage) {
  656. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  657. }
  658. func (user *User) HandleVideoMessage(message whatsapp.VideoMessage) {
  659. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  660. }
  661. func (user *User) HandleAudioMessage(message whatsapp.AudioMessage) {
  662. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  663. }
  664. func (user *User) HandleDocumentMessage(message whatsapp.DocumentMessage) {
  665. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  666. }
  667. func (user *User) HandleContactMessage(message whatsapp.ContactMessage) {
  668. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  669. }
  670. func (user *User) HandleLocationMessage(message whatsapp.LocationMessage) {
  671. user.putMessage(PortalMessage{message.Info.RemoteJid, user, message, message.Info.Timestamp})
  672. }
  673. func (user *User) HandleMessageRevoke(message whatsappExt.MessageRevocation) {
  674. user.putMessage(PortalMessage{message.RemoteJid, user, message, 0})
  675. }
  676. type FakeMessage struct {
  677. Text string
  678. ID string
  679. Alert bool
  680. }
  681. func (user *User) HandleCallInfo(info whatsappExt.CallInfo) {
  682. if info.Data != nil {
  683. return
  684. }
  685. data := FakeMessage{
  686. ID: info.ID,
  687. }
  688. switch info.Type {
  689. case whatsappExt.CallOffer:
  690. if !user.bridge.Config.Bridge.CallNotices.Start {
  691. return
  692. }
  693. data.Text = "Incoming call"
  694. data.Alert = true
  695. case whatsappExt.CallOfferVideo:
  696. if !user.bridge.Config.Bridge.CallNotices.Start {
  697. return
  698. }
  699. data.Text = "Incoming video call"
  700. data.Alert = true
  701. case whatsappExt.CallTerminate:
  702. if !user.bridge.Config.Bridge.CallNotices.End {
  703. return
  704. }
  705. data.Text = "Call ended"
  706. data.ID += "E"
  707. default:
  708. return
  709. }
  710. portal := user.GetPortalByJID(info.From)
  711. if portal != nil {
  712. portal.messages <- PortalMessage{info.From, user, data, 0}
  713. }
  714. }
  715. func (user *User) HandlePresence(info whatsappExt.Presence) {
  716. puppet := user.bridge.GetPuppetByJID(info.SenderJID)
  717. switch info.Status {
  718. case whatsapp.PresenceUnavailable:
  719. _ = puppet.DefaultIntent().SetPresence("offline")
  720. case whatsapp.PresenceAvailable:
  721. if len(puppet.typingIn) > 0 && puppet.typingAt+15 > time.Now().Unix() {
  722. portal := user.bridge.GetPortalByMXID(puppet.typingIn)
  723. _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  724. puppet.typingIn = ""
  725. puppet.typingAt = 0
  726. } else {
  727. _ = puppet.DefaultIntent().SetPresence("online")
  728. }
  729. case whatsapp.PresenceComposing:
  730. portal := user.GetPortalByJID(info.JID)
  731. if len(puppet.typingIn) > 0 && puppet.typingAt+15 > time.Now().Unix() {
  732. if puppet.typingIn == portal.MXID {
  733. return
  734. }
  735. _, _ = puppet.IntentFor(portal).UserTyping(puppet.typingIn, false, 0)
  736. }
  737. puppet.typingIn = portal.MXID
  738. puppet.typingAt = time.Now().Unix()
  739. _, _ = puppet.IntentFor(portal).UserTyping(portal.MXID, true, 15*1000)
  740. }
  741. }
  742. func (user *User) HandleMsgInfo(info whatsappExt.MsgInfo) {
  743. if (info.Command == whatsappExt.MsgInfoCommandAck || info.Command == whatsappExt.MsgInfoCommandAcks) && info.Acknowledgement == whatsappExt.AckMessageRead {
  744. portal := user.GetPortalByJID(info.ToJID)
  745. if len(portal.MXID) == 0 {
  746. return
  747. }
  748. go func() {
  749. intent := user.bridge.GetPuppetByJID(info.SenderJID).IntentFor(portal)
  750. for _, id := range info.IDs {
  751. msg := user.bridge.DB.Message.GetByJID(portal.Key, id)
  752. if msg == nil {
  753. continue
  754. }
  755. err := intent.MarkRead(portal.MXID, msg.MXID)
  756. if err != nil {
  757. user.log.Warnln("Failed to mark message %s as read by %s: %v", msg.MXID, info.SenderJID, err)
  758. }
  759. }
  760. }()
  761. }
  762. }
  763. func (user *User) HandleCommand(cmd whatsappExt.Command) {
  764. switch cmd.Type {
  765. case whatsappExt.CommandPicture:
  766. if strings.HasSuffix(cmd.JID, whatsappExt.NewUserSuffix) {
  767. puppet := user.bridge.GetPuppetByJID(cmd.JID)
  768. go puppet.UpdateAvatar(user, cmd.ProfilePicInfo)
  769. } else {
  770. portal := user.GetPortalByJID(cmd.JID)
  771. go portal.UpdateAvatar(user, cmd.ProfilePicInfo, true)
  772. }
  773. case whatsappExt.CommandDisconnect:
  774. user.cleanDisconnection = true
  775. if cmd.Kind == "replaced" {
  776. go user.sendMarkdownBridgeAlert("\u26a0 Your WhatsApp connection was closed by the server because you opened another WhatsApp Web client.\n\n" +
  777. "Use the `reconnect` command to disconnect the other client and resume bridging.")
  778. } else {
  779. user.log.Warnln("Unknown kind of disconnect:", string(cmd.Raw))
  780. go user.sendMarkdownBridgeAlert("\u26a0 Your WhatsApp connection was closed by the server (reason code: %s).\n\n"+
  781. "Use the `reconnect` command to reconnect.", cmd.Kind)
  782. }
  783. }
  784. }
  785. func (user *User) HandleChatUpdate(cmd whatsappExt.ChatUpdate) {
  786. if cmd.Command != whatsappExt.ChatUpdateCommandAction {
  787. return
  788. }
  789. portal := user.GetPortalByJID(cmd.JID)
  790. if len(portal.MXID) == 0 {
  791. if cmd.Data.Action == whatsappExt.ChatActionIntroduce || cmd.Data.Action == whatsappExt.ChatActionCreate {
  792. go func() {
  793. err := portal.CreateMatrixRoom(user)
  794. if err != nil {
  795. user.log.Errorln("Failed to create portal room after receiving join event:", err)
  796. }
  797. }()
  798. }
  799. return
  800. }
  801. switch cmd.Data.Action {
  802. case whatsappExt.ChatActionNameChange:
  803. go portal.UpdateName(cmd.Data.NameChange.Name, cmd.Data.SenderJID, true)
  804. case whatsappExt.ChatActionAddTopic:
  805. go portal.UpdateTopic(cmd.Data.AddTopic.Topic, cmd.Data.SenderJID, true)
  806. case whatsappExt.ChatActionRemoveTopic:
  807. go portal.UpdateTopic("", cmd.Data.SenderJID, true)
  808. case whatsappExt.ChatActionPromote:
  809. go portal.ChangeAdminStatus(cmd.Data.UserChange.JIDs, true)
  810. case whatsappExt.ChatActionDemote:
  811. go portal.ChangeAdminStatus(cmd.Data.UserChange.JIDs, false)
  812. case whatsappExt.ChatActionAnnounce:
  813. go portal.RestrictMessageSending(cmd.Data.Announce)
  814. case whatsappExt.ChatActionRestrict:
  815. go portal.RestrictMetadataChanges(cmd.Data.Restrict)
  816. case whatsappExt.ChatActionRemove:
  817. go portal.HandleWhatsAppKick(cmd.Data.SenderJID, cmd.Data.UserChange.JIDs)
  818. case whatsappExt.ChatActionAdd:
  819. go portal.HandleWhatsAppInvite(cmd.Data.SenderJID, cmd.Data.UserChange.JIDs)
  820. case whatsappExt.ChatActionIntroduce:
  821. if cmd.Data.SenderJID != "unknown" {
  822. go portal.Sync(user, whatsapp.Contact{Jid: portal.Key.JID})
  823. }
  824. }
  825. }
  826. func (user *User) HandleJsonMessage(message string) {
  827. var msg json.RawMessage
  828. err := json.Unmarshal([]byte(message), &msg)
  829. if err != nil {
  830. return
  831. }
  832. user.log.Debugln("JSON message:", message)
  833. user.updateLastConnectionIfNecessary()
  834. }
  835. func (user *User) HandleRawMessage(message *waProto.WebMessageInfo) {
  836. user.updateLastConnectionIfNecessary()
  837. }
  838. func (user *User) NeedsRelaybot(portal *Portal) bool {
  839. return !user.HasSession() || !user.IsInPortal(portal.Key)
  840. }