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