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 }