click the play button works

microsoft · Nov 19, 2024 · be40fb2 · be40fb2
1 parent d81fd81
commit be40fb2
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/src/cascadia/QueryExtension/ExtensionPalette.cpp b/src/cascadia/QueryExtension/ExtensionPalette.cpp
@@ -207,6 +207,10 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         std::vector<IInspectable> messageParts;
 
         const auto chatMsg = winrt::make<ChatMessage>(winrt::to_hstring(response.Message()), false, false);
+        chatMsg.RunCommandClicked([this](auto&&, const auto commandlines) {
+            _InputSuggestionRequestedHandlers(*this, commandlines);
+            _close();
+        });
         messageParts.push_back(chatMsg);
 
         const auto brandingData = _lmProvider ? _lmProvider.BrandingData() : nullptr;
@@ -496,6 +500,9 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
                                 {
                                     codeBlock.Foreground(foregroundBrush);
                                 }
+                                codeBlock.RequestRunCommands([this, commandlines = codeBlock.Commandlines()](auto&&, auto&&) {
+                                    _RunCommandClickedHandlers(*this, commandlines);
+                                });
                             }
                         }
                     }

diff --git a/src/cascadia/QueryExtension/ExtensionPalette.h b/src/cascadia/QueryExtension/ExtensionPalette.h
@@ -70,6 +70,8 @@ namespace winrt::Microsoft::Terminal::Query::Extension::implementation
         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:
         bool _isQuery;
         bool _isCode;

diff --git a/src/cascadia/QueryExtension/ExtensionPalette.idl b/src/cascadia/QueryExtension/ExtensionPalette.idl
@@ -12,6 +12,7 @@ namespace Microsoft.Terminal.Query.Extension
         Boolean IsQuery { get; };
         Boolean IsCode { get; };
         Windows.UI.Xaml.Controls.RichTextBlock RichBlock { get; };
+        event Windows.Foundation.TypedEventHandler<ChatMessage, String> RunCommandClicked;
     }
 
     runtimeclass GroupedChatMessages : Windows.Foundation.Collections.IVector<IInspectable>