[PatchWidget] Handle discard and saveAll actions

Fixed: 399559699
Change-Id: I14073f0e3b8107ff4b632c0b87c7c90aeec1feab
Reviewed-on: https://chromium-review.googlesource.com/c/devtools/devtools-frontend/+/6348560
Reviewed-by: Wolfgang Beyer <[email protected]>
Reviewed-by: Alex Rudenko <[email protected]>
Commit-Queue: Ergün Erdoğmuş <[email protected]>

Author: ergunsh

front_end/panels/ai_assistance/PatchWidget.test.ts

@@ -9,6 +9,7 @@ import * as PanelCommon from '../../panels/common/common.js';
 import {
   cleanup,
   createPatchWidget,
+  createPatchWidgetWithDiffView,
   createTestFilesystem,
   initializePersistenceImplForTests,
   mockAidaClient,
@@ -176,4 +177,51 @@ Files:
       assert.strictEqual(input.projectName, 'test2');
     });
   });
+
+  describe('diff view', () => {
+    let commitWorkingCopyStub:
+        sinon.SinonStub<Parameters<typeof Workspace.UISourceCode.UISourceCode.prototype.commitWorkingCopy>>;
+    let resetWorkingCopyStub:
+        sinon.SinonStub<Parameters<typeof Workspace.UISourceCode.UISourceCode.prototype.resetWorkingCopy>>;
+
+    beforeEach(() => {
+      createTestFilesystem('file://test');
+      updateHostConfig({
+        devToolsFreestyler: {
+          enabled: true,
+          patching: true,
+        },
+      });
+
+      commitWorkingCopyStub =
+          sinon.stub(Workspace.UISourceCode.UISourceCode.prototype, 'commitWorkingCopy').callThrough();
+      resetWorkingCopyStub =
+          sinon.stub(Workspace.UISourceCode.UISourceCode.prototype, 'resetWorkingCopy').callThrough();
+    });
+
+    it('save all should commit the working copy of the changed UI codes to the disk and render savedToDisk view',
+       async () => {
+         const {uiSourceCode} = createTestFilesystem('file://test');
+         const {view} = await createPatchWidgetWithDiffView();
+
+         uiSourceCode.setWorkingCopy('working copy');
+         view.input.onSaveAll();
+         const nextInput = await view.nextInput;
+
+         assert.isTrue(nextInput.savedToDisk);
+         assert.isTrue(commitWorkingCopyStub.called, 'Expected commitWorkingCopy to be called but it is not called');
+       });
+
+    it('discard should discard the working copy and render the view without patchSuggestion', async () => {
+      const {uiSourceCode} = createTestFilesystem('file://test');
+      const {view} = await createPatchWidgetWithDiffView();
+      uiSourceCode.setWorkingCopy('working copy');
+
+      view.input.onDiscard();
+      const nextInput = await view.nextInput;
+
+      assert.notExists(nextInput.patchSuggestion);
+      assert.isTrue(resetWorkingCopyStub.called, 'Expected resetWorkingCopy to be called but it is not called');
+    });
+  });
 });

front_end/panels/ai_assistance/PatchWidget.ts

@@ -501,14 +501,21 @@ ${processedFiles.map(filename => `* ${filename}`).join('\n')}`;
   }
 
   #onDiscard(): void {
-    // TODO: Remove changes from the working copies as well.
+    this.#workspaceDiff.modifiedUISourceCodes().forEach(modifiedUISourceCode => {
+      modifiedUISourceCode.resetWorkingCopy();
+    });
+
     this.#patchSuggestion = undefined;
     this.#patchSources = undefined;
     this.requestUpdate();
   }
 
   #onSaveAll(): void {
-    // TODO: Handle saving all the files.
+    // TODO: What should we do for the inspector stylesheet?
+    this.#workspaceDiff.modifiedUISourceCodes().forEach(modifiedUISourceCode => {
+      modifiedUISourceCode.commitWorkingCopy();
+    });
+
     this.#savedToDisk = true;
     this.requestUpdate();
   }

front_end/testing/AiAssistanceHelpers.ts

@@ -229,6 +229,16 @@ export async function createPatchWidget(options?: {
   };
 }
 
+export async function createPatchWidgetWithDiffView() {
+  const {view, panel, aidaClient} =
+      await createPatchWidget({aidaClient: mockAidaClient([[{explanation: 'patch applied'}]])});
+  panel.changeSummary = 'body { background-color: red; }';
+  view.input.onApplyToWorkspace();
+  assert.exists((await view.nextInput).patchSuggestion);
+
+  return {panel, view, aidaClient};
+}
+
 export function initializePersistenceImplForTests(): void {
   const workspace = Workspace.Workspace.WorkspaceImpl.instance({forceNew: true});
   const debuggerWorkspaceBinding = Bindings.DebuggerWorkspaceBinding.DebuggerWorkspaceBinding.instance({