wip

babel-plugin-expose-private-functions-and-variables.js

@@ -13,6 +13,20 @@ module.exports = function ({types: t}) {
 		}
 	}
 
+	function createCheckExistIfStatement(identifiers) {
+		const nestedExpression = createNestedMemberExpression(identifiers);
+		return t.ifStatement(
+			t.unaryExpression('!', nestedExpression),
+			t.expressionStatement(
+				t.assignmentExpression(
+					'=',
+					nestedExpression,
+					t.objectExpression([])
+				)
+			)
+		);
+	}
+
 	return {
 		visitor: {
 			Program: {
@@ -67,38 +81,12 @@ module.exports = function ({types: t}) {
 					});
 
 					// insert check-exist if statements in the start of the file
-					for (let i = 0; i < dir.length - 1; i++) {
-						const nestedExpression = createNestedMemberExpression(dir.slice(i));
-						path.unshiftContainer('body',
-							t.ifStatement(
-								t.unaryExpression('!', nestedExpression),
-								t.expressionStatement(
-									t.assignmentExpression(
-										'=',
-										nestedExpression,
-										t.objectExpression([])
-									)
-								)
-							)
-						);
-					}
+					for (let i = 0; i < dir.length - 1; i++)
+						path.unshiftContainer('body', createCheckExistIfStatement(dir.slice(i)));
 
 					// append check-exist if statements in the end of the file
-					for (let i = dir.length - 2; i >= 0; i--) {
-						const nestedExpression = createNestedMemberExpression(dir.slice(i));
-						path.pushContainer('body',
-							t.ifStatement(
-								t.unaryExpression('!', nestedExpression),
-								t.expressionStatement(
-									t.assignmentExpression(
-										'=',
-										nestedExpression,
-										t.objectExpression([])
-									)
-								)
-							)
-						);
-					}
+					for (let i = dir.length - 2; i >= 0; i--)
+						path.pushContainer('body', createCheckExistIfStatement(dir.slice(i)));
 
 					// export private variable to `window.PACKAGE._._var = _var`
 					privateVars.forEach(v => {