summaryrefslogtreecommitdiffstats
path: root/devtools/client/shared/sourceeditor/codemirror/addon/search/matchesonscrollbar.js
blob: 0b505947d969d2828861f242e48380d6fa4b68a9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
// CodeMirror, copyright (c) by Marijn Haverbeke and others
// Distributed under an MIT license: https://codemirror.net/LICENSE

(function(mod) {
    if (typeof exports == "object" && typeof module == "object") // CommonJS
      mod(require("../../lib/codemirror"), require("./searchcursor"), require("../scroll/annotatescrollbar"));
    else if (typeof define == "function" && define.amd) // AMD
      define(["../../lib/codemirror", "./searchcursor", "../scroll/annotatescrollbar"], mod);
    else // Plain browser env
      mod(CodeMirror);
  })(function(CodeMirror) {
    "use strict";
  
    CodeMirror.defineExtension("showMatchesOnScrollbar", function(query, caseFold, options) {
      if (typeof options == "string") options = {className: options};
      if (!options) options = {};
      return new SearchAnnotation(this, query, caseFold, options);
    });
  
    function SearchAnnotation(cm, query, caseFold, options) {
      this.cm = cm;
      this.options = options;
      var annotateOptions = {listenForChanges: false};
      for (var prop in options) annotateOptions[prop] = options[prop];
      if (!annotateOptions.className) annotateOptions.className = "CodeMirror-search-match";
      this.annotation = cm.annotateScrollbar(annotateOptions);
      this.query = query;
      this.caseFold = caseFold;
      this.gap = {from: cm.firstLine(), to: cm.lastLine() + 1};
      this.matches = [];
      this.update = null;
  
      this.findMatches();
      this.annotation.update(this.matches);
  
      var self = this;
      cm.on("change", this.changeHandler = function(_cm, change) { self.onChange(change); });
    }
  
    var MAX_MATCHES = 1000;
  
    SearchAnnotation.prototype.findMatches = function() {
      if (!this.gap) return;
      for (var i = 0; i < this.matches.length; i++) {
        var match = this.matches[i];
        if (match.from.line >= this.gap.to) break;
        if (match.to.line >= this.gap.from) this.matches.splice(i--, 1);
      }
      var cursor = this.cm.getSearchCursor(this.query, CodeMirror.Pos(this.gap.from, 0), {caseFold: this.caseFold, multiline: this.options.multiline});
      var maxMatches = this.options && this.options.maxMatches || MAX_MATCHES;
      while (cursor.findNext()) {
        var match = {from: cursor.from(), to: cursor.to()};
        if (match.from.line >= this.gap.to) break;
        this.matches.splice(i++, 0, match);
        if (this.matches.length > maxMatches) break;
      }
      this.gap = null;
    };
  
    function offsetLine(line, changeStart, sizeChange) {
      if (line <= changeStart) return line;
      return Math.max(changeStart, line + sizeChange);
    }
  
    SearchAnnotation.prototype.onChange = function(change) {
      var startLine = change.from.line;
      var endLine = CodeMirror.changeEnd(change).line;
      var sizeChange = endLine - change.to.line;
      if (this.gap) {
        this.gap.from = Math.min(offsetLine(this.gap.from, startLine, sizeChange), change.from.line);
        this.gap.to = Math.max(offsetLine(this.gap.to, startLine, sizeChange), change.from.line);
      } else {
        this.gap = {from: change.from.line, to: endLine + 1};
      }
  
      if (sizeChange) for (var i = 0; i < this.matches.length; i++) {
        var match = this.matches[i];
        var newFrom = offsetLine(match.from.line, startLine, sizeChange);
        if (newFrom != match.from.line) match.from = CodeMirror.Pos(newFrom, match.from.ch);
        var newTo = offsetLine(match.to.line, startLine, sizeChange);
        if (newTo != match.to.line) match.to = CodeMirror.Pos(newTo, match.to.ch);
      }
      clearTimeout(this.update);
      var self = this;
      this.update = setTimeout(function() { self.updateAfterChange(); }, 250);
    };
  
    SearchAnnotation.prototype.updateAfterChange = function() {
      this.findMatches();
      this.annotation.update(this.matches);
    };
  
    SearchAnnotation.prototype.clear = function() {
      this.cm.off("change", this.changeHandler);
      this.annotation.clear();
    };
  });