From f0bd43881e1100f25c6a51251e749b286c4b16fe Mon Sep 17 00:00:00 2001
From: Kyle Holmberg <inbox@kylemh.com>
Date: Sat, 17 Sep 2022 13:42:58 +0000
Subject: [PATCH 1/4] add new relevant fixtures

---
 .../data/MultipleAllWithDisplayName.tsx       | 22 +++++++++++++++++++
 .../data/MultipleSomeWithDisplayName.tsx      | 18 +++++++++++++++
 .../MultipleWithNoExplicitDisplayName.tsx     | 14 ++++++++++++
 3 files changed, 54 insertions(+)
 create mode 100644 src/__tests__/data/MultipleAllWithDisplayName.tsx
 create mode 100644 src/__tests__/data/MultipleSomeWithDisplayName.tsx
 create mode 100644 src/__tests__/data/MultipleWithNoExplicitDisplayName.tsx

diff --git a/src/__tests__/data/MultipleAllWithDisplayName.tsx b/src/__tests__/data/MultipleAllWithDisplayName.tsx
new file mode 100644
index 00000000..8c892833
--- /dev/null
+++ b/src/__tests__/data/MultipleAllWithDisplayName.tsx
@@ -0,0 +1,22 @@
+import * as React from 'react';
+
+interface ButtonProps
+  extends Omit<React.ButtonHTMLAttributes<HTMLButtonElement>, 'type'> {}
+
+export const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  (props, ref) => <button {...props} ref={ref} type="button" />
+);
+
+Button.displayName = 'First';
+
+export const SubmitButton = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  (props, ref) => <button {...props} ref={ref} type="submit" />
+);
+
+SubmitButton.displayName = 'Second';
+
+export const ResetButton = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  (props, ref) => <button {...props} ref={ref} type="reset" />
+);
+
+ResetButton.displayName = 'Third';
diff --git a/src/__tests__/data/MultipleSomeWithDisplayName.tsx b/src/__tests__/data/MultipleSomeWithDisplayName.tsx
new file mode 100644
index 00000000..94f47d90
--- /dev/null
+++ b/src/__tests__/data/MultipleSomeWithDisplayName.tsx
@@ -0,0 +1,18 @@
+import * as React from 'react';
+
+interface ButtonProps
+  extends Omit<React.ButtonHTMLAttributes<HTMLButtonElement>, 'type'> {}
+
+export const Button1 = React.forwardRef<HTMLButtonElement, ButtonProps>(
+  (props, ref) => <button {...props} ref={ref} type="button" />
+);
+Button1.displayName = 'First';
+
+export const NoExplicitDisplayName1 = (props: any) => <div>Some text</div>;
+
+export const Button2 = (props: ButtonProps) => (
+  <button {...props} type="button" />
+);
+Button2.displayName = 'Second';
+
+export const NoExplicitDisplayName2 = (props: any) => <div>Some more text</div>;
diff --git a/src/__tests__/data/MultipleWithNoExplicitDisplayName.tsx b/src/__tests__/data/MultipleWithNoExplicitDisplayName.tsx
new file mode 100644
index 00000000..c3b8c653
--- /dev/null
+++ b/src/__tests__/data/MultipleWithNoExplicitDisplayName.tsx
@@ -0,0 +1,14 @@
+import * as React from 'react';
+
+interface ButtonProps
+  extends Omit<React.ButtonHTMLAttributes<HTMLButtonElement>, 'type'> {}
+
+export const Button = (props: ButtonProps) => (
+  <button {...props} type="button" />
+);
+export const SubmitButton = (props: ButtonProps) => (
+  <button {...props} type="submit" />
+);
+export const ResetButton = (props: ButtonProps) => (
+  <button {...props} type="reset" />
+);

From f5021d0a44c6bf7d23432067ff8e8943d2d2906b Mon Sep 17 00:00:00 2001
From: Kyle Holmberg <inbox@kylemh.com>
Date: Sat, 17 Sep 2022 13:43:20 +0000
Subject: [PATCH 2/4] adjust parser

---
 src/parser.ts | 33 ++++++++++++++++++++++++++++++++-
 1 file changed, 32 insertions(+), 1 deletion(-)

diff --git a/src/parser.ts b/src/parser.ts
index 47a86e5d..8fdb6e9c 100644
--- a/src/parser.ts
+++ b/src/parser.ts
@@ -1200,11 +1200,42 @@ function getTextValueOfFunctionProperty(
     .filter(statement => {
       const expr = (statement as ts.ExpressionStatement)
         .expression as ts.BinaryExpression;
+
+      const locals = Array.from(
+        (source as any).locals as [string, ts.Symbol][]
+      );
+      const hasOneLocalExport =
+        locals.filter(local => !!local[1].exports).length === 1;
+
+      if (hasOneLocalExport) {
+        return (
+          expr.left &&
+          (expr.left as ts.PropertyAccessExpression).name &&
+          (expr.left as ts.PropertyAccessExpression).name.escapedText ===
+            propertyName
+        );
+      }
+
+      /**
+       * Ensure the .displayName is for the currently processing function.
+       *
+       * This avoids the following situations:
+       *
+       *  - A file has multiple functions, one has `.displayName`, and all
+       *    functions ends up with that same `.displayName` value.
+       *
+       *  - A file has multiple functions, each with a different
+       *    `.displayName`, but the first is applied to all of them.
+       */
+      const flowNodeNameEscapedText = (statement as any)?.flowNode?.node?.name
+        ?.escapedText as false | ts.__String | undefined;
+
       return (
         expr.left &&
         (expr.left as ts.PropertyAccessExpression).name &&
         (expr.left as ts.PropertyAccessExpression).name.escapedText ===
-          propertyName
+          propertyName &&
+        flowNodeNameEscapedText === exp.escapedName
       );
     })
     .filter(statement => {

From c0836ade0661be5e31dffc8fd02249fabd4f6bb8 Mon Sep 17 00:00:00 2001
From: Kyle Holmberg <inbox@kylemh.com>
Date: Sat, 17 Sep 2022 13:43:30 +0000
Subject: [PATCH 3/4] add new tests

---
 src/__tests__/parser.ts | 42 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 42 insertions(+)

diff --git a/src/__tests__/parser.ts b/src/__tests__/parser.ts
index fab7b73e..9025b3cc 100644
--- a/src/__tests__/parser.ts
+++ b/src/__tests__/parser.ts
@@ -989,6 +989,48 @@ describe('parser', () => {
       const [parsed] = parse(fixturePath('StatefulDisplayNameFolder/index'));
       assert.equal(parsed.displayName, 'StatefulDisplayNameFolder');
     });
+
+    describe('multiple components in one', () => {
+      it('should parse all `displayName` properties correctly when all are explicitly defined.', () => {
+        const result = parse(fixturePath('MultipleAllWithDisplayName'));
+
+        // Ensure we're not missing any exports.
+        assert.equal(result.length, 3);
+
+        const [parsed1, parsed2, parsed3] = result;
+
+        assert.equal(parsed1.displayName, 'First');
+        assert.equal(parsed2.displayName, 'Second');
+        assert.equal(parsed3.displayName, 'Third');
+      });
+
+      it('should parse all `displayName` properties correctly when some are explicitly defined.', () => {
+        const result = parse(fixturePath('MultipleSomeWithDisplayName'));
+
+        // Ensure we're not missing any exports.
+        assert.equal(result.length, 4);
+
+        const [parsed1, parsed2, parsed3, parsed4] = result;
+
+        assert.equal(parsed1.displayName, 'First');
+        assert.equal(parsed2.displayName, 'NoExplicitDisplayName1');
+        assert.equal(parsed3.displayName, 'Second');
+        assert.equal(parsed4.displayName, 'NoExplicitDisplayName2');
+      });
+
+      it('should parse all `displayName` properties correctly when none are explicitly defined.', () => {
+        const result = parse(fixturePath('MultipleWithNoExplicitDisplayName'));
+
+        // Ensure we're not missing any exports.
+        assert.equal(result.length, 3);
+
+        const [parsed1, parsed2, parsed3] = result;
+
+        assert.equal(parsed1.displayName, 'Button');
+        assert.equal(parsed2.displayName, 'SubmitButton');
+        assert.equal(parsed3.displayName, 'ResetButton');
+      });
+    });
   });
 
   describe('Parser options', () => {

From ebc80bc17f09dd8cc3ca96663c547b6639e58b21 Mon Sep 17 00:00:00 2001
From: Kyle Holmberg <inbox@kylemh.com>
Date: Sat, 17 Sep 2022 13:46:57 +0000
Subject: [PATCH 4/4] fix CI script to run on PR pushes

---
 .github/workflows/main.yml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml
index 9cb4038d..6257f6da 100644
--- a/.github/workflows/main.yml
+++ b/.github/workflows/main.yml
@@ -1,6 +1,9 @@
 name: Node.js Package
 
 on:
+  push:
+    branches:
+      - '**'
   release:
     types: [created]