[Patch agent] Skip workspace selection dialog for disconnected

automatic workspaces

Bug: 399560823
Change-Id: I1656bffcec9afd4d3237d00bb5465b003eab67ff
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6446876
Auto-Submit: Wolfgang Beyer <wolfi@chromium.org>
Reviewed-by: Alex Rudenko <alexrudenko@chromium.org>
Commit-Queue: Alex Rudenko <alexrudenko@chromium.org>

Author: wolfib

front_end/panels/ai_assistance/PatchWidget.test.ts

@@ -278,6 +278,27 @@ Files:
       // Assert that the project has been updated
       assert.strictEqual(input.projectName, 'test2');
     });
+
+    it('preselects an automatic workspace folder and allows connecting to it', async () => {
+      setupAutomaticFileSystem({hasFileSystem: true});
+      const showSelectWorkspaceDialogSpy = sinon.spy(AiAssistance.SelectWorkspaceDialog, 'show');
+      const applyChangesSpy = sinon.spy(AiAssistanceModel.PatchAgent.prototype, 'applyChanges');
+
+      const {view, widget} = await createPatchWidget();
+      widget.changeSummary = 'body { background-color: red; }';
+      assert.strictEqual(view.input.projectName, 'my-automatic-file-system');
+      assert.strictEqual(view.input.projectPath, '/path/to/my-automatic-file-system');
+
+      // Clicking on "Apply to workspace" does not open a SelectWorkspaceDialog
+      view.input.onApplyToWorkspace();
+      assert.isTrue(showSelectWorkspaceDialogSpy.notCalled);
+      await new Promise(resolve => setTimeout(resolve, 0));
+      assert.isTrue(applyChangesSpy.notCalled);
+
+      // This simulates the user confirming the file permissions dialog
+      createTestFilesystem('file:///path/to/my-automatic-file-system');
+      assert.isTrue(applyChangesSpy.calledOnce);
+    });
   });
 
   describe('diff view', () => {

front_end/panels/ai_assistance/PatchWidget.ts

@@ -209,6 +209,7 @@ export class PatchWidget extends UI.Widget.Widget {
   #workspace = Workspace.Workspace.WorkspaceImpl.instance();
   #automaticFileSystem =
       Persistence.AutomaticFileSystemManager.AutomaticFileSystemManager.instance().automaticFileSystem;
+  #applyToDisconnectedAutomaticWorkspace = false;
 
   constructor(element?: HTMLElement, view?: View, opts?: {
     aidaClient: Host.AidaClient.AidaClient,
@@ -518,6 +519,7 @@ export class PatchWidget extends UI.Widget.Widget {
   }
 
   override willHide(): void {
+    this.#applyToDisconnectedAutomaticWorkspace = false;
     if (isAiAssistancePatchingEnabled()) {
       this.#workspace.removeEventListener(Workspace.Workspace.Events.ProjectAdded, this.#onProjectAdded, this);
       this.#workspace.removeEventListener(Workspace.Workspace.Events.ProjectRemoved, this.#onProjectRemoved, this);
@@ -581,8 +583,14 @@ export class PatchWidget extends UI.Widget.Widget {
     this.requestUpdate();
   }
 
-  #onProjectAdded(): void {
-    if (this.#project === undefined) {
+  #onProjectAdded(event: Common.EventTarget.EventTargetEvent<Workspace.Workspace.Project>): void {
+    const addedProject = event.data;
+    if (this.#applyToDisconnectedAutomaticWorkspace && this.#automaticFileSystem &&
+        addedProject === this.#workspace.projectForFileSystemRoot(this.#automaticFileSystem.root)) {
+      this.#applyToDisconnectedAutomaticWorkspace = false;
+      this.#project = addedProject;
+      void this.#applyPatchAndUpdateUI();
+    } else if (this.#project === undefined) {
       this.#selectDefaultProject();
     }
   }
@@ -623,6 +631,10 @@ export class PatchWidget extends UI.Widget.Widget {
 
     if (this.#project) {
       await this.#applyPatchAndUpdateUI();
+    } else if (this.#automaticFileSystem) {
+      this.#applyToDisconnectedAutomaticWorkspace = true;
+      await Persistence.AutomaticFileSystemManager.AutomaticFileSystemManager.instance().connectAutomaticFileSystem(
+          /* addIfMissing= */ true);
     } else {
       this.#showSelectWorkspaceDialog({applyPatch: true});
     }