// Define search commands. Depends on dialog.js or another

// implementation of the openDialog method.



// Replace works a little oddly -- it will do the replace on the next

// Ctrl-G (or whatever is bound to findNext) press. You prevent a

// replace by making sure the match is no longer selected when hitting

// Ctrl-G.



(function() {

  function SearchState() {

    this.posFrom = this.posTo = this.query = null;

    this.marked = [];

  }

  function getSearchState(cm) {

    return cm._searchState || (cm._searchState = new SearchState());

  }

  function getSearchCursor(cm, query, pos) {

    // Heuristic: if the query string is all lowercase, do a case insensitive search.

    return cm.getSearchCursor(query, pos, typeof query == "string" && query == query.toLowerCase());

  }

  function dialog(cm, text, shortText, f) {

    if (cm.openDialog) cm.openDialog(text, f);

    else f(prompt(shortText, ""));

  }

  function confirmDialog(cm, text, shortText, fs) {

    if (cm.openConfirm) cm.openConfirm(text, fs);

    else if (confirm(shortText)) fs[0]();

  }

  function parseQuery(query) {

    var isRE = query.match(/^\/(.*)\/([a-z]*)$/);

    return isRE ? new RegExp(isRE[1], isRE[2].indexOf("i") == -1 ? "" : "i") : query;

  }

  var queryDialog =

    'Search: <input type="text" style="width: 10em"/> <span style="color: #888">(Use /re/ syntax for regexp search)</span>';

  function doSearch(cm, rev) {

    var state = getSearchState(cm);

    if (state.query) return findNext(cm, rev);

    dialog(cm, queryDialog, "Search for:", function(query) {

      cm.operation(function() {

        if (!query || state.query) return;

        state.query = parseQuery(query);

        if (cm.lineCount() < 2000) { // This is too expensive on big documents.

          for (var cursor = getSearchCursor(cm, state.query); cursor.findNext();)

            state.marked.push(cm.markText(cursor.from(), cursor.to(), "CodeMirror-searching"));

        }

        state.posFrom = state.posTo = cm.getCursor();

        findNext(cm, rev);

      });

    });

  }

  function findNext(cm, rev) {cm.operation(function() {

    var state = getSearchState(cm);

    var cursor = getSearchCursor(cm, state.query, rev ? state.posFrom : state.posTo);

    if (!cursor.find(rev)) {

      cursor = getSearchCursor(cm, state.query, rev ? {line: cm.lineCount() - 1} : {line: 0, ch: 0});

      if (!cursor.find(rev)) return;

    }

    cm.setSelection(cursor.from(), cursor.to());

    state.posFrom = cursor.from(); state.posTo = cursor.to();

  });}

  function clearSearch(cm) {cm.operation(function() {

    var state = getSearchState(cm);

    if (!state.query) return;

    state.query = null;

    for (var i = 0; i < state.marked.length; ++i) state.marked[i].clear();

    state.marked.length = 0;

  });}



  var replaceQueryDialog =

    'Replace: <input type="text" style="width: 10em"/> <span style="color: #888">(Use /re/ syntax for regexp search)</span>';

  var replacementQueryDialog = 'With: <input type="text" style="width: 10em"/>';

  var doReplaceConfirm = "Replace? <button>Yes</button> <button>No</button> <button>Stop</button>";

  function replace(cm, all) {

    dialog(cm, replaceQueryDialog, "Replace:", function(query) {

      if (!query) return;

      query = parseQuery(query);

      dialog(cm, replacementQueryDialog, "Replace with:", function(text) {

        if (all) {

          cm.compoundChange(function() { cm.operation(function() {

            for (var cursor = getSearchCursor(cm, query); cursor.findNext();) {

              if (typeof query != "string") {

                var match = cm.getRange(cursor.from(), cursor.to()).match(query);

                cursor.replace(text.replace(/\$(\d)/, function(w, i) {return match[i];}));

              } else cursor.replace(text);

            }

          });});

        } else {

          clearSearch(cm);

          var cursor = getSearchCursor(cm, query, cm.getCursor());

          function advance() {

            var start = cursor.from(), match;

            if (!(match = cursor.findNext())) {

              cursor = getSearchCursor(cm, query);

              if (!(match = cursor.findNext()) ||

                  (start && cursor.from().line == start.line && cursor.from().ch == start.ch)) return;

            }

            cm.setSelection(cursor.from(), cursor.to());

            confirmDialog(cm, doReplaceConfirm, "Replace?",

                          [function() {doReplace(match);}, advance]);

          }

          function doReplace(match) {

            cursor.replace(typeof query == "string" ? text :

                           text.replace(/\$(\d)/, function(w, i) {return match[i];}));

            advance();

          }

          advance();

        }

      });

    });

  }



  CodeMirror.commands.find = function(cm) {clearSearch(cm); doSearch(cm);};

  CodeMirror.commands.findNext = doSearch;

  CodeMirror.commands.findPrev = function(cm) {doSearch(cm, true);};

  CodeMirror.commands.clearSearch = clearSearch;

  CodeMirror.commands.replace = replace;

  CodeMirror.commands.replaceAll = function(cm) {replace(cm, true);};

})();

