diff options
Diffstat (limited to 'content_scripts')
| -rw-r--r-- | content_scripts/mode_find.coffee | 43 | ||||
| -rw-r--r-- | content_scripts/ui_component.coffee | 2 | 
2 files changed, 22 insertions, 23 deletions
| diff --git a/content_scripts/mode_find.coffee b/content_scripts/mode_find.coffee index 9b47cfbd..e863b553 100644 --- a/content_scripts/mode_find.coffee +++ b/content_scripts/mode_find.coffee @@ -115,30 +115,27 @@ class FindMode extends Mode      # default to 'smartcase' mode, unless noIgnoreCase is explicitly specified      @query.ignoreCase = !hasNoIgnoreCaseFlag && !Utils.hasUpperCase(@query.parsedQuery) -    # if we are dealing with a regex, grep for all matches in the text, and then call window.find() on them -    # sequentially so the browser handles the scrolling / text selection. -    if @query.isRegex -      try -        pattern = new RegExp(@query.parsedQuery, "g" + (if @query.ignoreCase then "i" else "")) -      catch error -        # if we catch a SyntaxError, assume the user is not done typing yet and return quietly -        return -      # innerText will not return the text of hidden elements, and strip out tags while preserving newlines -      text = document.body.innerText -      @query.regexMatches = text.match(pattern) -      @query.activeRegexIndex = 0 -      @query.matchCount = @query.regexMatches?.length -    # if we are doing a basic plain string match, we still want to grep for matches of the string, so we can -    # show a the number of results. We can grep on document.body.innerText, as it should be indistinguishable -    # from the internal representation used by window.find. +    regexPattern = if @query.isRegex +      @query.parsedQuery      else -      # escape all special characters, so RegExp just parses the string 'as is'. -      # Taken from http://stackoverflow.com/questions/3446170/escape-string-for-use-in-javascript-regex -      escapeRegExp = /[\-\[\]\/\{\}\(\)\*\+\?\.\\\^\$\|]/g -      parsedNonRegexQuery = @query.parsedQuery.replace(escapeRegExp, (char) -> "\\" + char) -      pattern = new RegExp(parsedNonRegexQuery, "g" + (if @query.ignoreCase then "i" else "")) -      text = document.body.innerText -      @query.matchCount = text.match(pattern)?.length +      Utils.escapeRegexSpecialCharacters @query.parsedQuery + +    # If we are dealing with a regex, grep for all matches in the text, and then call window.find() on them +    # sequentially so the browser handles the scrolling / text selection. +    # If we are doing a basic plain string match, we still want to grep for matches of the string, so we can +    # show a the number of results. +    try +      pattern = new RegExp regexPattern, "g#{if @query.ignoreCase then "i" else ""}" +    catch error +      return # If we catch a SyntaxError, assume the user is not done typing yet and return quietly. + +    # innerText will not return the text of hidden elements, and strip out tags while preserving newlines. +    # NOTE(mrmr1993): innerText doesn't include the text contents of <input>s and <textarea>s. See #1118. +    text = document.body.innerText +    regexMatches = text.match pattern +    @query.regexMatches = regexMatches if @query.isRegex +    @query.activeRegexIndex = 0 if @query.isRegex +    @query.matchCount = regexMatches?.length    @getNextQueryFromRegexMatches: (stepSize) ->      # find()ing an empty query always returns false diff --git a/content_scripts/ui_component.coffee b/content_scripts/ui_component.coffee index 72627f69..74b58b82 100644 --- a/content_scripts/ui_component.coffee +++ b/content_scripts/ui_component.coffee @@ -12,6 +12,8 @@ class UIComponent      # Default to everything hidden while the stylesheet loads.      styleSheet.innerHTML = "iframe {display: none;}" +    # Use an XMLHttpRequest, possibly via the background page, to fetch the stylesheet. This allows us to +    # catch and recover from failures that we could not have caught when using CSS @include (eg. #1817).      UIComponent::styleSheetGetter ?= new AsyncDataFetcher @fetchFileContents "content_scripts/vimium.css"      @styleSheetGetter.use (styles) -> styleSheet.innerHTML = styles | 
