X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.modeling.ui%2Fsrc%2Forg%2Fsimantics%2Fmodeling%2Fui%2FdiagramEditor%2FHasDiagramSourceListener.java;fp=bundles%2Forg.simantics.modeling.ui%2Fsrc%2Forg%2Fsimantics%2Fmodeling%2Fui%2FdiagramEditor%2FHasDiagramSourceListener.java;h=c7873676be1949c979dc102757272cda06543a24;hb=a18337dc03c999c2bda582822d53c10ebdd9a9ef;hp=0000000000000000000000000000000000000000;hpb=91d2a4c9212ecbb89656acaf5a01fb86a0aa5898;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/HasDiagramSourceListener.java b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/HasDiagramSourceListener.java
new file mode 100644
index 000000000..c7873676b
--- /dev/null
+++ b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/HasDiagramSourceListener.java
@@ -0,0 +1,58 @@
+/*******************************************************************************
+ * Copyright (c) 2019 Association for Decentralized Information Management
+ * in Industry THTH ry.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     Semantum Oy - initial API and implementation, gitlab #399
+ *******************************************************************************/
+package org.simantics.modeling.ui.diagramEditor;
+
+import org.simantics.db.Resource;
+import org.simantics.db.common.procedure.adapter.DisposableListener;
+import org.simantics.diagram.ui.DiagramModelHints;
+import org.simantics.g2d.diagram.IDiagram;
+import org.simantics.g2d.element.ElementUtils;
+import org.simantics.utils.DataContainer;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * @author Tuukka Lehtonen
+ * @since 1.41.0, 1.35.2
+ * @see gitlab #399
+ */
+class HasDiagramSourceListener extends DisposableListener<Resource> {
+
+    private static final Logger LOGGER = LoggerFactory.getLogger(HasDiagramSourceListener.class);
+
+    private DataContainer<IDiagram> diagramContainer;
+
+    public HasDiagramSourceListener(DataContainer<IDiagram> diagramContainer) {
+        this.diagramContainer = diagramContainer;
+    }
+
+    @Override
+    public void execute(Resource source) {
+        DataContainer<IDiagram> c = diagramContainer;
+        IDiagram d = c != null ? diagramContainer.get() : null;
+        if (d != null) {
+            ElementUtils.setOrRemoveHint(d, DiagramModelHints.KEY_HAS_DIAGRAM_SOURCE, source);
+        }
+    }
+
+    @Override
+    public void exception(Throwable t) {
+        LOGGER.error("Received unexpected exception", t);
+    }
+
+    @Override
+    public void dispose() {
+        super.dispose();
+        diagramContainer = null;
+    }
+
+}