Separate function to add code copy button

For neatness and also so it can show up separately in the profiler.
matrix-org · Sep 10, 2017 · 05a9863 · 05a9863
1 parent 876257f
commit 05a9863
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/src/components/views/messages/TextualBody.js b/src/components/views/messages/TextualBody.js
@@ -114,16 +114,7 @@ module.exports = React.createClass({
                 }, 10);
             }
 
-            // Add 'copy' buttons to pre blocks
-            ReactDOM.findDOMNode(this).querySelectorAll('.mx_EventTile_body pre').forEach((p) => {
-                const button = document.createElement("span");
-                button.className = "mx_EventTile_copyButton";
-                button.onclick = (e) => {
-                    const copyCode = button.parentNode.getElementsByTagName("code")[0];
-                    this.copyToClipboard(copyCode.textContent);
-                };
-                p.appendChild(button);
-            });
+            this._addCodeCopyButton();
         }
     },
 
@@ -260,6 +251,19 @@ module.exports = React.createClass({
         }
     },
 
+    _addCodeCopyButton() {
+        // Add 'copy' buttons to pre blocks
+        ReactDOM.findDOMNode(this).querySelectorAll('.mx_EventTile_body pre').forEach((p) => {
+            const button = document.createElement("span");
+            button.className = "mx_EventTile_copyButton";
+            button.onclick = (e) => {
+                const copyCode = button.parentNode.getElementsByTagName("code")[0];
+                this.copyToClipboard(copyCode.textContent);
+            };
+            p.appendChild(button);
+        });
+    },
+
     onCancelClick: function(event) {
         this.setState({ widgetHidden: true });
         // FIXME: persist this somewhere smarter than local storage