refined label provide for unsaved sketches

Signed-off-by: Akos Kitta <kittaakos@typefox.io>

arduino-ide-extension/src/browser/theia/editor/editor-widget-factory.ts

@@ -0,0 +1,38 @@
+import { inject, injectable } from 'inversify';
+import URI from '@theia/core/lib/common/uri';
+import { EditorWidget } from '@theia/editor/lib/browser';
+import { LabelProvider } from '@theia/core/lib/browser';
+import { EditorWidgetFactory as TheiaEditorWidgetFactory } from '@theia/editor/lib/browser/editor-widget-factory';
+import { SketchesServiceClientImpl } from '../../../common/protocol/sketches-service-client-impl';
+import { SketchesService, Sketch } from '../../../common/protocol';
+
+@injectable()
+export class EditorWidgetFactory extends TheiaEditorWidgetFactory {
+
+    @inject(SketchesService)
+    protected readonly sketchesService: SketchesService;
+
+    @inject(SketchesServiceClientImpl)
+    protected readonly sketchesServiceClient: SketchesServiceClientImpl;
+
+    @inject(LabelProvider)
+    protected readonly labelProvider: LabelProvider;
+
+    protected async createEditor(uri: URI): Promise<EditorWidget> {
+        const widget = await super.createEditor(uri);
+        return this.maybeUpdateCaption(widget);
+    }
+
+    protected async maybeUpdateCaption(widget: EditorWidget): Promise<EditorWidget> {
+        const sketch = await this.sketchesServiceClient.currentSketch();
+        const { uri } = widget.editor;
+        if (sketch && Sketch.isInSketch(uri, sketch)) {
+            const isTemp = await this.sketchesService.isTemp(sketch);
+            if (isTemp) {
+                widget.title.caption = `Unsaved – ${this.labelProvider.getName(uri)}`;
+            }
+        }
+        return widget;
+    }
+
+}