Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "extensions/browser/file_highlighter.h" | |
| 6 | |
| 5 #include <stack> | 7 #include <stack> |
| 6 | 8 |
| 7 #include "extensions/browser/manifest_highlighter.h" | 9 #include "base/strings/utf_string_conversions.h" |
| 10 #include "base/values.h" | |
| 8 | 11 |
| 9 namespace extensions { | 12 namespace extensions { |
| 10 | 13 |
| 11 namespace { | 14 namespace { |
| 12 | 15 |
| 16 // Keys for a highlighted dictionary. | |
| 17 const char kBeforeHighlightKey[] = "beforeHighlight"; | |
| 18 const char kHighlightKey[] = "highlight"; | |
| 19 const char kAfterHighlightKey[] = "afterHighlight"; | |
| 20 | |
| 13 // Increment |index| to the position of the next quote ('"') in |str|, skipping | 21 // Increment |index| to the position of the next quote ('"') in |str|, skipping |
| 14 // over any escaped quotes. If no next quote is found, |index| is set to | 22 // over any escaped quotes. If no next quote is found, |index| is set to |
| 15 // std::string::npos. Assumes |index| currently points to a quote. | 23 // std::string::npos. Assumes |index| currently points to a quote. |
| 16 void QuoteIncrement(const std::string& str, size_t* index) { | 24 void QuoteIncrement(const std::string& str, size_t* index) { |
| 17 size_t i = *index + 1; // Skip over the first quote. | 25 size_t i = *index + 1; // Skip over the first quote. |
| 18 bool found = false; | 26 bool found = false; |
| 19 while (!found && i < str.size()) { | 27 while (!found && i < str.size()) { |
| 20 if (str[i] == '\\') | 28 if (str[i] == '\\') |
| 21 i += 2; // if we find an escaped character, skip it. | 29 i += 2; // if we find an escaped character, skip it. |
| 22 else if (str[i] == '"') | 30 else if (str[i] == '"') |
| (...skipping 38 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 61 stack.push('}'); | 69 stack.push('}'); |
| 62 else if (!stack.empty() && c == stack.top()) | 70 else if (!stack.empty() && c == stack.top()) |
| 63 stack.pop(); | 71 stack.pop(); |
| 64 CommentSafeIncrement(str, index); | 72 CommentSafeIncrement(str, index); |
| 65 c = str[*index]; | 73 c = str[*index]; |
| 66 } while (!stack.empty() && *index < end); | 74 } while (!stack.empty() && *index < end); |
| 67 } | 75 } |
| 68 | 76 |
| 69 } // namespace | 77 } // namespace |
| 70 | 78 |
| 79 FileHighlighter::FileHighlighter(const std::string& contents) | |
| 80 : contents_(contents), start_(0u), end_(contents_.size()) { | |
| 81 } | |
| 82 | |
| 83 FileHighlighter::~FileHighlighter() { | |
| 84 } | |
| 85 | |
| 86 std::string FileHighlighter::GetBeforeFeature() const { | |
| 87 return contents_.substr(0, start_); | |
| 88 } | |
| 89 | |
| 90 std::string FileHighlighter::GetFeature() const { | |
| 91 return contents_.substr(start_, end_ - start_); | |
| 92 } | |
| 93 | |
| 94 std::string FileHighlighter::GetAfterFeature() const { | |
| 95 return contents_.substr(end_); | |
| 96 } | |
| 97 | |
| 98 void FileHighlighter::SetHighlightedRegions(base::DictionaryValue* dict) const { | |
| 99 std::string before_feature = GetBeforeFeature(); | |
| 100 if (!before_feature.empty()) | |
| 101 dict->SetString(kBeforeHighlightKey, base::UTF8ToUTF16(before_feature)); | |
| 102 | |
| 103 std::string feature = GetFeature(); | |
| 104 if (!feature.empty()) | |
| 105 dict->SetString(kHighlightKey, base::UTF8ToUTF16(feature)); | |
| 106 | |
| 107 std::string after_feature = GetAfterFeature(); | |
| 108 if (!after_feature.empty()) | |
| 109 dict->SetString(kAfterHighlightKey, base::UTF8ToUTF16(after_feature)); | |
| 110 } | |
| 111 | |
| 71 ManifestHighlighter::ManifestHighlighter(const std::string& manifest, | 112 ManifestHighlighter::ManifestHighlighter(const std::string& manifest, |
| 72 const std::string& key, | 113 const std::string& key, |
| 73 const std::string& specific) | 114 const std::string& specific) |
| 74 : manifest_(manifest), | 115 : FileHighlighter(manifest) { |
| 75 start_(manifest_.find('{') + 1), | 116 start_ = contents_.find('{') + 1; |
| 76 end_(manifest_.rfind('}')) { | 117 end_ = contents_.rfind('}'); |
| 77 Parse(key, specific); | 118 Parse(key, specific); |
| 78 } | 119 } |
| 79 | 120 |
| 80 ManifestHighlighter::~ManifestHighlighter() { | 121 ManifestHighlighter::~ManifestHighlighter() { |
| 81 } | 122 } |
| 82 | 123 |
| 83 std::string ManifestHighlighter::GetBeforeFeature() const { | |
| 84 return manifest_.substr(0, start_); | |
| 85 } | |
| 86 | |
| 87 std::string ManifestHighlighter::GetFeature() const { | |
| 88 return manifest_.substr(start_, end_ - start_); | |
| 89 } | |
| 90 | |
| 91 std::string ManifestHighlighter::GetAfterFeature() const { | |
| 92 return manifest_.substr(end_); | |
| 93 } | |
| 94 | 124 |
| 95 void ManifestHighlighter::Parse(const std::string& key, | 125 void ManifestHighlighter::Parse(const std::string& key, |
| 96 const std::string& specific) { | 126 const std::string& specific) { |
| 97 // First, try to find the bounds of the full key. | 127 // First, try to find the bounds of the full key. |
| 98 if (FindBounds(key, true) /* enforce at top level */ ) { | 128 if (FindBounds(key, true) /* enforce at top level */ ) { |
| 99 // If we succeed, and we have a specific location, find the bounds of the | 129 // If we succeed, and we have a specific location, find the bounds of the |
| 100 // specific. | 130 // specific. |
| 101 if (!specific.empty()) | 131 if (!specific.empty()) |
| 102 FindBounds(specific, false /* don't enforce at top level */ ); | 132 FindBounds(specific, false /* don't enforce at top level */ ); |
| 103 | 133 |
| 104 // We may have found trailing whitespace. Don't use base::TrimWhitespace, | 134 // We may have found trailing whitespace. Don't use base::TrimWhitespace, |
| 105 // because we want to keep any whitespace we find - just not highlight it. | 135 // because we want to keep any whitespace we find - just not highlight it. |
| 106 size_t trim = manifest_.find_last_not_of(" \t\n\r", end_ - 1); | 136 size_t trim = contents_.find_last_not_of(" \t\n\r", end_ - 1); |
| 107 if (trim < end_ && trim > start_) | 137 if (trim < end_ && trim > start_) |
| 108 end_ = trim + 1; | 138 end_ = trim + 1; |
| 109 } else { | 139 } else { |
| 110 // If we fail, then we set start to end so that the highlighted portion is | 140 // If we fail, then we set start to end so that the highlighted portion is |
| 111 // empty. | 141 // empty. |
| 112 start_ = end_; | 142 start_ = end_; |
| 113 } | 143 } |
| 114 } | 144 } |
| 115 | 145 |
| 116 bool ManifestHighlighter::FindBounds(const std::string& feature, | 146 bool ManifestHighlighter::FindBounds(const std::string& feature, |
| 117 bool enforce_at_top_level) { | 147 bool enforce_at_top_level) { |
| 118 char c = '\0'; | 148 char c = '\0'; |
| 119 while (start_ < end_) { | 149 while (start_ < end_) { |
| 120 c = manifest_[start_]; | 150 c = contents_[start_]; |
| 121 if (c == '"') { | 151 if (c == '"') { |
| 122 // The feature may be quoted. | 152 // The feature may be quoted. |
| 123 size_t quote_end = start_; | 153 size_t quote_end = start_; |
| 124 QuoteIncrement(manifest_, "e_end); | 154 QuoteIncrement(contents_, "e_end); |
| 125 if (manifest_.substr(start_ + 1, quote_end - 1 - start_) == feature) { | 155 if (contents_.substr(start_ + 1, quote_end - 1 - start_) == feature) { |
| 126 FindBoundsEnd(feature, quote_end + 1); | 156 FindBoundsEnd(feature, quote_end + 1); |
| 127 return true; | 157 return true; |
| 128 } else { | 158 } else { |
| 129 // If it's not the feature, then we can skip the quoted section. | 159 // If it's not the feature, then we can skip the quoted section. |
| 130 start_ = quote_end + 1; | 160 start_ = quote_end + 1; |
| 131 } | 161 } |
| 132 } else if (manifest_.substr(start_, feature.size()) == feature) { | 162 } else if (contents_.substr(start_, feature.size()) == feature) { |
| 133 FindBoundsEnd(feature, start_ + feature.size() + 1); | 163 FindBoundsEnd(feature, start_ + feature.size() + 1); |
| 134 return true; | 164 return true; |
| 135 } else if (enforce_at_top_level && (c == '{' || c == '[')) { | 165 } else if (enforce_at_top_level && (c == '{' || c == '[')) { |
| 136 // If we don't have to be at the top level, then we can skip any chunks | 166 // If we don't have to be at the top level, then we can skip any chunks |
| 137 // we find. | 167 // we find. |
| 138 ChunkIncrement(manifest_, &start_, end_); | 168 ChunkIncrement(contents_, &start_, end_); |
| 139 } else { | 169 } else { |
| 140 CommentSafeIncrement(manifest_, &start_); | 170 CommentSafeIncrement(contents_, &start_); |
| 141 } | 171 } |
| 142 } | 172 } |
| 143 return false; | 173 return false; |
| 144 } | 174 } |
| 145 | 175 |
| 146 void ManifestHighlighter::FindBoundsEnd(const std::string& feature, | 176 void ManifestHighlighter::FindBoundsEnd(const std::string& feature, |
| 147 size_t local_start) { | 177 size_t local_start) { |
| 148 char c = '\0'; | 178 char c = '\0'; |
| 149 while (local_start < end_) { | 179 while (local_start < end_) { |
| 150 c = manifest_[local_start]; | 180 c = contents_[local_start]; |
| 151 // We're done when we find a terminating character (i.e., either a comma or | 181 // We're done when we find a terminating character (i.e., either a comma or |
| 152 // an ending bracket. | 182 // an ending bracket. |
| 153 if (c == ',' || c == '}' || c == ']') { | 183 if (c == ',' || c == '}' || c == ']') { |
| 154 end_ = local_start; | 184 end_ = local_start; |
| 155 return; | 185 return; |
| 156 } | 186 } |
| 157 // We can skip any chunks we find, since we are looking for the end of the | 187 // We can skip any chunks we find, since we are looking for the end of the |
| 158 // current feature, and don't want to go any deeper. | 188 // current feature, and don't want to go any deeper. |
| 159 if (c == '"' || c == '{' || c == '[') | 189 if (c == '"' || c == '{' || c == '[') |
| 160 ChunkIncrement(manifest_, &local_start, end_); | 190 ChunkIncrement(contents_, &local_start, end_); |
| 161 else | 191 else |
| 162 CommentSafeIncrement(manifest_, &local_start); | 192 CommentSafeIncrement(contents_, &local_start); |
| 163 } | 193 } |
| 164 } | 194 } |
| 165 | 195 |
| 196 SourceHighlighter::SourceHighlighter(const std::string& contents, | |
| 197 size_t line_number) | |
| 198 : FileHighlighter(contents) { | |
| 199 Parse(line_number); | |
| 200 } | |
| 201 | |
| 202 SourceHighlighter::~SourceHighlighter() { | |
| 203 } | |
| 204 | |
| 205 void SourceHighlighter::Parse(size_t line_number) { | |
| 206 for (size_t i = 1; i < line_number; ++i) | |
| 207 start_ = contents_.find('\n', start_) + 1; | |
|
Finnur
2013/09/09 09:15:56
The +1 makes the check for npos on line 213 invali
Devlin
2013/09/09 16:56:03
Whoops! It'll actually work, given valid input (t
| |
| 208 end_ = contents_.find('\n', start_); | |
| 209 | |
| 210 // If we went off the end of the string (i.e., the line number was invalid), | |
| 211 // then move start and end to the end of the string, so that the highlighted | |
| 212 // portion is empty. | |
| 213 start_ = start_ == std::string::npos ? contents_.size() : start_; | |
| 214 end_ = end_ == std::string::npos ? contents_.size() : end_; | |
| 215 } | |
| 216 | |
| 166 } // namespace extensions | 217 } // namespace extensions |
| OLD | NEW |