Implement stub of receive functionality

bryanjos · bryanjos · commit 4385a59ddee4 · 2017-09-10T09:31:08.000-05:00
diff --git a/lib/elixir_script/passes/translate/forms/receive.ex b/lib/elixir_script/passes/translate/forms/receive.ex
@@ -1,26 +1,48 @@
 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 = Keyword.get(blocks, :do)
+    after_block = Keyword.get(blocks, :after, nil)
+
+    receive_block = Enum.map(receive_block, 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")
     )
 
+    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/src/javascript/lib/core/special_forms.js b/src/javascript/lib/core/special_forms.js
@@ -16,7 +16,7 @@ function cond(...clauses) {
 
 function run_list_generators(generator, generators) {
   if (generators.length === 0) {
-    return generator.map((x) => {
+    return generator.map(x => {
       if (Array.isArray(x)) {
         return x;
       }
@@ -42,14 +42,23 @@ function _for(expression, generators, collectable_protocol, into = []) {
 
   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)));
+      result = fun(
+        result,
+        new Core.Tuple(Symbol.for('cont'), expression.fn.apply(this, value))
+      );
     }
   }
 
   return fun(result, Symbol.for('done'));
 }
 
-function _try(do_fun, rescue_function, catch_fun, else_function, after_function) {
+function _try(
+  do_fun,
+  rescue_function,
+  catch_fun,
+  else_function,
+  after_function
+) {
   let result = null;
 
   try {
@@ -132,8 +141,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();
+
+  for (let i = 0; i < messages.length; i++) {
+    for (const clause of clauses) {
+      const value = 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 {
@@ -142,5 +172,5 @@ export default {
   _for,
   _try,
   _with,
-  receive,
+  receive
 };
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
 
