Migrated source code from Simantics SVN

[simantics/platform.git] / bundles / org.simantics.scl.compiler / src / org / simantics / scl / compiler / source / repository / SourceRepositories.java

diff --git a/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/source/repository/SourceRepositories.java b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/source/repository/SourceRepositories.java
new file mode 100644 (file)
index 0000000..4e89a47
--- /dev/null
+++ b/bundles/org.simantics.scl.compiler/src/org/simantics/scl/compiler/source/repository/SourceRepositories.java
@@ -0,0 +1,69 @@
+package org.simantics.scl.compiler.source.repository;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.URL;
+import java.util.ArrayList;
+import java.util.List;
+
+import org.simantics.scl.compiler.module.ImportDeclaration;
+import org.simantics.scl.runtime.SCLContext;
+
+import gnu.trove.procedure.TObjectProcedure;
+
+public class SourceRepositories {
+    public static final ModuleSourceRepository BUILTIN_SOURCE_REPOSITORY = new BuiltinModuleSourceRepository();
+    
+    public static final ModuleSourceRepository PRELUDE_SOURCE_REPOSITORY = createPreludeSourceEnvironment();
+
+    private static ModuleSourceRepository createPreludeSourceEnvironment() {
+        try {
+            File directory = getPreludeDirectory();
+            if(directory != null) {
+                return new FileModuleSourceRepository(directory, SourceRepositories.class.getClassLoader()) {
+                    protected ImportDeclaration[] getBuiltinImports() {
+                        return ImportDeclaration.ONLY_BUILTINS;
+                    }
+                };
+            }
+            
+            // Hope that ClassSourceEnvironment works
+            return new ClassModuleSourceRepository(SCLContext.class, SourceRepositories.class.getClassLoader(), "/scl") {
+                protected ImportDeclaration[] getBuiltinImports() {
+                    return ImportDeclaration.ONLY_BUILTINS;
+                }
+            };
+        } catch(IOException e) {
+            e.printStackTrace();
+            return null;
+        }
+    }
+    
+    private static File getPreludeDirectory() throws IOException {
+        URL url = SCLContext.class.getResource(".");
+        if(url == null)
+            return null;
+        File file = ModuleBrowsingUtil.toFile(url);
+        if(file == null)
+            return null;
+        File preludeDir = new File(file, "../../../../scl");
+        if(preludeDir.exists())
+            return preludeDir.getCanonicalFile();
+        preludeDir = new File(file, "../../../../../scl");
+        if(preludeDir.exists())
+            return preludeDir.getCanonicalFile();
+        return null;
+    }
+    
+    public static List<String> getModuleNames(ModuleSourceRepository repository) {
+        final ArrayList<String> result = new ArrayList<String>(); 
+        repository.forAllModules(new TObjectProcedure<String>() {
+            @Override
+            public boolean execute(String moduleName) {
+                result.add(moduleName);
+                return true;
+            }
+        });
+        return result;
+    }
+}