Merge pull request elixirscript#399 from elixirscript/implement_receive_stub

bryanjos · web-flow · commit b8f2478f92f7 · 2017-09-12T14:09:55.000-05:00
Implement stub of receive functionality
diff --git a/lib/elixir_script/passes/translate/forms/receive.ex b/lib/elixir_script/passes/translate/forms/receive.ex
@@ -1,26 +1,51 @@
 defmodule ElixirScript.Translate.Forms.Receive do
   @moduledoc false
   alias ESTree.Tools.Builder, as: J
-  alias ElixirScript.Translate.Helpers
+  alias ElixirScript.Translate.{Helpers, Form, Function, Clause}
 
   @doc """
   receive is not supported just yet, but we compile it
   to a stub function for now
   """
   def compile(blocks, state) do
-    _receive_block = Keyword.get(blocks, :do)
-    _after_block = Keyword.get(blocks, :after, nil)
+    receive_block = blocks
+    |> Keyword.get(:do)
+    |> Enum.map(fn x ->
+      Clause.compile(x, state)
+      |> elem(0)
+    end)
+    |> List.flatten
+    |> J.array_expression()
 
     receive_function = J.member_expression(
       Helpers.special_forms(),
       J.identifier("receive")
     )
 
+    after_block = Keyword.get(blocks, :after, nil)
+    args = [receive_block] ++ process_after(after_block, state)
+
     ast = Helpers.call(
       receive_function,
-      []
+      args
     )
 
     { ast, state }
   end
+
+  defp process_after(nil, _) do
+    []
+  end
+
+  defp process_after([{:->, _, [[timeout], body]}], state) do
+    timeout = Form.compile!(timeout, state)
+    {body, _state} = Function.compile_block(body, state)
+
+    function = Helpers.arrow_function(
+      [],
+      J.block_statement(List.wrap(body))
+    )
+
+    [timeout, function]
+  end
 end
diff --git a/rollup.config.js b/rollup.config.js
@@ -9,16 +9,16 @@ export default {
   plugins: [
     nodeResolve({
       jsnext: true,
-      main: true
+      main: true,
     }),
     commonjs(),
     babel({
-      babelrc: false
+      babelrc: false,
     }),
     minify({
       keepFnName: true,
-      keepClassName: true
-    })
+      keepClassName: true,
+    }),
   ],
-  output: [{ file: 'priv/build/iife/ElixirScript.Core.js', format: 'iife' }]
+  output: [{ file: 'priv/build/iife/ElixirScript.Core.js', format: 'iife' }],
 };
diff --git a/src/javascript/lib/core/functions.js b/src/javascript/lib/core/functions.js
@@ -95,7 +95,10 @@ function map_to_object(map, options = []) {
 
   const object = {};
 
-  for (let [key, value] of map.entries()) {
+  for (const entry of map.entries()) {
+    let key = entry[0];
+    const value = entry[1];
+
     if (opt_keys === Symbol.for('string') && typeof key === 'number') {
       key = key.toString();
     } else if (
diff --git a/src/javascript/lib/core/special_forms.js b/src/javascript/lib/core/special_forms.js
@@ -36,17 +36,21 @@ function run_list_generators(generator, generators) {
 }
 
 function _for(expression, generators, collectable_protocol, into = []) {
-  let [result, fun] = collectable_protocol.into(into);
+  const [result, fun] = collectable_protocol.into(into);
+  let accumulatingResult = result;
 
   const generatedValues = run_list_generators(generators.pop()(), generators);
 
   for (const value of generatedValues) {
     if (expression.guard.apply(this, value)) {
-      result = fun(result, new Core.Tuple(Symbol.for('cont'), expression.fn.apply(this, value)));
+      accumulatingResult = fun(
+        accumulatingResult,
+        new Core.Tuple(Symbol.for('cont'), expression.fn.apply(this, value)),
+      );
     }
   }
 
-  return fun(result, Symbol.for('done'));
+  return fun(accumulatingResult, Symbol.for('done'));
 }
 
 function _try(do_fun, rescue_function, catch_fun, else_function, after_function) {
@@ -132,8 +136,29 @@ function _with(...args) {
   return successFunction(...argsToPass);
 }
 
-function receive() {
+function receive(clauses, timeout = 0, timeoutFn = () => true) {
   console.warn('Receive not supported');
+
+  const messages = []; // this.mailbox.get();
+  const NOMATCH = Symbol('NOMATCH');
+
+  for (let i = 0; i < messages.length; i++) {
+    for (const clause of clauses) {
+      const value = Core.Patterns.match_or_default(
+        clause.pattern,
+        messages[i],
+        clause.guard,
+        NOMATCH,
+      );
+
+      if (value !== NOMATCH) {
+        this.mailbox.removeAt(i);
+        return clause.fn.apply(null, value);
+      }
+    }
+  }
+
+  return null;
 }
 
 export default {
diff --git a/test/passes/translate/forms/receive_test.exs b/test/passes/translate/forms/receive_test.exs
@@ -4,16 +4,14 @@ defmodule ElixirScript.Translate.Forms.Receive.Test do
   alias ESTree.Tools.Builder, as: J
 
   test "receive translation" do
-    ast = {:receive, [line: 644], [[do: 1, after: 2]]}
-    state = %{function: {:each, nil}, module: __MODULE__}
+    clause = {:->, [line: 644], [[], [{:__block__, [], [1]}]]}
+    ast = {:receive, [line: 644], [[do: [clause], after: nil]]}
+    state = %{function: {:each, nil}, module: __MODULE__, anonymous_fn: false}
 
     {js_ast, _} = Form.compile(ast, state)
-    assert js_ast == J.call_expression(
-      J.member_expression(
-        Helpers.special_forms(),
-        J.identifier("receive")
-      ),
-      []
+    assert js_ast.callee == J.member_expression(
+      Helpers.special_forms(),
+      J.identifier("receive")
     )
   end
 
