Merge pull request #87 from LittleBigRefresh/patchwork-config

Add pipeline for reconfiguring Patchwork

Author: jvyden

Refresher.Core/Pipelines/Lbp/PatchworkPs3ConfigPipeline.cs

@@ -0,0 +1,16 @@
+using Refresher.Core.Pipelines.Steps;
+
+namespace Refresher.Core.Pipelines.Lbp;
+
+public class PatchworkPs3ConfigPipeline : Pipeline
+{
+    public override string Id => "patchwork-config-ps3";
+    public override string Name => "Patchwork PS3 Config";
+
+    protected override Type SetupAccessorStepType => typeof(SetupPS3AccessorStep);
+
+    protected override List<Type> StepTypes =>
+    [
+        typeof(UploadPatchworkConfigurationStep),
+    ];
+}

Refresher.Core/Pipelines/Pipeline.cs

@@ -202,6 +202,27 @@ public async Task<List<GameInformation>> DownloadGameListAsync(CancellationToken
         return this.GameList;
     }
 
+    public async Task ConnectAsync(CancellationToken cancellationToken = default)
+    {
+        if (this.State != PipelineState.NotStarted)
+        {
+            this.State = PipelineState.Error;
+            throw new InvalidOperationException("Pipeline must be in a clean state before downloading games.");
+        }
+        
+        if(this.SetupAccessorStepType == null)
+            throw new InvalidOperationException("This pipeline doesn't have accessors configured.");
+
+        List<Step> stepTypes = [
+            (Step)Activator.CreateInstance(this.SetupAccessorStepType, this)!,
+        ];
+        
+        this.State = PipelineState.Running;
+        
+        await this.RunListOfSteps(stepTypes, cancellationToken);
+        this.Reset();
+    }
+    
     public async Task RevertGameEbootAsync(CancellationToken cancellationToken = default)
     {
         if (this.State != PipelineState.NotStarted)

Refresher/UI/MainForm.cs

@@ -20,6 +20,7 @@ public class MainForm : RefresherForm
             new Label { Text = "LittleBigPlanet:" },
             this.PipelineButton<LbpPS3PatchPipeline>("Patch LBP1/2/3 for PS3"),
 
+            this.PipelineButton<PatchworkPs3ConfigPipeline>("Reconfigure Patch for PS3"),
             new Label { Text = "General (for non-LBP games):" },
             new Button((_, _) => this.ShowChild<FilePatchForm>()) { Text = "File Patch (using a .ELF)" },
             this.PipelineButton<RPCS3PatchPipeline>("Patch any RPCS3 game"),

Refresher/UI/PipelineForm.cs

@@ -140,10 +140,10 @@ private void InitializePipeline()
                 case StepInputType.Directory:
                     row = AddField<FilePicker>(input, value);
                     if (input.ShouldCauseGameDownloadWhenChanged)
-                        (row.Cells[1].Control as FilePicker)!.FilePathChanged += this.OnDownloadGameList;
+                        (row.Cells[1].Control as FilePicker)!.FilePathChanged += this.OnConnect;
                     break;
                 case StepInputType.ConsoleIp:
-                    row = AddField<TextBox>(input, value, this._connectButton = new Button(this.OnDownloadGameList) { Text = "Connect" });
+                    row = AddField<TextBox>(input, value, this._connectButton = new Button(this.OnConnect) { Text = "Connect" });
                     break;
                 default:
                     throw new ArgumentOutOfRangeException();
@@ -232,7 +232,7 @@ private void OnButtonClick(object? sender, EventArgs e)
         this.UpdateFormState();
     }
 
-    private void OnDownloadGameList(object? sender, EventArgs e)
+    private void OnConnect(object? sender, EventArgs e)
     {
         if (this._pipeline == null)
             return;
@@ -246,11 +246,20 @@ private void OnDownloadGameList(object? sender, EventArgs e)
         {
             try
             {
-                List<GameInformation> games = await this._pipeline!.DownloadGameListAsync();
-                await Application.Instance.InvokeAsync(() =>
+                if (this._gamesDropDown != null)
                 {
-                    this.HandleGameList(games);
-                });
+                    List<GameInformation> games = await this._pipeline!.DownloadGameListAsync();
+                    await Application.Instance.InvokeAsync(() =>
+                    {
+                        this.HandleConnection();
+                        this.HandleGameList(games);
+                    });
+                }
+                else
+                {
+                    await this._pipeline.ConnectAsync();
+                    await Application.Instance.InvokeAsync(this.HandleConnection);
+                }
             }
             catch (DirectoryNotFoundException)
             {
@@ -264,15 +273,17 @@ await Application.Instance.InvokeAsync(() =>
         });
     }
 
+    private void HandleConnection()
+    {
+        if (this._connectButton == null) return;
+
+        this._connectButton.Enabled = false;
+        this._connectButton.Text = "Connected!";
+    }
+
     private void HandleGameList(List<GameInformation> games)
     {
         Debug.Assert(this._gamesDropDown != null);
-    
-        if (this._connectButton != null)
-        {
-            this._connectButton.Enabled = false;
-            this._connectButton.Text = "Connected!";
-        }
 
         this._gamesDropDown.Items.Clear();