matrix.go 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117
  1. // mautrix-whatsapp - A Matrix-WhatsApp puppeting bridge.
  2. // Copyright (C) 2018 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. "maunium.net/go/gomatrix"
  19. "maunium.net/go/mautrix-whatsapp/types"
  20. "maunium.net/go/mautrix-appservice"
  21. "maunium.net/go/maulogger"
  22. "strings"
  23. )
  24. type MatrixHandler struct {
  25. bridge *Bridge
  26. as *appservice.AppService
  27. log maulogger.Logger
  28. cmd *CommandHandler
  29. }
  30. func NewMatrixHandler(bridge *Bridge) *MatrixHandler {
  31. handler := &MatrixHandler{
  32. bridge: bridge,
  33. as: bridge.AppService,
  34. log: bridge.Log.Sub("Matrix"),
  35. cmd: NewCommandHandler(bridge),
  36. }
  37. bridge.EventProcessor.On(gomatrix.EventMessage, handler.HandleMessage)
  38. bridge.EventProcessor.On(gomatrix.StateMember, handler.HandleMembership)
  39. return handler
  40. }
  41. func (mx *MatrixHandler) HandleBotInvite(evt *gomatrix.Event) {
  42. intent := mx.as.BotIntent()
  43. resp, err := intent.JoinRoom(evt.RoomID, "", nil)
  44. if err != nil {
  45. mx.log.Debugln("Failed to join room", evt.RoomID, "with invite from", evt.Sender)
  46. return
  47. }
  48. members, err := intent.JoinedMembers(resp.RoomID)
  49. if err != nil {
  50. mx.log.Debugln("Failed to get members in room", resp.RoomID, "after accepting invite from", evt.Sender)
  51. intent.LeaveRoom(resp.RoomID)
  52. return
  53. }
  54. if len(members.Joined) < 2 {
  55. mx.log.Debugln("Leaving empty room", resp.RoomID, "after accepting invite from", evt.Sender)
  56. intent.LeaveRoom(resp.RoomID)
  57. return
  58. }
  59. hasPuppets := false
  60. for mxid, _ := range members.Joined {
  61. if mxid == intent.UserID || mxid == evt.Sender {
  62. continue
  63. } else if _, _, ok := mx.bridge.ParsePuppetMXID(types.MatrixUserID(mxid)); ok {
  64. hasPuppets = true
  65. continue
  66. }
  67. mx.log.Debugln("Leaving multi-user room", resp.RoomID, "after accepting invite from", evt.Sender)
  68. intent.SendNotice(resp.RoomID, "This bridge is user-specific, please don't invite me into rooms with other users.")
  69. intent.LeaveRoom(resp.RoomID)
  70. return
  71. }
  72. if !hasPuppets {
  73. user := mx.bridge.GetUser(types.MatrixUserID(evt.Sender))
  74. user.SetManagementRoom(types.MatrixRoomID(resp.RoomID))
  75. intent.SendNotice(string(user.ManagementRoom), "This room has been registered as your bridge management/status room.")
  76. mx.log.Debugln(resp.RoomID, "registered as a management room with", evt.Sender)
  77. }
  78. }
  79. func (mx *MatrixHandler) HandleMembership(evt *gomatrix.Event) {
  80. mx.log.Debugln(evt.Content, evt.Content.Membership, evt.GetStateKey())
  81. if evt.Content.Membership == "invite" && evt.GetStateKey() == mx.as.BotMXID() {
  82. mx.HandleBotInvite(evt)
  83. }
  84. }
  85. func (mx *MatrixHandler) HandleMessage(evt *gomatrix.Event) {
  86. roomID := types.MatrixRoomID(evt.RoomID)
  87. user := mx.bridge.GetUser(types.MatrixUserID(evt.Sender))
  88. if evt.Content.MsgType == gomatrix.MsgText {
  89. commandPrefix := mx.bridge.Config.Bridge.CommandPrefix
  90. hasCommandPrefix := strings.HasPrefix(evt.Content.Body, commandPrefix)
  91. if hasCommandPrefix {
  92. evt.Content.Body = strings.TrimLeft(evt.Content.Body[len(commandPrefix):], " ")
  93. }
  94. if hasCommandPrefix || roomID == user.ManagementRoom {
  95. mx.cmd.Handle(roomID, user, evt.Content.Body)
  96. return
  97. }
  98. }
  99. portal := user.GetPortalByMXID(roomID)
  100. if portal != nil {
  101. portal.HandleMatrixMessage(evt)
  102. }
  103. }