user.go 43 KB

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