From 6f9082db10e5823633ffc6cedc53f7e9760c0525 Mon Sep 17 00:00:00 2001
From: Michelle Tilley <binarymuse@github.com>
Date: Sun, 14 May 2017 13:11:52 +0200
Subject: [PATCH 1/3] Don't display large diffs by default

---
 lib/controllers/file-patch-controller.js      | 33 +++++++++
 styles/pane-view.less                         |  6 +-
 .../controllers/file-patch-controller.test.js | 73 +++++++++++++++++++
 3 files changed, 111 insertions(+), 1 deletion(-)

diff --git a/lib/controllers/file-patch-controller.js b/lib/controllers/file-patch-controller.js
index 1ba40a897d..13bb1c3d95 100644
--- a/lib/controllers/file-patch-controller.js
+++ b/lib/controllers/file-patch-controller.js
@@ -1,3 +1,5 @@
+import path from 'path';
+
 import React from 'react';
 import PropTypes from 'prop-types';
 
@@ -10,6 +12,7 @@ import FilePatchView from '../views/file-patch-view';
 
 export default class FilePatchController extends React.Component {
   static propTypes = {
+    largeDiffLineThreshold: PropTypes.number,
     activeWorkingDirectory: PropTypes.string,
     repository: PropTypes.object.isRequired,
     commandRegistry: PropTypes.object.isRequired,
@@ -27,9 +30,16 @@ export default class FilePatchController extends React.Component {
   }
 
   static defaultProps = {
+    largeDiffLineThreshold: 1000,
     switchboard: new Switchboard(),
   }
 
+  static confirmedLargeFilePatches = new Set()
+
+  static resetConfirmedLargeFilePatches() {
+    this.confirmedLargeFilePatches = new Set();
+  }
+
   constructor(props, context) {
     super(props, context);
 
@@ -55,6 +65,13 @@ export default class FilePatchController extends React.Component {
           <span className="icon icon-info">File has no contents</span>
         </div>
       );
+    } else if (!this.shouldDisplayLargeDiff(hunks)) {
+      return (
+        <div className="github-PaneView pane-item large-file-patch">
+          <p>This is a large diff. For performance reasons, it is not rendered by default.</p>
+          <button className="btn btn-primary" onClick={this.handleShowDiffClick}>Show Diff</button>
+        </div>
+      );
     } else {
       // NOTE: Outer div is required for etch to render elements correctly
       const filePath = this.props.filePatch.getPath();
@@ -83,6 +100,13 @@ export default class FilePatchController extends React.Component {
     }
   }
 
+  shouldDisplayLargeDiff(hunks) {
+    const lineCount = hunks.reduce((acc, hunk) => hunk.lines.length, 0);
+    const fullPath = path.join(this.props.repository.getWorkingDirectoryPath(), this.props.filePatch.getPath());
+    return lineCount < this.props.largeDiffLineThreshold ||
+      FilePatchController.confirmedLargeFilePatches.has(fullPath);
+  }
+
   onDidChangeTitle(callback) {
     return this.emitter.on('did-change-title', callback);
   }
@@ -91,6 +115,15 @@ export default class FilePatchController extends React.Component {
     return this.emitter.on('did-destroy', callback);
   }
 
+  @autobind
+  handleShowDiffClick() {
+    if (this.props.repository) {
+      const fullPath = path.join(this.props.repository.getWorkingDirectoryPath(), this.props.filePatch.getPath());
+      FilePatchController.confirmedLargeFilePatches.add(fullPath);
+      this.forceUpdate();
+    }
+  }
+
   async stageHunk(hunk) {
     this.props.switchboard.didBeginStageOperation({stage: true, hunk: true});
 
diff --git a/styles/pane-view.less b/styles/pane-view.less
index d5160d6b09..e18fdf118d 100644
--- a/styles/pane-view.less
+++ b/styles/pane-view.less
@@ -5,11 +5,15 @@
   display: flex;
   height: 100%;
 
-  &.is-blank {
+  &.is-blank, &.large-file-patch {
     align-items: center;
     justify-content: center;
     text-align: center;
     font-size: 1.2em;
     padding: @component-padding;
   }
+
+  &.large-file-patch {
+    flex-direction: column;
+  }
 }
diff --git a/test/controllers/file-patch-controller.test.js b/test/controllers/file-patch-controller.test.js
index 3641e36546..4b6f627d8d 100644
--- a/test/controllers/file-patch-controller.test.js
+++ b/test/controllers/file-patch-controller.test.js
@@ -33,8 +33,11 @@ describe('FilePatchController', function() {
 
     const filePatch = new FilePatch('a.txt', 'a.txt', 'modified', [new Hunk(1, 1, 1, 3, '', [])]);
     const repository = Repository.absent();
+    sinon.stub(repository, 'getWorkingDirectoryPath').returns('/absent');
     const resolutionProgress = new ResolutionProgress();
 
+    FilePatchController.resetConfirmedLargeFilePatches();
+
     component = (
       <FilePatchController
         commandRegistry={commandRegistry}
@@ -79,6 +82,76 @@ describe('FilePatchController', function() {
     assert.isTrue(changeHandler.calledWith(actualTitle));
   });
 
+  describe('when the FilePatch has many lines', function() {
+    it('renders a confirmation widget', function() {
+      const hunk1 = new Hunk(0, 0, 1, 1, '', [
+        new HunkLine('line-1', 'added', 1, 1),
+        new HunkLine('line-2', 'added', 2, 2),
+        new HunkLine('line-3', 'added', 3, 3),
+        new HunkLine('line-4', 'added', 4, 4),
+        new HunkLine('line-5', 'added', 5, 5),
+        new HunkLine('line-6', 'added', 6, 6),
+      ]);
+      const filePatch = new FilePatch('a.txt', 'a.txt', 'modified', [hunk1]);
+
+      const wrapper = mount(React.cloneElement(component, {
+        filePatch, largeDiffLineThreshold: 5,
+      }));
+
+      assert.isFalse(wrapper.find('FilePatchView').exists());
+      assert.match(wrapper.text(), /large diff/);
+    });
+
+    it('renders the full diff when the confirmation is clicked', function() {
+      const hunk = new Hunk(0, 0, 1, 1, '', [
+        new HunkLine('line-1', 'added', 1, 1),
+        new HunkLine('line-2', 'added', 2, 2),
+        new HunkLine('line-3', 'added', 3, 3),
+        new HunkLine('line-4', 'added', 4, 4),
+        new HunkLine('line-5', 'added', 5, 5),
+        new HunkLine('line-6', 'added', 6, 6),
+      ]);
+      const filePatch = new FilePatch('a.txt', 'a.txt', 'modified', [hunk]);
+
+      const wrapper = mount(React.cloneElement(component, {
+        filePatch, largeDiffLineThreshold: 5,
+      }));
+
+      assert.isFalse(wrapper.find('FilePatchView').exists());
+
+      // wrapper.find('button').simulate('click');
+      // assert.isTrue(wrapper.find('FilePatchView').exists());
+    });
+
+    it('renders the full diff if the file has been confirmed before', async function() {
+      const hunk = new Hunk(0, 0, 1, 1, '', [
+        new HunkLine('line-1', 'added', 1, 1),
+        new HunkLine('line-2', 'added', 2, 2),
+        new HunkLine('line-3', 'added', 3, 3),
+        new HunkLine('line-4', 'added', 4, 4),
+        new HunkLine('line-5', 'added', 5, 5),
+        new HunkLine('line-6', 'added', 6, 6),
+      ]);
+      const filePatch1 = new FilePatch('a.txt', 'a.txt', 'modified', [hunk]);
+      const filePatch2 = new FilePatch('b.txt', 'b.txt', 'modified', [hunk]);
+
+      const wrapper = mount(React.cloneElement(component, {
+        filePatch: filePatch1, largeDiffLineThreshold: 5,
+      }));
+
+      assert.isFalse(wrapper.find('FilePatchView').exists());
+
+      wrapper.find('button').simulate('click');
+      assert.isTrue(wrapper.find('FilePatchView').exists());
+
+      wrapper.setProps({filePatch: filePatch2});
+      await assert.async.isFalse(wrapper.find('FilePatchView').exists());
+
+      wrapper.setProps({filePatch: filePatch1});
+      assert.isTrue(wrapper.find('FilePatchView').exists());
+    });
+  });
+
   it('renders FilePatchView only if FilePatch has hunks', function() {
     const emptyFilePatch = new FilePatch('a.txt', 'a.txt', 'modified', []);
 

From 4d53fdc6f15e01dd5b6ae1e6d1c1b9165c2ac672 Mon Sep 17 00:00:00 2001
From: Michelle Tilley <binarymuse@github.com>
Date: Sun, 14 May 2017 13:17:02 +0200
Subject: [PATCH 2/3] :keyboard:

---
 test/controllers/file-patch-controller.test.js | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/test/controllers/file-patch-controller.test.js b/test/controllers/file-patch-controller.test.js
index 4b6f627d8d..bc5e1e4ef4 100644
--- a/test/controllers/file-patch-controller.test.js
+++ b/test/controllers/file-patch-controller.test.js
@@ -119,8 +119,8 @@ describe('FilePatchController', function() {
 
       assert.isFalse(wrapper.find('FilePatchView').exists());
 
-      // wrapper.find('button').simulate('click');
-      // assert.isTrue(wrapper.find('FilePatchView').exists());
+      wrapper.find('button').simulate('click');
+      assert.isTrue(wrapper.find('FilePatchView').exists());
     });
 
     it('renders the full diff if the file has been confirmed before', async function() {

From d41c422d92a504590ff0326c8ac8e1e553f24c2c Mon Sep 17 00:00:00 2001
From: Michelle Tilley <binarymuse@github.com>
Date: Sun, 14 May 2017 22:12:53 +0200
Subject: [PATCH 3/3] :art:

---
 lib/controllers/file-patch-controller.js | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/lib/controllers/file-patch-controller.js b/lib/controllers/file-patch-controller.js
index 13bb1c3d95..a0953eba60 100644
--- a/lib/controllers/file-patch-controller.js
+++ b/lib/controllers/file-patch-controller.js
@@ -65,7 +65,7 @@ export default class FilePatchController extends React.Component {
           <span className="icon icon-info">File has no contents</span>
         </div>
       );
-    } else if (!this.shouldDisplayLargeDiff(hunks)) {
+    } else if (!this.shouldDisplayLargeDiff(this.props.filePatch)) {
       return (
         <div className="github-PaneView pane-item large-file-patch">
           <p>This is a large diff. For performance reasons, it is not rendered by default.</p>
@@ -100,11 +100,14 @@ export default class FilePatchController extends React.Component {
     }
   }
 
-  shouldDisplayLargeDiff(hunks) {
-    const lineCount = hunks.reduce((acc, hunk) => hunk.lines.length, 0);
+  shouldDisplayLargeDiff(filePatch) {
     const fullPath = path.join(this.props.repository.getWorkingDirectoryPath(), this.props.filePatch.getPath());
-    return lineCount < this.props.largeDiffLineThreshold ||
-      FilePatchController.confirmedLargeFilePatches.has(fullPath);
+    if (FilePatchController.confirmedLargeFilePatches.has(fullPath)) {
+      return true;
+    }
+
+    const lineCount = filePatch.getHunks().reduce((acc, hunk) => hunk.getLines().length, 0);
+    return lineCount < this.props.largeDiffLineThreshold;
   }
 
   onDidChangeTitle(callback) {