Fixed #44, broken test, Fixed #33 update core

Author: Ivan Demidov

package.json

@@ -43,6 +43,7 @@
     "babel-register": "^6.7.2",
     "coveralls": "^2.11.8",
     "husky": "^0.11.4",
+    "lodash.merge": "^4.3.4",
     "nyc": "^6.1.1",
     "updtr": "^0.1.10",
     "xo": "^0.13.0"

src/index.js

@@ -1,46 +1,52 @@
 import postcss from 'postcss';
-import deepAssign from 'deep-assign';
+import {merge as assign} from 'lodash';
 
-const VAR_FUNC_IDENTIFIER = 'var';
-const maps = {};
+function hasVar(str) {
+	return str.indexOf('var(');
+}
 
-function resolveValue(value) {
-	const hasVar = value.indexOf(`${VAR_FUNC_IDENTIFIER}(`);
-	if (hasVar === -1) {
+function resolveValue(value, maps) {
+	if (hasVar(value) === -1) {
 		return value;
 	}
 
 	return value.replace(/var\(--.*?\)/g, (match) => maps[match.slice(4, -1)] || match);
 }
 
+function getProperty(nodes) {
+	let propertys = {};
+
+	nodes.walkRules(rule => {
+		if (rule.selector !== ':root') {
+			return;
+		}
+
+		rule.each(decl => {
+			const {prop, value} = decl;
+			propertys[prop] = value;
+		});
+	});
+
+	return propertys;
+}
+
 export default postcss.plugin('postcss-at-rules-variables', options => {
 	const DEFAULT = {
 		atRules: ['for', 'if', 'else', 'each']
 	};
 
-	const mergeOptions = deepAssign(DEFAULT, options, (a, b) => {
+	options = assign(DEFAULT, options, (a, b) => {
 		if (Array.isArray(a)) {
 			return a.concat(b);
 		}
 	});
 
 	return nodes => {
-		const reg = new RegExp(mergeOptions.atRules.join('|'));
-
-		nodes.walkRules(rule => {
-			if (rule.selector !== ':root') {
-				return;
-			}
-
-			rule.each(decl => {
-				const prop = decl.prop;
-				const value = decl.value;
-				maps[prop] = value;
-			});
-		});
+		const maps = getProperty(nodes);
+		const {atRules} = options;
 
-		nodes.walkAtRules(reg, rules => {
-			rules.params = resolveValue(rules.params);
+		nodes.walkAtRules(new RegExp(atRules.join('|')), rules => {
+			rules.params = resolveValue(rules.params, maps);
 		});
 	};
 });

test/test.js

@@ -43,8 +43,8 @@ test('it change two properties for @if, @else if', t => {
 });
 
 test('it change multi properties for @each', t => {
-	const expected = ':root{ --array: foo, bar, baz; } @each $val in foo, bar, baz @for foo, bar, baz';
-	const value = ':root{ --array: foo, bar, baz; } @each $val in var(--array) @for var(--array)';
+	const expected = ':root{ --array: foo, bar, baz; } @each $val in foo, bar, baz {} @for foo, bar, baz {}';
+	const value = ':root{ --array: foo, bar, baz; } @each $val in var(--array) {} @for var(--array) {}';
 	t.is(processing(value, {atRules: ['each']}), expected);
 });