fixed #165, export const

satyr · Sep 5, 2012 · bb5b5a3 · bb5b5a3
1 parent 3737564
commit bb5b5a3
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 2 deletions.
diff --git a/lib/ast.js b/lib/ast.js
@@ -3223,6 +3223,10 @@ DECLS = {
     i = -1;
     out = Util('out');
     while (node = lines[++i]) {
+      if (node instanceof Block) {
+        lines.splice.apply(lines, [i--, 1].concat(slice$.call(node.lines)));
+        continue;
+      }
       if (that = node instanceof Fun && node.name) {
         lines.splice(i++, 0, Assign(Chain(out, [Index(Key(that))]), Var(that)));
         continue;

diff --git a/src/ast.co b/src/ast.co
@@ -1978,6 +1978,9 @@ DECLS =
   export: (lines) ->
     i = -1; out = Util \out
     while node = lines[++i]
+      if node instanceof Block
+        lines.splice i-- 1 ...node.lines
+        continue
       if node instanceof Fun and node.name
         lines.splice i++ 0 Assign Chain(out, [Index Key that]), Var that
         continue

diff --git a/test/declaration.co b/test/declaration.co
@@ -44,8 +44,13 @@ let
     d = 4
     [e, f] = [5, 6]
   export const g = 7
-  eq '1,2,3,4,5,6,7' [a, b, c, d, e, f, g]join!
-  ok out.g, g
+  export
+    const h = 8
+    const i = 9
+  eq '1,2,3,4,5,6,7,8,9' ''+[a, b, c, d, e, f, g, h, i]
+  eq out.g, g
+  eq out.h, h
+  eq out.i, i
 
 compileThrows 'redeclaration of constant "a"' 2 '''
   const a = 0