From 8fa1174b32ec6bb31d32843334c921d0e97da663 Mon Sep 17 00:00:00 2001
From: Alexander Kotliarskyi <alex.frantic@gmail.com>
Date: Mon, 16 May 2016 13:45:33 -0700
Subject: [PATCH] Skip adding __source if it already exists (#3504)

---
 .../src/index.js                              | 24 ++++++++++++++-----
 .../fixtures/react-source/with-source/exec.js |  8 +++++++
 2 files changed, 26 insertions(+), 6 deletions(-)
 create mode 100644 packages/babel-plugin-transform-react-jsx-source/test/fixtures/react-source/with-source/exec.js

diff --git a/packages/babel-plugin-transform-react-jsx-source/src/index.js b/packages/babel-plugin-transform-react-jsx-source/src/index.js
index c1cfae6fd7..cd296af979 100644
--- a/packages/babel-plugin-transform-react-jsx-source/src/index.js
+++ b/packages/babel-plugin-transform-react-jsx-source/src/index.js
@@ -28,6 +28,22 @@ export default function ({ types: t }) {
 
   let visitor = {
     JSXOpeningElement(path, state) {
+      const id = t.jSXIdentifier(TRACE_ID);
+      const location = path.container.openingElement.loc;
+      if (!location) {
+        // the element was generated and doesn't have location information
+        return;
+      }
+
+      const attributes = path.container.openingElement.attributes;
+      for (let i = 0; i < attributes.length; i++) {
+        const name = attributes[i].name;
+        if (name && name.name === TRACE_ID) {
+          // The __source attibute already exists
+          return;
+        }
+      }
+
       if (!state.fileNameIdentifier) {
         const fileName = state.file.log.filename !== "unknown"
           ? state.file.log.filename
@@ -38,12 +54,8 @@ export default function ({ types: t }) {
         state.fileNameIdentifier = fileNameIdentifier;
       }
 
-      const id = t.jSXIdentifier(TRACE_ID);
-      const location = path.container.openingElement.loc; // undefined for generated elements
-      if (location) {
-        const trace = makeTrace(state.fileNameIdentifier, location.start.line);
-        path.container.openingElement.attributes.push(t.jSXAttribute(id, t.jSXExpressionContainer(trace)));
-      }
+      const trace = makeTrace(state.fileNameIdentifier, location.start.line);
+      attributes.push(t.jSXAttribute(id, t.jSXExpressionContainer(trace)));
     }
   };
 
diff --git a/packages/babel-plugin-transform-react-jsx-source/test/fixtures/react-source/with-source/exec.js b/packages/babel-plugin-transform-react-jsx-source/test/fixtures/react-source/with-source/exec.js
new file mode 100644
index 0000000000..eaf60634ff
--- /dev/null
+++ b/packages/babel-plugin-transform-react-jsx-source/test/fixtures/react-source/with-source/exec.js
@@ -0,0 +1,8 @@
+var actual = transform(
+  'var x = <sometag __source="custom" />;',
+  Object.assign({}, opts, { filename: '/fake/path/mock.js' })
+).code;
+
+var expected = 'var x = <sometag __source="custom" />;';
+
+assert.equal(actual, expected);