set dirty icon

mathansen · Aug 29, 2018 · eaf4478 · eaf4478
1 parent 71afecb
commit eaf4478
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 5 deletions.
diff --git a/src/browser/ResourceSaveable.ts b/src/browser/ResourceSaveable.ts
@@ -1,5 +1,8 @@
 import { Disposable, Event, MaybePromise, Resource } from '@theia/core/lib/common';
-import { Saveable } from "@theia/core/lib/browser";
+import {Saveable, WidgetManager} from "@theia/core/lib/browser";
+import { DIRTY_CLASS, TreeEditorWidget } from './theia-tree-editor-widget';
+import * as _ from 'lodash';
+import { unmanaged } from 'inversify';
 
 export class ResourceSaveable implements Saveable {
   autoSave;
@@ -16,7 +19,9 @@ export class ResourceSaveable implements Saveable {
     }
   );
 
-  constructor(private resource: Resource, private getData: () => any) {}
+  constructor(private resource: Resource,
+              private getData: () => any,
+              @unmanaged() protected readonly widgetManager: WidgetManager) {}
 
   save(): MaybePromise<void> {
     return this.onSave(this.getData()).then(this.doSave)
@@ -27,6 +32,9 @@ export class ResourceSaveable implements Saveable {
       return this.resource.saveContents(content, { encoding: 'UTF-8' })
         .then(() => {
           this.dirty = false;
+          const widget = _.head(this.widgetManager.getWidgets('theia-tree-editor')) as TreeEditorWidget;
+          const dirtyClass = ` ${DIRTY_CLASS}`;
+          widget.title.className = widget.title.className.replace(dirtyClass, '');
       });
     } else {
       console.warn('resource cannot save');

diff --git a/src/browser/theia-tree-editor-contribution.ts b/src/browser/theia-tree-editor-contribution.ts
@@ -16,7 +16,6 @@ import { OpenerOptions, WidgetManager, WidgetOpenHandler } from '@theia/core/lib
 import { EditorContextMenu } from '@theia/editor/lib/browser';
 import { TreeEditorWidget } from './theia-tree-editor-widget';
 
-
 export const UISchemaDownloadCommand = {
   id: 'UISchemaDownload.command',
   label: "Download UI Schema"

diff --git a/src/browser/theia-tree-editor-widget.tsx b/src/browser/theia-tree-editor-widget.tsx
@@ -29,6 +29,7 @@ const defaultResourceParser = (content: string): Promise<any> => {
 };
 
 let widgetCounter = 0;
+export const DIRTY_CLASS = 'theia-mod-dirty';
 @injectable()
 export class TreeEditorWidget extends BaseWidget implements SaveableSource {
   saveable: Saveable;
@@ -57,7 +58,7 @@ export class TreeEditorWidget extends BaseWidget implements SaveableSource {
       .then(parsedContent => {
         Promise.resolve(this.store).then(initializedStore => {
           initializedStore.dispatch(Actions.update('', () => parsedContent));
-          const Editor = withProps({'saveable': this.saveable})(this.options.EditorComponent);
+          const Editor = withProps({'widget': this})(this.options.EditorComponent);
           ReactDOM.render(
             <Provider store={initializedStore}>
               <Editor/>

diff --git a/src/browser/tree-editor-utils.ts b/src/browser/tree-editor-utils.ts
@@ -9,7 +9,7 @@ export interface TreeEditorProps {
   filterPredicate: any;
   labelProvider: any;
   imageProvider: any;
-  saveable: any;
+  widget: any;
 }
 
 export interface LabelDefinition {