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