atom/packages/git-diff/lib/git-diff-view.js

const { CompositeDisposable } = require('atom');
const { repositoryForPath } = require('./helpers');

const MAX_BUFFER_LENGTH_TO_DIFF = 2 * 1024 * 1024;

module.exports = class GitDiffView {
  constructor(editor) {
    this.updateDiffs = this.updateDiffs.bind(this);
    this.editor = editor;
    this.subscriptions = new CompositeDisposable();
    this.decorations = {};
    this.markers = [];
  }

  start() {
    const editorElement = this.editor.getElement();

    this.subscribeToRepository();

    this.subscriptions.add(
      this.editor.onDidStopChanging(this.updateDiffs),
      this.editor.onDidChangePath(this.updateDiffs),
      atom.project.onDidChangePaths(() => this.subscribeToRepository()),
      atom.commands.add(editorElement, 'git-diff:move-to-next-diff', () =>
        this.moveToNextDiff()
      ),
      atom.commands.add(editorElement, 'git-diff:move-to-previous-diff', () =>
        this.moveToPreviousDiff()
      ),
      atom.config.onDidChange('git-diff.showIconsInEditorGutter', () =>
        this.updateIconDecoration()
      ),
      atom.config.onDidChange('editor.showLineNumbers', () =>
        this.updateIconDecoration()
      ),
      editorElement.onDidAttach(() => this.updateIconDecoration()),
      this.editor.onDidDestroy(() => {
        this.cancelUpdate();
        this.removeDecorations();
        this.subscriptions.dispose();
      })
    );

    this.updateIconDecoration();
    this.scheduleUpdate();
  }

  moveToNextDiff() {
    const cursorLineNumber = this.editor.getCursorBufferPosition().row + 1;
    let nextDiffLineNumber = null;
    let firstDiffLineNumber = null;
    if (this.diffs) {
      for (const { newStart } of this.diffs) {
        if (newStart > cursorLineNumber) {
          if (nextDiffLineNumber == null) nextDiffLineNumber = newStart - 1;
          nextDiffLineNumber = Math.min(newStart - 1, nextDiffLineNumber);
        }

        if (firstDiffLineNumber == null) firstDiffLineNumber = newStart - 1;
        firstDiffLineNumber = Math.min(newStart - 1, firstDiffLineNumber);
      }
    }

    // Wrap around to the first diff in the file
    if (
      atom.config.get('git-diff.wrapAroundOnMoveToDiff') &&
      nextDiffLineNumber == null
    ) {
      nextDiffLineNumber = firstDiffLineNumber;
    }

    this.moveToLineNumber(nextDiffLineNumber);
  }

  updateIconDecoration() {
    const gutter = this.editor.getElement().querySelector('.gutter');
    if (gutter) {
      if (
        atom.config.get('editor.showLineNumbers') &&
        atom.config.get('git-diff.showIconsInEditorGutter')
      ) {
        gutter.classList.add('git-diff-icon');
      } else {
        gutter.classList.remove('git-diff-icon');
      }
    }
  }

  moveToPreviousDiff() {
    const cursorLineNumber = this.editor.getCursorBufferPosition().row + 1;
    let previousDiffLineNumber = -1;
    let lastDiffLineNumber = -1;
    if (this.diffs) {
      for (const { newStart } of this.diffs) {
        if (newStart < cursorLineNumber) {
          previousDiffLineNumber = Math.max(
            newStart - 1,
            previousDiffLineNumber
          );
        }
        lastDiffLineNumber = Math.max(newStart - 1, lastDiffLineNumber);
      }
    }

    // Wrap around to the last diff in the file
    if (
      atom.config.get('git-diff.wrapAroundOnMoveToDiff') &&
      previousDiffLineNumber === -1
    ) {
      previousDiffLineNumber = lastDiffLineNumber;
    }

    this.moveToLineNumber(previousDiffLineNumber);
  }

  moveToLineNumber(lineNumber) {
    if (lineNumber != null && lineNumber >= 0) {
      this.editor.setCursorBufferPosition([lineNumber, 0]);
      this.editor.moveToFirstCharacterOfLine();
    }
  }

  subscribeToRepository() {
    this.repository = repositoryForPath(this.editor.getPath());
    if (this.repository) {
      this.subscriptions.add(
        this.repository.onDidChangeStatuses(() => {
          this.scheduleUpdate();
        })
      );
      this.subscriptions.add(
        this.repository.onDidChangeStatus(changedPath => {
          if (changedPath === this.editor.getPath()) this.scheduleUpdate();
        })
      );
    }
  }

  cancelUpdate() {
    clearImmediate(this.immediateId);
  }

  scheduleUpdate() {
    this.cancelUpdate();
    this.immediateId = setImmediate(this.updateDiffs);
  }

  updateDiffs() {
    if (this.editor.isDestroyed()) return;
    this.removeDecorations();
    const path = this.editor && this.editor.getPath();
    if (
      path &&
      this.editor.getBuffer().getLength() < MAX_BUFFER_LENGTH_TO_DIFF
    ) {
      this.diffs =
        this.repository &&
        this.repository.getLineDiffs(path, this.editor.getText());
      if (this.diffs) this.addDecorations(this.diffs);
    }
  }

  addDecorations(diffs) {
    for (const { newStart, oldLines, newLines } of diffs) {
      const startRow = newStart - 1;
      const endRow = newStart + newLines - 1;
      if (oldLines === 0 && newLines > 0) {
        this.markRange(startRow, endRow, 'git-line-added');
      } else if (newLines === 0 && oldLines > 0) {
        if (startRow < 0) {
          this.markRange(0, 0, 'git-previous-line-removed');
        } else {
          this.markRange(startRow, startRow, 'git-line-removed');
        }
      } else {
        this.markRange(startRow, endRow, 'git-line-modified');
      }
    }
  }

  removeDecorations() {
    for (let marker of this.markers) marker.destroy();
    this.markers = [];
  }

  markRange(startRow, endRow, klass) {
    const marker = this.editor.markBufferRange([[startRow, 0], [endRow, 0]], {
      invalidate: 'never'
    });
    this.editor.decorateMarker(marker, { type: 'line-number', class: klass });
    this.markers.push(marker);
  }
};