feature: add Use back to apply ai generated commit message (#2287)

Signed-off-by: leo <longshuang@msn.cn>

Author: love-linger

src/Resources/Locales/zh_CN.axaml

@@ -25,7 +25,7 @@
   <x:String x:Key="Text.AIAssistant.Model" xml:space="preserve">模型</x:String>
   <x:String x:Key="Text.AIAssistant.Regen" xml:space="preserve">重新生成</x:String>
   <x:String x:Key="Text.AIAssistant.Tip" xml:space="preserve">使用AI助手生成提交信息</x:String>
-  <x:String x:Key="Text.AIAssistant.Use" xml:space="preserve">应用所选</x:String>
+  <x:String x:Key="Text.AIAssistant.Use" xml:space="preserve">应用</x:String>
   <x:String x:Key="Text.App.Hide" xml:space="preserve">隐藏 SourceGit</x:String>
   <x:String x:Key="Text.App.HideOthers" xml:space="preserve">隐藏其他</x:String>
   <x:String x:Key="Text.App.ShowAll" xml:space="preserve">显示全部</x:String>

src/Resources/Locales/zh_TW.axaml

@@ -25,7 +25,7 @@
   <x:String x:Key="Text.AIAssistant.Model" xml:space="preserve">模型</x:String>
   <x:String x:Key="Text.AIAssistant.Regen" xml:space="preserve">重新產生</x:String>
   <x:String x:Key="Text.AIAssistant.Tip" xml:space="preserve">使用 AI 產生提交訊息</x:String>
-  <x:String x:Key="Text.AIAssistant.Use" xml:space="preserve">套用選取</x:String>
+  <x:String x:Key="Text.AIAssistant.Use" xml:space="preserve">套用</x:String>
   <x:String x:Key="Text.App.Hide" xml:space="preserve">隱藏 SourceGit</x:String>
   <x:String x:Key="Text.App.HideOthers" xml:space="preserve">隱藏其他</x:String>
   <x:String x:Key="Text.App.ShowAll" xml:space="preserve">顯示全部</x:String>

src/ViewModels/AIAssistant.cs

@@ -33,6 +33,12 @@ public string Text
             private set => SetProperty(ref _text, value);
         }
 
+        public string Response
+        {
+            get => _response;
+            private set => SetProperty(ref _response, value);
+        }
+
         public AIAssistant(Repository repo, AI.Service service, List<Models.Change> changes)
         {
             _repo = repo;
@@ -55,16 +61,35 @@ public async Task GenAsync()
             var builder = new StringBuilder();
             builder.AppendLine("Asking AI to generate commit message...").AppendLine();
 
+            var responseBuilder = new StringBuilder();
+            var foundResponse = false;
+
             Text = builder.ToString();
+            Response = string.Empty;
             IsGenerating = true;
 
             try
             {
                 await agent.GenerateCommitMessageAsync(_repo.FullPath, _changeList, message =>
                 {
                     builder.AppendLine(message);
+
+                    if (foundResponse)
+                    {
+                        if (message.Equals("# Token Usage", StringComparison.Ordinal))
+                            foundResponse = false;
+                        else
+                            responseBuilder.AppendLine(message);
+                    }
+                    else if (message.Equals("# Assistant", StringComparison.Ordinal))
+                    {
+                        foundResponse = true;
+                    }
+
                     Dispatcher.UIThread.Post(() => Text = builder.ToString());
                 }, _cancel.Token);
+
+                Response = responseBuilder.ToString().Trim();
             }
             catch (OperationCanceledException)
             {
@@ -77,7 +102,9 @@ await agent.GenerateCommitMessageAsync(_repo.FullPath, _changeList, message =>
                     .AppendLine()
                     .AppendLine("[ERROR]")
                     .Append(e.Message);
+
                 Text = builder.ToString();
+                Response = string.Empty;
             }
 
             IsGenerating = false;
@@ -120,5 +147,6 @@ private void SerializeChange(Models.Change c, StringBuilder builder)
         private CancellationTokenSource _cancel = null;
         private bool _isGenerating = false;
         private string _text = string.Empty;
+        private string _response = string.Empty;
     }
 }

src/Views/AIAssistant.axaml

@@ -45,7 +45,7 @@
                       Content="{Binding Text}"/>
 
     <!-- Options -->
-    <Grid Grid.Row="2" Margin="8,0,8,8" ColumnDefinitions="Auto,*,18,Auto">
+    <Grid Grid.Row="2" Margin="8,0,8,8" ColumnDefinitions="Auto,*,18,Auto,Auto">
       <TextBlock Grid.Column="0"
                  Classes="group_header_label"
                  Text="{DynamicResource Text.AIAssistant.Model}"/>
@@ -62,10 +62,17 @@
 
       <v:LoadingIcon Grid.Column="2"
                      Width="14" Height="14"
-                     Margin="0,0,8,0"
                      IsVisible="{Binding IsGenerating}"/>
-      
+
       <Button Grid.Column="3"
+              Classes="flat"
+              Height="28"
+              Margin="8,0" Padding="12,0"
+              Content="{DynamicResource Text.AIAssistant.Use}"
+              IsEnabled="{Binding Response, Converter={x:Static StringConverters.IsNotNullOrEmpty}}"
+              Click="OnUseClicked"/>
+      
+      <Button Grid.Column="4"
               Classes="flat"
               Height="28"
               Padding="12,0"

src/Views/AIAssistant.axaml.cs

@@ -153,6 +153,15 @@ private async void OnModelChanged(object sender, SelectionChangedEventArgs e)
         {
             if (DataContext is ViewModels.AIAssistant vm && IsLoaded)
                 await vm.GenAsync();
+
+            e.Handled = true;
+        }
+
+        private void OnUseClicked(object sender, RoutedEventArgs e)
+        {
+            if (DataContext is ViewModels.AIAssistant vm && !string.IsNullOrEmpty(vm.Response))
+                vm.Use(vm.Response);
+
             e.Handled = true;
         }