moved update to transform

Author: leafo

moonscript/compile/line.lua

@@ -82,27 +82,6 @@ line_compile = {
       return _with_0
     end
   end,
-  update = function(self, node)
-    local _, name, op, exp = unpack(node)
-    local op_final = op:match("^(.+)=$")
-    if not op_final then
-      error("Unknown op: " .. op)
-    end
-    return self:stm({
-      "assign",
-      {
-        name
-      },
-      {
-        {
-          "exp",
-          name,
-          op_final,
-          exp
-        }
-      }
-    })
-  end,
   ["return"] = function(self, node)
     return self:line("return ", (function()
       if node[2] ~= "" then

moonscript/compile/line.moon

@@ -46,12 +46,6 @@ line_compile =
       \append " = "
       \append_list [@value v for v in *values], ", "
 
-  update: (node) =>
-    _, name, op, exp = unpack node
-    op_final = op\match "^(.+)=$"
-    error "Unknown op: "..op if not op_final
-    @stm {"assign", {name}, {{"exp", name, op_final, exp}}}
-
   return: (node) =>
     @line "return ", if node[2] != "" then @value node[2]
 

moonscript/transform.lua

@@ -215,6 +215,19 @@ Statement = Transformer({
       return node
     end
   end,
+  update = function(node)
+    local _, name, op, exp = unpack(node)
+    local op_final = op:match("^(.+)=$")
+    if not op_final then
+      error("Unknown op: " .. op)
+    end
+    return build.assign_one(name, {
+      "exp",
+      name,
+      op_final,
+      exp
+    })
+  end,
   comprehension = function(node, action)
     local _, exp, clauses = unpack(node)
     action = action or function(exp)

moonscript/transform.moon

@@ -124,6 +124,12 @@ Statement = Transformer {
     else
       node
 
+  update: (node) ->
+    _, name, op, exp = unpack node
+    op_final = op\match "^(.+)=$"
+    error "Unknown op: "..op if not op_final
+    build.assign_one name, {"exp", name, op_final, exp}
+
   comprehension: (node, action) ->
     _, exp, clauses = unpack node