support unary and binary expressions

__tests__/__fixtures__/complex/conditional-binary/code.js

@@ -0,0 +1,5 @@
+import { reduce } from '../../../../src/inline-loops.macro';
+
+function getStuff(array) {
+  return array ** reduce(array, (a, v) => a + v * 2);
+}

__tests__/__fixtures__/complex/conditional-binary/output.js

@@ -0,0 +1,13 @@
+function getStuff(array) {
+  return (
+    array **
+    (() => {
+      let _a = array[0];
+      for (let _key = 1, _length = array.length, _v; _key < _length; ++_key) {
+        _v = array[_key];
+        _a = _a + _v * 2;
+      }
+      return _a;
+    })()
+  );
+}

__tests__/__fixtures__/complex/conditional-unary/code.js

@@ -0,0 +1,5 @@
+import { reduce } from '../../../../src/inline-loops.macro';
+
+function getStuff(array) {
+  return array + +reduce(array, (a, v) => a + v * 2);
+}

__tests__/__fixtures__/complex/conditional-unary/output.js

@@ -0,0 +1,13 @@
+function getStuff(array) {
+  return (
+    array +
+    +(() => {
+      let _a = array[0];
+      for (let _key = 1, _length = array.length, _v; _key < _length; ++_key) {
+        _v = array[_key];
+        _a = _a + _v * 2;
+      }
+      return _a;
+    })()
+  );
+}

src/utils.ts

@@ -132,7 +132,10 @@ export function isConditionalUsage(path: Path<CallExpression>): boolean {
   const parentPath = path.parentPath;
 
   return (
-    parentPath.isConditionalExpression() || parentPath.isLogicalExpression()
+    parentPath.isConditionalExpression() ||
+    parentPath.isLogicalExpression() ||
+    parentPath.isUnaryExpression() ||
+    parentPath.isBinaryExpression()
   );
 }