aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--src/test.cpp6
-rw-r--r--src/utility_string.h15
2 files changed, 21 insertions, 0 deletions
diff --git a/src/test.cpp b/src/test.cpp
index 6588f113c..1b9dfcb5d 100644
--- a/src/test.cpp
+++ b/src/test.cpp
@@ -35,6 +35,7 @@ with this program; if not, write to the Free Software Foundation, Inc.,
#include "mapsector.h"
#include "settings.h"
#include "log.h"
+#include "utility_string.h"
/*
Asserts that the exception occurs
@@ -120,6 +121,11 @@ struct TestUtilities
assert(is_yes("YeS") == true);
assert(is_yes("") == false);
assert(is_yes("FAlse") == false);
+ const char *ends[] = {"abc", "c", "bc", NULL};
+ assert(removeStringEnd("abc", ends) == "");
+ assert(removeStringEnd("bc", ends) == "b");
+ assert(removeStringEnd("12c", ends) == "12");
+ assert(removeStringEnd("foo", ends) == "");
}
};
diff --git a/src/utility_string.h b/src/utility_string.h
index f29057ad7..df283ba69 100644
--- a/src/utility_string.h
+++ b/src/utility_string.h
@@ -31,5 +31,20 @@ static inline std::string padStringRight(std::string s, size_t len)
return s;
}
+// ends: NULL- or ""-terminated array of strings
+// Returns "" if no end could be removed.
+static inline std::string removeStringEnd(const std::string &s, const char *ends[])
+{
+ const char **p = ends;
+ for(; (*p) && (*p)[0] != '\0'; p++){
+ std::string end = *p;
+ if(s.size() < end.size())
+ continue;
+ if(s.substr(s.size()-end.size(), end.size()) == end)
+ return s.substr(0, s.size() - end.size());
+ }
+ return "";
+}
+
#endif