Merge pull request #108 from ony3000/dogfooding

Fix line wrapping inside 'if' blocks

Author: ony3000

src/core-parts/finder.ts

@@ -162,7 +162,8 @@ export function findTargetClassNameNodes(ast: AST, options: ResolvedOptions): Cl
         recursiveProps = ['arguments'];
         break;
       }
-      case 'ConditionalExpression': {
+      case 'ConditionalExpression':
+      case 'IfStatement': {
         recursiveProps = ['consequent', 'alternate'];
         break;
       }

tests/babel/others/absolute.test.ts

@@ -290,6 +290,40 @@ const Foo = forwardRef(function Foo() {
     </div>
   );
 });
+`,
+    options: {
+      printWidth: 80,
+    },
+  },
+  {
+    name: "class name inside the 'if' block",
+    input: `
+function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div className="lorem ipsum dolor sit amet consectetur adipiscing elit proin ex massa hendrerit eu posuere eu volutpat id neque pellentesque">
+        content
+      </div>
+    );
+  }
+  return elem;
+}
+`,
+    output: `function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div
+        className="lorem ipsum dolor sit amet consectetur adipiscing elit proin
+          ex massa hendrerit eu posuere eu volutpat id neque pellentesque"
+      >
+        content
+      </div>
+    );
+  }
+  return elem;
+}
 `,
     options: {
       printWidth: 80,

tests/babel/others/relative.test.ts

@@ -285,6 +285,40 @@ const Foo = forwardRef(function Foo() {
     </div>
   );
 });
+`,
+    options: {
+      printWidth: 80,
+    },
+  },
+  {
+    name: "class name inside the 'if' block",
+    input: `
+function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div className="lorem ipsum dolor sit amet consectetur adipiscing elit proin ex massa hendrerit eu posuere eu volutpat id neque pellentesque">
+        content
+      </div>
+    );
+  }
+  return elem;
+}
+`,
+    output: `function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div
+        className="lorem ipsum dolor sit amet consectetur adipiscing elit proin ex massa hendrerit
+          eu posuere eu volutpat id neque pellentesque"
+      >
+        content
+      </div>
+    );
+  }
+  return elem;
+}
 `,
     options: {
       printWidth: 80,

tests/typescript/others/absolute.test.ts

@@ -290,6 +290,40 @@ const Foo = forwardRef(function Foo() {
     </div>
   );
 });
+`,
+    options: {
+      printWidth: 80,
+    },
+  },
+  {
+    name: "class name inside the 'if' block",
+    input: `
+function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div className="lorem ipsum dolor sit amet consectetur adipiscing elit proin ex massa hendrerit eu posuere eu volutpat id neque pellentesque">
+        content
+      </div>
+    );
+  }
+  return elem;
+}
+`,
+    output: `function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div
+        className="lorem ipsum dolor sit amet consectetur adipiscing elit proin
+          ex massa hendrerit eu posuere eu volutpat id neque pellentesque"
+      >
+        content
+      </div>
+    );
+  }
+  return elem;
+}
 `,
     options: {
       printWidth: 80,

tests/typescript/others/relative.test.ts

@@ -285,6 +285,40 @@ const Foo = forwardRef(function Foo() {
     </div>
   );
 });
+`,
+    options: {
+      printWidth: 80,
+    },
+  },
+  {
+    name: "class name inside the 'if' block",
+    input: `
+function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div className="lorem ipsum dolor sit amet consectetur adipiscing elit proin ex massa hendrerit eu posuere eu volutpat id neque pellentesque">
+        content
+      </div>
+    );
+  }
+  return elem;
+}
+`,
+    output: `function Foo() {
+  let elem;
+  if (true) {
+    elem = (
+      <div
+        className="lorem ipsum dolor sit amet consectetur adipiscing elit proin ex massa hendrerit
+          eu posuere eu volutpat id neque pellentesque"
+      >
+        content
+      </div>
+    );
+  }
+  return elem;
+}
 `,
     options: {
       printWidth: 80,