user.go 42 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436
  1. package main
  2. import (
  3. "errors"
  4. "fmt"
  5. "math/rand"
  6. "net/http"
  7. "os"
  8. "sort"
  9. "strconv"
  10. "strings"
  11. "sync"
  12. "sync/atomic"
  13. "time"
  14. "github.com/bwmarrin/discordgo"
  15. "github.com/gorilla/websocket"
  16. "github.com/rs/zerolog"
  17. "maunium.net/go/mautrix"
  18. "maunium.net/go/mautrix/appservice"
  19. "maunium.net/go/mautrix/bridge"
  20. "maunium.net/go/mautrix/bridge/bridgeconfig"
  21. "maunium.net/go/mautrix/bridge/status"
  22. "maunium.net/go/mautrix/event"
  23. "maunium.net/go/mautrix/id"
  24. "maunium.net/go/mautrix/pushrules"
  25. "go.mau.fi/mautrix-discord/database"
  26. )
  27. var (
  28. ErrNotConnected = errors.New("not connected")
  29. ErrNotLoggedIn = errors.New("not logged in")
  30. )
  31. type User struct {
  32. *database.User
  33. sync.Mutex
  34. bridge *DiscordBridge
  35. log zerolog.Logger
  36. PermissionLevel bridgeconfig.PermissionLevel
  37. spaceCreateLock sync.Mutex
  38. spaceMembershipChecked bool
  39. dmSpaceMembershipChecked bool
  40. Session *discordgo.Session
  41. BridgeState *bridge.BridgeStateQueue
  42. bridgeStateLock sync.Mutex
  43. wasDisconnected bool
  44. wasLoggedOut bool
  45. markedOpened map[string]time.Time
  46. markedOpenedLock sync.Mutex
  47. pendingInteractions map[string]*WrappedCommandEvent
  48. pendingInteractionsLock sync.Mutex
  49. nextDiscordUploadID atomic.Int32
  50. relationships map[string]*discordgo.Relationship
  51. }
  52. func (user *User) GetRemoteID() string {
  53. return user.DiscordID
  54. }
  55. func (user *User) GetRemoteName() string {
  56. if user.Session != nil && user.Session.State != nil && user.Session.State.User != nil {
  57. return fmt.Sprintf("%s#%s", user.Session.State.User.Username, user.Session.State.User.Discriminator)
  58. }
  59. return user.DiscordID
  60. }
  61. var discordLog zerolog.Logger
  62. func discordToZeroLevel(level int) zerolog.Level {
  63. switch level {
  64. case discordgo.LogError:
  65. return zerolog.ErrorLevel
  66. case discordgo.LogWarning:
  67. return zerolog.WarnLevel
  68. case discordgo.LogInformational:
  69. return zerolog.InfoLevel
  70. case discordgo.LogDebug:
  71. fallthrough
  72. default:
  73. return zerolog.DebugLevel
  74. }
  75. }
  76. func init() {
  77. discordgo.Logger = func(msgL, caller int, format string, a ...interface{}) {
  78. discordLog.WithLevel(discordToZeroLevel(msgL)).Caller(caller+1).Msgf(strings.TrimSpace(format), a...)
  79. }
  80. }
  81. func (user *User) GetPermissionLevel() bridgeconfig.PermissionLevel {
  82. return user.PermissionLevel
  83. }
  84. func (user *User) GetManagementRoomID() id.RoomID {
  85. return user.ManagementRoom
  86. }
  87. func (user *User) GetMXID() id.UserID {
  88. return user.MXID
  89. }
  90. func (user *User) GetCommandState() map[string]interface{} {
  91. return nil
  92. }
  93. func (user *User) GetIDoublePuppet() bridge.DoublePuppet {
  94. p := user.bridge.GetPuppetByCustomMXID(user.MXID)
  95. if p == nil || p.CustomIntent() == nil {
  96. return nil
  97. }
  98. return p
  99. }
  100. func (user *User) GetIGhost() bridge.Ghost {
  101. if user.DiscordID == "" {
  102. return nil
  103. }
  104. p := user.bridge.GetPuppetByID(user.DiscordID)
  105. if p == nil {
  106. return nil
  107. }
  108. return p
  109. }
  110. var _ bridge.User = (*User)(nil)
  111. func (br *DiscordBridge) loadUser(dbUser *database.User, mxid *id.UserID) *User {
  112. if dbUser == nil {
  113. if mxid == nil {
  114. return nil
  115. }
  116. dbUser = br.DB.User.New()
  117. dbUser.MXID = *mxid
  118. dbUser.Insert()
  119. }
  120. user := br.NewUser(dbUser)
  121. br.usersByMXID[user.MXID] = user
  122. if user.DiscordID != "" {
  123. br.usersByID[user.DiscordID] = user
  124. }
  125. if user.ManagementRoom != "" {
  126. br.managementRoomsLock.Lock()
  127. br.managementRooms[user.ManagementRoom] = user
  128. br.managementRoomsLock.Unlock()
  129. }
  130. return user
  131. }
  132. func (br *DiscordBridge) GetUserByMXID(userID id.UserID) *User {
  133. if userID == br.Bot.UserID || br.IsGhost(userID) {
  134. return nil
  135. }
  136. br.usersLock.Lock()
  137. defer br.usersLock.Unlock()
  138. user, ok := br.usersByMXID[userID]
  139. if !ok {
  140. return br.loadUser(br.DB.User.GetByMXID(userID), &userID)
  141. }
  142. return user
  143. }
  144. func (br *DiscordBridge) GetUserByID(id string) *User {
  145. br.usersLock.Lock()
  146. defer br.usersLock.Unlock()
  147. user, ok := br.usersByID[id]
  148. if !ok {
  149. return br.loadUser(br.DB.User.GetByID(id), nil)
  150. }
  151. return user
  152. }
  153. func (br *DiscordBridge) NewUser(dbUser *database.User) *User {
  154. user := &User{
  155. User: dbUser,
  156. bridge: br,
  157. log: br.ZLog.With().Str("user_id", string(dbUser.MXID)).Logger(),
  158. markedOpened: make(map[string]time.Time),
  159. PermissionLevel: br.Config.Bridge.Permissions.Get(dbUser.MXID),
  160. pendingInteractions: make(map[string]*WrappedCommandEvent),
  161. relationships: make(map[string]*discordgo.Relationship),
  162. }
  163. user.nextDiscordUploadID.Store(rand.Int31n(100))
  164. user.BridgeState = br.NewBridgeStateQueue(user)
  165. return user
  166. }
  167. func (br *DiscordBridge) getAllUsersWithToken() []*User {
  168. br.usersLock.Lock()
  169. defer br.usersLock.Unlock()
  170. dbUsers := br.DB.User.GetAllWithToken()
  171. users := make([]*User, len(dbUsers))
  172. for idx, dbUser := range dbUsers {
  173. user, ok := br.usersByMXID[dbUser.MXID]
  174. if !ok {
  175. user = br.loadUser(dbUser, nil)
  176. }
  177. users[idx] = user
  178. }
  179. return users
  180. }
  181. func (br *DiscordBridge) startUsers() {
  182. br.ZLog.Debug().Msg("Starting users")
  183. usersWithToken := br.getAllUsersWithToken()
  184. for _, u := range usersWithToken {
  185. go u.startupTryConnect(0)
  186. }
  187. if len(usersWithToken) == 0 {
  188. br.SendGlobalBridgeState(status.BridgeState{StateEvent: status.StateUnconfigured}.Fill(nil))
  189. }
  190. br.ZLog.Debug().Msg("Starting custom puppets")
  191. for _, customPuppet := range br.GetAllPuppetsWithCustomMXID() {
  192. go func(puppet *Puppet) {
  193. br.ZLog.Debug().Str("user_id", puppet.CustomMXID.String()).Msg("Starting custom puppet")
  194. if err := puppet.StartCustomMXID(true); err != nil {
  195. puppet.log.Error().Err(err).Msg("Failed to start custom puppet")
  196. }
  197. }(customPuppet)
  198. }
  199. }
  200. func (user *User) startupTryConnect(retryCount int) {
  201. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnecting})
  202. err := user.Connect()
  203. if err != nil {
  204. user.log.Error().Err(err).Msg("Error connecting on startup")
  205. closeErr := &websocket.CloseError{}
  206. if errors.As(err, &closeErr) && closeErr.Code == 4004 {
  207. user.invalidAuthHandler(nil, nil)
  208. } else if retryCount < 6 {
  209. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateTransientDisconnect, Error: "dc-unknown-websocket-error", Message: err.Error()})
  210. retryInSeconds := 2 << retryCount
  211. user.log.Debug().Int("retry_in_seconds", retryInSeconds).Msg("Sleeping and retrying connection")
  212. time.Sleep(time.Duration(retryInSeconds) * time.Second)
  213. user.startupTryConnect(retryCount + 1)
  214. } else {
  215. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateUnknownError, Error: "dc-unknown-websocket-error", Message: err.Error()})
  216. }
  217. }
  218. }
  219. func (user *User) SetManagementRoom(roomID id.RoomID) {
  220. user.bridge.managementRoomsLock.Lock()
  221. defer user.bridge.managementRoomsLock.Unlock()
  222. existing, ok := user.bridge.managementRooms[roomID]
  223. if ok {
  224. existing.ManagementRoom = ""
  225. existing.Update()
  226. }
  227. user.ManagementRoom = roomID
  228. user.bridge.managementRooms[user.ManagementRoom] = user
  229. user.Update()
  230. }
  231. func (user *User) getSpaceRoom(ptr *id.RoomID, name, topic string, parent id.RoomID) id.RoomID {
  232. if len(*ptr) > 0 {
  233. return *ptr
  234. }
  235. user.spaceCreateLock.Lock()
  236. defer user.spaceCreateLock.Unlock()
  237. if len(*ptr) > 0 {
  238. return *ptr
  239. }
  240. initialState := []*event.Event{{
  241. Type: event.StateRoomAvatar,
  242. Content: event.Content{
  243. Parsed: &event.RoomAvatarEventContent{
  244. URL: user.bridge.Config.AppService.Bot.ParsedAvatar,
  245. },
  246. },
  247. }}
  248. if parent != "" {
  249. parentIDStr := parent.String()
  250. initialState = append(initialState, &event.Event{
  251. Type: event.StateSpaceParent,
  252. StateKey: &parentIDStr,
  253. Content: event.Content{
  254. Parsed: &event.SpaceParentEventContent{
  255. Canonical: true,
  256. Via: []string{user.bridge.AS.HomeserverDomain},
  257. },
  258. },
  259. })
  260. }
  261. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  262. Visibility: "private",
  263. Name: name,
  264. Topic: topic,
  265. InitialState: initialState,
  266. CreationContent: map[string]interface{}{
  267. "type": event.RoomTypeSpace,
  268. },
  269. PowerLevelOverride: &event.PowerLevelsEventContent{
  270. Users: map[id.UserID]int{
  271. user.bridge.Bot.UserID: 9001,
  272. user.MXID: 50,
  273. },
  274. },
  275. })
  276. if err != nil {
  277. user.log.Error().Err(err).Msg("Failed to auto-create space room")
  278. } else {
  279. *ptr = resp.RoomID
  280. user.Update()
  281. user.ensureInvited(nil, *ptr, false)
  282. if parent != "" {
  283. _, err = user.bridge.Bot.SendStateEvent(parent, event.StateSpaceChild, resp.RoomID.String(), &event.SpaceChildEventContent{
  284. Via: []string{user.bridge.AS.HomeserverDomain},
  285. Order: " 0000",
  286. })
  287. if err != nil {
  288. user.log.Error().Err(err).
  289. Str("created_space_id", resp.RoomID.String()).
  290. Str("parent_space_id", parent.String()).
  291. Msg("Failed to add created space room to parent space")
  292. }
  293. }
  294. }
  295. return *ptr
  296. }
  297. func (user *User) GetSpaceRoom() id.RoomID {
  298. return user.getSpaceRoom(&user.SpaceRoom, "Discord", "Your Discord bridged chats", "")
  299. }
  300. func (user *User) GetDMSpaceRoom() id.RoomID {
  301. return user.getSpaceRoom(&user.DMSpaceRoom, "Direct Messages", "Your Discord direct messages", user.GetSpaceRoom())
  302. }
  303. func (user *User) tryAutomaticDoublePuppeting() {
  304. user.Lock()
  305. defer user.Unlock()
  306. if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) {
  307. return
  308. }
  309. user.log.Debug().Msg("Checking if double puppeting needs to be enabled")
  310. puppet := user.bridge.GetPuppetByID(user.DiscordID)
  311. if puppet.CustomMXID != "" {
  312. user.log.Debug().Msg("User already has double-puppeting enabled")
  313. return
  314. }
  315. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  316. if err != nil {
  317. user.log.Warn().Err(err).Msg("Failed to login with shared secret")
  318. return
  319. }
  320. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  321. if err != nil {
  322. puppet.log.Warn().Err(err).Msg("Failed to switch to auto-logined custom puppet")
  323. return
  324. }
  325. user.log.Info().Msg("Successfully automatically enabled custom puppet")
  326. }
  327. func (user *User) ViewingChannel(portal *Portal) bool {
  328. if portal.GuildID != "" || !user.Session.IsUser {
  329. return false
  330. }
  331. user.markedOpenedLock.Lock()
  332. defer user.markedOpenedLock.Unlock()
  333. ts := user.markedOpened[portal.Key.ChannelID]
  334. // TODO is there an expiry time?
  335. if ts.IsZero() {
  336. user.markedOpened[portal.Key.ChannelID] = time.Now()
  337. err := user.Session.MarkViewing(portal.Key.ChannelID)
  338. if err != nil {
  339. user.log.Error().Err(err).
  340. Str("channel_id", portal.Key.ChannelID).
  341. Msg("Failed to mark user as viewing channel")
  342. }
  343. return true
  344. }
  345. return false
  346. }
  347. func (user *User) mutePortal(intent *appservice.IntentAPI, portal *Portal, unmute bool) {
  348. if len(portal.MXID) == 0 || !user.bridge.Config.Bridge.MuteChannelsOnCreate {
  349. return
  350. }
  351. var err error
  352. if unmute {
  353. user.log.Debug().Str("room_id", portal.MXID.String()).Msg("Unmuting portal")
  354. err = intent.DeletePushRule("global", pushrules.RoomRule, string(portal.MXID))
  355. } else {
  356. user.log.Debug().Str("room_id", portal.MXID.String()).Msg("Muting portal")
  357. err = intent.PutPushRule("global", pushrules.RoomRule, string(portal.MXID), &mautrix.ReqPutPushRule{
  358. Actions: []pushrules.PushActionType{pushrules.ActionDontNotify},
  359. })
  360. }
  361. if err != nil && !errors.Is(err, mautrix.MNotFound) {
  362. user.log.Warn().Err(err).
  363. Str("room_id", portal.MXID.String()).
  364. Msg("Failed to update push rule through double puppet")
  365. }
  366. }
  367. func (user *User) syncChatDoublePuppetDetails(portal *Portal, justCreated bool) {
  368. doublePuppetIntent := portal.bridge.GetPuppetByCustomMXID(user.MXID).CustomIntent()
  369. if doublePuppetIntent == nil || portal.MXID == "" {
  370. return
  371. }
  372. // TODO sync mute status properly
  373. if portal.GuildID != "" && user.bridge.Config.Bridge.MuteChannelsOnCreate {
  374. user.mutePortal(doublePuppetIntent, portal, false)
  375. }
  376. }
  377. func (user *User) NextDiscordUploadID() string {
  378. val := user.nextDiscordUploadID.Add(2)
  379. return strconv.Itoa(int(val))
  380. }
  381. func (user *User) Login(token string) error {
  382. user.bridgeStateLock.Lock()
  383. user.wasLoggedOut = false
  384. user.bridgeStateLock.Unlock()
  385. user.DiscordToken = token
  386. var err error
  387. const maxRetries = 3
  388. Loop:
  389. for i := 0; i < maxRetries; i++ {
  390. err = user.Connect()
  391. if err == nil {
  392. user.Update()
  393. return nil
  394. }
  395. user.log.Error().Err(err).Msg("Error connecting for login")
  396. closeErr := &websocket.CloseError{}
  397. errors.As(err, &closeErr)
  398. switch closeErr.Code {
  399. case 4004, 4010, 4011, 4012, 4013, 4014:
  400. break Loop
  401. case 4000:
  402. fallthrough
  403. default:
  404. if i < maxRetries-1 {
  405. time.Sleep(time.Duration(i+1) * 2 * time.Second)
  406. }
  407. }
  408. }
  409. user.DiscordToken = ""
  410. return err
  411. }
  412. func (user *User) IsLoggedIn() bool {
  413. user.Lock()
  414. defer user.Unlock()
  415. return user.DiscordToken != ""
  416. }
  417. func (user *User) Logout(isOverwriting bool) {
  418. user.Lock()
  419. defer user.Unlock()
  420. if user.DiscordID != "" {
  421. puppet := user.bridge.GetPuppetByID(user.DiscordID)
  422. if puppet.CustomMXID != "" {
  423. err := puppet.SwitchCustomMXID("", "")
  424. if err != nil {
  425. user.log.Warn().Err(err).Msg("Failed to disable custom puppet while logging out of Discord")
  426. }
  427. }
  428. }
  429. if user.Session != nil {
  430. if err := user.Session.Close(); err != nil {
  431. user.log.Warn().Err(err).Msg("Error closing session")
  432. }
  433. }
  434. user.Session = nil
  435. user.DiscordToken = ""
  436. user.ReadStateVersion = 0
  437. if !isOverwriting {
  438. user.bridge.usersLock.Lock()
  439. if user.bridge.usersByID[user.DiscordID] == user {
  440. delete(user.bridge.usersByID, user.DiscordID)
  441. }
  442. user.bridge.usersLock.Unlock()
  443. }
  444. user.DiscordID = ""
  445. user.Update()
  446. user.log.Info().Msg("User logged out")
  447. }
  448. func (user *User) Connected() bool {
  449. user.Lock()
  450. defer user.Unlock()
  451. return user.Session != nil
  452. }
  453. const BotIntents = discordgo.IntentGuilds |
  454. discordgo.IntentGuildMessages |
  455. discordgo.IntentGuildMessageReactions |
  456. discordgo.IntentGuildMessageTyping |
  457. discordgo.IntentGuildBans |
  458. discordgo.IntentGuildEmojis |
  459. discordgo.IntentGuildIntegrations |
  460. discordgo.IntentGuildInvites |
  461. //discordgo.IntentGuildVoiceStates |
  462. //discordgo.IntentGuildScheduledEvents |
  463. discordgo.IntentDirectMessages |
  464. discordgo.IntentDirectMessageTyping |
  465. discordgo.IntentDirectMessageTyping |
  466. // Privileged intents
  467. discordgo.IntentMessageContent |
  468. //discordgo.IntentGuildPresences |
  469. discordgo.IntentGuildMembers
  470. func (user *User) Connect() error {
  471. user.Lock()
  472. defer user.Unlock()
  473. if user.DiscordToken == "" {
  474. return ErrNotLoggedIn
  475. }
  476. user.log.Debug().Msg("Connecting to discord")
  477. session, err := discordgo.New(user.DiscordToken)
  478. if err != nil {
  479. return err
  480. }
  481. // TODO move to config
  482. if os.Getenv("DISCORD_DEBUG") == "1" {
  483. session.LogLevel = discordgo.LogDebug
  484. } else {
  485. session.LogLevel = discordgo.LogInformational
  486. }
  487. userDiscordLog := user.log.With().Str("component", "discordgo").Logger()
  488. session.Logger = func(msgL, caller int, format string, a ...interface{}) {
  489. userDiscordLog.WithLevel(discordToZeroLevel(msgL)).Caller(caller+1).Msgf(strings.TrimSpace(format), a...)
  490. }
  491. if !session.IsUser {
  492. session.Identify.Intents = BotIntents
  493. }
  494. user.Session = session
  495. user.Session.AddHandler(user.readyHandler)
  496. user.Session.AddHandler(user.resumeHandler)
  497. user.Session.AddHandler(user.connectedHandler)
  498. user.Session.AddHandler(user.disconnectedHandler)
  499. user.Session.AddHandler(user.invalidAuthHandler)
  500. user.Session.AddHandler(user.guildCreateHandler)
  501. user.Session.AddHandler(user.guildDeleteHandler)
  502. user.Session.AddHandler(user.guildUpdateHandler)
  503. user.Session.AddHandler(user.guildRoleCreateHandler)
  504. user.Session.AddHandler(user.guildRoleUpdateHandler)
  505. user.Session.AddHandler(user.guildRoleDeleteHandler)
  506. user.Session.AddHandler(user.channelCreateHandler)
  507. user.Session.AddHandler(user.channelDeleteHandler)
  508. user.Session.AddHandler(user.channelPinsUpdateHandler)
  509. user.Session.AddHandler(user.channelUpdateHandler)
  510. user.Session.AddHandler(user.channelRecipientAdd)
  511. user.Session.AddHandler(user.channelRecipientRemove)
  512. user.Session.AddHandler(user.relationshipAddHandler)
  513. user.Session.AddHandler(user.relationshipRemoveHandler)
  514. user.Session.AddHandler(user.relationshipUpdateHandler)
  515. user.Session.AddHandler(user.messageCreateHandler)
  516. user.Session.AddHandler(user.messageDeleteHandler)
  517. user.Session.AddHandler(user.messageUpdateHandler)
  518. user.Session.AddHandler(user.reactionAddHandler)
  519. user.Session.AddHandler(user.reactionRemoveHandler)
  520. user.Session.AddHandler(user.messageAckHandler)
  521. user.Session.AddHandler(user.typingStartHandler)
  522. user.Session.AddHandler(user.interactionSuccessHandler)
  523. return user.Session.Open()
  524. }
  525. func (user *User) Disconnect() error {
  526. user.Lock()
  527. defer user.Unlock()
  528. if user.Session == nil {
  529. return ErrNotConnected
  530. }
  531. user.log.Info().Msg("Disconnecting session manually")
  532. if err := user.Session.Close(); err != nil {
  533. return err
  534. }
  535. user.Session = nil
  536. return nil
  537. }
  538. func (user *User) getGuildBridgingMode(guildID string) database.GuildBridgingMode {
  539. if guildID == "" {
  540. return database.GuildBridgeEverything
  541. }
  542. guild := user.bridge.GetGuildByID(guildID, false)
  543. if guild == nil {
  544. return database.GuildBridgeNothing
  545. }
  546. return guild.BridgingMode
  547. }
  548. type ChannelSlice []*discordgo.Channel
  549. func (s ChannelSlice) Len() int {
  550. return len(s)
  551. }
  552. func (s ChannelSlice) Less(i, j int) bool {
  553. if s[i].Position != 0 || s[j].Position != 0 {
  554. return s[i].Position < s[j].Position
  555. }
  556. return compareMessageIDs(s[i].LastMessageID, s[j].LastMessageID) == 1
  557. }
  558. func (s ChannelSlice) Swap(i, j int) {
  559. s[i], s[j] = s[j], s[i]
  560. }
  561. func (user *User) readyHandler(_ *discordgo.Session, r *discordgo.Ready) {
  562. user.log.Debug().Msg("Discord connection ready")
  563. user.bridgeStateLock.Lock()
  564. user.wasLoggedOut = false
  565. user.bridgeStateLock.Unlock()
  566. if user.DiscordID != r.User.ID {
  567. user.bridge.usersLock.Lock()
  568. user.DiscordID = r.User.ID
  569. if previousUser, ok := user.bridge.usersByID[user.DiscordID]; ok && previousUser != user {
  570. user.log.Warn().
  571. Str("previous_user_id", previousUser.MXID.String()).
  572. Msg("Another user is logged in with same Discord ID, logging them out")
  573. // TODO send notice?
  574. previousUser.Logout(true)
  575. }
  576. user.bridge.usersByID[user.DiscordID] = user
  577. user.bridge.usersLock.Unlock()
  578. user.Update()
  579. }
  580. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateBackfilling})
  581. user.tryAutomaticDoublePuppeting()
  582. for _, relationship := range r.Relationships {
  583. user.relationships[relationship.ID] = relationship
  584. }
  585. updateTS := time.Now()
  586. portalsInSpace := make(map[string]bool)
  587. for _, guild := range user.GetPortals() {
  588. portalsInSpace[guild.DiscordID] = guild.InSpace
  589. }
  590. for _, guild := range r.Guilds {
  591. user.handleGuild(guild, updateTS, portalsInSpace[guild.ID])
  592. }
  593. // The private channel list doesn't seem to be sorted by default, so sort it by message IDs (highest=newest first)
  594. sort.Sort(ChannelSlice(r.PrivateChannels))
  595. for i, ch := range r.PrivateChannels {
  596. portal := user.GetPortalByMeta(ch)
  597. user.handlePrivateChannel(portal, ch, updateTS, i < user.bridge.Config.Bridge.PrivateChannelCreateLimit, portalsInSpace[portal.Key.ChannelID])
  598. }
  599. user.PrunePortalList(updateTS)
  600. if r.ReadState != nil && r.ReadState.Version > user.ReadStateVersion {
  601. // TODO can we figure out which read states are actually new?
  602. for _, entry := range r.ReadState.Entries {
  603. user.messageAckHandler(nil, &discordgo.MessageAck{
  604. MessageID: string(entry.LastMessageID),
  605. ChannelID: entry.ID,
  606. })
  607. }
  608. user.ReadStateVersion = r.ReadState.Version
  609. user.Update()
  610. }
  611. go user.subscribeGuilds(2 * time.Second)
  612. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnected})
  613. }
  614. func (user *User) subscribeGuilds(delay time.Duration) {
  615. if !user.Session.IsUser {
  616. return
  617. }
  618. for _, guildMeta := range user.Session.State.Guilds {
  619. guild := user.bridge.GetGuildByID(guildMeta.ID, false)
  620. if guild != nil && guild.MXID != "" {
  621. user.log.Debug().Str("guild_id", guild.ID).Msg("Subscribing to guild")
  622. dat := discordgo.GuildSubscribeData{
  623. GuildID: guild.ID,
  624. Typing: true,
  625. Activities: true,
  626. Threads: true,
  627. }
  628. err := user.Session.SubscribeGuild(dat)
  629. if err != nil {
  630. user.log.Warn().Err(err).Str("guild_id", guild.ID).Msg("Failed to subscribe to guild")
  631. }
  632. time.Sleep(delay)
  633. }
  634. }
  635. }
  636. func (user *User) resumeHandler(_ *discordgo.Session, r *discordgo.Resumed) {
  637. user.log.Debug().Msg("Discord connection resumed")
  638. user.subscribeGuilds(0 * time.Second)
  639. }
  640. func (user *User) addPrivateChannelToSpace(portal *Portal) bool {
  641. if portal.MXID == "" {
  642. return false
  643. }
  644. _, err := user.bridge.Bot.SendStateEvent(user.GetDMSpaceRoom(), event.StateSpaceChild, portal.MXID.String(), &event.SpaceChildEventContent{
  645. Via: []string{user.bridge.AS.HomeserverDomain},
  646. })
  647. if err != nil {
  648. user.log.Error().Err(err).
  649. Str("room_id", portal.MXID.String()).
  650. Msg("Failed to add DMM room to user DM space")
  651. return false
  652. } else {
  653. return true
  654. }
  655. }
  656. func (user *User) relationshipAddHandler(_ *discordgo.Session, r *discordgo.RelationshipAdd) {
  657. user.log.Debug().Interface("relationship", r.Relationship).Msg("Relationship added")
  658. user.relationships[r.ID] = r.Relationship
  659. user.handleRelationshipChange(r.ID, r.Nickname)
  660. }
  661. func (user *User) relationshipUpdateHandler(_ *discordgo.Session, r *discordgo.RelationshipUpdate) {
  662. user.log.Debug().Interface("relationship", r.Relationship).Msg("Relationship update")
  663. user.relationships[r.ID] = r.Relationship
  664. user.handleRelationshipChange(r.ID, r.Nickname)
  665. }
  666. func (user *User) relationshipRemoveHandler(_ *discordgo.Session, r *discordgo.RelationshipRemove) {
  667. user.log.Debug().Str("other_user_id", r.ID).Msg("Relationship removed")
  668. delete(user.relationships, r.ID)
  669. user.handleRelationshipChange(r.ID, "")
  670. }
  671. func (user *User) handleRelationshipChange(userID, nickname string) {
  672. puppet := user.bridge.GetPuppetByID(userID)
  673. portal := user.FindPrivateChatWith(userID)
  674. if portal == nil || puppet == nil {
  675. return
  676. }
  677. updated := portal.FriendNick == (nickname != "")
  678. portal.FriendNick = nickname != ""
  679. if nickname != "" {
  680. updated = portal.UpdateNameDirect(nickname, true)
  681. } else if portal.Name != puppet.Name {
  682. if portal.shouldSetDMRoomMetadata() {
  683. updated = portal.UpdateNameDirect(puppet.Name, false)
  684. } else if portal.NameSet {
  685. _, err := portal.MainIntent().SendStateEvent(portal.MXID, event.StateRoomName, "", map[string]any{})
  686. if err != nil {
  687. portal.zlog.Warn().Err(err).Msg("Failed to clear room name after friend nickname was removed")
  688. } else {
  689. portal.zlog.Debug().Msg("Cleared room name after friend nickname was removed")
  690. portal.NameSet = false
  691. portal.Update()
  692. updated = true
  693. }
  694. }
  695. }
  696. if !updated {
  697. portal.Update()
  698. }
  699. }
  700. func (user *User) handlePrivateChannel(portal *Portal, meta *discordgo.Channel, timestamp time.Time, create, isInSpace bool) {
  701. if create && portal.MXID == "" {
  702. err := portal.CreateMatrixRoom(user, meta)
  703. if err != nil {
  704. user.log.Error().Err(err).
  705. Str("channel_id", portal.Key.ChannelID).
  706. Msg("Failed to create portal for private channel in create handler")
  707. }
  708. } else {
  709. portal.UpdateInfo(user, meta)
  710. portal.ForwardBackfillMissed(user, meta)
  711. }
  712. user.MarkInPortal(database.UserPortal{
  713. DiscordID: portal.Key.ChannelID,
  714. Type: database.UserPortalTypeDM,
  715. Timestamp: timestamp,
  716. InSpace: isInSpace || user.addPrivateChannelToSpace(portal),
  717. })
  718. }
  719. func (user *User) addGuildToSpace(guild *Guild, isInSpace bool, timestamp time.Time) bool {
  720. if len(guild.MXID) > 0 && !isInSpace {
  721. _, err := user.bridge.Bot.SendStateEvent(user.GetSpaceRoom(), event.StateSpaceChild, guild.MXID.String(), &event.SpaceChildEventContent{
  722. Via: []string{user.bridge.AS.HomeserverDomain},
  723. })
  724. if err != nil {
  725. user.log.Error().Err(err).
  726. Str("guild_space_id", guild.MXID.String()).
  727. Msg("Failed to add guild space to user space")
  728. } else {
  729. isInSpace = true
  730. }
  731. }
  732. user.MarkInPortal(database.UserPortal{
  733. DiscordID: guild.ID,
  734. Type: database.UserPortalTypeGuild,
  735. Timestamp: timestamp,
  736. InSpace: isInSpace,
  737. })
  738. return isInSpace
  739. }
  740. func (user *User) discordRoleToDB(guildID string, role *discordgo.Role, dbRole *database.Role) (*database.Role, bool) {
  741. var changed bool
  742. if dbRole == nil {
  743. dbRole = user.bridge.DB.Role.New()
  744. dbRole.ID = role.ID
  745. dbRole.GuildID = guildID
  746. changed = true
  747. } else {
  748. changed = dbRole.Name != role.Name ||
  749. dbRole.Icon != role.Icon ||
  750. dbRole.Mentionable != role.Mentionable ||
  751. dbRole.Managed != role.Managed ||
  752. dbRole.Hoist != role.Hoist ||
  753. dbRole.Color != role.Color ||
  754. dbRole.Position != role.Position ||
  755. dbRole.Permissions != role.Permissions
  756. }
  757. dbRole.Role = *role
  758. return dbRole, changed
  759. }
  760. func (user *User) handleGuildRoles(guildID string, newRoles []*discordgo.Role) {
  761. existingRoles := user.bridge.DB.Role.GetAll(guildID)
  762. existingRoleMap := make(map[string]*database.Role, len(existingRoles))
  763. for _, role := range existingRoles {
  764. existingRoleMap[role.ID] = role
  765. }
  766. txn, err := user.bridge.DB.Begin()
  767. if err != nil {
  768. user.log.Error().Err(err).Msg("Failed to start transaction for guild role sync")
  769. panic(err)
  770. }
  771. for _, role := range newRoles {
  772. dbRole, changed := user.discordRoleToDB(guildID, role, existingRoleMap[role.ID])
  773. delete(existingRoleMap, role.ID)
  774. if changed {
  775. dbRole.Upsert(txn)
  776. }
  777. }
  778. for _, removeRole := range existingRoleMap {
  779. removeRole.Delete(txn)
  780. }
  781. err = txn.Commit()
  782. if err != nil {
  783. user.log.Error().Err(err).Msg("Failed to commit guild role sync transaction")
  784. rollbackErr := txn.Rollback()
  785. if rollbackErr != nil {
  786. user.log.Error().Err(rollbackErr).Msg("Failed to rollback errored guild role sync transaction")
  787. }
  788. panic(err)
  789. }
  790. }
  791. func (user *User) guildRoleCreateHandler(_ *discordgo.Session, r *discordgo.GuildRoleCreate) {
  792. dbRole, _ := user.discordRoleToDB(r.GuildID, r.Role, nil)
  793. dbRole.Upsert(nil)
  794. }
  795. func (user *User) guildRoleUpdateHandler(_ *discordgo.Session, r *discordgo.GuildRoleUpdate) {
  796. dbRole, _ := user.discordRoleToDB(r.GuildID, r.Role, nil)
  797. dbRole.Upsert(nil)
  798. }
  799. func (user *User) guildRoleDeleteHandler(_ *discordgo.Session, r *discordgo.GuildRoleDelete) {
  800. user.bridge.DB.Role.DeleteByID(r.GuildID, r.RoleID)
  801. }
  802. func (user *User) handleGuild(meta *discordgo.Guild, timestamp time.Time, isInSpace bool) {
  803. guild := user.bridge.GetGuildByID(meta.ID, true)
  804. guild.UpdateInfo(user, meta)
  805. if len(meta.Channels) > 0 {
  806. for _, ch := range meta.Channels {
  807. portal := user.GetPortalByMeta(ch)
  808. if guild.BridgingMode >= database.GuildBridgeEverything && portal.MXID == "" && user.channelIsBridgeable(ch) {
  809. err := portal.CreateMatrixRoom(user, ch)
  810. if err != nil {
  811. user.log.Error().Err(err).
  812. Str("guild_id", guild.ID).
  813. Str("channel_id", ch.ID).
  814. Msg("Failed to create portal for guild channel in guild handler")
  815. }
  816. } else {
  817. portal.UpdateInfo(user, ch)
  818. if user.bridge.Config.Bridge.Backfill.MaxGuildMembers < 0 || meta.MemberCount < user.bridge.Config.Bridge.Backfill.MaxGuildMembers {
  819. portal.ForwardBackfillMissed(user, ch)
  820. }
  821. }
  822. }
  823. }
  824. if len(meta.Roles) > 0 {
  825. user.handleGuildRoles(meta.ID, meta.Roles)
  826. }
  827. user.addGuildToSpace(guild, isInSpace, timestamp)
  828. }
  829. func (user *User) connectedHandler(_ *discordgo.Session, _ *discordgo.Connect) {
  830. user.bridgeStateLock.Lock()
  831. defer user.bridgeStateLock.Unlock()
  832. user.log.Debug().Msg("Connected to Discord")
  833. if user.wasDisconnected {
  834. user.wasDisconnected = false
  835. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateConnected})
  836. }
  837. }
  838. func (user *User) disconnectedHandler(_ *discordgo.Session, _ *discordgo.Disconnect) {
  839. user.bridgeStateLock.Lock()
  840. defer user.bridgeStateLock.Unlock()
  841. if user.wasLoggedOut {
  842. user.log.Debug().Msg("Disconnected from Discord (not updating bridge state as user was just logged out)")
  843. return
  844. }
  845. user.log.Debug().Msg("Disconnected from Discord")
  846. user.wasDisconnected = true
  847. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateTransientDisconnect, Error: "dc-transient-disconnect", Message: "Temporarily disconnected from Discord, trying to reconnect"})
  848. }
  849. func (user *User) invalidAuthHandler(_ *discordgo.Session, _ *discordgo.InvalidAuth) {
  850. user.bridgeStateLock.Lock()
  851. defer user.bridgeStateLock.Unlock()
  852. user.log.Info().Msg("Got logged out from Discord due to invalid token")
  853. user.wasLoggedOut = true
  854. user.BridgeState.Send(status.BridgeState{StateEvent: status.StateBadCredentials, Error: "dc-websocket-disconnect-4004", Message: "Discord access token is no longer valid, please log in again"})
  855. go user.Logout(false)
  856. }
  857. func (user *User) guildCreateHandler(_ *discordgo.Session, g *discordgo.GuildCreate) {
  858. user.log.Info().
  859. Str("guild_id", g.ID).
  860. Str("name", g.Name).
  861. Bool("unavailable", g.Unavailable).
  862. Msg("Got guild create event")
  863. user.handleGuild(g.Guild, time.Now(), false)
  864. }
  865. func (user *User) guildDeleteHandler(_ *discordgo.Session, g *discordgo.GuildDelete) {
  866. user.log.Info().Str("guild_id", g.ID).Msg("Got guild delete event")
  867. user.MarkNotInPortal(g.ID)
  868. guild := user.bridge.GetGuildByID(g.ID, false)
  869. if guild == nil || guild.MXID == "" {
  870. return
  871. }
  872. if user.bridge.Config.Bridge.DeleteGuildOnLeave && !user.PortalHasOtherUsers(g.ID) {
  873. user.log.Debug().Str("guild_id", g.ID).Msg("No other users in guild, cleaning up all portals")
  874. err := user.unbridgeGuild(g.ID)
  875. if err != nil {
  876. user.log.Warn().Err(err).Msg("Failed to unbridge guild that was deleted")
  877. }
  878. }
  879. }
  880. func (user *User) guildUpdateHandler(_ *discordgo.Session, g *discordgo.GuildUpdate) {
  881. user.log.Debug().Str("guild_id", g.ID).Msg("Got guild update event")
  882. user.handleGuild(g.Guild, time.Now(), user.IsInSpace(g.ID))
  883. }
  884. func (user *User) channelCreateHandler(_ *discordgo.Session, c *discordgo.ChannelCreate) {
  885. if user.getGuildBridgingMode(c.GuildID) < database.GuildBridgeEverything {
  886. user.log.Debug().
  887. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  888. Msg("Ignoring channel create event in unbridged guild")
  889. return
  890. }
  891. user.log.Info().
  892. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  893. Msg("Got channel create event")
  894. portal := user.GetPortalByMeta(c.Channel)
  895. if portal.MXID != "" {
  896. return
  897. }
  898. if c.GuildID == "" {
  899. user.handlePrivateChannel(portal, c.Channel, time.Now(), true, user.IsInSpace(portal.Key.String()))
  900. } else if user.channelIsBridgeable(c.Channel) {
  901. err := portal.CreateMatrixRoom(user, c.Channel)
  902. if err != nil {
  903. user.log.Error().Err(err).
  904. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  905. Msg("Error creating Matrix room after channel create event")
  906. }
  907. } else {
  908. user.log.Debug().
  909. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  910. Msg("Got channel create event, but it's not bridgeable, ignoring")
  911. }
  912. }
  913. func (user *User) channelDeleteHandler(_ *discordgo.Session, c *discordgo.ChannelDelete) {
  914. portal := user.GetExistingPortalByID(c.ID)
  915. if portal == nil {
  916. user.log.Debug().
  917. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  918. Msg("Ignoring channel delete event of unknown channel")
  919. return
  920. }
  921. user.log.Info().
  922. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  923. Msg("Got channel delete event, cleaning up portal")
  924. portal.Delete()
  925. portal.cleanup(!user.bridge.Config.Bridge.DeletePortalOnChannelDelete)
  926. if c.GuildID == "" {
  927. user.MarkNotInPortal(portal.Key.ChannelID)
  928. }
  929. user.log.Debug().
  930. Str("guild_id", c.GuildID).Str("channel_id", c.ID).
  931. Msg("Completed cleaning up channel")
  932. }
  933. func (user *User) channelPinsUpdateHandler(_ *discordgo.Session, c *discordgo.ChannelPinsUpdate) {
  934. user.log.Debug().Msg("channel pins update")
  935. }
  936. func (user *User) channelUpdateHandler(_ *discordgo.Session, c *discordgo.ChannelUpdate) {
  937. portal := user.GetPortalByMeta(c.Channel)
  938. if c.GuildID == "" {
  939. user.handlePrivateChannel(portal, c.Channel, time.Now(), true, user.IsInSpace(portal.Key.String()))
  940. } else {
  941. portal.UpdateInfo(user, c.Channel)
  942. }
  943. }
  944. func (user *User) channelRecipientAdd(_ *discordgo.Session, c *discordgo.ChannelRecipientAdd) {
  945. portal := user.GetExistingPortalByID(c.ChannelID)
  946. if portal != nil {
  947. portal.syncParticipant(user, c.User, false)
  948. }
  949. }
  950. func (user *User) channelRecipientRemove(_ *discordgo.Session, c *discordgo.ChannelRecipientRemove) {
  951. portal := user.GetExistingPortalByID(c.ChannelID)
  952. if portal != nil {
  953. portal.syncParticipant(user, c.User, true)
  954. }
  955. }
  956. func (user *User) findPortal(channelID string) (*Portal, *Thread) {
  957. portal := user.GetExistingPortalByID(channelID)
  958. if portal != nil {
  959. return portal, nil
  960. }
  961. thread := user.bridge.GetThreadByID(channelID, nil)
  962. if thread != nil && thread.Parent != nil {
  963. return thread.Parent, thread
  964. }
  965. if !user.Session.IsUser {
  966. channel, _ := user.Session.State.Channel(channelID)
  967. if channel == nil {
  968. user.log.Debug().Str("channel_id", channelID).Msg("Fetching info of unknown channel to handle message")
  969. var err error
  970. channel, err = user.Session.Channel(channelID)
  971. if err != nil {
  972. user.log.Warn().Err(err).Str("channel_id", channelID).Msg("Failed to get info of unknown channel")
  973. } else {
  974. user.log.Debug().Str("channel_id", channelID).Msg("Got info for channel to handle message")
  975. _ = user.Session.State.ChannelAdd(channel)
  976. }
  977. }
  978. if channel != nil && user.channelIsBridgeable(channel) {
  979. user.log.Debug().Str("channel_id", channelID).Msg("Creating portal and updating info to handle message")
  980. portal = user.GetPortalByMeta(channel)
  981. if channel.GuildID == "" {
  982. user.handlePrivateChannel(portal, channel, time.Now(), false, false)
  983. } else {
  984. user.log.Warn().
  985. Str("channel_id", channel.ID).Str("guild_id", channel.GuildID).
  986. Msg("Unexpected unknown guild channel")
  987. }
  988. return portal, nil
  989. }
  990. }
  991. return nil, nil
  992. }
  993. func (user *User) pushPortalMessage(msg interface{}, typeName, channelID, guildID string) {
  994. if user.getGuildBridgingMode(guildID) <= database.GuildBridgeNothing {
  995. // If guild bridging mode is nothing, don't even check if the portal exists
  996. return
  997. }
  998. portal, thread := user.findPortal(channelID)
  999. if portal == nil {
  1000. user.log.Debug().
  1001. Str("discord_event", typeName).
  1002. Str("guild_id", guildID).
  1003. Str("channel_id", channelID).
  1004. Msg("Dropping event in unknown channel")
  1005. return
  1006. }
  1007. if mode := user.getGuildBridgingMode(portal.GuildID); mode <= database.GuildBridgeNothing || (portal.MXID == "" && mode <= database.GuildBridgeIfPortalExists) {
  1008. return
  1009. }
  1010. portal.discordMessages <- portalDiscordMessage{
  1011. msg: msg,
  1012. user: user,
  1013. thread: thread,
  1014. }
  1015. }
  1016. func (user *User) messageCreateHandler(_ *discordgo.Session, m *discordgo.MessageCreate) {
  1017. user.pushPortalMessage(m, "message create", m.ChannelID, m.GuildID)
  1018. }
  1019. func (user *User) messageDeleteHandler(_ *discordgo.Session, m *discordgo.MessageDelete) {
  1020. user.pushPortalMessage(m, "message delete", m.ChannelID, m.GuildID)
  1021. }
  1022. func (user *User) messageUpdateHandler(_ *discordgo.Session, m *discordgo.MessageUpdate) {
  1023. user.pushPortalMessage(m, "message update", m.ChannelID, m.GuildID)
  1024. }
  1025. func (user *User) reactionAddHandler(_ *discordgo.Session, m *discordgo.MessageReactionAdd) {
  1026. user.pushPortalMessage(m, "reaction add", m.ChannelID, m.GuildID)
  1027. }
  1028. func (user *User) reactionRemoveHandler(_ *discordgo.Session, m *discordgo.MessageReactionRemove) {
  1029. user.pushPortalMessage(m, "reaction remove", m.ChannelID, m.GuildID)
  1030. }
  1031. type CustomReadReceipt struct {
  1032. Timestamp int64 `json:"ts,omitempty"`
  1033. DoublePuppetSource string `json:"fi.mau.double_puppet_source,omitempty"`
  1034. }
  1035. type CustomReadMarkers struct {
  1036. mautrix.ReqSetReadMarkers
  1037. ReadExtra CustomReadReceipt `json:"com.beeper.read.extra"`
  1038. FullyReadExtra CustomReadReceipt `json:"com.beeper.fully_read.extra"`
  1039. }
  1040. func (user *User) makeReadMarkerContent(eventID id.EventID) *CustomReadMarkers {
  1041. var extra CustomReadReceipt
  1042. extra.DoublePuppetSource = user.bridge.Name
  1043. return &CustomReadMarkers{
  1044. ReqSetReadMarkers: mautrix.ReqSetReadMarkers{
  1045. Read: eventID,
  1046. FullyRead: eventID,
  1047. },
  1048. ReadExtra: extra,
  1049. FullyReadExtra: extra,
  1050. }
  1051. }
  1052. func (user *User) messageAckHandler(_ *discordgo.Session, m *discordgo.MessageAck) {
  1053. portal := user.GetExistingPortalByID(m.ChannelID)
  1054. if portal == nil || portal.MXID == "" {
  1055. return
  1056. }
  1057. dp := user.GetIDoublePuppet()
  1058. if dp == nil {
  1059. return
  1060. }
  1061. msg := user.bridge.DB.Message.GetLastByDiscordID(portal.Key, m.MessageID)
  1062. if msg == nil {
  1063. user.log.Debug().
  1064. Str("channel_id", m.ChannelID).Str("message_id", m.MessageID).
  1065. Msg("Dropping message ack event for unknown message")
  1066. return
  1067. }
  1068. err := dp.CustomIntent().SetReadMarkers(portal.MXID, user.makeReadMarkerContent(msg.MXID))
  1069. if err != nil {
  1070. user.log.Error().Err(err).
  1071. Str("event_id", msg.MXID.String()).Str("message_id", msg.DiscordID).
  1072. Msg("Failed to mark event as read")
  1073. } else {
  1074. user.log.Debug().
  1075. Str("event_id", msg.MXID.String()).Str("message_id", msg.DiscordID).
  1076. Msg("Marked event as read after Discord message ack")
  1077. if user.ReadStateVersion < m.Version {
  1078. user.ReadStateVersion = m.Version
  1079. // TODO maybe don't update every time?
  1080. user.Update()
  1081. }
  1082. }
  1083. }
  1084. func (user *User) typingStartHandler(_ *discordgo.Session, t *discordgo.TypingStart) {
  1085. portal := user.GetExistingPortalByID(t.ChannelID)
  1086. if portal == nil || portal.MXID == "" {
  1087. return
  1088. }
  1089. portal.handleDiscordTyping(t)
  1090. }
  1091. func (user *User) interactionSuccessHandler(_ *discordgo.Session, s *discordgo.InteractionSuccess) {
  1092. user.pendingInteractionsLock.Lock()
  1093. defer user.pendingInteractionsLock.Unlock()
  1094. ce, ok := user.pendingInteractions[s.Nonce]
  1095. if !ok {
  1096. user.log.Debug().Str("nonce", s.Nonce).Str("id", s.ID).Msg("Got interaction success for unknown interaction")
  1097. } else {
  1098. user.log.Debug().Str("nonce", s.Nonce).Str("id", s.ID).Msg("Got interaction success for pending interaction")
  1099. ce.React("✅")
  1100. delete(user.pendingInteractions, s.Nonce)
  1101. }
  1102. }
  1103. func (user *User) ensureInvited(intent *appservice.IntentAPI, roomID id.RoomID, isDirect bool) bool {
  1104. if intent == nil {
  1105. intent = user.bridge.Bot
  1106. }
  1107. ret := false
  1108. inviteContent := event.Content{
  1109. Parsed: &event.MemberEventContent{
  1110. Membership: event.MembershipInvite,
  1111. IsDirect: isDirect,
  1112. },
  1113. Raw: map[string]interface{}{},
  1114. }
  1115. customPuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  1116. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  1117. inviteContent.Raw["fi.mau.will_auto_accept"] = true
  1118. }
  1119. _, err := intent.SendStateEvent(roomID, event.StateMember, user.MXID.String(), &inviteContent)
  1120. var httpErr mautrix.HTTPError
  1121. if err != nil && errors.As(err, &httpErr) && httpErr.RespError != nil && strings.Contains(httpErr.RespError.Err, "is already in the room") {
  1122. user.bridge.StateStore.SetMembership(roomID, user.MXID, event.MembershipJoin)
  1123. ret = true
  1124. } else if err != nil {
  1125. user.log.Error().Err(err).Str("room_id", roomID.String()).Msg("Failed to invite user to room")
  1126. } else {
  1127. ret = true
  1128. }
  1129. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  1130. err = customPuppet.CustomIntent().EnsureJoined(roomID, appservice.EnsureJoinedParams{IgnoreCache: true})
  1131. if err != nil {
  1132. user.log.Warn().Err(err).Str("room_id", roomID.String()).Msg("Failed to auto-join room")
  1133. ret = false
  1134. } else {
  1135. ret = true
  1136. }
  1137. }
  1138. return ret
  1139. }
  1140. func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
  1141. chats := map[id.UserID][]id.RoomID{}
  1142. privateChats := user.bridge.DB.Portal.FindPrivateChatsOf(user.DiscordID)
  1143. for _, portal := range privateChats {
  1144. if portal.MXID != "" {
  1145. puppetMXID := user.bridge.FormatPuppetMXID(portal.Key.Receiver)
  1146. chats[puppetMXID] = []id.RoomID{portal.MXID}
  1147. }
  1148. }
  1149. return chats
  1150. }
  1151. func (user *User) updateDirectChats(chats map[id.UserID][]id.RoomID) {
  1152. if !user.bridge.Config.Bridge.SyncDirectChatList {
  1153. return
  1154. }
  1155. puppet := user.bridge.GetPuppetByMXID(user.MXID)
  1156. if puppet == nil {
  1157. return
  1158. }
  1159. intent := puppet.CustomIntent()
  1160. if intent == nil {
  1161. return
  1162. }
  1163. method := http.MethodPatch
  1164. if chats == nil {
  1165. chats = user.getDirectChats()
  1166. method = http.MethodPut
  1167. }
  1168. user.log.Debug().Msg("Updating m.direct list on homeserver")
  1169. var err error
  1170. if user.bridge.Config.Homeserver.Software == bridgeconfig.SoftwareAsmux {
  1171. urlPath := intent.BuildURL(mautrix.ClientURLPath{"unstable", "com.beeper.asmux", "dms"})
  1172. _, err = intent.MakeFullRequest(mautrix.FullRequest{
  1173. Method: method,
  1174. URL: urlPath,
  1175. Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
  1176. RequestJSON: chats,
  1177. })
  1178. } else {
  1179. existingChats := map[id.UserID][]id.RoomID{}
  1180. err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
  1181. if err != nil {
  1182. user.log.Warn().Err(err).Msg("Failed to get m.direct event to update it")
  1183. return
  1184. }
  1185. for userID, rooms := range existingChats {
  1186. if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
  1187. // This is not a ghost user, include it in the new list
  1188. chats[userID] = rooms
  1189. } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
  1190. // This is a ghost user, but we're not replacing the whole list, so include it too
  1191. chats[userID] = rooms
  1192. }
  1193. }
  1194. err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
  1195. }
  1196. if err != nil {
  1197. user.log.Warn().Err(err).Msg("Failed to update m.direct event")
  1198. }
  1199. }
  1200. func (user *User) bridgeGuild(guildID string, everything bool) error {
  1201. guild := user.bridge.GetGuildByID(guildID, false)
  1202. if guild == nil {
  1203. return errors.New("guild not found")
  1204. }
  1205. meta, _ := user.Session.State.Guild(guildID)
  1206. err := guild.CreateMatrixRoom(user, meta)
  1207. if err != nil {
  1208. return err
  1209. }
  1210. log := user.log.With().Str("guild_id", guild.ID).Logger()
  1211. user.addGuildToSpace(guild, false, time.Now())
  1212. for _, ch := range meta.Channels {
  1213. portal := user.GetPortalByMeta(ch)
  1214. if (everything && user.channelIsBridgeable(ch)) || ch.Type == discordgo.ChannelTypeGuildCategory {
  1215. err = portal.CreateMatrixRoom(user, ch)
  1216. if err != nil {
  1217. log.Error().Err(err).Str("channel_id", ch.ID).
  1218. Msg("Failed to create room for guild channel while bridging guild")
  1219. }
  1220. }
  1221. }
  1222. if everything {
  1223. guild.BridgingMode = database.GuildBridgeEverything
  1224. }
  1225. guild.Update()
  1226. if user.Session.IsUser {
  1227. log.Debug().Msg("Subscribing to guild after bridging")
  1228. err = user.Session.SubscribeGuild(discordgo.GuildSubscribeData{
  1229. GuildID: guild.ID,
  1230. Typing: true,
  1231. Activities: true,
  1232. Threads: true,
  1233. })
  1234. if err != nil {
  1235. log.Warn().Err(err).Msg("Failed to subscribe to guild")
  1236. }
  1237. }
  1238. return nil
  1239. }
  1240. func (user *User) unbridgeGuild(guildID string) error {
  1241. if user.PermissionLevel < bridgeconfig.PermissionLevelAdmin && user.PortalHasOtherUsers(guildID) {
  1242. return errors.New("only bridge admins can unbridge guilds with other users")
  1243. }
  1244. guild := user.bridge.GetGuildByID(guildID, false)
  1245. if guild == nil {
  1246. return errors.New("guild not found")
  1247. }
  1248. guild.roomCreateLock.Lock()
  1249. defer guild.roomCreateLock.Unlock()
  1250. if guild.BridgingMode == database.GuildBridgeNothing && guild.MXID == "" {
  1251. return errors.New("that guild is not bridged")
  1252. }
  1253. guild.BridgingMode = database.GuildBridgeNothing
  1254. guild.Update()
  1255. for _, portal := range user.bridge.GetAllPortalsInGuild(guild.ID) {
  1256. portal.cleanup(false)
  1257. portal.RemoveMXID()
  1258. }
  1259. guild.cleanup()
  1260. guild.RemoveMXID()
  1261. return nil
  1262. }