🎨 ASF 增加一键下载功能

Author: jinzaz

src/BD.WTTS.Client.Plugins.ArchiSteamFarmPlus/Services.Implementation/ArchiSteamFarmServiceImpl.cs

@@ -69,7 +69,7 @@ public async Task ShellMessageInput(string data)
                 isFirstStart = false;
             }
 
-            (var isSuccess, var ipcUrl) = await StartProcess();
+            (var isSuccess, var ipcUrl) = await StartProcessAsync();
             if (!isSuccess && !isFirstStart)
                 throw new ArgumentException();
             else
@@ -85,7 +85,7 @@ public async Task ShellMessageInput(string data)
 
     #region StartProcess
 
-    private async Task<(bool IsSuccess, string Message)> StartProcess()
+    private async Task<(bool IsSuccess, string Message)> StartProcessAsync()
     {
         var isStartSuccess = false;
         var ipcUrl = string.Empty;
@@ -306,7 +306,7 @@ public async Task StopAsync()
         }
     }
 
-    public async Task Restart()
+    public async Task RestartAsync()
     {
         Toast.Show(ToastIcon.Info, AppResources.ASF_Restarting, ToastLength.Short);
 

src/BD.WTTS.Client.Plugins.ArchiSteamFarmPlus/Services/Mvvm/ASFService.cs

@@ -47,7 +47,7 @@ private ASFService()
     {
         SteamBotsSourceList = new SourceCache<BotViewModel, string>(t => t.Bot.BotName);
 
-        archiSteamFarmService.OnConsoleWirteLine += OnConsoleWirteLine;
+        archiSteamFarmService.OnConsoleWirteLine += OnConsoleWriteLine;
 
         ASFSettings.ConsoleMaxLine.Subscribe(x =>
         {
@@ -58,9 +58,9 @@ private ASFService()
         });
     }
 
-    void OnConsoleWirteLine(string message)
+    void OnConsoleWriteLine(string message)
     {
-        MainThread2.InvokeOnMainThreadAsync(() =>
+        MainThread2.BeginInvokeOnMainThread(() =>
         {
             //ConsoleLogBuilder.Append(message); // message 包含换行
             //var text = ConsoleLogBuilder.ToString();
@@ -74,7 +74,7 @@ public void ShellMessageInput(string? input)
         {
             return;
         }
-        archiSteamFarmService.ShellMessageInput(input);
+        _ = archiSteamFarmService.ShellMessageInput(input);
     }
 
     /// <summary>
@@ -287,7 +287,7 @@ async Task ImportJsonFileAsync(IEnumerable<string?>? files, bool allowBot, bool
         Toast.Show(ToastIcon.Success, string.Format(AppResources.LocalAuth_ImportSuccessTip_, num));
     }
 
-    public async Task SelectASFProgramLocation()
+    public async Task SelectASFProgramLocationAsync()
     {
         AvaloniaFilePickerFileTypeFilter fileTypes = new AvaloniaFilePickerFileTypeFilter.Item[] {
             new("ArchiSteamFarm") {
@@ -302,5 +302,77 @@ await FilePicker2.PickAsync((path) =>
                 ASFSettings.ArchiSteamFarmExePath.Value = path;
         }, fileTypes);
     }
+
+    public async void DownloadASFAsync(string variant = "win-x64", IProgress<float>? progress = null, CancellationToken cancellationToken = default)
+    {
+        try
+        {
+            var assetName = $"{nameof(ASF)}-{variant}.zip";
+            using var httpClient = new HttpClient();
+            httpClient.DefaultRequestHeaders.TryAddWithoutValidation("User-Agent", IHttpPlatformHelperService.Instance.UserAgent);
+            using var latestReleaseStream = await httpClient.GetStreamAsync("https://api.github.com/repos/JustArchiNET/ArchiSteamFarm/releases/latest", cancellationToken);
+            using var element = await JsonDocument.ParseAsync(latestReleaseStream, cancellationToken: cancellationToken);
+            if (element.RootElement.TryGetProperty("assets", out var assets) &&
+                assets.EnumerateArray().FirstOrDefault(s => s.GetProperty("name").ValueEquals(assetName))
+                .TryGetProperty("browser_download_url", out var downloadUrl))
+            {
+                var tag_name = element.RootElement.GetProperty("tag_name").GetString();
+                var downloadSavingPath = Path.Combine(Plugin.Instance.AppDataDirectory, $"ASF-{tag_name}", assetName);
+                var downloadSavingDir = Path.GetDirectoryName(downloadSavingPath)!;
+                Directory.CreateDirectory(downloadSavingDir);
+
+                var message = new HttpRequestMessage(HttpMethod.Get, downloadUrl.GetString());
+                var result = await httpClient.SendAsync(message, HttpCompletionOption.ResponseHeadersRead, cancellationToken);
+                using var contentStream = await result.Content.ReadAsStreamAsync(cancellationToken);
+
+                byte batch = 0;
+                long readThisBatch = 0;
+                long batchIncreaseSize = result.Content.Headers.ContentLength.GetValueOrDefault() / 100;
+                await using (FileStream fileStream = new(downloadSavingPath, FileMode.Create, FileAccess.Write, FileShare.None, 4096, true))
+                {
+                    ArrayPool<byte> bytePool = ArrayPool<byte>.Shared;
+                    byte[] buffer = bytePool.Rent(4096);
+
+                    try
+                    {
+                        while (contentStream.CanRead)
+                        {
+                            int read = await contentStream.ReadAsync(buffer.AsMemory(0, buffer.Length), cancellationToken);
+
+                            if (read == 0)
+                                break;
+
+                            await fileStream.WriteAsync(buffer.AsMemory(0, read), cancellationToken);
+                            if ((progress == null) || (batchIncreaseSize == 0) || (batch >= 99))
+                            {
+                                continue;
+                            }
+
+                            readThisBatch += read;
+
+                            while ((readThisBatch >= batchIncreaseSize) && (batch < 99))
+                            {
+                                readThisBatch -= batchIncreaseSize;
+                                progress.Report(++batch);
+                            }
+                        }
+                    }
+                    finally
+                    {
+                        bytePool.Return(buffer);
+                    }
+                }
+
+                ZipFile.ExtractToDirectory(downloadSavingPath, downloadSavingDir);
+                ASFSettings.ArchiSteamFarmExePath.Value = Path.Combine(downloadSavingDir, "ArchiSteamFarm.exe");
+                progress?.Report(100);
+                Toast.Show(ToastIcon.Success, "ASF 下载成功");
+            }
+        }
+        catch (Exception ex)
+        {
+            Toast.LogAndShowT(ex, "ASF 文件下载异常");
+        }
+    }
 }
 #endif

src/BD.WTTS.Client.Plugins.ArchiSteamFarmPlus/UI/ViewModels/ArchiSteamFarmExePathSettingsPageViewModel.cs

@@ -4,8 +4,16 @@ public partial class ArchiSteamFarmExePathSettingsPageViewModel : ViewModelBase
 {
     public ICommand SelectProgramPath { get; }
 
+    public ICommand DownloadASF { get; }
+
     public ArchiSteamFarmExePathSettingsPageViewModel()
     {
-        SelectProgramPath = ReactiveCommand.Create(ASFService.Current.SelectASFProgramLocation);
+        SelectProgramPath = ReactiveCommand.Create(ASFService.Current.SelectASFProgramLocationAsync);
+        DownloadASF = ReactiveCommand.Create(() =>
+        {
+            var progress = INotificationService.Instance.NotifyDownload(() => "开始下载 ASF ", NotificationType.NewVersion);
+            ASFService.Current.DownloadASFAsync(progress: progress);
+            Toast.Show(ToastIcon.Info, "等待后台下载文件", ToastLength.Short);
+        });
     }
 }

src/BD.WTTS.Client.Plugins.ArchiSteamFarmPlus/UI/ViewModels/ArchiSteamFarmPlusPageViewModel.cs

@@ -9,7 +9,7 @@ public sealed partial class ArchiSteamFarmPlusPageViewModel : ViewModelBase
 
     public ArchiSteamFarmPlusPageViewModel()
     {
-        SelectASFExePath = ReactiveCommand.CreateFromTask(ASFService.Current.SelectASFProgramLocation);
+        SelectASFExePath = ReactiveCommand.CreateFromTask(ASFService.Current.SelectASFProgramLocationAsync);
 
         OpenWebUIConsole = ReactiveCommand.Create(() => ASFService.Current.OpenBrowser(null));
 
@@ -18,12 +18,12 @@ public ArchiSteamFarmPlusPageViewModel()
         RunOrStop = ReactiveCommand.Create(RunOrStopASF);
     }
 
-    public override void Activation()
+    public override async void Activation()
     {
         base.Activation();
         if (string.IsNullOrEmpty(ASFSettings.ArchiSteamFarmExePath.Value))
         {
-            MainThread2.InvokeOnMainThreadAsync(async () =>
+            await MainThread2.InvokeOnMainThreadAsync(async () =>
             {
                 var model = new ArchiSteamFarmExePathSettingsPageViewModel();
                 await IWindowManager.Instance.ShowTaskDialogAsync(model,
@@ -44,7 +44,16 @@ await IWindowManager.Instance.ShowTaskDialogAsync(model,
             StartOrStopASF_Click();
     }
 
-    public static void StartOrStopASF_Click(bool? startOrStop = null) => Task.Run(async () =>
+    protected override void Dispose(bool disposing)
+    {
+        if (disposing)
+        {
+            ASFService.Current.StopASFAsync().Wait();
+        }
+        base.Dispose(disposing);
+    }
+
+    public void StartOrStopASF_Click(bool? startOrStop = null) => _ = Task.Run(async () =>
     {
         var s = ASFService.Current;
         if (!s.IsASFRuning)

src/BD.WTTS.Client.Plugins.ArchiSteamFarmPlus/UI/Views/Controls/ArchiSteamFarmExePathSettingsPage.axaml

@@ -26,5 +26,16 @@
                 <Button Command="{Binding SelectProgramPath}" Content="选择路径" />
             </ui:SettingsExpander.Footer>
         </ui:SettingsExpander>
+        <ui:SettingsExpander
+            Description="下载并设置 ASF 路径"
+            Header="还没有下载过ASF？请点击下载"
+            IsClickEnabled="True">
+            <ui:SettingsExpander.IconSource>
+                <ui:FontIconSource FontFamily="{StaticResource SymbolThemeFontFamily}" Glyph="&#xE896;" />
+            </ui:SettingsExpander.IconSource>
+            <ui:SettingsExpander.Footer>
+                <Button Command="{Binding DownloadASF}" Content="一键下载" />
+            </ui:SettingsExpander.Footer>
+        </ui:SettingsExpander>
     </StackPanel>
 </UserControl>