[Truffle] Add Kernel.global_variables

jruby · Sep 24, 2016 · f85827b · eregon · Sep 26, 2016 · bjfish
1 parent 37852c0
commit f85827b
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 2 deletions.
diff --git a/spec/truffle/tags/core/kernel/global_variables_tags.txt b/spec/truffle/tags/core/kernel/global_variables_tags.txt
diff --git a/truffle/src/main/java/org/jruby/truffle/core/kernel/KernelNodes.java b/truffle/src/main/java/org/jruby/truffle/core/kernel/KernelNodes.java
@@ -48,6 +48,8 @@
 import org.jruby.truffle.builtins.CoreMethod;
 import org.jruby.truffle.builtins.CoreMethodArrayArgumentsNode;
 import org.jruby.truffle.builtins.CoreMethodNode;
+import org.jruby.truffle.builtins.Primitive;
+import org.jruby.truffle.builtins.PrimitiveArrayArgumentsNode;
 import org.jruby.truffle.builtins.UnaryCoreMethodNode;
 import org.jruby.truffle.core.ObjectNodes;
 import org.jruby.truffle.core.ObjectNodesFactory;
@@ -148,6 +150,7 @@
 import java.lang.ProcessBuilder.Redirect;
 import java.nio.charset.StandardCharsets;
 import java.util.Arrays;
+import java.util.Collection;
 import java.util.Map;
 
 @CoreClass("Kernel")
@@ -1777,6 +1780,23 @@ protected CallTarget compileFormat(DynamicObject format) {
 
     }
 
+    @Primitive(name = "kernel_global_variables")
+    public abstract static class KernelGlobalVariablesPrimitiveNode extends PrimitiveArrayArgumentsNode {
+
+        @Specialization
+        public DynamicObject globalVariables() {
+            final Collection<String> keys = coreLibrary().getGlobalVariables().dynamicObjectKeys();
+            final Object[] store = new Object[keys.size()];
+            int i = 0;
+            for (String key : keys) {
+                store[i] = getSymbol(key);
+                i++;
+            }
+            return createArray(store, store.length);
+        }
+
+    }
+
     @CoreMethod(names = "taint")
     public abstract static class KernelTaintNode extends CoreMethodArrayArgumentsNode {
 

diff --git a/truffle/src/main/java/org/jruby/truffle/language/globals/GlobalVariables.java b/truffle/src/main/java/org/jruby/truffle/language/globals/GlobalVariables.java
@@ -60,6 +60,10 @@ public void alias(String name, GlobalVariableStorage storage) {
         variables.put(name, storage);
     }
 
+    public Collection<String> dynamicObjectKeys() {
+        return variables.keySet();
+    }
+
     public Collection<DynamicObject> dynamicObjectValues() {
         final Collection<GlobalVariableStorage> storages = variables.values();
         final ArrayList<DynamicObject> values = new ArrayList<>(storages.size());

diff --git a/truffle/src/main/ruby/core/kernel.rb b/truffle/src/main/ruby/core/kernel.rb
@@ -666,4 +666,11 @@ def at_exit(&block)
     Truffle::Kernel.at_exit false, &block
   end
   module_function :at_exit
+
+  def global_variables
+    Truffle.primitive :kernel_global_variables
+    raise PrimitiveFailure, "Kernel.global_variables primitive failed"
+  end
+  module_function :global_variables
+
 end