Unfurling Discord Message Links

docker-compose.yml

@@ -22,6 +22,7 @@ services:
         - SLACK_TEAM_ID
         - SLACK_REDIRECT_URI
         - SLACK_BOT_USER_OAUTH_TOKEN
+        - DISCORD_BOT_TOKEN
         - SENTRY_BACKEND_DSN
         - SENTRY_FRONTEND_DSN
         - SENTRY_AUTH_TOKEN

potal/handler.go

@@ -223,6 +223,32 @@ func EventsHandler(w http.ResponseWriter, r *http.Request, _ httprouter.Params)
 					return
 				}
 
+				txn.Status = sentry.SpanStatusOK
+			}()
+		case *slackevents.LinkSharedEvent:
+			go event.ProcessLinkSharedEvent(r.Context(), ev)
+			go func() {
+				hub := sentry.CurrentHub().Clone()
+				ctx := sentry.SetHubOnContext(context.Background(), hub)
+
+				options := []sentry.SpanOption{
+					sentry.WithOpName("event.handler"),
+					sentry.WithTransactionSource(sentry.SourceTask),
+					sentry.ContinueFromHeaders(transaction.ToSentryTrace(), transaction.ToBaggage()),
+				}
+				txn := sentry.StartTransaction(ctx, "EVENT link_shared", options...)
+				defer txn.Finish()
+
+				processedEvent := event.ProcessLinkSharedEvent(txn.Context(), ev)
+				if processedEvent == nil {
+					return
+				}
+				err := potalhttp.SendRequest(txn.Context(), processedEvent)
+				if err != nil {
+					txn.Status = sentry.SpanStatusInternalError
+					return
+				}
+
 				txn.Status = sentry.SpanStatusOK
 			}()
 		}

potal/internal/event/event.go

@@ -14,6 +14,7 @@ const (
 	appHomeOpened       PotalEventType = "app_home_opened"
 	slashCommand        PotalEventType = "slash_command"
 	interactionCallback PotalEventType = "interaction_callback"
+	linkShared          PotalEventType = "link_shared"
 )
 
 func (e PotalEventType) String() string {

potal/internal/event/linkshared.go

@@ -0,0 +1,66 @@
+package event
+
+import (
+	"context"
+
+	"github.com/getsentry/sentry-go"
+	"github.com/slack-go/slack/slackevents"
+)
+
+type LinkSharedEvent struct {
+	Type           PotalEventType `json:"type"`
+	User           string         `json:"user"`
+	TimeStamp      string         `json:"ts"`
+	Channel        string         `json:"channel"`
+	MessageTimeStamp string        `json:"message_ts"`
+	ThreadTimeStamp  string        `json:"thread_ts"`
+	Links            []Link `json:"links"`
+	EventTimestamp   string        `json:"event_ts"`
+}
+
+type Link struct {
+	Domain string `json:"domain"`
+	URL    string `json:"url"`
+}
+
+func (e LinkSharedEvent) isValid() bool {
+	return len(e.Links) > 0
+}
+
+func ProcessLinkSharedEvent(ctx context.Context, e *slackevents.LinkSharedEvent) *LinkSharedEvent {
+	hub := sentry.GetHubFromContext(ctx)
+	txn := sentry.TransactionFromContext(ctx)
+
+	span := txn.StartChild("event.process", sentry.WithDescription("Process LinkShared Event"))
+	defer span.Finish()
+
+	links := make([]Link, len(e.Links))
+	for i, link := range e.Links {
+		links[i] = Link{
+			Domain: link.Domain,
+			URL:    link.URL,
+		}
+	}
+
+	linkSharedEvent := LinkSharedEvent{
+		Type:           linkShared,
+		User:           e.User,
+		TimeStamp:      e.TimeStamp,
+		Channel:        e.Channel,
+		MessageTimeStamp: e.MessageTimeStamp,
+		ThreadTimeStamp:  e.ThreadTimeStamp,
+		Links:            links,
+		EventTimestamp:   e.EventTimestamp,
+	}
+
+	if !linkSharedEvent.isValid() {
+		span.Status = sentry.SpanStatusInvalidArgument
+		return nil
+	}
+	span.Status = sentry.SpanStatusOK
+
+	hub.Scope().SetExtra("event", linkSharedEvent)
+	hub.Scope().SetTag("event_type", linkShared.String())
+
+	return &linkSharedEvent
+}

src/Event/AbstractEvent.php

@@ -3,6 +3,7 @@
 
 namespace App\Event;
 
+use App\Http\DiscordClient;
 use App\Http\SlackClient;
 use Cake\ORM\Locator\LocatorAwareTrait;
 
@@ -17,8 +18,10 @@ abstract class AbstractEvent
     public const TYPE_APP_MENTION = 'app_mention';
     public const TYPE_SLASH_COMMAND = 'slash_command';
     public const TYPE_INTERACTIONS_CALLBACK = 'interaction_callback';
+    public const TYPE_LINK_SHARED = 'link_shared';
 
     protected SlackClient $slackClient;
+    protected DiscordClient $discordClient;
 
     public string $type;
     public string $eventTimestamp;
@@ -29,6 +32,7 @@ abstract class AbstractEvent
     public function __construct()
     {
         $this->slackClient = new SlackClient();
+        $this->discordClient = new DiscordClient();
     }
 
     /**

src/Event/EventFactory.php

@@ -43,6 +43,8 @@
                 return new SlashCommandEvent($data);
             case AbstractEvent::TYPE_INTERACTIONS_CALLBACK:
                 return new InteractionsCallbackEvent($data);
+            case AbstractEvent::TYPE_LINK_SHARED:
+                return new LinkSharedEvent($data);
             default:
                 throw new Exception('Unknown event type');
         }

src/Event/LinkSharedEvent.php

@@ -0,0 +1,85 @@
+<?php
+declare(strict_types=1);
+
+namespace App\Event;
+
+class LinkSharedEvent extends AbstractEvent
+{
+    public string $user;
+    public string $timestamp;
+    public string $channel;
+    public string $messageTimeStamp;
+    public string $threadTimeStamp;
+    public array $links;
+    public string $eventTimestamp;
+
+    /**
+     * Constructor
+     *
+     * @param array $event Event data.
+     */
+    public function __construct(array $event)
+    {
+        parent::__construct();
+
+        $this->type = self::TYPE_LINK_SHARED;
+        $this->user = $event['user'];
+        $this->timestamp = $event['ts'];
+        $this->channel = $event['channel'];
+        $this->messageTimeStamp = $event['message_ts'];
+        $this->threadTimeStamp = $event['thread_ts'];
+        $this->eventTimestamp = $event['event_ts'];
+        $this->links = $event['links'];
+    }
+
+    /**
+     * @inheritDoc
+     */
+    public function process(): void
+    {
+        foreach ($this->links as $link) {
+            // if Discord link, unfurl and fetch the message and return it
+            if (str_starts_with($link['url'], 'https://discord.com/')) {
+                $message = $this->fetchDiscordMessage($link['url']);
+                $this->slackClient->unfurl(
+                    channel: $this->channel,
+                    timestamp: $this->messageTimeStamp,
+                    unfurls: [
+                        $link['url'] => [
+                            'blocks' => [
+                                [
+                                    'type' => 'section',
+                                    'text' => [
+                                        'type' => 'mrkdwn',
+                                        'text' => $message,
+                                    ],
+                                ],
+                            ],
+                        ],
+                    ],
+                );
+            }
+        }
+    }
+
+    /**
+     * Fetch a Discord message
+     *
+     * @param string $url The Discord message URL
+     * @return string The message content
+     */
+    private function fetchDiscordMessage(string $url): string
+    {
+        if (!preg_match('#^https://discord\.com/channels/\d+/\d+/\d+$#', $url)) {
+            return $url;
+        }
+
+        $parts = explode('/', $url);
+        $channelId = $parts[5];
+        $messageId = $parts[6];
+
+        $message = $this->discordClient->getMessage($channelId, $messageId);
+
+        return $message;
+    }
+}

src/Http/DiscordClient.php

@@ -0,0 +1,59 @@
+<?php
+declare(strict_types=1);
+
+namespace App\Http;
+
+use function Cake\Core\env;
+use function Sentry\captureMessage;
+use function Sentry\withScope;
+
+class DiscordClient
+{
+    protected const DISCORD_API_URL = 'discord.com/api/v10';
+
+    protected Client $client;
+
+    /**
+     * Constructor.
+     */
+    public function __construct()
+    {
+        $this->client = new Client([
+            'host' => self::DISCORD_API_URL,
+            'scheme' => 'https',
+            'headers' => [
+                'Authorization' => 'Bot ' . env('DISCORD_BOT_TOKEN'),
+                'Content-Type' => 'application/json',
+            ],
+        ]);
+    }
+
+    /**
+     * Get a message from a channel
+     *
+     * @param string $channelId The channel ID
+     * @param string $messageId The message ID
+     * @return string The message content
+     */
+    public function getMessage(string $channelId, string $messageId): string
+    {
+        $response = $this->client->get("channels/{$channelId}/messages/{$messageId}");
+
+        if ($response->isSuccess()) {
+            $json = $response->getJson();
+
+            return $json['content'] ?? '';
+        }
+
+        withScope(function ($scope) use ($response, $channelId, $messageId): void {
+            $scope->setContext('Discord API', [
+                'channel_id' => $channelId,
+                'message_id' => $messageId,
+                'discord_response' => $response->getJson(),
+            ]);
+            captureMessage('Discord API error: Failed to fetch message');
+        });
+
+        return '';
+    }
+}

src/Http/SlackClient.php

@@ -207,4 +207,34 @@
 
         return [];
     }
+
+    /**
+     * @param string $channel The channel to unfurl the links in.
+     * @param string $timestamp The timestamp of the message to unfurl the links in.
+     * @param array $unfurls The unfurls to send.
+     * @return void
+     * @see https://api.slack.com/methods/chat.unfurl
+     */
+    public function unfurl(string $channel, string $timestamp, array $unfurls): void
+    {
+        $response = $this->client->post('chat.unfurl', [
+            'channel' => $channel,
+            'ts' => $timestamp,
+            'unfurls' => json_encode($unfurls),
+        ]);
+        if ($response->isSuccess()) {
+            $json = $response->getJson();
+            if ($json['ok'] === false) {
+                withScope(function ($scope) use ($json, $channel, $timestamp, $unfurls): void {
+                    $scope->setContext('Slack API', [
+                        'channel' => $channel,
+                        'timestamp' => $timestamp,
+                        'unfurls' => $unfurls,
+                        'slack_response' => $json,
+                    ]);
+                    captureMessage('Slack API error: https://api.slack.com/methods/chat.unfurl');
+                });
+            }
+        }
+    }
 }