Merge pull request #375 from elixirscript/add_erlang_functions

bryanjos · web-flow · commit cc874f31df37 · 2017-08-31T20:23:27.000-05:00
Add math.log2/1, erlang.nodes/0 and erlang.nodes/1
diff --git a/src/javascript/lib/core.js b/src/javascript/lib/core.js
@@ -11,6 +11,7 @@ import io from './core/erlang_compat/io';
 import binary from './core/erlang_compat/binary';
 import unicode from './core/erlang_compat/unicode';
 import Store from './core/store';
+import math from './core/erlang_compat/math';
 
 class Integer {}
 class Float {}
@@ -51,5 +52,6 @@ export default {
   io,
   binary,
   unicode,
-  elixir_config
+  elixir_config,
+  math
 };
diff --git a/src/javascript/lib/core/erlang_compat/erlang.js b/src/javascript/lib/core/erlang_compat/erlang.js
@@ -360,6 +360,19 @@ function node() {
   return Symbol.for('nonode@nohost');
 }
 
+function nodes(arg = []) {
+  const nodeTypes = Array.isArray(arg) ? arg : [arg];
+  const nodesFound = [];
+
+  for (const nodeType of nodeTypes) {
+    if (nodeType === Symbol.for('this')) {
+      nodesFound.push(Symbol.for('nonode@nohost'));
+      console.log(nodesFound);
+    }
+  }
+  return nodesFound;
+}
+
 function self() {
   return selfPID;
 }
@@ -458,5 +471,6 @@ export default {
   error,
   exit,
   raise,
-  list_to_binary
+  list_to_binary,
+  nodes
 };
diff --git a/src/javascript/lib/core/erlang_compat/math.js b/src/javascript/lib/core/erlang_compat/math.js
@@ -0,0 +1,7 @@
+function log2(x) {
+  return Math.log2(x);
+}
+
+export default {
+  log2
+};
diff --git a/src/javascript/tests/core/erlang_compat/erlang_spec.js b/src/javascript/tests/core/erlang_compat/erlang_spec.js
@@ -95,3 +95,23 @@ test('list_subtraction', t => {
   t.deepEqual(Core.erlang.list_subtraction([1, 2, 3], [1, 2, 3]), []);
   t.deepEqual(Core.erlang.list_subtraction([1, 2, 3], [1, 2]), [3]);
 });
+
+test('node', t => {
+  t.deepEqual(Core.erlang.node(), Symbol.for('nonode@nohost'));
+});
+
+test('nodes/0', t => {
+  t.deepEqual(Core.erlang.nodes(), []);
+});
+
+test('nodes/1', t => {
+  t.deepEqual(Core.erlang.nodes(Symbol.for('this')), [
+    Symbol.for('nonode@nohost')
+  ]);
+
+  t.deepEqual(Core.erlang.nodes([Symbol.for('this')]), [
+    Symbol.for('nonode@nohost')
+  ]);
+
+  t.deepEqual(Core.erlang.nodes([Symbol.for('connected')]), []);
+});
diff --git a/src/javascript/tests/core/erlang_compat/math_spec.js b/src/javascript/tests/core/erlang_compat/math_spec.js
@@ -0,0 +1,13 @@
+import test from 'ava';
+import Core from '../../../lib/core';
+
+test('log2/1', t => {
+  let result = Core.math.log2(1);
+  t.is(result, 0);
+
+  result = Core.math.log2(2);
+  t.is(result, 1);
+
+  result = Core.math.log2(4);
+  t.is(result, 2);
+});
