--- /dev/null
+package org.simantics.modeling.ui.componentTypeEditor;\r
+\r
+import java.io.PrintWriter;\r
+import java.io.StringWriter;\r
+import java.util.Arrays;\r
+import java.util.Collection;\r
+import java.util.Collections;\r
+import java.util.List;\r
+\r
+import org.eclipse.core.runtime.CoreException;\r
+import org.eclipse.core.runtime.IProgressMonitor;\r
+import org.eclipse.jface.operation.IRunnableContext;\r
+import org.eclipse.jface.text.Document;\r
+import org.eclipse.jface.text.IDocument;\r
+import org.eclipse.jface.text.Position;\r
+import org.eclipse.jface.text.source.Annotation;\r
+import org.eclipse.jface.text.source.AnnotationModel;\r
+import org.eclipse.jface.text.source.IAnnotationModel;\r
+import org.eclipse.ui.texteditor.AbstractDocumentProvider;\r
+import org.simantics.Simantics;\r
+import org.simantics.databoard.Bindings;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.Statement;\r
+import org.simantics.db.WriteGraph;\r
+import org.simantics.db.common.request.ReadRequest;\r
+import org.simantics.db.common.request.UniqueRead;\r
+import org.simantics.db.common.request.WriteRequest;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.layer0.util.Layer0Utils;\r
+import org.simantics.layer0.Layer0;\r
+import org.simantics.modeling.ModelingResources;\r
+import org.simantics.scl.compiler.errors.CompilationError;\r
+import org.simantics.scl.compiler.errors.Failable;\r
+import org.simantics.scl.compiler.errors.Failure;\r
+import org.simantics.scl.compiler.errors.Locations;\r
+import org.simantics.scl.compiler.module.Module;\r
+import org.simantics.scl.compiler.module.repository.UpdateListener;\r
+import org.simantics.scl.osgi.SCLOsgi;\r
+import org.simantics.scl.runtime.SCLContext;\r
+import org.simantics.ui.workbench.ResourceEditorInput;\r
+import org.simantics.utils.logging.TimeLogger;\r
+\r
+public class SCLQueryEditorDocumentProvider extends AbstractDocumentProvider {\r
+ \r
+ Resource resource;\r
+ String currentText;\r
+ boolean errorHappened;\r
+ \r
+ AnnotationModel annotationModel = new AnnotationModel();\r
+ SCLQueryEditor editor;\r
+ \r
+ public SCLQueryEditorDocumentProvider(SCLQueryEditor editor) {\r
+ this.editor = editor;\r
+ }\r
+ \r
+ private boolean isType(ReadGraph graph) throws DatabaseException {\r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+ return graph.isInstanceOf(resource, MOD.SCLQueryType);\r
+ }\r
+ \r
+ @Override\r
+ protected IDocument createDocument(Object element) throws CoreException {\r
+ ResourceEditorInput input = (ResourceEditorInput)element;\r
+ resource = input.getResource();\r
+ try {\r
+ return Simantics.getSession().syncRequest(new UniqueRead<Document>() {\r
+ @Override\r
+ public Document perform(ReadGraph graph) throws DatabaseException {\r
+ Layer0 L0 = Layer0.getInstance(graph);\r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+ if(isType(graph)) {\r
+ Collection<Resource> assertions = graph.getAssertedObjects(resource, MOD.SCLQuery_values);\r
+ if(assertions.size() != 1) throw new DatabaseException("No query text assertion defined in Query Type");\r
+ Resource value = assertions.iterator().next();\r
+ currentText = graph.getRelatedValue(value, L0.SCLValue_expression, Bindings.STRING);\r
+ errorHappened = false;\r
+ return new Document(currentText != null ? currentText : "");\r
+ } else {\r
+ Resource value = graph.getSingleObject(resource, MOD.SCLQuery_values);\r
+ currentText = graph.getRelatedValue(value, L0.SCLValue_expression, Bindings.STRING);\r
+ errorHappened = false;\r
+ return new Document(currentText != null ? currentText : "");\r
+ }\r
+ }\r
+ });\r
+ } catch (DatabaseException e) {\r
+ StringWriter sw = new StringWriter();\r
+ PrintWriter pw = new PrintWriter(sw);\r
+ e.printStackTrace(pw);\r
+ errorHappened = false;\r
+ return new Document(sw.toString());\r
+ }\r
+ }\r
+ \r
+ // While this editor is active we do not want that this listener gets collected\r
+ private UpdateListener listener = new UpdateListener() {\r
+ @Override\r
+ public void notifyAboutUpdate() {\r
+ updateAnnotations();\r
+ }\r
+ };\r
+ \r
+ private void updateAnnotations() {\r
+ Simantics.getSession().asyncRequest(new ReadRequest() {\r
+ @Override\r
+ public void run(ReadGraph graph) throws DatabaseException {\r
+ String moduleName = graph.getURI(resource);\r
+ SCLContext context = SCLContext.getCurrent();\r
+ context.put("graph", graph);\r
+ Failable<Module> result = SCLOsgi.MODULE_REPOSITORY.getModule(moduleName, listener);\r
+ \r
+ if(result instanceof Failure) {\r
+ Failure failure = (Failure)result;\r
+ setAnnotations(Arrays.asList(failure.errors));\r
+ }\r
+ else {\r
+ setAnnotations(Collections.<CompilationError>emptyList());\r
+ }\r
+ }\r
+ });\r
+ }\r
+ \r
+ private void setAnnotations(List<CompilationError> errors) {\r
+ synchronized(annotationModel.getLockObject()) {\r
+ annotationModel.removeAllAnnotations();\r
+ for(CompilationError error : errors) {\r
+ Annotation annotation = new Annotation("org.eclipse.ui.workbench.texteditor.error", true,\r
+ error.description);\r
+ int begin = Locations.beginOf(error.location);\r
+ int end = Locations.endOf(error.location);\r
+ Position position = new Position(begin, end - begin);\r
+ annotationModel.addAnnotation(annotation, position);\r
+ }\r
+ }\r
+ }\r
+ \r
+ boolean annotationsInitialized = false;\r
+ \r
+ @Override\r
+ protected IAnnotationModel createAnnotationModel(Object element)\r
+ throws CoreException {\r
+ if(!annotationsInitialized) {\r
+ updateAnnotations();\r
+ annotationsInitialized = true;\r
+ }\r
+ return annotationModel;\r
+ }\r
+\r
+ @Override\r
+ protected void doSaveDocument(IProgressMonitor monitor, Object element,\r
+ IDocument document, boolean overwrite) throws CoreException {\r
+ TimeLogger.resetTimeAndLog("SCLModuleEditorDocumentProvider.doSaveDocument");\r
+ currentText = document.get();\r
+ Simantics.getSession().asyncRequest(new WriteRequest() {\r
+ \r
+ private Resource getValue(WriteGraph graph) throws DatabaseException {\r
+\r
+ Layer0 L0 = Layer0.getInstance(graph);\r
+ ModelingResources MOD = ModelingResources.getInstance(graph);\r
+\r
+ if(isType(graph)) {\r
+ \r
+ Collection<Resource> assertions = graph.getAssertedObjects(resource, MOD.SCLQuery_values);\r
+ if(assertions.size() > 1) throw new DatabaseException("Invalid query text assertions in Query Type");\r
+ if(assertions.size() == 1) return assertions.iterator().next();\r
+ \r
+ Resource value = graph.newResource();\r
+ graph.claim(value, L0.InstanceOf, MOD.SCLQuery_Value);\r
+ Layer0Utils.assert_(graph, resource, MOD.SCLQuery_values, value);\r
+ return value;\r
+ \r
+ } else {\r
+ \r
+ Statement stm = graph.getSingleStatement(resource, MOD.SCLQuery_values);\r
+ if(stm.isAsserted(resource)) {\r
+\r
+ Resource value = graph.newResource();\r
+ graph.claim(value, L0.InstanceOf, MOD.SCLQuery_Value);\r
+ graph.claim(resource, MOD.SCLQuery_values, value);\r
+ return value;\r
+\r
+ } else {\r
+ return stm.getObject();\r
+ }\r
+ \r
+ }\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public void perform(WriteGraph graph) throws DatabaseException {\r
+ graph.markUndoPoint();\r
+ Layer0 L0 = Layer0.getInstance(graph);\r
+ Resource value = getValue(graph);\r
+ graph.claimLiteral(value, L0.SCLValue_expression, currentText, Bindings.STRING);\r
+ Layer0Utils.addCommentMetadata(graph, "Saved SCL Query " + graph.getRelatedValue2(resource, Layer0.getInstance(graph).HasName, Bindings.STRING));\r
+ }\r
+ });\r
+ }\r
+\r
+ @Override\r
+ protected IRunnableContext getOperationRunner(IProgressMonitor monitor) {\r
+ return null;\r
+ }\r
+ \r
+ @Override\r
+ public boolean isModifiable(Object element) {\r
+ return !errorHappened;\r
+ }\r
+ \r
+ @Override\r
+ public boolean isReadOnly(Object element) {\r
+ return errorHappened;\r
+ }\r
+ \r
+ @Override\r
+ public boolean canSaveDocument(Object element) {\r
+ return !errorHappened && !getDocument(element).get().equals(currentText);\r
+ }\r
+\r
+}\r