Save dialog for closing temp sketch

Author: msujew

Diff for: arduino-ide-extension/src/browser/arduino-frontend-contribution.tsx

@@ -20,6 +20,7 @@ import {
   FrontendApplication,
   FrontendApplicationContribution,
   LocalStorageService,
+  OnWillStopAction,
   StatusBar,
   StatusBarAlignment,
 } from '@theia/core/lib/browser';
@@ -70,6 +71,7 @@ import { SaveAsSketch } from './contributions/save-as-sketch';
 import { SketchbookWidgetContribution } from './widgets/sketchbook/sketchbook-widget-contribution';
 import { IDEUpdaterDialog } from './dialogs/ide-updater/ide-updater-dialog';
 import { IDEUpdater } from '../common/protocol/ide-updater';
+import { TempSketchDialog } from './dialogs/temp-sketch-dialog';
 
 const INIT_LIBS_AND_PACKAGES = 'initializedLibsAndPackages';
 export const SKIP_IDE_VERSION = 'skipIDEVersion';
@@ -549,6 +551,39 @@ export class ArduinoFrontendContribution
     }
   }
 
+  onWillStop(): OnWillStopAction {
+    return {
+      reason: 'Temp Sketch',
+      action: async () => {
+        const sketch = await this.sketchServiceClient.currentSketch();
+        if (!sketch) {
+          return true;
+        }
+        const isTemp = await this.sketchService.isTemp(sketch);
+        if (!isTemp) {
+          return true;
+        }
+        const dialog = new TempSketchDialog({
+          title: 'Arduino-IDE'
+        })
+        const result = await dialog.open();
+        if (result === "Don't Save") {
+          return true;
+        } else if (result === 'Save As...') {
+          return !!(await this.commandRegistry.executeCommand(
+            SaveAsSketch.Commands.SAVE_AS_SKETCH.id,
+            {
+              execOnlyIfTemp: false,
+              openAfterMove: false,
+              wipeOriginal: true,
+            }
+          ));
+        }
+        return false;
+      }
+    }
+  }
+
   registerColors(colors: ColorRegistry): void {
     colors.register(
       {

Diff for: arduino-ide-extension/src/browser/dialogs/temp-sketch-dialog.ts

@@ -0,0 +1,54 @@
+import { nls } from "@theia/core";
+import { AbstractDialog, Dialog, DialogProps, Message } from "@theia/core/lib/browser";
+import { inject } from "@theia/core/shared/inversify";
+
+export class TempSketchDialog extends AbstractDialog<TempSketchDialog.Options> {
+  protected readonly dontSaveButton: HTMLButtonElement;
+  protected _value: TempSketchDialog.Options = 'Cancel';
+
+  get value(): TempSketchDialog.Options {
+    return this._value;
+  }
+
+  constructor(
+    @inject(DialogProps) props: DialogProps
+  ) {
+    super(props);
+    const messageNode = document.createElement('div');
+    messageNode.textContent = nls.localize('arduino/sketch/saveTempSketch', 'Save your sketch to open it again later.');
+    this.contentNode.appendChild(messageNode);
+    this.dontSaveButton = this.createButton(nls.localizeByDefault(TempSketchDialog.Values["Don't Save"]));
+    this.dontSaveButton.classList.add('secondary');
+    this.controlPanel.appendChild(this.dontSaveButton);
+    this.appendCloseButton(Dialog.CANCEL);
+    this.appendAcceptButton(nls.localizeByDefault(TempSketchDialog.Values['Save As...']));
+  }
+
+  protected onAfterAttach(msg: Message): void {
+    super.onAfterAttach(msg);
+    this.addAction(this.dontSaveButton, () => this.dontSave(), 'click');
+  }
+
+  protected addAcceptAction<K extends keyof HTMLElementEventMap>(element: HTMLElement, ...additionalEventTypes: K[]): void {
+    this.addAction(element, () => this.doSave(), 'click');
+  }
+
+  protected dontSave(): void {
+    this._value = TempSketchDialog.Values["Don't Save"];
+    this.accept();
+  }
+
+  protected doSave(): void {
+    this._value = TempSketchDialog.Values['Save As...'];
+    this.accept();
+  }
+}
+
+export namespace TempSketchDialog {
+  export const enum Values {
+    "Don't Save" = "Don't Save",
+    Cancel = 'Cancel',
+    'Save As...' = 'Save As...',
+  };
+  export type Options = keyof typeof Values;
+}

Diff for: arduino-ide-extension/src/electron-main/theia/electron-main-application.ts

@@ -272,6 +272,7 @@ export class ElectronMainApplication extends TheiaElectronMainApplication {
       }
     });
     this.attachClosedWorkspace(electronWindow);
+    this.attachCloseListeners(electronWindow, options);
     this.attachReadyToShow(electronWindow);
     this.attachSaveWindowState(electronWindow);
     this.attachGlobalShortcuts(electronWindow);