mirror of
https://github.com/RGBCube/serenity
synced 2025-07-26 02:17:34 +00:00
AK: Add String::replace() functionality
This adds a replace functionality that replaces a string that contains occurences of a "needle" by a "replacement" value. With "all_occurences" enabled, all occurences are being replaced, otherwise only the first occurence is being replaced.
This commit is contained in:
parent
9d5d0261e1
commit
2577712a1c
3 changed files with 62 additions and 0 deletions
|
@ -320,6 +320,41 @@ bool String::equals_ignoring_case(const StringView& other) const
|
||||||
return StringUtils::equals_ignoring_case(view(), other);
|
return StringUtils::equals_ignoring_case(view(), other);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int String::replace(const String& needle, const String& replacement, bool all_occurences)
|
||||||
|
{
|
||||||
|
if (is_empty())
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
Vector<size_t> positions;
|
||||||
|
size_t start = 0, pos;
|
||||||
|
for (;;) {
|
||||||
|
const char* ptr = strstr(characters() + start, needle.characters());
|
||||||
|
if (!ptr)
|
||||||
|
break;
|
||||||
|
|
||||||
|
pos = ptr - characters();
|
||||||
|
positions.append(pos);
|
||||||
|
if (!all_occurences)
|
||||||
|
break;
|
||||||
|
|
||||||
|
start = pos + 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!positions.size())
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
StringBuilder b;
|
||||||
|
size_t lastpos = 0;
|
||||||
|
for (auto& pos : positions) {
|
||||||
|
b.append(substring_view(lastpos, pos - lastpos));
|
||||||
|
b.append(replacement);
|
||||||
|
lastpos = pos + needle.length();
|
||||||
|
}
|
||||||
|
b.append(substring_view(lastpos, length() - lastpos));
|
||||||
|
m_impl = StringImpl::create(b.build().characters());
|
||||||
|
return positions.size();
|
||||||
|
}
|
||||||
|
|
||||||
String escape_html_entities(const StringView& html)
|
String escape_html_entities(const StringView& html)
|
||||||
{
|
{
|
||||||
StringBuilder builder;
|
StringBuilder builder;
|
||||||
|
|
|
@ -214,6 +214,8 @@ public:
|
||||||
|
|
||||||
StringView view() const;
|
StringView view() const;
|
||||||
|
|
||||||
|
int replace(const String& needle, const String& replacement, bool all_occurences = false);
|
||||||
|
|
||||||
private:
|
private:
|
||||||
RefPtr<StringImpl> m_impl;
|
RefPtr<StringImpl> m_impl;
|
||||||
};
|
};
|
||||||
|
|
|
@ -159,4 +159,29 @@ TEST_CASE(flystring)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
TEST_CASE(replace)
|
||||||
|
{
|
||||||
|
String test_string = "Well, hello Friends!";
|
||||||
|
u32 replacements = test_string.replace("Friends", "Testers");
|
||||||
|
EXPECT(replacements == 1);
|
||||||
|
EXPECT(test_string == "Well, hello Testers!");
|
||||||
|
|
||||||
|
replacements = test_string.replace("ell", "e're", true);
|
||||||
|
EXPECT(replacements == 2);
|
||||||
|
EXPECT(test_string == "We're, he'reo Testers!");
|
||||||
|
|
||||||
|
replacements = test_string.replace("!", " :^)");
|
||||||
|
EXPECT(replacements == 1);
|
||||||
|
EXPECT(test_string == "We're, he'reo Testers :^)");
|
||||||
|
|
||||||
|
test_string = String("111._.111._.111");
|
||||||
|
replacements = test_string.replace("111", "|||", true);
|
||||||
|
EXPECT(replacements == 3);
|
||||||
|
EXPECT(test_string == "|||._.|||._.|||");
|
||||||
|
|
||||||
|
replacements = test_string.replace("|||", "111");
|
||||||
|
EXPECT(replacements == 1);
|
||||||
|
EXPECT(test_string == "111._.|||._.|||");
|
||||||
|
}
|
||||||
|
|
||||||
TEST_MAIN(String)
|
TEST_MAIN(String)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue