user.go 42 KB

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