user.go 43 KB

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