--wip--

tonyhffong · May 5, 2017 · cb528c5 · cb528c5
1 parent d5578f3
commit cb528c5
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/SemanticASTs/functiondef.jl b/src/SemanticASTs/functiondef.jl
@@ -151,3 +151,16 @@ function show(io::IO, ast::MethodDefinition)
     end
     print("code=", code(ast), ")")
 end
+
+function annotateⁿ(::Type{FunctionDefinition}, ast::AST)
+    if isexpr(ast, :function) && length(ast.args) == 2
+        fn = ast.args[1]
+        body = ast.args[2]
+        if isexpr(fn, :tuple)
+            Lambda(signature=fn, code=SemanticAST(body))
+        elseif isexpr(fn, :call)
+            fnname = annotateⁿ(FunctionName, fn)
+            MethodDefinition(func=fnname, signature=fn, code=SemanticAST(body))
+        end
+    end
+end