From 0be93563ddca84fb251f3566f654c2aa16b15a56 Mon Sep 17 00:00:00 2001 From: Sebastian McKenzie Date: Sun, 26 Apr 2015 21:15:08 +0100 Subject: [PATCH] fix spec.blockScopedFunctions shouldVisit method --- .../transformers/spec/block-scoped-functions.js | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/src/babel/transformation/transformers/spec/block-scoped-functions.js b/src/babel/transformation/transformers/spec/block-scoped-functions.js index a1b17960f0..b5327d8c0f 100644 --- a/src/babel/transformation/transformers/spec/block-scoped-functions.js +++ b/src/babel/transformation/transformers/spec/block-scoped-functions.js @@ -24,12 +24,17 @@ function statementList(key, path, file) { } export function shouldVisit(node) { - if (node.type !== "BlockStatement") return false; - - for (var i = 0; i < node.body.length; i++) { - if (node.body[i].type === "FunctionDeclaration") return true; + var body; + if (node.type === "SwitchCase") { + body = node.consequent; + } else if (node.type === "BlockStatement") { + body = node.body; + } + if (body) { + for (var i = 0; i < body.length; i++) { + if (body[i].type === "FunctionDeclaration") return true; + } } - return false; }