string: add replaceInString

hyprwm · Jun 8, 2024 · f73a28c · f73a28c
1 parent 1384081
commit f73a28c
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 1 deletion.
diff --git a/include/hyprutils/string/String.hpp b/include/hyprutils/string/String.hpp
@@ -6,5 +6,6 @@ namespace Hyprutils {
         // trims beginning and end of whitespace characters
         std::string trim(const std::string& in);
         bool isNumber(const std::string& str, bool allowfloat = false);
+        void replaceInString(std::string& string, const std::string& what, const std::string& to);
     };
 };
diff --git a/src/string/String.cpp b/src/string/String.cpp
@@ -56,3 +56,13 @@ bool Hyprutils::String::isNumber(const std::string& str, bool allowfloat) {
 
     return true;
 }
+
+void Hyprutils::String::replaceInString(std::string& string, const std::string& what, const std::string& to) {
+    if (string.empty())
+        return;
+    size_t pos = 0;
+    while ((pos = string.find(what, pos)) != std::string::npos) {
+        string.replace(pos, what.length(), to);
+        pos += to.length();
+    }
+}
diff --git a/tests/shared.hpp b/tests/shared.hpp
@@ -17,4 +17,4 @@ namespace Colors {
         ret = 1;                                                                                                                                                                   \
     } else {                                                                                                                                                                       \
         std::cout << Colors::GREEN << "Passed " << Colors::RESET << #expr << ". Got " << val << "\n";                                                                              \
-    }
+    }
diff --git a/tests/string.cpp b/tests/string.cpp
@@ -34,5 +34,9 @@ int main(int argc, char** argv, char** envp) {
     EXPECT(list[0], "hello");
     EXPECT(list[1], "world!");
 
+    std::string hello = "hello world!";
+    replaceInString(hello, "hello", "hi");
+    EXPECT(hello, "hi world!");
+
     return ret;
 }