commands.go 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860
  1. // mautrix-discord - A Matrix-Discord puppeting bridge.
  2. // Copyright (C) 2022 Tulir Asokan
  3. //
  4. // This program is free software: you can redistribute it and/or modify
  5. // it under the terms of the GNU Affero General Public License as published by
  6. // the Free Software Foundation, either version 3 of the License, or
  7. // (at your option) any later version.
  8. //
  9. // This program is distributed in the hope that it will be useful,
  10. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. // GNU Affero General Public License for more details.
  13. //
  14. // You should have received a copy of the GNU Affero General Public License
  15. // along with this program. If not, see <https://www.gnu.org/licenses/>.
  16. package main
  17. import (
  18. "bytes"
  19. "context"
  20. "encoding/base64"
  21. "errors"
  22. "fmt"
  23. "html"
  24. "net/http"
  25. "strconv"
  26. "strings"
  27. "github.com/bwmarrin/discordgo"
  28. "github.com/skip2/go-qrcode"
  29. "maunium.net/go/mautrix"
  30. "maunium.net/go/mautrix/appservice"
  31. "maunium.net/go/mautrix/bridge/bridgeconfig"
  32. "maunium.net/go/mautrix/bridge/commands"
  33. "maunium.net/go/mautrix/event"
  34. "maunium.net/go/mautrix/id"
  35. "go.mau.fi/mautrix-discord/database"
  36. "go.mau.fi/mautrix-discord/remoteauth"
  37. )
  38. type WrappedCommandEvent struct {
  39. *commands.Event
  40. Bridge *DiscordBridge
  41. User *User
  42. Portal *Portal
  43. }
  44. var HelpSectionPortalManagement = commands.HelpSection{Name: "Portal management", Order: 20}
  45. func (br *DiscordBridge) RegisterCommands() {
  46. proc := br.CommandProcessor.(*commands.Processor)
  47. proc.AddHandlers(
  48. cmdLoginToken,
  49. cmdLoginQR,
  50. cmdLogout,
  51. cmdPing,
  52. cmdReconnect,
  53. cmdDisconnect,
  54. cmdBridge,
  55. cmdUnbridge,
  56. cmdDeletePortal,
  57. cmdSetRelay,
  58. cmdUnsetRelay,
  59. cmdGuilds,
  60. cmdRejoinSpace,
  61. cmdDeleteAllPortals,
  62. cmdExec,
  63. cmdCommands,
  64. )
  65. }
  66. func wrapCommand(handler func(*WrappedCommandEvent)) func(*commands.Event) {
  67. return func(ce *commands.Event) {
  68. user := ce.User.(*User)
  69. var portal *Portal
  70. if ce.Portal != nil {
  71. portal = ce.Portal.(*Portal)
  72. }
  73. br := ce.Bridge.Child.(*DiscordBridge)
  74. handler(&WrappedCommandEvent{ce, br, user, portal})
  75. }
  76. }
  77. var cmdLoginToken = &commands.FullHandler{
  78. Func: wrapCommand(fnLoginToken),
  79. Name: "login-token",
  80. Help: commands.HelpMeta{
  81. Section: commands.HelpSectionAuth,
  82. Description: "Link the bridge to your Discord account by extracting the access token manually.",
  83. Args: "<user/bot/oauth> <_token_>",
  84. },
  85. }
  86. const discordTokenEpoch = 1293840000
  87. func decodeToken(token string) (userID int64, err error) {
  88. parts := strings.Split(token, ".")
  89. if len(parts) != 3 {
  90. err = fmt.Errorf("invalid number of parts in token")
  91. return
  92. }
  93. var userIDStr []byte
  94. userIDStr, err = base64.RawURLEncoding.DecodeString(parts[0])
  95. if err != nil {
  96. err = fmt.Errorf("invalid base64 in user ID part: %w", err)
  97. return
  98. }
  99. _, err = base64.RawURLEncoding.DecodeString(parts[1])
  100. if err != nil {
  101. err = fmt.Errorf("invalid base64 in random part: %w", err)
  102. return
  103. }
  104. _, err = base64.RawURLEncoding.DecodeString(parts[2])
  105. if err != nil {
  106. err = fmt.Errorf("invalid base64 in checksum part: %w", err)
  107. return
  108. }
  109. userID, err = strconv.ParseInt(string(userIDStr), 10, 64)
  110. if err != nil {
  111. err = fmt.Errorf("invalid number in decoded user ID part: %w", err)
  112. return
  113. }
  114. return
  115. }
  116. func fnLoginToken(ce *WrappedCommandEvent) {
  117. if len(ce.Args) != 2 {
  118. ce.Reply("**Usage**: `$cmdprefix login-token <user/bot/oauth> <token>`")
  119. return
  120. }
  121. ce.MarkRead()
  122. defer ce.Redact()
  123. if ce.User.IsLoggedIn() {
  124. ce.Reply("You're already logged in")
  125. return
  126. }
  127. token := ce.Args[1]
  128. userID, err := decodeToken(token)
  129. if err != nil {
  130. ce.Reply("Invalid token")
  131. return
  132. }
  133. switch strings.ToLower(ce.Args[0]) {
  134. case "user":
  135. // Token is used as-is
  136. case "bot":
  137. token = "Bot " + token
  138. case "oauth":
  139. token = "Bearer " + token
  140. default:
  141. ce.Reply("Token type must be `user`, `bot` or `oauth`")
  142. return
  143. }
  144. ce.Reply("Connecting to Discord as user ID %d", userID)
  145. if err = ce.User.Login(token); err != nil {
  146. ce.Reply("Error connecting to Discord: %v", err)
  147. return
  148. }
  149. ce.Reply("Successfully logged in as @%s", ce.User.Session.State.User.Username)
  150. }
  151. var cmdLoginQR = &commands.FullHandler{
  152. Func: wrapCommand(fnLoginQR),
  153. Name: "login-qr",
  154. Aliases: []string{"login"},
  155. Help: commands.HelpMeta{
  156. Section: commands.HelpSectionAuth,
  157. Description: "Link the bridge to your Discord account by scanning a QR code.",
  158. },
  159. }
  160. func fnLoginQR(ce *WrappedCommandEvent) {
  161. if ce.User.IsLoggedIn() {
  162. ce.Reply("You're already logged in")
  163. return
  164. }
  165. client, err := remoteauth.New()
  166. if err != nil {
  167. ce.Reply("Failed to prepare login: %v", err)
  168. return
  169. }
  170. qrChan := make(chan string)
  171. doneChan := make(chan struct{})
  172. var qrCodeEvent id.EventID
  173. go func() {
  174. code := <-qrChan
  175. resp := sendQRCode(ce, code)
  176. qrCodeEvent = resp
  177. }()
  178. ctx := context.Background()
  179. if err = client.Dial(ctx, qrChan, doneChan); err != nil {
  180. close(qrChan)
  181. close(doneChan)
  182. ce.Reply("Error connecting to login websocket: %v", err)
  183. return
  184. }
  185. <-doneChan
  186. if qrCodeEvent != "" {
  187. _, _ = ce.MainIntent().RedactEvent(ce.RoomID, qrCodeEvent)
  188. }
  189. user, err := client.Result()
  190. if err != nil || len(user.Token) == 0 {
  191. if restErr := (&discordgo.RESTError{}); errors.As(err, &restErr) &&
  192. restErr.Response.StatusCode == http.StatusBadRequest &&
  193. bytes.Contains(restErr.ResponseBody, []byte("captcha-required")) {
  194. ce.Reply("Error logging in: %v\n\nCAPTCHAs are currently not supported - use token login instead", err)
  195. } else {
  196. ce.Reply("Error logging in: %v", err)
  197. }
  198. return
  199. } else if err = ce.User.Login(user.Token); err != nil {
  200. ce.Reply("Error connecting after login: %v", err)
  201. return
  202. }
  203. ce.User.Lock()
  204. ce.User.DiscordID = user.UserID
  205. ce.User.Update()
  206. ce.User.Unlock()
  207. ce.Reply("Successfully logged in as @%s", user.Username)
  208. }
  209. func sendQRCode(ce *WrappedCommandEvent, code string) id.EventID {
  210. url, ok := uploadQRCode(ce, code)
  211. if !ok {
  212. return ""
  213. }
  214. content := event.MessageEventContent{
  215. MsgType: event.MsgImage,
  216. Body: code,
  217. URL: url.CUString(),
  218. }
  219. resp, err := ce.Bot.SendMessageEvent(ce.RoomID, event.EventMessage, &content)
  220. if err != nil {
  221. ce.Log.Errorfln("Failed to send QR code: %v", err)
  222. return ""
  223. }
  224. return resp.EventID
  225. }
  226. func uploadQRCode(ce *WrappedCommandEvent, code string) (id.ContentURI, bool) {
  227. qrCode, err := qrcode.Encode(code, qrcode.Low, 256)
  228. if err != nil {
  229. ce.Log.Errorln("Failed to encode QR code:", err)
  230. ce.Reply("Failed to encode QR code: %v", err)
  231. return id.ContentURI{}, false
  232. }
  233. resp, err := ce.Bot.UploadBytes(qrCode, "image/png")
  234. if err != nil {
  235. ce.Log.Errorln("Failed to upload QR code:", err)
  236. ce.Reply("Failed to upload QR code: %v", err)
  237. return id.ContentURI{}, false
  238. }
  239. return resp.ContentURI, true
  240. }
  241. var cmdLogout = &commands.FullHandler{
  242. Func: wrapCommand(fnLogout),
  243. Name: "logout",
  244. Help: commands.HelpMeta{
  245. Section: commands.HelpSectionAuth,
  246. Description: "Forget the stored Discord auth token.",
  247. },
  248. }
  249. func fnLogout(ce *WrappedCommandEvent) {
  250. wasLoggedIn := ce.User.DiscordID != ""
  251. ce.User.Logout(false)
  252. if wasLoggedIn {
  253. ce.Reply("Logged out successfully.")
  254. } else {
  255. ce.Reply("You weren't logged in, but data was re-cleared just to be safe.")
  256. }
  257. }
  258. var cmdPing = &commands.FullHandler{
  259. Func: wrapCommand(fnPing),
  260. Name: "ping",
  261. Help: commands.HelpMeta{
  262. Section: commands.HelpSectionAuth,
  263. Description: "Check your connection to Discord",
  264. },
  265. }
  266. func fnPing(ce *WrappedCommandEvent) {
  267. if ce.User.Session == nil {
  268. if ce.User.DiscordToken == "" {
  269. ce.Reply("You're not logged in")
  270. } else {
  271. ce.Reply("You have a Discord token stored, but are not connected for some reason 🤔")
  272. }
  273. } else if ce.User.wasDisconnected {
  274. ce.Reply("You're logged in, but the Discord connection seems to be dead 💥")
  275. } else {
  276. ce.Reply("You're logged in as @%s (`%s`)", ce.User.Session.State.User.Username, ce.User.DiscordID)
  277. }
  278. }
  279. var cmdDisconnect = &commands.FullHandler{
  280. Func: wrapCommand(fnDisconnect),
  281. Name: "disconnect",
  282. Help: commands.HelpMeta{
  283. Section: commands.HelpSectionAuth,
  284. Description: "Disconnect from Discord (without logging out)",
  285. },
  286. RequiresLogin: true,
  287. }
  288. func fnDisconnect(ce *WrappedCommandEvent) {
  289. if !ce.User.Connected() {
  290. ce.Reply("You're already not connected")
  291. } else if err := ce.User.Disconnect(); err != nil {
  292. ce.Reply("Error while disconnecting: %v", err)
  293. } else {
  294. ce.Reply("Successfully disconnected")
  295. }
  296. }
  297. var cmdReconnect = &commands.FullHandler{
  298. Func: wrapCommand(fnReconnect),
  299. Name: "reconnect",
  300. Aliases: []string{"connect"},
  301. Help: commands.HelpMeta{
  302. Section: commands.HelpSectionAuth,
  303. Description: "Reconnect to Discord after disconnecting",
  304. },
  305. RequiresLogin: true,
  306. }
  307. func fnReconnect(ce *WrappedCommandEvent) {
  308. if ce.User.Connected() {
  309. ce.Reply("You're already connected")
  310. } else if err := ce.User.Connect(); err != nil {
  311. ce.Reply("Error while reconnecting: %v", err)
  312. } else {
  313. ce.Reply("Successfully reconnected")
  314. }
  315. }
  316. var cmdRejoinSpace = &commands.FullHandler{
  317. Func: wrapCommand(fnRejoinSpace),
  318. Name: "rejoin-space",
  319. Help: commands.HelpMeta{
  320. Section: HelpSectionPortalManagement,
  321. Description: "Ask the bridge for an invite to a space you left",
  322. Args: "<_guild ID_/main/dms>",
  323. },
  324. RequiresLogin: true,
  325. }
  326. func fnRejoinSpace(ce *WrappedCommandEvent) {
  327. if len(ce.Args) == 0 {
  328. ce.Reply("**Usage**: `$cmdprefix rejoin-space <guild ID/main/dms>`")
  329. return
  330. }
  331. user := ce.User
  332. if ce.Args[0] == "main" {
  333. user.ensureInvited(nil, user.GetSpaceRoom(), false, true)
  334. ce.Reply("Invited you to your main space ([link](%s))", user.GetSpaceRoom().URI(ce.Bridge.AS.HomeserverDomain).MatrixToURL())
  335. } else if ce.Args[0] == "dms" {
  336. user.ensureInvited(nil, user.GetDMSpaceRoom(), false, true)
  337. ce.Reply("Invited you to your DM space ([link](%s))", user.GetDMSpaceRoom().URI(ce.Bridge.AS.HomeserverDomain).MatrixToURL())
  338. } else if _, err := strconv.Atoi(ce.Args[0]); err == nil {
  339. ce.Reply("Rejoining guild spaces is not yet implemented")
  340. } else {
  341. ce.Reply("**Usage**: `$cmdprefix rejoin-space <guild ID/main/dms>`")
  342. return
  343. }
  344. }
  345. var roomModerator = event.Type{Type: "fi.mau.discord.admin", Class: event.StateEventType}
  346. var cmdSetRelay = &commands.FullHandler{
  347. Func: wrapCommand(fnSetRelay),
  348. Name: "set-relay",
  349. Help: commands.HelpMeta{
  350. Section: HelpSectionPortalManagement,
  351. Description: "Create or set a relay webhook for a portal",
  352. Args: "[room ID] <​--url URL> OR <​--create [name]>",
  353. },
  354. RequiresLogin: true,
  355. RequiresEventLevel: roomModerator,
  356. }
  357. const webhookURLFormat = "https://discord.com/api/webhooks/%d/%s"
  358. const selectRelayHelp = "Usage: `$cmdprefix [room ID] <​--url URL> OR <​--create [name]>`"
  359. func fnSetRelay(ce *WrappedCommandEvent) {
  360. portal := ce.Portal
  361. if len(ce.Args) > 0 && strings.HasPrefix(ce.Args[0], "!") {
  362. portal = ce.Bridge.GetPortalByMXID(id.RoomID(ce.Args[0]))
  363. if portal == nil {
  364. ce.Reply("Portal with room ID %s not found", ce.Args[0])
  365. return
  366. }
  367. if ce.User.PermissionLevel < bridgeconfig.PermissionLevelAdmin {
  368. levels, err := portal.MainIntent().PowerLevels(ce.RoomID)
  369. if err != nil {
  370. ce.ZLog.Warn().Err(err).Msg("Failed to check room power levels")
  371. ce.Reply("Failed to get room power levels to see if you're allowed to use that command")
  372. return
  373. } else if levels.GetUserLevel(ce.User.GetMXID()) < levels.GetEventLevel(roomModerator) {
  374. ce.Reply("You don't have admin rights in that room")
  375. return
  376. }
  377. }
  378. ce.Args = ce.Args[1:]
  379. } else if portal == nil {
  380. ce.Reply("You must either run the command in a portal, or specify an internal room ID as the first parameter")
  381. return
  382. }
  383. log := ce.ZLog.With().Str("channel_id", portal.Key.ChannelID).Logger()
  384. if portal.GuildID == "" {
  385. ce.Reply("Only guild channels can have relays")
  386. return
  387. } else if portal.RelayWebhookID != "" {
  388. webhookMeta, err := relayClient.WebhookWithToken(portal.RelayWebhookID, portal.RelayWebhookSecret)
  389. if err != nil {
  390. log.Warn().Err(err).Msg("Failed to get existing webhook info")
  391. ce.Reply("This channel has a relay webhook set, but getting its info failed: %v", err)
  392. return
  393. }
  394. ce.Reply("This channel already has a relay webhook %s (%s)", webhookMeta.Name, webhookMeta.ID)
  395. return
  396. } else if len(ce.Args) == 0 {
  397. ce.Reply(selectRelayHelp)
  398. return
  399. }
  400. createType := strings.ToLower(strings.TrimLeft(ce.Args[0], "-"))
  401. var webhookMeta *discordgo.Webhook
  402. switch createType {
  403. case "url":
  404. if len(ce.Args) < 2 {
  405. ce.Reply("Usage: `$cmdprefix [room ID] --url <URL>")
  406. return
  407. }
  408. ce.Redact()
  409. var webhookID int64
  410. var webhookSecret string
  411. _, err := fmt.Sscanf(ce.Args[1], webhookURLFormat, &webhookID, &webhookSecret)
  412. if err != nil {
  413. log.Warn().Str("webhook_url", ce.Args[1]).Err(err).Msg("Failed to parse provided webhook URL")
  414. ce.Reply("Invalid webhook URL")
  415. return
  416. }
  417. webhookMeta, err = relayClient.WebhookWithToken(strconv.FormatInt(webhookID, 10), webhookSecret)
  418. if err != nil {
  419. log.Warn().Err(err).Msg("Failed to get webhook info")
  420. ce.Reply("Failed to get webhook info: %v", err)
  421. return
  422. }
  423. case "create":
  424. perms, err := ce.User.Session.UserChannelPermissions(ce.User.DiscordID, portal.Key.ChannelID)
  425. if err != nil {
  426. log.Warn().Err(err).Msg("Failed to check user permissions")
  427. ce.Reply("Failed to check if you have permission to create webhooks")
  428. return
  429. } else if perms&discordgo.PermissionManageWebhooks == 0 {
  430. log.Debug().Int64("perms", perms).Msg("User doesn't have permissions to manage webhooks in channel")
  431. ce.Reply("You don't have permission to manage webhooks in that channel")
  432. return
  433. }
  434. name := "mautrix"
  435. if len(ce.Args) > 1 {
  436. name = strings.Join(ce.Args[1:], " ")
  437. }
  438. log.Debug().Str("webhook_name", name).Msg("Creating webhook")
  439. webhookMeta, err = ce.User.Session.WebhookCreate(portal.Key.ChannelID, name, "")
  440. if err != nil {
  441. log.Warn().Err(err).Msg("Failed to create webhook")
  442. ce.Reply("Failed to create webhook: %v", err)
  443. return
  444. }
  445. default:
  446. ce.Reply(selectRelayHelp)
  447. return
  448. }
  449. if portal.Key.ChannelID != webhookMeta.ChannelID {
  450. log.Debug().
  451. Str("portal_channel_id", portal.Key.ChannelID).
  452. Str("webhook_channel_id", webhookMeta.ChannelID).
  453. Msg("Provided webhook is for wrong channel")
  454. ce.Reply("That webhook is not for the right channel (expected %s, webhook is for %s)", portal.Key.ChannelID, webhookMeta.ChannelID)
  455. return
  456. }
  457. log.Debug().Str("webhook_id", webhookMeta.ID).Msg("Setting portal relay webhook")
  458. portal.RelayWebhookID = webhookMeta.ID
  459. portal.RelayWebhookSecret = webhookMeta.Token
  460. portal.Update()
  461. ce.Reply("Saved webhook %s (%s) as portal relay webhook", webhookMeta.Name, portal.RelayWebhookID)
  462. }
  463. var cmdUnsetRelay = &commands.FullHandler{
  464. Func: wrapCommand(fnUnsetRelay),
  465. Name: "unset-relay",
  466. Help: commands.HelpMeta{
  467. Section: HelpSectionPortalManagement,
  468. Description: "Disable the relay webhook and optionally delete it on Discord",
  469. Args: "[--delete]",
  470. },
  471. RequiresPortal: true,
  472. RequiresEventLevel: roomModerator,
  473. }
  474. func fnUnsetRelay(ce *WrappedCommandEvent) {
  475. if ce.Portal.RelayWebhookID == "" {
  476. ce.Reply("This portal doesn't have a relay webhook")
  477. return
  478. }
  479. if len(ce.Args) > 0 && ce.Args[0] == "--delete" {
  480. err := relayClient.WebhookDeleteWithToken(ce.Portal.RelayWebhookID, ce.Portal.RelayWebhookSecret)
  481. if err != nil {
  482. ce.Reply("Failed to delete webhook: %v", err)
  483. return
  484. } else {
  485. ce.Reply("Successfully deleted webhook")
  486. }
  487. } else {
  488. ce.Reply("Relay webhook disabled")
  489. }
  490. ce.Portal.RelayWebhookID = ""
  491. ce.Portal.RelayWebhookSecret = ""
  492. ce.Portal.Update()
  493. }
  494. var cmdGuilds = &commands.FullHandler{
  495. Func: wrapCommand(fnGuilds),
  496. Name: "guilds",
  497. Aliases: []string{"servers", "guild", "server"},
  498. Help: commands.HelpMeta{
  499. Section: HelpSectionPortalManagement,
  500. Description: "Guild bridging management",
  501. Args: "<status/bridge/unbridge/bridging-mode> [_guild ID_] [...]",
  502. },
  503. RequiresLogin: true,
  504. }
  505. const smallGuildsHelp = "**Usage**: `$cmdprefix guilds <help/status/bridge/unbridge> [guild ID] [...]`"
  506. const fullGuildsHelp = smallGuildsHelp + `
  507. * **help** - View this help message.
  508. * **status** - View the list of guilds and their bridging status.
  509. * **bridge <_guild ID_> [--entire]** - Enable bridging for a guild. The --entire flag auto-creates portals for all channels.
  510. * **bridging-mode <_guild ID_> <_mode_>** - Set the mode for bridging messages and new channels in a guild.
  511. * **unbridge <_guild ID_>** - Unbridge a guild and delete all channel portal rooms.`
  512. func fnGuilds(ce *WrappedCommandEvent) {
  513. if len(ce.Args) == 0 {
  514. ce.Reply(fullGuildsHelp)
  515. return
  516. }
  517. subcommand := strings.ToLower(ce.Args[0])
  518. ce.Args = ce.Args[1:]
  519. switch subcommand {
  520. case "status", "list":
  521. fnListGuilds(ce)
  522. case "bridge":
  523. fnBridgeGuild(ce)
  524. case "unbridge", "delete":
  525. fnUnbridgeGuild(ce)
  526. case "bridging-mode", "mode":
  527. fnGuildBridgingMode(ce)
  528. case "help":
  529. ce.Reply(fullGuildsHelp)
  530. default:
  531. ce.Reply("Unknown subcommand `%s`\n\n"+smallGuildsHelp, subcommand)
  532. }
  533. }
  534. func fnListGuilds(ce *WrappedCommandEvent) {
  535. var items []string
  536. for _, userGuild := range ce.User.GetPortals() {
  537. guild := ce.Bridge.GetGuildByID(userGuild.DiscordID, false)
  538. if guild == nil {
  539. continue
  540. }
  541. var avatarHTML string
  542. if !guild.AvatarURL.IsEmpty() {
  543. avatarHTML = fmt.Sprintf(`<img data-mx-emoticon height="24" src="%s" alt="" title="Guild avatar"> `, guild.AvatarURL.String())
  544. }
  545. items = append(items, fmt.Sprintf("<li>%s%s (<code>%s</code>) - %s</li>", avatarHTML, html.EscapeString(guild.Name), guild.ID, guild.BridgingMode.Description()))
  546. }
  547. if len(items) == 0 {
  548. ce.Reply("No guilds found")
  549. } else {
  550. ce.ReplyAdvanced(fmt.Sprintf("<p>List of guilds:</p><ul>%s</ul>", strings.Join(items, "")), false, true)
  551. }
  552. }
  553. func fnBridgeGuild(ce *WrappedCommandEvent) {
  554. if len(ce.Args) == 0 || len(ce.Args) > 2 {
  555. ce.Reply("**Usage**: `$cmdprefix guilds bridge <guild ID> [--entire]")
  556. } else if err := ce.User.bridgeGuild(ce.Args[0], len(ce.Args) == 2 && strings.ToLower(ce.Args[1]) == "--entire"); err != nil {
  557. ce.Reply("Error bridging guild: %v", err)
  558. } else {
  559. ce.Reply("Successfully bridged guild")
  560. }
  561. }
  562. func fnUnbridgeGuild(ce *WrappedCommandEvent) {
  563. if len(ce.Args) != 1 {
  564. ce.Reply("**Usage**: `$cmdprefix guilds unbridge <guild ID>")
  565. } else if err := ce.User.unbridgeGuild(ce.Args[0]); err != nil {
  566. ce.Reply("Error unbridging guild: %v", err)
  567. } else {
  568. ce.Reply("Successfully unbridged guild")
  569. }
  570. }
  571. const availableModes = "Available modes:\n" +
  572. "* `nothing` to never bridge any messages (default when unbridged)\n" +
  573. "* `if-portal-exists` to bridge messages in existing portals, but drop messages in unbridged channels\n" +
  574. "* `create-on-message` to bridge all messages and create portals if necessary on incoming messages (default after bridging)\n" +
  575. "* `everything` to bridge all messages and create portals proactively on bridge startup (default if bridged with `--entire`)\n"
  576. func fnGuildBridgingMode(ce *WrappedCommandEvent) {
  577. if len(ce.Args) == 0 || len(ce.Args) > 2 {
  578. ce.Reply("**Usage**: `$cmdprefix guilds bridging-mode <guild ID> [mode]`\n\n" + availableModes)
  579. return
  580. }
  581. guild := ce.Bridge.GetGuildByID(ce.Args[0], false)
  582. if guild == nil {
  583. ce.Reply("Guild not found")
  584. return
  585. }
  586. if len(ce.Args) == 1 {
  587. ce.Reply("%s (%s) is currently set to %s (`%s`)\n\n%s", guild.PlainName, guild.ID, guild.BridgingMode.Description(), guild.BridgingMode.String(), availableModes)
  588. return
  589. }
  590. mode := database.ParseGuildBridgingMode(ce.Args[1])
  591. if mode == database.GuildBridgeInvalid {
  592. ce.Reply("Invalid guild bridging mode `%s`", ce.Args[1])
  593. return
  594. }
  595. guild.BridgingMode = mode
  596. guild.Update()
  597. ce.Reply("Set guild bridging mode to %s", mode.Description())
  598. }
  599. var cmdBridge = &commands.FullHandler{
  600. Func: wrapCommand(fnBridge),
  601. Name: "bridge",
  602. Help: commands.HelpMeta{
  603. Section: HelpSectionPortalManagement,
  604. Description: "Bridge this room to a specific Discord channel",
  605. Args: "[--replace[=delete]] <_channel ID_>",
  606. },
  607. RequiresEventLevel: roomModerator,
  608. }
  609. func isNumber(str string) bool {
  610. for _, chr := range str {
  611. if chr < '0' || chr > '9' {
  612. return false
  613. }
  614. }
  615. return true
  616. }
  617. func fnBridge(ce *WrappedCommandEvent) {
  618. if ce.Portal != nil {
  619. ce.Reply("This is already a portal room. Unbridge with `$cmdprefix unbridge` first if you want to link it to a different channel.")
  620. return
  621. }
  622. var channelID string
  623. var unbridgeOld, deleteOld bool
  624. fail := true
  625. for _, arg := range ce.Args {
  626. arg = strings.ToLower(arg)
  627. if arg == "--replace" {
  628. unbridgeOld = true
  629. } else if arg == "--replace=delete" {
  630. unbridgeOld = true
  631. deleteOld = true
  632. } else if channelID == "" && isNumber(arg) {
  633. channelID = arg
  634. fail = false
  635. } else {
  636. fail = true
  637. break
  638. }
  639. }
  640. if fail {
  641. ce.Reply("**Usage**: `$cmdprefix bridge [--replace[=delete]] <channel ID>`")
  642. return
  643. }
  644. portal := ce.User.GetExistingPortalByID(channelID)
  645. if portal == nil {
  646. ce.Reply("Channel not found")
  647. return
  648. }
  649. portal.roomCreateLock.Lock()
  650. defer portal.roomCreateLock.Unlock()
  651. if portal.MXID != "" {
  652. hasUnbridgePermission := ce.User.PermissionLevel >= bridgeconfig.PermissionLevelAdmin
  653. if !hasUnbridgePermission {
  654. levels, err := portal.MainIntent().PowerLevels(portal.MXID)
  655. if errors.Is(err, mautrix.MNotFound) {
  656. ce.ZLog.Debug().Err(err).Msg("Got M_NOT_FOUND trying to get power levels to check if user can unbridge it, assuming the room is gone")
  657. hasUnbridgePermission = true
  658. } else if err != nil {
  659. ce.ZLog.Warn().Err(err).Msg("Failed to check room power levels")
  660. ce.Reply("Failed to get power levels in old room to see if you're allowed to unbridge it")
  661. return
  662. } else {
  663. hasUnbridgePermission = levels.GetUserLevel(ce.User.GetMXID()) >= levels.GetEventLevel(roomModerator)
  664. }
  665. }
  666. if !unbridgeOld || !hasUnbridgePermission {
  667. extraHelp := "Rerun the command with `--replace` or `--replace=delete` to unbridge the old room."
  668. if !hasUnbridgePermission {
  669. extraHelp = "Additionally, you do not have the permissions to unbridge the old room."
  670. }
  671. ce.Reply("That channel is already bridged to [%s](https://matrix.to/#/%s). %s", portal.Name, portal.MXID, extraHelp)
  672. return
  673. }
  674. ce.ZLog.Debug().
  675. Str("old_room_id", portal.MXID.String()).
  676. Bool("delete", deleteOld).
  677. Msg("Unbridging old room")
  678. portal.removeFromSpace()
  679. portal.cleanup(!deleteOld)
  680. portal.RemoveMXID()
  681. ce.ZLog.Info().
  682. Str("old_room_id", portal.MXID.String()).
  683. Bool("delete", deleteOld).
  684. Msg("Unbridged old room to make space for new bridge")
  685. }
  686. if portal.Guild != nil && portal.Guild.BridgingMode < database.GuildBridgeIfPortalExists {
  687. ce.ZLog.Debug().Str("guild_id", portal.Guild.ID).Msg("Bumping bridging mode of portal guild to if-portal-exists")
  688. portal.Guild.BridgingMode = database.GuildBridgeIfPortalExists
  689. portal.Guild.Update()
  690. }
  691. ce.ZLog.Debug().Str("channel_id", portal.Key.ChannelID).Msg("Bridging room")
  692. portal.MXID = ce.RoomID
  693. portal.bridge.portalsLock.Lock()
  694. portal.bridge.portalsByMXID[portal.MXID] = portal
  695. portal.bridge.portalsLock.Unlock()
  696. portal.updateRoomName()
  697. portal.updateRoomAvatar()
  698. portal.updateRoomTopic()
  699. portal.updateSpace()
  700. portal.UpdateBridgeInfo()
  701. state, err := portal.MainIntent().State(portal.MXID)
  702. if err != nil {
  703. ce.ZLog.Error().Err(err).Msg("Failed to update state cache for room")
  704. } else {
  705. encryptionEvent, isEncrypted := state[event.StateEncryption][""]
  706. portal.Encrypted = isEncrypted && encryptionEvent.Content.AsEncryption().Algorithm == id.AlgorithmMegolmV1
  707. }
  708. portal.Update()
  709. ce.Reply("Room successfully bridged")
  710. ce.ZLog.Info().
  711. Str("channel_id", portal.Key.ChannelID).
  712. Bool("encrypted", portal.Encrypted).
  713. Msg("Manual bridging complete")
  714. }
  715. var cmdUnbridge = &commands.FullHandler{
  716. Func: wrapCommand(fnUnbridge),
  717. Name: "unbridge",
  718. Help: commands.HelpMeta{
  719. Section: HelpSectionPortalManagement,
  720. Description: "Unbridge this room from the linked Discord channel",
  721. },
  722. RequiresPortal: true,
  723. RequiresEventLevel: roomModerator,
  724. }
  725. var cmdDeletePortal = &commands.FullHandler{
  726. Func: wrapCommand(fnUnbridge),
  727. Name: "delete-portal",
  728. Help: commands.HelpMeta{
  729. Section: HelpSectionPortalManagement,
  730. Description: "Unbridge this room and kick all Matrix users",
  731. },
  732. RequiresPortal: true,
  733. RequiresEventLevel: roomModerator,
  734. }
  735. func fnUnbridge(ce *WrappedCommandEvent) {
  736. ce.Portal.roomCreateLock.Lock()
  737. defer ce.Portal.roomCreateLock.Unlock()
  738. ce.Portal.removeFromSpace()
  739. ce.Portal.cleanup(ce.Command == "unbridge")
  740. ce.Portal.RemoveMXID()
  741. }
  742. var cmdDeleteAllPortals = &commands.FullHandler{
  743. Func: wrapCommand(fnDeleteAllPortals),
  744. Name: "delete-all-portals",
  745. Help: commands.HelpMeta{
  746. Section: commands.HelpSectionAdmin,
  747. Description: "Delete all portals.",
  748. },
  749. RequiresAdmin: true,
  750. }
  751. func fnDeleteAllPortals(ce *WrappedCommandEvent) {
  752. portals := ce.Bridge.GetAllPortals()
  753. guilds := ce.Bridge.GetAllGuilds()
  754. if len(portals) == 0 && len(guilds) == 0 {
  755. ce.Reply("Didn't find any portals")
  756. return
  757. }
  758. leave := func(mxid id.RoomID, intent *appservice.IntentAPI) {
  759. if len(mxid) > 0 {
  760. _, _ = intent.KickUser(mxid, &mautrix.ReqKickUser{
  761. Reason: "Deleting portal",
  762. UserID: ce.User.MXID,
  763. })
  764. }
  765. }
  766. customPuppet := ce.Bridge.GetPuppetByCustomMXID(ce.User.MXID)
  767. if customPuppet != nil && customPuppet.CustomIntent() != nil {
  768. intent := customPuppet.CustomIntent()
  769. leave = func(mxid id.RoomID, _ *appservice.IntentAPI) {
  770. if len(mxid) > 0 {
  771. _, _ = intent.LeaveRoom(mxid)
  772. _, _ = intent.ForgetRoom(mxid)
  773. }
  774. }
  775. }
  776. ce.Reply("Found %d channel portals and %d guild portals, deleting...", len(portals), len(guilds))
  777. for _, portal := range portals {
  778. portal.Delete()
  779. leave(portal.MXID, portal.MainIntent())
  780. }
  781. for _, guild := range guilds {
  782. guild.Delete()
  783. leave(guild.MXID, ce.Bot)
  784. }
  785. ce.Reply("Finished deleting portal info. Now cleaning up rooms in background. You'll have to restart the bridge or relogin before rooms can be bridged again.")
  786. go func() {
  787. for _, portal := range portals {
  788. portal.cleanup(false)
  789. }
  790. ce.Reply("Finished background cleanup of deleted portal rooms.")
  791. }()
  792. }