diff options
| -rw-r--r-- | src/str.cc | 15 | ||||
| -rw-r--r-- | src/str.hh | 6 | ||||
| -rw-r--r-- | test/str.cc | 58 |
3 files changed, 79 insertions, 0 deletions
@@ -76,4 +76,19 @@ std::string_view trim(std::string_view str) { return str.substr(s, e - s); } +std::string_view ltrim(std::string_view str) { + size_t s = 0; + size_t const e = str.size(); + while (s < e && is_space(str[s])) + ++s; + return str.substr(s, e - s); +} + +std::string_view rtrim(std::string_view str) { + size_t e = str.size(); + while (e > 0 && is_space(str[e - 1])) + --e; + return str.substr(0, e); +} + } // namespace str @@ -23,6 +23,12 @@ void split(std::string_view str, std::vector<std::string_view>& out, [[nodiscard]] std::string_view trim(std::string_view str); +[[nodiscard]] +std::string_view ltrim(std::string_view str); + +[[nodiscard]] +std::string_view rtrim(std::string_view str); + } // namespace str #endif // STR_HH diff --git a/test/str.cc b/test/str.cc index 6d7edf2..d7d331f 100644 --- a/test/str.cc +++ b/test/str.cc @@ -65,3 +65,61 @@ TEST(str, trim) { ret = str::trim("\tfoo bar\r\n"); EXPECT_EQ("foo bar", ret); } + +TEST(str, ltrim) { + auto ret = str::ltrim(""); + EXPECT_EQ("", ret); + + ret = str::ltrim(" "); + EXPECT_EQ("", ret); + + ret = str::ltrim(" "); + EXPECT_EQ("", ret); + + ret = str::ltrim("foo"); + EXPECT_EQ("foo", ret); + + ret = str::ltrim(" foo"); + EXPECT_EQ("foo", ret); + + ret = str::ltrim(" foo "); + EXPECT_EQ("foo ", ret); + + ret = str::ltrim("foo "); + EXPECT_EQ("foo ", ret); + + ret = str::ltrim(" foo bar "); + EXPECT_EQ("foo bar ", ret); + + ret = str::ltrim("\tfoo bar\r\n"); + EXPECT_EQ("foo bar\r\n", ret); +} + +TEST(str, rtrim) { + auto ret = str::rtrim(""); + EXPECT_EQ("", ret); + + ret = str::rtrim(" "); + EXPECT_EQ("", ret); + + ret = str::rtrim(" "); + EXPECT_EQ("", ret); + + ret = str::rtrim("foo"); + EXPECT_EQ("foo", ret); + + ret = str::rtrim(" foo"); + EXPECT_EQ(" foo", ret); + + ret = str::rtrim(" foo "); + EXPECT_EQ(" foo", ret); + + ret = str::rtrim("foo "); + EXPECT_EQ("foo", ret); + + ret = str::rtrim(" foo bar "); + EXPECT_EQ(" foo bar", ret); + + ret = str::rtrim("\tfoo bar\r\n"); + EXPECT_EQ("\tfoo bar", ret); +} |
