user.go 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882
  1. package main
  2. import (
  3. "errors"
  4. "fmt"
  5. "net/http"
  6. "os"
  7. "runtime"
  8. "strings"
  9. "sync"
  10. "time"
  11. log "maunium.net/go/maulogger/v2"
  12. "github.com/bwmarrin/discordgo"
  13. "maunium.net/go/mautrix"
  14. "maunium.net/go/mautrix/appservice"
  15. "maunium.net/go/mautrix/bridge"
  16. "maunium.net/go/mautrix/bridge/bridgeconfig"
  17. "maunium.net/go/mautrix/event"
  18. "maunium.net/go/mautrix/id"
  19. "go.mau.fi/mautrix-discord/database"
  20. )
  21. var (
  22. ErrNotConnected = errors.New("not connected")
  23. ErrNotLoggedIn = errors.New("not logged in")
  24. )
  25. type User struct {
  26. *database.User
  27. sync.Mutex
  28. bridge *DiscordBridge
  29. log log.Logger
  30. PermissionLevel bridgeconfig.PermissionLevel
  31. spaceCreateLock sync.Mutex
  32. spaceMembershipChecked bool
  33. dmSpaceMembershipChecked bool
  34. Session *discordgo.Session
  35. }
  36. var discordLog log.Logger
  37. func init() {
  38. discordgo.Logger = func(msgL, caller int, format string, a ...interface{}) {
  39. pc, file, line, _ := runtime.Caller(caller + 1)
  40. files := strings.Split(file, "/")
  41. file = files[len(files)-1]
  42. name := runtime.FuncForPC(pc).Name()
  43. fns := strings.Split(name, ".")
  44. name = fns[len(fns)-1]
  45. msg := fmt.Sprintf(format, a...)
  46. var level log.Level
  47. switch msgL {
  48. case discordgo.LogError:
  49. level = log.LevelError
  50. case discordgo.LogWarning:
  51. level = log.LevelWarn
  52. case discordgo.LogInformational:
  53. level = log.LevelInfo
  54. case discordgo.LogDebug:
  55. level = log.LevelDebug
  56. }
  57. discordLog.Logfln(level, "%s:%d:%s() %s", file, line, name, msg)
  58. }
  59. }
  60. func (user *User) GetPermissionLevel() bridgeconfig.PermissionLevel {
  61. return user.PermissionLevel
  62. }
  63. func (user *User) GetManagementRoomID() id.RoomID {
  64. return user.ManagementRoom
  65. }
  66. func (user *User) GetMXID() id.UserID {
  67. return user.MXID
  68. }
  69. func (user *User) GetCommandState() map[string]interface{} {
  70. return nil
  71. }
  72. func (user *User) GetIDoublePuppet() bridge.DoublePuppet {
  73. p := user.bridge.GetPuppetByCustomMXID(user.MXID)
  74. if p == nil || p.CustomIntent() == nil {
  75. return nil
  76. }
  77. return p
  78. }
  79. func (user *User) GetIGhost() bridge.Ghost {
  80. if user.DiscordID == "" {
  81. return nil
  82. }
  83. p := user.bridge.GetPuppetByID(user.DiscordID)
  84. if p == nil {
  85. return nil
  86. }
  87. return p
  88. }
  89. var _ bridge.User = (*User)(nil)
  90. func (br *DiscordBridge) loadUser(dbUser *database.User, mxid *id.UserID) *User {
  91. // If we weren't passed in a user we attempt to create one if we were given
  92. // a matrix id.
  93. if dbUser == nil {
  94. if mxid == nil {
  95. return nil
  96. }
  97. dbUser = br.DB.User.New()
  98. dbUser.MXID = *mxid
  99. dbUser.Insert()
  100. }
  101. user := br.NewUser(dbUser)
  102. // We assume the usersLock was acquired by our caller.
  103. br.usersByMXID[user.MXID] = user
  104. if user.DiscordID != "" {
  105. br.usersByID[user.DiscordID] = user
  106. }
  107. if user.ManagementRoom != "" {
  108. // Lock the management rooms for our update
  109. br.managementRoomsLock.Lock()
  110. br.managementRooms[user.ManagementRoom] = user
  111. br.managementRoomsLock.Unlock()
  112. }
  113. return user
  114. }
  115. func (br *DiscordBridge) GetUserByMXID(userID id.UserID) *User {
  116. br.usersLock.Lock()
  117. defer br.usersLock.Unlock()
  118. user, ok := br.usersByMXID[userID]
  119. if !ok {
  120. return br.loadUser(br.DB.User.GetByMXID(userID), &userID)
  121. }
  122. return user
  123. }
  124. func (br *DiscordBridge) GetUserByID(id string) *User {
  125. br.usersLock.Lock()
  126. defer br.usersLock.Unlock()
  127. user, ok := br.usersByID[id]
  128. if !ok {
  129. return br.loadUser(br.DB.User.GetByID(id), nil)
  130. }
  131. return user
  132. }
  133. func (br *DiscordBridge) NewUser(dbUser *database.User) *User {
  134. user := &User{
  135. User: dbUser,
  136. bridge: br,
  137. log: br.Log.Sub("User").Sub(string(dbUser.MXID)),
  138. }
  139. user.PermissionLevel = br.Config.Bridge.Permissions.Get(user.MXID)
  140. return user
  141. }
  142. func (br *DiscordBridge) getAllUsersWithToken() []*User {
  143. br.usersLock.Lock()
  144. defer br.usersLock.Unlock()
  145. dbUsers := br.DB.User.GetAllWithToken()
  146. users := make([]*User, len(dbUsers))
  147. for idx, dbUser := range dbUsers {
  148. user, ok := br.usersByMXID[dbUser.MXID]
  149. if !ok {
  150. user = br.loadUser(dbUser, nil)
  151. }
  152. users[idx] = user
  153. }
  154. return users
  155. }
  156. func (br *DiscordBridge) startUsers() {
  157. br.Log.Debugln("Starting users")
  158. for _, u := range br.getAllUsersWithToken() {
  159. go func(user *User) {
  160. err := user.Connect()
  161. if err != nil {
  162. user.log.Errorfln("Error connecting: %v", err)
  163. }
  164. }(u)
  165. }
  166. br.Log.Debugln("Starting custom puppets")
  167. for _, customPuppet := range br.GetAllPuppetsWithCustomMXID() {
  168. go func(puppet *Puppet) {
  169. br.Log.Debugln("Starting custom puppet", puppet.CustomMXID)
  170. if err := puppet.StartCustomMXID(true); err != nil {
  171. puppet.log.Errorln("Failed to start custom puppet:", err)
  172. }
  173. }(customPuppet)
  174. }
  175. }
  176. func (user *User) SetManagementRoom(roomID id.RoomID) {
  177. user.bridge.managementRoomsLock.Lock()
  178. defer user.bridge.managementRoomsLock.Unlock()
  179. existing, ok := user.bridge.managementRooms[roomID]
  180. if ok {
  181. existing.ManagementRoom = ""
  182. existing.Update()
  183. }
  184. user.ManagementRoom = roomID
  185. user.bridge.managementRooms[user.ManagementRoom] = user
  186. user.Update()
  187. }
  188. func (user *User) getSpaceRoom(ptr *id.RoomID, checked *bool, name, topic string, parent id.RoomID) id.RoomID {
  189. if len(*ptr) == 0 {
  190. user.spaceCreateLock.Lock()
  191. defer user.spaceCreateLock.Unlock()
  192. if len(*ptr) > 0 {
  193. return *ptr
  194. }
  195. initialState := []*event.Event{{
  196. Type: event.StateRoomAvatar,
  197. Content: event.Content{
  198. Parsed: &event.RoomAvatarEventContent{
  199. URL: user.bridge.Config.AppService.Bot.ParsedAvatar,
  200. },
  201. },
  202. }}
  203. if parent != "" {
  204. parentIDStr := parent.String()
  205. initialState = append(initialState, &event.Event{
  206. Type: event.StateSpaceParent,
  207. StateKey: &parentIDStr,
  208. Content: event.Content{
  209. Parsed: &event.SpaceParentEventContent{
  210. Canonical: true,
  211. Via: []string{user.bridge.AS.HomeserverDomain},
  212. },
  213. },
  214. })
  215. }
  216. resp, err := user.bridge.Bot.CreateRoom(&mautrix.ReqCreateRoom{
  217. Visibility: "private",
  218. Name: name,
  219. Topic: topic,
  220. InitialState: initialState,
  221. CreationContent: map[string]interface{}{
  222. "type": event.RoomTypeSpace,
  223. },
  224. PowerLevelOverride: &event.PowerLevelsEventContent{
  225. Users: map[id.UserID]int{
  226. user.bridge.Bot.UserID: 9001,
  227. user.MXID: 50,
  228. },
  229. },
  230. })
  231. if err != nil {
  232. user.log.Errorln("Failed to auto-create space room:", err)
  233. } else {
  234. *ptr = resp.RoomID
  235. user.Update()
  236. user.ensureInvited(nil, *ptr, false)
  237. if parent != "" {
  238. _, err = user.bridge.Bot.SendStateEvent(parent, event.StateSpaceChild, resp.RoomID.String(), &event.SpaceChildEventContent{
  239. Via: []string{user.bridge.AS.HomeserverDomain},
  240. Order: " 0000",
  241. })
  242. if err != nil {
  243. user.log.Errorfln("Failed to add space room %s to parent space %s: %v", resp.RoomID, parent, err)
  244. }
  245. }
  246. }
  247. } else if !*checked && !user.bridge.StateStore.IsInRoom(*ptr, user.MXID) {
  248. user.ensureInvited(nil, *ptr, false)
  249. }
  250. *checked = true
  251. return *ptr
  252. }
  253. func (user *User) GetSpaceRoom() id.RoomID {
  254. return user.getSpaceRoom(&user.SpaceRoom, &user.spaceMembershipChecked, "Discord", "Your Discord bridged chats", "")
  255. }
  256. func (user *User) GetDMSpaceRoom() id.RoomID {
  257. return user.getSpaceRoom(&user.DMSpaceRoom, &user.dmSpaceMembershipChecked, "Direct Messages", "Your Discord direct messages", user.GetSpaceRoom())
  258. }
  259. func (user *User) tryAutomaticDoublePuppeting() {
  260. user.Lock()
  261. defer user.Unlock()
  262. if !user.bridge.Config.CanAutoDoublePuppet(user.MXID) {
  263. return
  264. }
  265. user.log.Debugln("Checking if double puppeting needs to be enabled")
  266. puppet := user.bridge.GetPuppetByID(user.DiscordID)
  267. if puppet.CustomMXID != "" {
  268. user.log.Debugln("User already has double-puppeting enabled")
  269. return
  270. }
  271. accessToken, err := puppet.loginWithSharedSecret(user.MXID)
  272. if err != nil {
  273. user.log.Warnln("Failed to login with shared secret:", err)
  274. return
  275. }
  276. err = puppet.SwitchCustomMXID(accessToken, user.MXID)
  277. if err != nil {
  278. puppet.log.Warnln("Failed to switch to auto-logined custom puppet:", err)
  279. return
  280. }
  281. user.log.Infoln("Successfully automatically enabled custom puppet")
  282. }
  283. func (user *User) syncChatDoublePuppetDetails(portal *Portal, justCreated bool) {
  284. doublePuppet := portal.bridge.GetPuppetByCustomMXID(user.MXID)
  285. if doublePuppet == nil {
  286. return
  287. }
  288. if doublePuppet == nil || doublePuppet.CustomIntent() == nil || portal.MXID == "" {
  289. return
  290. }
  291. // TODO sync mute status
  292. }
  293. func (user *User) Login(token string) error {
  294. user.DiscordToken = token
  295. user.Update()
  296. return user.Connect()
  297. }
  298. func (user *User) IsLoggedIn() bool {
  299. user.Lock()
  300. defer user.Unlock()
  301. return user.DiscordToken != ""
  302. }
  303. func (user *User) Logout() error {
  304. user.Lock()
  305. defer user.Unlock()
  306. if user.Session == nil {
  307. return ErrNotLoggedIn
  308. }
  309. puppet := user.bridge.GetPuppetByID(user.DiscordID)
  310. if puppet.CustomMXID != "" {
  311. err := puppet.SwitchCustomMXID("", "")
  312. if err != nil {
  313. user.log.Warnln("Failed to logout-matrix while logging out of Discord:", err)
  314. }
  315. }
  316. if err := user.Session.Close(); err != nil {
  317. return err
  318. }
  319. user.Session = nil
  320. user.DiscordToken = ""
  321. user.Update()
  322. return nil
  323. }
  324. func (user *User) Connected() bool {
  325. user.Lock()
  326. defer user.Unlock()
  327. return user.Session != nil
  328. }
  329. func (user *User) Connect() error {
  330. user.Lock()
  331. defer user.Unlock()
  332. if user.DiscordToken == "" {
  333. return ErrNotLoggedIn
  334. }
  335. user.log.Debugln("Connecting to discord")
  336. session, err := discordgo.New(user.DiscordToken)
  337. if err != nil {
  338. return err
  339. }
  340. // TODO move to config
  341. if os.Getenv("DISCORD_DEBUG") == "1" {
  342. session.LogLevel = discordgo.LogDebug
  343. }
  344. user.Session = session
  345. // Add our event handlers
  346. user.Session.AddHandler(user.readyHandler)
  347. user.Session.AddHandler(user.connectedHandler)
  348. user.Session.AddHandler(user.disconnectedHandler)
  349. user.Session.AddHandler(user.guildCreateHandler)
  350. user.Session.AddHandler(user.guildDeleteHandler)
  351. user.Session.AddHandler(user.guildUpdateHandler)
  352. user.Session.AddHandler(user.channelCreateHandler)
  353. user.Session.AddHandler(user.channelDeleteHandler)
  354. user.Session.AddHandler(user.channelPinsUpdateHandler)
  355. user.Session.AddHandler(user.channelUpdateHandler)
  356. user.Session.AddHandler(user.messageCreateHandler)
  357. user.Session.AddHandler(user.messageDeleteHandler)
  358. user.Session.AddHandler(user.messageUpdateHandler)
  359. user.Session.AddHandler(user.reactionAddHandler)
  360. user.Session.AddHandler(user.reactionRemoveHandler)
  361. user.Session.Identify.Presence.Status = "online"
  362. return user.Session.Open()
  363. }
  364. func (user *User) Disconnect() error {
  365. user.Lock()
  366. defer user.Unlock()
  367. if user.Session == nil {
  368. return ErrNotConnected
  369. }
  370. if err := user.Session.Close(); err != nil {
  371. return err
  372. }
  373. user.Session = nil
  374. return nil
  375. }
  376. func (user *User) bridgeMessage(guildID string) bool {
  377. // Non guild message always get bridged.
  378. if guildID == "" {
  379. return true
  380. }
  381. guild := user.bridge.GetGuildByID(guildID, false)
  382. if guild.MXID != "" {
  383. return true
  384. }
  385. user.log.Debugfln("Cgnoring message for non-bridged guild %s", guildID)
  386. return false
  387. }
  388. func (user *User) readyHandler(_ *discordgo.Session, r *discordgo.Ready) {
  389. user.log.Debugln("Discord connection ready")
  390. // Update our user fields
  391. if user.DiscordID != r.User.ID {
  392. user.DiscordID = r.User.ID
  393. user.Update()
  394. }
  395. updateTS := time.Now()
  396. portalsInSpace := make(map[string]bool)
  397. for _, guild := range user.GetPortals() {
  398. portalsInSpace[guild.DiscordID] = guild.InSpace
  399. }
  400. for _, guild := range r.Guilds {
  401. user.handleGuild(guild, updateTS, portalsInSpace[guild.ID])
  402. }
  403. user.PrunePortalList(updateTS)
  404. const maxCreate = 5
  405. for i, ch := range r.PrivateChannels {
  406. portal := user.GetPortalByMeta(ch)
  407. user.handlePrivateChannel(portal, ch, updateTS, i < maxCreate, portalsInSpace[portal.Key.String()])
  408. }
  409. }
  410. func (user *User) handlePrivateChannel(portal *Portal, meta *discordgo.Channel, timestamp time.Time, create, isInSpace bool) {
  411. if create && portal.MXID == "" {
  412. err := portal.CreateMatrixRoom(user, meta)
  413. if err != nil {
  414. user.log.Errorfln("Failed to create portal for private channel %s in initial sync: %v", meta.ID, err)
  415. }
  416. } else {
  417. portal.UpdateInfo(user, meta)
  418. }
  419. if len(portal.MXID) > 0 && !isInSpace {
  420. _, err := user.bridge.Bot.SendStateEvent(user.GetDMSpaceRoom(), event.StateSpaceChild, portal.MXID.String(), &event.SpaceChildEventContent{
  421. Via: []string{user.bridge.AS.HomeserverDomain},
  422. })
  423. if err != nil {
  424. user.log.Errorfln("Failed to add DM room %s to user DM space: %v", portal.MXID, err)
  425. } else {
  426. isInSpace = true
  427. }
  428. }
  429. user.MarkInPortal(database.UserPortal{
  430. DiscordID: meta.ID,
  431. Type: database.UserPortalTypeDM,
  432. Timestamp: timestamp,
  433. InSpace: isInSpace,
  434. })
  435. }
  436. func (user *User) addGuildToSpace(guild *Guild) bool {
  437. if len(guild.MXID) > 0 {
  438. _, err := user.bridge.Bot.SendStateEvent(user.GetSpaceRoom(), event.StateSpaceChild, guild.MXID.String(), &event.SpaceChildEventContent{
  439. Via: []string{user.bridge.AS.HomeserverDomain},
  440. })
  441. if err != nil {
  442. user.log.Errorfln("Failed to add guild space %s to user space: %v", guild.MXID, err)
  443. } else {
  444. return true
  445. }
  446. }
  447. return false
  448. }
  449. func (user *User) handleGuild(meta *discordgo.Guild, timestamp time.Time, isInSpace bool) {
  450. guild := user.bridge.GetGuildByID(meta.ID, true)
  451. guild.UpdateInfo(user, meta)
  452. if len(meta.Channels) > 0 {
  453. for _, ch := range meta.Channels {
  454. portal := user.GetPortalByMeta(ch)
  455. if guild.AutoBridgeChannels && portal.MXID == "" {
  456. err := portal.CreateMatrixRoom(user, ch)
  457. if err != nil {
  458. user.log.Errorfln("Failed to create portal for guild channel %s/%s in initial sync: %v", guild.ID, ch.ID, err)
  459. }
  460. } else {
  461. portal.UpdateInfo(user, ch)
  462. }
  463. }
  464. }
  465. if !isInSpace {
  466. isInSpace = user.addGuildToSpace(guild)
  467. }
  468. user.MarkInPortal(database.UserPortal{
  469. DiscordID: meta.ID,
  470. Type: database.UserPortalTypeGuild,
  471. Timestamp: timestamp,
  472. InSpace: isInSpace,
  473. })
  474. }
  475. func (user *User) connectedHandler(_ *discordgo.Session, c *discordgo.Connect) {
  476. user.log.Debugln("Connected to discord")
  477. user.tryAutomaticDoublePuppeting()
  478. }
  479. func (user *User) disconnectedHandler(_ *discordgo.Session, d *discordgo.Disconnect) {
  480. user.log.Debugln("Disconnected from discord")
  481. }
  482. func (user *User) guildCreateHandler(_ *discordgo.Session, g *discordgo.GuildCreate) {
  483. user.handleGuild(g.Guild, time.Now(), false)
  484. }
  485. func (user *User) guildDeleteHandler(_ *discordgo.Session, g *discordgo.GuildDelete) {
  486. user.MarkNotInPortal(g.ID)
  487. guild := user.bridge.GetGuildByID(g.ID, false)
  488. if guild == nil || guild.MXID == "" {
  489. return
  490. }
  491. // TODO clean up?
  492. }
  493. func (user *User) guildUpdateHandler(_ *discordgo.Session, g *discordgo.GuildUpdate) {
  494. user.handleGuild(g.Guild, time.Now(), user.IsInSpace(g.ID))
  495. }
  496. func (user *User) channelCreateHandler(_ *discordgo.Session, c *discordgo.ChannelCreate) {
  497. if !user.bridgeMessage(c.GuildID) {
  498. return
  499. }
  500. portal := user.GetPortalByMeta(c.Channel)
  501. if portal.MXID != "" {
  502. return
  503. }
  504. if c.GuildID == "" {
  505. user.handlePrivateChannel(portal, c.Channel, time.Now(), true, user.IsInSpace(portal.Key.String()))
  506. } else {
  507. err := portal.CreateMatrixRoom(user, c.Channel)
  508. if err != nil {
  509. user.log.Errorfln("Error creating Matrix room for %s on channel create event: %v", c.ID, err)
  510. }
  511. }
  512. }
  513. func (user *User) channelDeleteHandler(_ *discordgo.Session, c *discordgo.ChannelDelete) {
  514. user.log.Debugln("channel delete handler")
  515. }
  516. func (user *User) channelPinsUpdateHandler(_ *discordgo.Session, c *discordgo.ChannelPinsUpdate) {
  517. user.log.Debugln("channel pins update")
  518. }
  519. func (user *User) channelUpdateHandler(_ *discordgo.Session, c *discordgo.ChannelUpdate) {
  520. portal := user.GetPortalByMeta(c.Channel)
  521. if c.GuildID == "" {
  522. user.handlePrivateChannel(portal, c.Channel, time.Now(), true, user.IsInSpace(portal.Key.String()))
  523. } else {
  524. portal.UpdateInfo(user, c.Channel)
  525. }
  526. }
  527. func (user *User) pushPortalMessage(msg interface{}, typeName, channelID, guildID string) {
  528. if user.Session.LogLevel == discordgo.LogDebug {
  529. fmt.Printf("%+v\n", msg)
  530. }
  531. if !user.bridgeMessage(guildID) {
  532. return
  533. }
  534. portal := user.GetExistingPortalByID(channelID)
  535. var thread *Thread
  536. if portal == nil {
  537. thread = user.bridge.GetThreadByID(channelID, nil)
  538. if thread == nil || thread.Parent == nil {
  539. user.log.Debugfln("Dropping %s in unknown channel %s/%s", typeName, guildID, channelID)
  540. return
  541. }
  542. portal = thread.Parent
  543. }
  544. portal.discordMessages <- portalDiscordMessage{
  545. msg: msg,
  546. user: user,
  547. thread: thread,
  548. }
  549. }
  550. func (user *User) messageCreateHandler(_ *discordgo.Session, m *discordgo.MessageCreate) {
  551. user.pushPortalMessage(m, "message create", m.ChannelID, m.GuildID)
  552. }
  553. func (user *User) messageDeleteHandler(_ *discordgo.Session, m *discordgo.MessageDelete) {
  554. user.pushPortalMessage(m, "message delete", m.ChannelID, m.GuildID)
  555. }
  556. func (user *User) messageUpdateHandler(_ *discordgo.Session, m *discordgo.MessageUpdate) {
  557. user.pushPortalMessage(m, "message update", m.ChannelID, m.GuildID)
  558. }
  559. func (user *User) reactionAddHandler(_ *discordgo.Session, m *discordgo.MessageReactionAdd) {
  560. user.pushPortalMessage(m, "reaction add", m.ChannelID, m.GuildID)
  561. }
  562. func (user *User) reactionRemoveHandler(_ *discordgo.Session, m *discordgo.MessageReactionRemove) {
  563. user.pushPortalMessage(m, "reaction remove", m.ChannelID, m.GuildID)
  564. }
  565. func (user *User) ensureInvited(intent *appservice.IntentAPI, roomID id.RoomID, isDirect bool) bool {
  566. if intent == nil {
  567. intent = user.bridge.Bot
  568. }
  569. ret := false
  570. inviteContent := event.Content{
  571. Parsed: &event.MemberEventContent{
  572. Membership: event.MembershipInvite,
  573. IsDirect: isDirect,
  574. },
  575. Raw: map[string]interface{}{},
  576. }
  577. customPuppet := user.bridge.GetPuppetByCustomMXID(user.MXID)
  578. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  579. inviteContent.Raw["fi.mau.will_auto_accept"] = true
  580. }
  581. _, err := intent.SendStateEvent(roomID, event.StateMember, user.MXID.String(), &inviteContent)
  582. var httpErr mautrix.HTTPError
  583. if err != nil && errors.As(err, &httpErr) && httpErr.RespError != nil && strings.Contains(httpErr.RespError.Err, "is already in the room") {
  584. user.bridge.StateStore.SetMembership(roomID, user.MXID, event.MembershipJoin)
  585. ret = true
  586. } else if err != nil {
  587. user.log.Warnfln("Failed to invite user to %s: %v", roomID, err)
  588. } else {
  589. ret = true
  590. }
  591. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  592. err = customPuppet.CustomIntent().EnsureJoined(roomID, appservice.EnsureJoinedParams{IgnoreCache: true})
  593. if err != nil {
  594. user.log.Warnfln("Failed to auto-join %s: %v", roomID, err)
  595. ret = false
  596. } else {
  597. ret = true
  598. }
  599. }
  600. return ret
  601. }
  602. func (user *User) getDirectChats() map[id.UserID][]id.RoomID {
  603. chats := map[id.UserID][]id.RoomID{}
  604. privateChats := user.bridge.DB.Portal.FindPrivateChatsOf(user.DiscordID)
  605. for _, portal := range privateChats {
  606. if portal.MXID != "" {
  607. puppetMXID := user.bridge.FormatPuppetMXID(portal.Key.Receiver)
  608. chats[puppetMXID] = []id.RoomID{portal.MXID}
  609. }
  610. }
  611. return chats
  612. }
  613. func (user *User) updateDirectChats(chats map[id.UserID][]id.RoomID) {
  614. if !user.bridge.Config.Bridge.SyncDirectChatList {
  615. return
  616. }
  617. puppet := user.bridge.GetPuppetByMXID(user.MXID)
  618. if puppet == nil {
  619. return
  620. }
  621. intent := puppet.CustomIntent()
  622. if intent == nil {
  623. return
  624. }
  625. method := http.MethodPatch
  626. if chats == nil {
  627. chats = user.getDirectChats()
  628. method = http.MethodPut
  629. }
  630. user.log.Debugln("Updating m.direct list on homeserver")
  631. var err error
  632. if user.bridge.Config.Homeserver.Asmux {
  633. urlPath := intent.BuildURL(mautrix.ClientURLPath{"unstable", "com.beeper.asmux", "dms"})
  634. _, err = intent.MakeFullRequest(mautrix.FullRequest{
  635. Method: method,
  636. URL: urlPath,
  637. Headers: http.Header{"X-Asmux-Auth": {user.bridge.AS.Registration.AppToken}},
  638. RequestJSON: chats,
  639. })
  640. } else {
  641. existingChats := map[id.UserID][]id.RoomID{}
  642. err = intent.GetAccountData(event.AccountDataDirectChats.Type, &existingChats)
  643. if err != nil {
  644. user.log.Warnln("Failed to get m.direct list to update it:", err)
  645. return
  646. }
  647. for userID, rooms := range existingChats {
  648. if _, ok := user.bridge.ParsePuppetMXID(userID); !ok {
  649. // This is not a ghost user, include it in the new list
  650. chats[userID] = rooms
  651. } else if _, ok := chats[userID]; !ok && method == http.MethodPatch {
  652. // This is a ghost user, but we're not replacing the whole list, so include it too
  653. chats[userID] = rooms
  654. }
  655. }
  656. err = intent.SetAccountData(event.AccountDataDirectChats.Type, &chats)
  657. }
  658. if err != nil {
  659. user.log.Warnln("Failed to update m.direct list:", err)
  660. }
  661. }
  662. func (user *User) bridgeGuild(guildID string, everything bool) error {
  663. guild := user.bridge.GetGuildByID(guildID, false)
  664. if guild == nil {
  665. return errors.New("guild not found")
  666. }
  667. meta, _ := user.Session.State.Guild(guildID)
  668. err := guild.CreateMatrixRoom(user, meta)
  669. if err != nil {
  670. return err
  671. }
  672. user.addGuildToSpace(guild)
  673. user.MarkInPortal(database.UserPortal{
  674. DiscordID: meta.ID,
  675. Type: database.UserPortalTypeGuild,
  676. Timestamp: time.Now(),
  677. InSpace: true,
  678. })
  679. for _, ch := range meta.Channels {
  680. portal := user.GetPortalByMeta(ch)
  681. if (everything && channelIsBridgeable(ch)) || ch.Type == discordgo.ChannelTypeGuildCategory {
  682. err = portal.CreateMatrixRoom(user, ch)
  683. if err != nil {
  684. user.log.Warnfln("Error creating room for guild channel %s: %v", ch.ID, err)
  685. }
  686. }
  687. }
  688. return nil
  689. }
  690. func (user *User) unbridgeGuild(guildID string) error {
  691. //user.guildsLock.Lock()
  692. //defer user.guildsLock.Unlock()
  693. //
  694. //guild, exists := user.guilds[guildID]
  695. //if !exists {
  696. // return fmt.Errorf("guildID not found")
  697. //}
  698. //
  699. //if !guild.Bridge {
  700. // return fmt.Errorf("guild not bridged")
  701. //}
  702. //
  703. //// First update the guild so we don't have any other go routines recreating
  704. //// channels we're about to destroy.
  705. //guild.Bridge = false
  706. //guild.Upsert()
  707. //
  708. //// Now run through the channels in the guild and remove any portals we
  709. //// have for them.
  710. //channels, err := user.Session.GuildChannels(guildID)
  711. //if err != nil {
  712. // return err
  713. //}
  714. //
  715. //for _, channel := range channels {
  716. // if channelIsBridgeable(channel) {
  717. // key := database.PortalKey{
  718. // ChannelID: channel.ID,
  719. // Receiver: user.DiscordID,
  720. // }
  721. //
  722. // portal := user.bridge.GetPortalByID(key)
  723. // portal.leave(user)
  724. // }
  725. //}
  726. return nil
  727. }