portal.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2019 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. "bytes"
  19. "encoding/gob"
  20. "encoding/hex"
  21. "fmt"
  22. "image"
  23. "image/gif"
  24. "image/jpeg"
  25. "image/png"
  26. "math/rand"
  27. "mime"
  28. "net/http"
  29. "strings"
  30. "sync"
  31. "github.com/Rhymen/go-whatsapp"
  32. waProto "github.com/Rhymen/go-whatsapp/binary/proto"
  33. log "maunium.net/go/maulogger/v2"
  34. "maunium.net/go/mautrix"
  35. "maunium.net/go/mautrix-appservice"
  36. "maunium.net/go/mautrix-whatsapp/database"
  37. "maunium.net/go/mautrix-whatsapp/types"
  38. "maunium.net/go/mautrix-whatsapp/whatsapp-ext"
  39. )
  40. func (bridge *Bridge) GetPortalByMXID(mxid types.MatrixRoomID) *Portal {
  41. bridge.portalsLock.Lock()
  42. defer bridge.portalsLock.Unlock()
  43. portal, ok := bridge.portalsByMXID[mxid]
  44. if !ok {
  45. dbPortal := bridge.DB.Portal.GetByMXID(mxid)
  46. if dbPortal == nil {
  47. return nil
  48. }
  49. portal = bridge.NewPortal(dbPortal)
  50. bridge.portalsByJID[portal.Key] = portal
  51. if len(portal.MXID) > 0 {
  52. bridge.portalsByMXID[portal.MXID] = portal
  53. }
  54. }
  55. return portal
  56. }
  57. func (bridge *Bridge) GetPortalByJID(key database.PortalKey) *Portal {
  58. bridge.portalsLock.Lock()
  59. defer bridge.portalsLock.Unlock()
  60. portal, ok := bridge.portalsByJID[key]
  61. if !ok {
  62. dbPortal := bridge.DB.Portal.GetByJID(key)
  63. if dbPortal == nil {
  64. dbPortal = bridge.DB.Portal.New()
  65. dbPortal.Key = key
  66. dbPortal.Insert()
  67. }
  68. portal = bridge.NewPortal(dbPortal)
  69. bridge.portalsByJID[portal.Key] = portal
  70. if len(portal.MXID) > 0 {
  71. bridge.portalsByMXID[portal.MXID] = portal
  72. }
  73. }
  74. return portal
  75. }
  76. func (bridge *Bridge) GetAllPortals() []*Portal {
  77. bridge.portalsLock.Lock()
  78. defer bridge.portalsLock.Unlock()
  79. dbPortals := bridge.DB.Portal.GetAll()
  80. output := make([]*Portal, len(dbPortals))
  81. for index, dbPortal := range dbPortals {
  82. portal, ok := bridge.portalsByJID[dbPortal.Key]
  83. if !ok {
  84. portal = bridge.NewPortal(dbPortal)
  85. bridge.portalsByJID[portal.Key] = portal
  86. if len(dbPortal.MXID) > 0 {
  87. bridge.portalsByMXID[dbPortal.MXID] = portal
  88. }
  89. }
  90. output[index] = portal
  91. }
  92. return output
  93. }
  94. func (bridge *Bridge) NewPortal(dbPortal *database.Portal) *Portal {
  95. return &Portal{
  96. Portal: dbPortal,
  97. bridge: bridge,
  98. log: bridge.Log.Sub(fmt.Sprintf("Portal/%s", dbPortal.Key)),
  99. messageLocks: make(map[types.WhatsAppMessageID]sync.Mutex),
  100. recentlyHandled: [recentlyHandledLength]types.WhatsAppMessageID{},
  101. }
  102. }
  103. const recentlyHandledLength = 100
  104. type Portal struct {
  105. *database.Portal
  106. bridge *Bridge
  107. log log.Logger
  108. roomCreateLock sync.Mutex
  109. messageLocksLock sync.Mutex
  110. messageLocks map[types.WhatsAppMessageID]sync.Mutex
  111. recentlyHandled [recentlyHandledLength]types.WhatsAppMessageID
  112. recentlyHandledLock sync.Mutex
  113. recentlyHandledIndex uint8
  114. isPrivate *bool
  115. }
  116. func (portal *Portal) getMessageLock(messageID types.WhatsAppMessageID) sync.Mutex {
  117. portal.messageLocksLock.Lock()
  118. defer portal.messageLocksLock.Unlock()
  119. lock, ok := portal.messageLocks[messageID]
  120. if !ok {
  121. portal.messageLocks[messageID] = lock
  122. }
  123. return lock
  124. }
  125. func (portal *Portal) deleteMessageLock(messageID types.WhatsAppMessageID) {
  126. portal.messageLocksLock.Lock()
  127. delete(portal.messageLocks, messageID)
  128. portal.messageLocksLock.Unlock()
  129. }
  130. func (portal *Portal) isRecentlyHandled(id types.WhatsAppMessageID) bool {
  131. start := portal.recentlyHandledIndex
  132. for i := start; i != start; i = (i - 1) % recentlyHandledLength {
  133. if portal.recentlyHandled[i] == id {
  134. return true
  135. }
  136. }
  137. return false
  138. }
  139. func (portal *Portal) isDuplicate(id types.WhatsAppMessageID) bool {
  140. msg := portal.bridge.DB.Message.GetByJID(portal.Key, id)
  141. if msg != nil {
  142. return true
  143. }
  144. return false
  145. }
  146. func init() {
  147. gob.Register(&waProto.Message{})
  148. }
  149. func (portal *Portal) markHandled(source *User, message *waProto.WebMessageInfo, mxid types.MatrixEventID) {
  150. msg := portal.bridge.DB.Message.New()
  151. msg.Chat = portal.Key
  152. msg.JID = message.GetKey().GetId()
  153. msg.MXID = mxid
  154. if message.GetKey().GetFromMe() {
  155. msg.Sender = source.JID
  156. } else if portal.IsPrivateChat() {
  157. msg.Sender = portal.Key.JID
  158. } else {
  159. msg.Sender = message.GetKey().GetParticipant()
  160. if len(msg.Sender) == 0 {
  161. msg.Sender = message.GetParticipant()
  162. }
  163. }
  164. msg.Content = message.Message
  165. msg.Insert()
  166. portal.recentlyHandledLock.Lock()
  167. index := portal.recentlyHandledIndex
  168. portal.recentlyHandledIndex = (portal.recentlyHandledIndex + 1) % recentlyHandledLength
  169. portal.recentlyHandledLock.Unlock()
  170. portal.recentlyHandled[index] = msg.JID
  171. }
  172. func (portal *Portal) startHandling(id types.WhatsAppMessageID) (*sync.Mutex, bool) {
  173. if portal.isRecentlyHandled(id) {
  174. return nil, false
  175. }
  176. lock := portal.getMessageLock(id)
  177. lock.Lock()
  178. if portal.isDuplicate(id) {
  179. lock.Unlock()
  180. return nil, false
  181. }
  182. return &lock, true
  183. }
  184. func (portal *Portal) finishHandling(source *User, message *waProto.WebMessageInfo, mxid types.MatrixEventID) {
  185. portal.markHandled(source, message, mxid)
  186. id := message.GetKey().GetId()
  187. portal.deleteMessageLock(id)
  188. portal.log.Debugln("Handled message", id, "->", mxid)
  189. }
  190. func (portal *Portal) SyncParticipants(metadata *whatsappExt.GroupInfo) {
  191. changed := false
  192. levels, err := portal.MainIntent().PowerLevels(portal.MXID)
  193. if err != nil {
  194. levels = portal.GetBasePowerLevels()
  195. changed = true
  196. }
  197. for _, participant := range metadata.Participants {
  198. puppet := portal.bridge.GetPuppetByJID(participant.JID)
  199. err := puppet.Intent().EnsureJoined(portal.MXID)
  200. if err != nil {
  201. portal.log.Warnfln("Failed to make puppet of %s join %s: %v", participant.JID, portal.MXID, err)
  202. }
  203. user := portal.bridge.GetUserByJID(participant.JID)
  204. if user != nil && !portal.bridge.AS.StateStore.IsInvited(portal.MXID, user.MXID) {
  205. _, err = portal.MainIntent().InviteUser(portal.MXID, &mautrix.ReqInviteUser{
  206. UserID: user.MXID,
  207. })
  208. if err != nil {
  209. portal.log.Warnfln("Failed to invite %s to %s: %v", user.MXID, portal.MXID, err)
  210. }
  211. }
  212. expectedLevel := 0
  213. if participant.IsSuperAdmin {
  214. expectedLevel = 95
  215. } else if participant.IsAdmin {
  216. expectedLevel = 50
  217. }
  218. changed = levels.EnsureUserLevel(puppet.MXID, expectedLevel) || changed
  219. if user != nil {
  220. changed = levels.EnsureUserLevel(user.MXID, expectedLevel) || changed
  221. }
  222. }
  223. if changed {
  224. _, err = portal.MainIntent().SetPowerLevels(portal.MXID, levels)
  225. if err != nil {
  226. portal.log.Errorln("Failed to change power levels:", err)
  227. }
  228. }
  229. }
  230. func (portal *Portal) UpdateAvatar(user *User, avatar *whatsappExt.ProfilePicInfo) bool {
  231. if avatar == nil {
  232. var err error
  233. avatar, err = user.Conn.GetProfilePicThumb(portal.Key.JID)
  234. if err != nil {
  235. portal.log.Errorln(err)
  236. return false
  237. }
  238. }
  239. if avatar.Status != 0 {
  240. // 401: ???
  241. // 404: ???
  242. return false
  243. }
  244. if portal.Avatar == avatar.Tag {
  245. return false
  246. }
  247. data, err := avatar.DownloadBytes()
  248. if err != nil {
  249. portal.log.Warnln("Failed to download avatar:", err)
  250. return false
  251. }
  252. mimeType := http.DetectContentType(data)
  253. resp, err := portal.MainIntent().UploadBytes(data, mimeType)
  254. if err != nil {
  255. portal.log.Warnln("Failed to upload avatar:", err)
  256. return false
  257. }
  258. _, err = portal.MainIntent().SetRoomAvatar(portal.MXID, resp.ContentURI)
  259. if err != nil {
  260. portal.log.Warnln("Failed to set room topic:", err)
  261. return false
  262. }
  263. portal.Avatar = avatar.Tag
  264. return true
  265. }
  266. func (portal *Portal) UpdateName(name string, setBy types.WhatsAppID) bool {
  267. if portal.Name != name {
  268. intent := portal.bridge.GetPuppetByJID(setBy).Intent()
  269. _, err := intent.SetRoomName(portal.MXID, name)
  270. if err == nil {
  271. portal.Name = name
  272. return true
  273. }
  274. portal.log.Warnln("Failed to set room name:", err)
  275. }
  276. return false
  277. }
  278. func (portal *Portal) UpdateTopic(topic string, setBy types.WhatsAppID) bool {
  279. if portal.Topic != topic {
  280. intent := portal.bridge.GetPuppetByJID(setBy).Intent()
  281. _, err := intent.SetRoomTopic(portal.MXID, topic)
  282. if err == nil {
  283. portal.Topic = topic
  284. return true
  285. }
  286. portal.log.Warnln("Failed to set room topic:", err)
  287. }
  288. return false
  289. }
  290. func (portal *Portal) UpdateMetadata(user *User) bool {
  291. metadata, err := user.Conn.GetGroupMetaData(portal.Key.JID)
  292. if err != nil {
  293. portal.log.Errorln(err)
  294. return false
  295. }
  296. if metadata.Status != 0 {
  297. // 401: access denied
  298. // 404: group does (no longer) exist
  299. // 500: ??? happens with status@broadcast
  300. // TODO: update the room, e.g. change priority level
  301. // to send messages to moderator
  302. return false
  303. }
  304. portal.SyncParticipants(metadata)
  305. update := false
  306. update = portal.UpdateName(metadata.Name, metadata.NameSetBy) || update
  307. update = portal.UpdateTopic(metadata.Topic, metadata.TopicSetBy) || update
  308. return update
  309. }
  310. func (portal *Portal) Sync(user *User, contact whatsapp.Contact) {
  311. if portal.IsPrivateChat() {
  312. return
  313. }
  314. if len(portal.MXID) == 0 {
  315. portal.Name = contact.Name
  316. err := portal.CreateMatrixRoom([]string{user.MXID})
  317. if err != nil {
  318. portal.log.Errorln("Failed to create portal room:", err)
  319. return
  320. }
  321. } else {
  322. err := portal.MainIntent().EnsureInvited(portal.MXID, user.MXID)
  323. if err != nil {
  324. portal.log.Warnfln("Failed to ensure %s is invited to %s: %v", user.MXID, portal.MXID, err)
  325. }
  326. }
  327. update := false
  328. update = portal.UpdateMetadata(user) || update
  329. update = portal.UpdateAvatar(user, nil) || update
  330. if update {
  331. portal.Update()
  332. }
  333. }
  334. func (portal *Portal) GetBasePowerLevels() *mautrix.PowerLevels {
  335. anyone := 0
  336. nope := 99
  337. return &mautrix.PowerLevels{
  338. UsersDefault: anyone,
  339. EventsDefault: anyone,
  340. RedactPtr: &anyone,
  341. StateDefaultPtr: &nope,
  342. BanPtr: &nope,
  343. InvitePtr: &nope,
  344. Users: map[string]int{
  345. portal.MainIntent().UserID: 100,
  346. },
  347. Events: map[string]int{
  348. mautrix.StateRoomName.Type: anyone,
  349. mautrix.StateRoomAvatar.Type: anyone,
  350. mautrix.StateTopic.Type: anyone,
  351. },
  352. }
  353. }
  354. func (portal *Portal) ChangeAdminStatus(jids []string, setAdmin bool) {
  355. levels, err := portal.MainIntent().PowerLevels(portal.MXID)
  356. if err != nil {
  357. levels = portal.GetBasePowerLevels()
  358. }
  359. newLevel := 0
  360. if setAdmin {
  361. newLevel = 50
  362. }
  363. changed := false
  364. for _, jid := range jids {
  365. puppet := portal.bridge.GetPuppetByJID(jid)
  366. changed = levels.EnsureUserLevel(puppet.MXID, newLevel) || changed
  367. user := portal.bridge.GetUserByJID(jid)
  368. if user != nil {
  369. changed = levels.EnsureUserLevel(user.MXID, newLevel) || changed
  370. }
  371. }
  372. if changed {
  373. _, err = portal.MainIntent().SetPowerLevels(portal.MXID, levels)
  374. if err != nil {
  375. portal.log.Errorln("Failed to change power levels:", err)
  376. }
  377. }
  378. }
  379. func (portal *Portal) RestrictMessageSending(restrict bool) {
  380. levels, err := portal.MainIntent().PowerLevels(portal.MXID)
  381. if err != nil {
  382. levels = portal.GetBasePowerLevels()
  383. }
  384. if restrict {
  385. levels.EventsDefault = 50
  386. } else {
  387. levels.EventsDefault = 0
  388. }
  389. _, err = portal.MainIntent().SetPowerLevels(portal.MXID, levels)
  390. if err != nil {
  391. portal.log.Errorln("Failed to change power levels:", err)
  392. }
  393. }
  394. func (portal *Portal) RestrictMetadataChanges(restrict bool) {
  395. levels, err := portal.MainIntent().PowerLevels(portal.MXID)
  396. if err != nil {
  397. levels = portal.GetBasePowerLevels()
  398. }
  399. newLevel := 0
  400. if restrict {
  401. newLevel = 50
  402. }
  403. changed := false
  404. changed = levels.EnsureEventLevel(mautrix.StateRoomName, newLevel) || changed
  405. changed = levels.EnsureEventLevel(mautrix.StateRoomAvatar, newLevel) || changed
  406. changed = levels.EnsureEventLevel(mautrix.StateTopic, newLevel) || changed
  407. if changed {
  408. _, err = portal.MainIntent().SetPowerLevels(portal.MXID, levels)
  409. if err != nil {
  410. portal.log.Errorln("Failed to change power levels:", err)
  411. }
  412. }
  413. }
  414. func (portal *Portal) CreateMatrixRoom(invite []string) error {
  415. portal.roomCreateLock.Lock()
  416. defer portal.roomCreateLock.Unlock()
  417. if len(portal.MXID) > 0 {
  418. return nil
  419. }
  420. intent := portal.MainIntent()
  421. if err := intent.EnsureRegistered(); err != nil {
  422. return err
  423. }
  424. name := portal.Name
  425. topic := portal.Topic
  426. isPrivateChat := false
  427. if portal.IsPrivateChat() {
  428. name = ""
  429. topic = "WhatsApp private chat"
  430. isPrivateChat = true
  431. }
  432. resp, err := intent.CreateRoom(&mautrix.ReqCreateRoom{
  433. Visibility: "private",
  434. Name: name,
  435. Topic: topic,
  436. Invite: invite,
  437. Preset: "private_chat",
  438. IsDirect: isPrivateChat,
  439. InitialState: []*mautrix.Event{{
  440. Type: mautrix.StatePowerLevels,
  441. Content: mautrix.Content{
  442. PowerLevels: portal.GetBasePowerLevels(),
  443. },
  444. }},
  445. })
  446. if err != nil {
  447. return err
  448. }
  449. portal.MXID = resp.RoomID
  450. portal.Update()
  451. return nil
  452. }
  453. func (portal *Portal) IsPrivateChat() bool {
  454. if portal.isPrivate == nil {
  455. val := strings.HasSuffix(portal.Key.JID, whatsappExt.NewUserSuffix)
  456. portal.isPrivate = &val
  457. }
  458. return *portal.isPrivate
  459. }
  460. func (portal *Portal) MainIntent() *appservice.IntentAPI {
  461. if portal.IsPrivateChat() {
  462. return portal.bridge.GetPuppetByJID(portal.Key.JID).Intent()
  463. }
  464. return portal.bridge.Bot
  465. }
  466. func (portal *Portal) GetMessageIntent(user *User, info whatsapp.MessageInfo) *appservice.IntentAPI {
  467. if info.FromMe {
  468. if portal.IsPrivateChat() {
  469. // TODO handle own messages in private chats properly
  470. return nil
  471. }
  472. return portal.bridge.GetPuppetByJID(user.JID).Intent()
  473. } else if portal.IsPrivateChat() {
  474. return portal.MainIntent()
  475. } else if len(info.SenderJid) == 0 {
  476. if len(info.Source.GetParticipant()) != 0 {
  477. info.SenderJid = info.Source.GetParticipant()
  478. } else {
  479. return nil
  480. }
  481. }
  482. return portal.bridge.GetPuppetByJID(info.SenderJid).Intent()
  483. }
  484. func (portal *Portal) SetReply(content *mautrix.Content, info whatsapp.MessageInfo) {
  485. if len(info.QuotedMessageID) == 0 {
  486. return
  487. }
  488. message := portal.bridge.DB.Message.GetByJID(portal.Key, info.QuotedMessageID)
  489. if message != nil {
  490. event, err := portal.MainIntent().GetEvent(portal.MXID, message.MXID)
  491. if err != nil {
  492. portal.log.Warnln("Failed to get reply target:", err)
  493. return
  494. }
  495. event.Content.RemoveReplyFallback()
  496. content.SetReply(event)
  497. }
  498. return
  499. }
  500. func (portal *Portal) HandleTextMessage(source *User, message whatsapp.TextMessage) {
  501. lock, ok := portal.startHandling(message.Info.Id)
  502. if !ok {
  503. return
  504. }
  505. defer lock.Unlock()
  506. err := portal.CreateMatrixRoom([]string{source.MXID})
  507. if err != nil {
  508. portal.log.Errorln("Failed to create portal room:", err)
  509. return
  510. }
  511. intent := portal.GetMessageIntent(source, message.Info)
  512. if intent == nil {
  513. return
  514. }
  515. content := &mautrix.Content{
  516. Body: message.Text,
  517. MsgType: mautrix.MsgText,
  518. }
  519. portal.bridge.Formatter.ParseWhatsApp(content)
  520. portal.SetReply(content, message.Info)
  521. _, _ = intent.UserTyping(portal.MXID, false, 0)
  522. resp, err := intent.SendMassagedMessageEvent(portal.MXID, mautrix.EventMessage, content, int64(message.Info.Timestamp*1000))
  523. if err != nil {
  524. portal.log.Errorfln("Failed to handle message %s: %v", message.Info.Id, err)
  525. return
  526. }
  527. portal.finishHandling(source, message.Info.Source, resp.EventID)
  528. }
  529. func (portal *Portal) HandleMediaMessage(source *User, download func() ([]byte, error), thumbnail []byte, info whatsapp.MessageInfo, mimeType, caption string) {
  530. lock, ok := portal.startHandling(info.Id)
  531. if !ok {
  532. return
  533. }
  534. defer lock.Unlock()
  535. err := portal.CreateMatrixRoom([]string{source.MXID})
  536. if err != nil {
  537. portal.log.Errorln("Failed to create portal room:", err)
  538. return
  539. }
  540. intent := portal.GetMessageIntent(source, info)
  541. if intent == nil {
  542. return
  543. }
  544. data, err := download()
  545. if err != nil {
  546. portal.log.Errorln("Failed to download media:", err)
  547. return
  548. }
  549. uploaded, err := intent.UploadBytes(data, mimeType)
  550. if err != nil {
  551. portal.log.Errorln("Failed to upload media:", err)
  552. return
  553. }
  554. fileName := info.Id
  555. exts, _ := mime.ExtensionsByType(mimeType)
  556. if exts != nil && len(exts) > 0 {
  557. fileName += exts[0]
  558. }
  559. content := &mautrix.Content{
  560. Body: fileName,
  561. URL: uploaded.ContentURI,
  562. Info: &mautrix.FileInfo{
  563. Size: len(data),
  564. MimeType: mimeType,
  565. },
  566. }
  567. portal.SetReply(content, info)
  568. if thumbnail != nil {
  569. thumbnailMime := http.DetectContentType(thumbnail)
  570. uploadedThumbnail, _ := intent.UploadBytes(thumbnail, thumbnailMime)
  571. if uploadedThumbnail != nil {
  572. content.Info.ThumbnailURL = uploadedThumbnail.ContentURI
  573. cfg, _, _ := image.DecodeConfig(bytes.NewReader(data))
  574. content.Info.ThumbnailInfo = &mautrix.FileInfo{
  575. Size: len(thumbnail),
  576. Width: cfg.Width,
  577. Height: cfg.Height,
  578. MimeType: thumbnailMime,
  579. }
  580. }
  581. }
  582. switch strings.ToLower(strings.Split(mimeType, "/")[0]) {
  583. case "image":
  584. content.MsgType = mautrix.MsgImage
  585. cfg, _, _ := image.DecodeConfig(bytes.NewReader(data))
  586. content.Info.Width = cfg.Width
  587. content.Info.Height = cfg.Height
  588. case "video":
  589. content.MsgType = mautrix.MsgVideo
  590. case "audio":
  591. content.MsgType = mautrix.MsgAudio
  592. default:
  593. content.MsgType = mautrix.MsgFile
  594. }
  595. _, _ = intent.UserTyping(portal.MXID, false, 0)
  596. ts := int64(info.Timestamp * 1000)
  597. resp, err := intent.SendMassagedMessageEvent(portal.MXID, mautrix.EventMessage, content, ts)
  598. if err != nil {
  599. portal.log.Errorfln("Failed to handle message %s: %v", info.Id, err)
  600. return
  601. }
  602. if len(caption) > 0 {
  603. captionContent := &mautrix.Content{
  604. Body: caption,
  605. MsgType: mautrix.MsgNotice,
  606. }
  607. portal.bridge.Formatter.ParseWhatsApp(captionContent)
  608. _, err := intent.SendMassagedMessageEvent(portal.MXID, mautrix.EventMessage, captionContent, ts)
  609. if err != nil {
  610. portal.log.Warnfln("Failed to handle caption of message %s: %v", info.Id, err)
  611. }
  612. // TODO store caption mxid?
  613. }
  614. portal.finishHandling(source, info.Source, resp.EventID)
  615. }
  616. func makeMessageID() *string {
  617. b := make([]byte, 10)
  618. rand.Read(b)
  619. str := strings.ToUpper(hex.EncodeToString(b))
  620. return &str
  621. }
  622. func (portal *Portal) downloadThumbnail(evt *mautrix.Event) []byte {
  623. if evt.Content.Info == nil || len(evt.Content.Info.ThumbnailURL) == 0 {
  624. return nil
  625. }
  626. thumbnail, err := portal.MainIntent().DownloadBytes(evt.Content.Info.ThumbnailURL)
  627. if err != nil {
  628. portal.log.Errorln("Failed to download thumbnail in %s: %v", evt.ID, err)
  629. return nil
  630. }
  631. thumbnailType := http.DetectContentType(thumbnail)
  632. var img image.Image
  633. switch thumbnailType {
  634. case "image/png":
  635. img, err = png.Decode(bytes.NewReader(thumbnail))
  636. case "image/gif":
  637. img, err = gif.Decode(bytes.NewReader(thumbnail))
  638. case "image/jpeg":
  639. return thumbnail
  640. default:
  641. return nil
  642. }
  643. var buf bytes.Buffer
  644. err = jpeg.Encode(&buf, img, &jpeg.Options{
  645. Quality: jpeg.DefaultQuality,
  646. })
  647. if err != nil {
  648. portal.log.Errorln("Failed to re-encode thumbnail in %s: %v", evt.ID, err)
  649. return nil
  650. }
  651. return buf.Bytes()
  652. }
  653. func (portal *Portal) preprocessMatrixMedia(sender *User, evt *mautrix.Event, mediaType whatsapp.MediaType) *MediaUpload {
  654. if evt.Content.Info == nil {
  655. evt.Content.Info = &mautrix.FileInfo{}
  656. }
  657. caption := evt.Content.Body
  658. exts, err := mime.ExtensionsByType(evt.Content.Info.MimeType)
  659. for _, ext := range exts {
  660. if strings.HasSuffix(caption, ext) {
  661. caption = ""
  662. break
  663. }
  664. }
  665. content, err := portal.MainIntent().DownloadBytes(evt.Content.URL)
  666. if err != nil {
  667. portal.log.Errorfln("Failed to download media in %s: %v", evt.ID, err)
  668. return nil
  669. }
  670. url, mediaKey, fileEncSHA256, fileSHA256, fileLength, err := sender.Conn.Upload(bytes.NewReader(content), mediaType)
  671. if err != nil {
  672. portal.log.Errorfln("Failed to upload media in %s: %v", evt.ID, err)
  673. return nil
  674. }
  675. return &MediaUpload{
  676. Caption: caption,
  677. URL: url,
  678. MediaKey: mediaKey,
  679. FileEncSHA256: fileEncSHA256,
  680. FileSHA256: fileSHA256,
  681. FileLength: fileLength,
  682. Thumbnail: portal.downloadThumbnail(evt),
  683. }
  684. }
  685. type MediaUpload struct {
  686. Caption string
  687. URL string
  688. MediaKey []byte
  689. FileEncSHA256 []byte
  690. FileSHA256 []byte
  691. FileLength uint64
  692. Thumbnail []byte
  693. }
  694. func (portal *Portal) HandleMatrixMessage(sender *User, evt *mautrix.Event) {
  695. if portal.IsPrivateChat() && sender.JID != portal.Key.Receiver {
  696. return
  697. }
  698. portal.log.Debugfln("Received event %s", evt.ID)
  699. ts := uint64(evt.Timestamp / 1000)
  700. status := waProto.WebMessageInfo_ERROR
  701. fromMe := true
  702. info := &waProto.WebMessageInfo{
  703. Key: &waProto.MessageKey{
  704. FromMe: &fromMe,
  705. Id: makeMessageID(),
  706. RemoteJid: &portal.Key.JID,
  707. },
  708. MessageTimestamp: &ts,
  709. Message: &waProto.Message{},
  710. Status: &status,
  711. }
  712. ctxInfo := &waProto.ContextInfo{}
  713. replyToID := evt.Content.GetReplyTo()
  714. if len(replyToID) > 0 {
  715. evt.Content.RemoveReplyFallback()
  716. msg := portal.bridge.DB.Message.GetByMXID(replyToID)
  717. if msg != nil && msg.Content != nil {
  718. ctxInfo.StanzaId = &msg.JID
  719. ctxInfo.Participant = &msg.Sender
  720. ctxInfo.QuotedMessage = []*waProto.Message{msg.Content}
  721. }
  722. }
  723. var err error
  724. switch evt.Content.MsgType {
  725. case mautrix.MsgText, mautrix.MsgEmote:
  726. text := evt.Content.Body
  727. if evt.Content.Format == mautrix.FormatHTML {
  728. text = portal.bridge.Formatter.ParseMatrix(evt.Content.FormattedBody)
  729. }
  730. if evt.Content.MsgType == mautrix.MsgEmote {
  731. text = "/me " + text
  732. }
  733. ctxInfo.MentionedJid = mentionRegex.FindAllString(text, -1)
  734. for index, mention := range ctxInfo.MentionedJid {
  735. ctxInfo.MentionedJid[index] = mention[1:] + whatsappExt.NewUserSuffix
  736. }
  737. if ctxInfo.StanzaId != nil || ctxInfo.MentionedJid != nil {
  738. info.Message.ExtendedTextMessage = &waProto.ExtendedTextMessage{
  739. Text: &text,
  740. ContextInfo: ctxInfo,
  741. }
  742. } else {
  743. info.Message.Conversation = &text
  744. }
  745. case mautrix.MsgImage:
  746. media := portal.preprocessMatrixMedia(sender, evt, whatsapp.MediaImage)
  747. if media == nil {
  748. return
  749. }
  750. info.Message.ImageMessage = &waProto.ImageMessage{
  751. Caption: &media.Caption,
  752. JpegThumbnail: media.Thumbnail,
  753. Url: &media.URL,
  754. MediaKey: media.MediaKey,
  755. Mimetype: &evt.Content.GetInfo().MimeType,
  756. FileEncSha256: media.FileEncSHA256,
  757. FileSha256: media.FileSHA256,
  758. FileLength: &media.FileLength,
  759. }
  760. case mautrix.MsgVideo:
  761. media := portal.preprocessMatrixMedia(sender, evt, whatsapp.MediaVideo)
  762. if media == nil {
  763. return
  764. }
  765. duration := uint32(evt.Content.GetInfo().Duration)
  766. info.Message.VideoMessage = &waProto.VideoMessage{
  767. Caption: &media.Caption,
  768. JpegThumbnail: media.Thumbnail,
  769. Url: &media.URL,
  770. MediaKey: media.MediaKey,
  771. Mimetype: &evt.Content.GetInfo().MimeType,
  772. Seconds: &duration,
  773. FileEncSha256: media.FileEncSHA256,
  774. FileSha256: media.FileSHA256,
  775. FileLength: &media.FileLength,
  776. }
  777. case mautrix.MsgAudio:
  778. media := portal.preprocessMatrixMedia(sender, evt, whatsapp.MediaAudio)
  779. if media == nil {
  780. return
  781. }
  782. duration := uint32(evt.Content.GetInfo().Duration)
  783. info.Message.AudioMessage = &waProto.AudioMessage{
  784. Url: &media.URL,
  785. MediaKey: media.MediaKey,
  786. Mimetype: &evt.Content.GetInfo().MimeType,
  787. Seconds: &duration,
  788. FileEncSha256: media.FileEncSHA256,
  789. FileSha256: media.FileSHA256,
  790. FileLength: &media.FileLength,
  791. }
  792. case mautrix.MsgFile:
  793. media := portal.preprocessMatrixMedia(sender, evt, whatsapp.MediaDocument)
  794. if media == nil {
  795. return
  796. }
  797. info.Message.DocumentMessage = &waProto.DocumentMessage{
  798. Url: &media.URL,
  799. MediaKey: media.MediaKey,
  800. Mimetype: &evt.Content.GetInfo().MimeType,
  801. FileEncSha256: media.FileEncSHA256,
  802. FileSha256: media.FileSHA256,
  803. FileLength: &media.FileLength,
  804. }
  805. default:
  806. portal.log.Debugln("Unhandled Matrix event:", evt)
  807. return
  808. }
  809. portal.markHandled(sender, info, evt.ID)
  810. portal.log.Debugln("Sending event", evt.ID, "to WhatsApp")
  811. err = sender.Conn.Send(info)
  812. if err != nil {
  813. portal.log.Errorfln("Error handling Matrix event %s: %v", evt.ID, err)
  814. } else {
  815. portal.log.Debugln("Handled Matrix event:", evt)
  816. }
  817. }