8eed51aca7ed70401b7ddafb7da5a13bab85b2ff
[simantics/platform.git] / bundles / org.simantics.modeling / src / org / simantics / modeling / scl / GraphModuleSourceRepository.java
1 package org.simantics.modeling.scl;
2
3 import java.util.ArrayList;
4 import java.util.Collection;
5
6 import org.simantics.Simantics;
7 import org.simantics.db.ReadGraph;
8 import org.simantics.db.Resource;
9 import org.simantics.db.WriteGraph;
10 import org.simantics.db.common.request.UnaryRead;
11 import org.simantics.db.common.request.WriteRequest;
12 import org.simantics.db.exception.DatabaseException;
13 import org.simantics.db.procedure.SyncListener;
14 import org.simantics.db.request.Read;
15 import org.simantics.layer0.Layer0;
16 import org.simantics.modeling.ModelingUtils;
17 import org.simantics.modeling.internal.Activator;
18 import org.simantics.modeling.scl.ontologymodule.OntologyModuleSourceRepository;
19 import org.simantics.scl.compiler.internal.codegen.types.JavaReferenceValidatorFactory;
20 import org.simantics.scl.compiler.module.repository.UpdateListener;
21 import org.simantics.scl.compiler.module.repository.UpdateListener.Observable;
22 import org.simantics.scl.compiler.source.ModuleSource;
23 import org.simantics.scl.compiler.source.StringModuleSource;
24 import org.simantics.scl.compiler.source.repository.ModuleSourceRepository;
25 import org.simantics.scl.osgi.internal.OsgiJavaReferenceValidatorFactory;
26 import org.simantics.scl.runtime.SCLContext;
27 import org.simantics.scl.runtime.tuple.Tuple0;
28 import org.simantics.structural2.utils.StructuralUtils;
29 import org.slf4j.Logger;
30 import org.slf4j.LoggerFactory;
31
32 import gnu.trove.procedure.TObjectProcedure;
33 import gnu.trove.set.hash.THashSet;
34
35 public enum GraphModuleSourceRepository implements ModuleSourceRepository {
36     INSTANCE; 
37     
38     private static final Logger LOGGER = LoggerFactory.getLogger(OntologyModuleSourceRepository.class);
39
40     private static final OsgiJavaReferenceValidatorFactory REFERENCE_VALIDATOR_FACTORY = new OsgiJavaReferenceValidatorFactory(Activator.getContext().getBundle());
41
42     @Override
43     public ModuleSource getModuleSource(final String moduleName, UpdateListener listener) {
44         if(!moduleName.startsWith("http://"))
45             return null; // Don't do a graph request if this cannot be a resource
46
47         Read<ModuleSource> request = new ReadModuleSource(moduleName);
48
49         try {
50             if(listener != null)
51                 return Simantics.getAvailableRequestProcessor().syncRequest(request, new ModuleListener(listener, moduleName));
52             else
53                 return Simantics.getAvailableRequestProcessor().syncRequest(request);
54         } catch (DatabaseException e) {
55             LOGGER.error("Failed to read graph module {}.",  moduleName, e);
56             return null;
57         }
58     }
59
60     static class ModuleListener implements SyncListener<ModuleSource>, Observable {
61         UpdateListener listener;
62         boolean alreadyExecutedOnce;
63         final String moduleName;
64         public ModuleListener(UpdateListener listener, String moduleName) {
65             this.listener = listener;
66             this.listener.addObservable(this);
67             this.moduleName = moduleName;
68         }
69         @Override
70         public void removeListener(UpdateListener listener) {
71             this.listener = null;
72         }
73         @Override
74         public boolean isDisposed() {
75             return listener == null;
76         }
77         private void fireUpdate(ReadGraph graph) {
78             if(listener != null) {
79                 SCLContext context = SCLContext.getCurrent();
80                 Object oldGraph = context.put("graph", graph);
81                 try {
82                     listener.notifyAboutUpdate();
83                 } finally {
84                     listener = null;
85                     context.put("graph", oldGraph);
86                 }
87             }
88         }
89         @Override
90         public void execute(ReadGraph graph, ModuleSource result)
91                 throws DatabaseException {
92             if(alreadyExecutedOnce)
93                 fireUpdate(graph);
94             else
95                 alreadyExecutedOnce = true;
96         }
97         @Override
98         public void exception(ReadGraph graph, Throwable t)
99                 throws DatabaseException {
100             LOGGER.error("Could not listen {}", this, t);
101             if(alreadyExecutedOnce && listener != null)
102                 fireUpdate(graph);
103         }
104
105         @Override
106         public String toString() {
107             return moduleName + " " + listener + " (" + alreadyExecutedOnce + ") [" + getClass().toString() + "]";
108         }
109     };
110     
111     public static class GraphModuleSource extends StringModuleSource {
112
113         private final boolean immutable;
114
115         public GraphModuleSource(String moduleName, ClassLoader classLoader, String moduleText, boolean immutable) {
116             super(moduleName, classLoader, moduleText);
117             this.immutable = immutable;
118         }
119         
120         @Override
121         public boolean isUpdateable() {
122             return !immutable;
123         }
124         
125         @Override
126         public void update(String newSourceText) {
127             try {
128                 Simantics.getSession().syncRequest(new WriteModuleSource(getModuleName(), newSourceText));
129             } catch (DatabaseException e) {
130                 LOGGER.error("Could not update {} with newSourceText {}", this, newSourceText);
131             }
132         }
133         
134         @Override
135         public JavaReferenceValidatorFactory getJavaReferenceValidatorFactory() {
136             return REFERENCE_VALIDATOR_FACTORY;
137         }
138     }
139
140     static class ReadModuleSource extends UnaryRead<String, ModuleSource> {
141         public ReadModuleSource(String moduleName) {
142             super(moduleName);
143         }
144
145         @Override
146         public ModuleSource perform(ReadGraph graph) throws DatabaseException {
147             Resource moduleResource = graph.getPossibleResource(parameter);
148             if(moduleResource == null)
149                 return null;
150             Layer0 L0 = Layer0.getInstance(graph);
151             if(!graph.isInstanceOf(moduleResource, L0.SCLModule))
152                 return null;
153             String text = graph.getRelatedValue(moduleResource, L0.SCLModule_definition);
154             boolean immutable = StructuralUtils.isImmutable(graph, moduleResource);
155             return new GraphModuleSource(parameter, getClass().getClassLoader(), text, immutable);
156         }
157     }
158     
159     static class WriteModuleSource extends WriteRequest {
160         private final String moduleURI;
161         private final String sourceText;
162         
163         public WriteModuleSource(String moduleURI, String sourceText) {
164             this.moduleURI = moduleURI;
165             this.sourceText = sourceText;
166         }
167
168         @Override
169         public void perform(WriteGraph graph) throws DatabaseException {
170             Resource moduleResource = graph.getPossibleResource(moduleURI);
171             if(moduleResource == null)
172                 return;
173             Layer0 L0 = Layer0.getInstance(graph);
174             if(!graph.isInstanceOf(moduleResource, L0.SCLModule))
175                 return;
176             graph.claimLiteral(moduleResource, L0.SCLModule_definition, sourceText);
177         }
178     }
179
180     private THashSet<String> getAllModules(ReadGraph graph) throws DatabaseException {
181         THashSet<String> result = new THashSet<String>(); 
182         Resource projectResource = Simantics.getProjectResource();
183         Layer0 L0 = Layer0.getInstance(graph);
184         for(Resource model : graph.getObjects(projectResource, L0.ConsistsOf)) {
185             if(graph.isInstanceOf(model, L0.IndexRoot)) {
186                 for(Resource module : ModelingUtils.searchByType(graph, model, L0.SCLModule))
187                     result.add(graph.getURI(module));
188             }
189         }
190         
191         Collection<Resource> ontologies = Simantics.applySCL("Simantics/SharedOntologies", "getSharedOntologies", graph, Tuple0.INSTANCE);
192         for (Resource ontology : ontologies) {
193             for(Resource module : ModelingUtils.searchByType(graph, ontology, L0.SCLModule))
194                 try {
195                     result.add(graph.getURI(module));
196                 } catch(DatabaseException e) {
197                     LOGGER.error("Failed to find uri for " + module + ".");
198                 }
199         }
200         
201         return result;
202     }
203     
204     @Override
205     public void forAllModules(TObjectProcedure<String> procedure) {
206         THashSet<String> moduleURIs = null;
207         try {
208             moduleURIs = Simantics.getAvailableRequestProcessor().syncRequest(new Read<THashSet<String>>() {
209                 @Override
210                 public THashSet<String> perform(ReadGraph graph)
211                         throws DatabaseException {
212                     return getAllModules(graph);
213                 }
214             });
215             moduleURIs.forEach(procedure);
216         } catch (DatabaseException e) {
217             LOGGER.error("Could not execute procedure {} for all modules {}", procedure, String.valueOf(moduleURIs), e);
218         }
219     }
220     
221     @Override
222     public Collection<String> getModuleNames() {
223         ArrayList<String> result = new ArrayList<>();
224         forAllModules((String name) -> {
225             result.add(name);
226             return true;
227         });
228         return result;
229     }
230 }