Enable terminal chat message copy

src/cascadia/QueryExtension/ExtensionPalette.cpp

@@ -7,6 +7,8 @@
 #include "LibraryResources.h"
 #include <winrt/Windows.UI.Xaml.Media.Imaging.h>
 
+#include <winrt/Windows.ApplicationModel.DataTransfer.h>
+
 #include "ExtensionPalette.g.cpp"
 #include "ChatMessage.g.cpp"
 #include "GroupedChatMessages.g.cpp"
@@ -16,6 +18,7 @@ using namespace winrt::Windows::Foundation::Collections;
 using namespace winrt::Windows::UI::Core;
 using namespace winrt::Windows::UI::Xaml;
 using namespace winrt::Windows::UI::Xaml::Controls;
+using namespace winrt::Windows::UI::Xaml::Controls::Primitives;
 using namespace winrt::Windows::System;
 namespace WWH = ::winrt::Windows::Web::Http;
 namespace WSS = ::winrt::Windows::Storage::Streams;
@@ -340,6 +343,14 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
     void ExtensionPalette::_lostFocusHandler(const Windows::Foundation::IInspectable& /*sender*/,
                                              const Windows::UI::Xaml::RoutedEventArgs& /*args*/)
     {
+        const auto focusedElement = Input::FocusManager::GetFocusedElement(this->XamlRoot());
+        if (focusedElement && (focusedElement.try_as<RichTextBlock>() || focusedElement.try_as<MenuFlyoutPresenter>() || focusedElement.try_as<Popup>()))
+        {
+            // The context menu for the message don't seem to be found when the VisualTreeHelper walks the visual tree. So we check here
+            // if one of the focused elements is a message or a context menu of one of those messages and return early to support
+            // copy and select all using a mouse
+            return;
+        }
         const auto flyout = _queryBox().ContextFlyout();
         if (flyout && flyout.IsOpen())
         {
@@ -404,6 +415,13 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         }
         else if (key == VirtualKey::C && ctrlDown)
         {
+            // Get the focused element. If it is a chat message copy its selection (if any) to the clipboard.
+            const auto focusedElement = Input::FocusManager::GetFocusedElement(this->XamlRoot());
+            if (focusedElement && focusedElement.try_as<RichTextBlock>())
+            {
+                const auto textBlock = focusedElement.as<RichTextBlock>();
+                textBlock.CopySelectionToClipboard();
+            }
             _queryBox().CopySelectionToClipboard();
             e.Handled(true);
         }
@@ -440,6 +458,37 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         _richBlock{ nullptr }
     {
         _richBlock = Microsoft::Terminal::UI::Markdown::Builder::Convert(_messageContent, L"");
+        _richBlock.IsTextSelectionEnabled(true);
+        _richBlock.ContextMenuOpening([this](const Windows::Foundation::IInspectable& /*sender*/, const Windows::UI::Xaml::Controls::ContextMenuEventArgs& e) {
+            // If the context menu is opening we want to show our custom copy option
+            MenuFlyout menuFlyout;
+            Windows::UI::Xaml::Controls::FontIcon copyIcon;
+            copyIcon.Glyph(L"\uE8C8"); // Copy icon
+
+            auto copyItem = MenuFlyoutItem();
+            copyItem.Text(RS_(L"CopyMessage"));
+            copyItem.Icon(copyIcon);
+            copyItem.Click([this](auto&, auto&) {
+                const auto messageContent = this->MessageContent();
+
+                if (!messageContent.empty())
+                {
+                    // Create a DataPackage and set the content to the message content
+                    Windows::ApplicationModel::DataTransfer::DataPackage package{};
+                    package.SetText(messageContent);
+                    Windows::ApplicationModel::DataTransfer::Clipboard::SetContent(package);
+                }
+            });
+            menuFlyout.Items().Append(copyItem);
+
+            menuFlyout.ShouldConstrainToRootBounds(true);
+            menuFlyout.Placement(Windows::UI::Xaml::Controls::Primitives::FlyoutPlacementMode::BottomEdgeAlignedRight);
+            menuFlyout.ShowAt(_richBlock);
+            e.Handled(true);
+        });
+        _richBlock.AllowFocusWhenDisabled(true);
+        _richBlock.AllowFocusOnInteraction(true);
+
         const auto resources = Application::Current().Resources();
         const auto textBrushObj = _isQuery ? resources.Lookup(box_value(L"TextOnAccentFillColorPrimaryBrush")) : resources.Lookup(box_value(L"TextFillColorPrimaryBrush"));
         if (const auto textBrush = textBrushObj.try_as<Windows::UI::Xaml::Media::SolidColorBrush>())

src/cascadia/QueryExtension/ExtensionPalette.h

@@ -56,7 +56,6 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         void _previewKeyDownHandler(const Windows::Foundation::IInspectable& sender,
                                     const Windows::UI::Xaml::Input::KeyRoutedEventArgs& e);
         void _setUpAIProviderInSettings(const Windows::Foundation::IInspectable& sender, const Windows::UI::Xaml::RoutedEventArgs& args);
-
         void _close();
     };
 
@@ -67,7 +66,6 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         bool IsQuery() const { return _isQuery; };
         winrt::hstring MessageContent() const { return _messageContent; };
         winrt::Windows::UI::Xaml::Controls::RichTextBlock RichBlock() const { return _richBlock; };
-
         TYPED_EVENT(RunCommandClicked, winrt::Microsoft::Terminal::Query::Extension::ChatMessage, winrt::hstring);
 
     private:

src/cascadia/QueryExtension/Resources/en-US/Resources.resw

@@ -197,4 +197,8 @@
     <value>GitHub Copilot</value>
     <comment>The metadata string to display whenever a response is received from the GitHub Copilot service provider</comment>
   </data>
-</root>
+  <data name="CopyMessage" xml:space="preserve">
+    <value>Copy Message</value>
+    <comment>Copy terminal chat message</comment>
+  </data>
+</root>