diff --git a/src/main/java/net/starlark/java/eval/StringModule.java b/src/main/java/net/starlark/java/eval/StringModule.java
index f08113a635778c..5e6939f70964c4 100644
--- a/src/main/java/net/starlark/java/eval/StringModule.java
+++ b/src/main/java/net/starlark/java/eval/StringModule.java
@@ -1023,4 +1023,34 @@ public boolean startsWith(String self, Object sub, Object start, Object end)
private static boolean substringStartsWith(String str, int start, int end, String prefix) {
return start + prefix.length() <= end && str.startsWith(prefix, start);
}
+
+ @StarlarkMethod(
+ name = "removeprefix",
+ doc = "If the string starts with prefix
, returns a new string with the prefix "
+ + "removed. Otherwise, returns the string.",
+ parameters = {
+ @Param(name = "self", doc = "This string."),
+ @Param(name = "prefix", doc = "The prefix to remove if present."),
+ })
+ public String removePrefix(String self, String prefix) {
+ if (self.startsWith(prefix)) {
+ return self.substring(prefix.length());
+ }
+ return self;
+ }
+
+ @StarlarkMethod(
+ name = "removesuffix",
+ doc = "If the string ends with suffix
, returns a new string with the suffix "
+ + "removed. Otherwise, returns the string.",
+ parameters = {
+ @Param(name = "self", doc = "This string."),
+ @Param(name = "suffix", doc = "The suffix to remove if present."),
+ })
+ public String removeSuffix(String self, String suffix) {
+ if (self.endsWith(suffix)) {
+ return self.substring(0, self.length() - suffix.length());
+ }
+ return self;
+ }
}
diff --git a/src/test/java/net/starlark/java/eval/testdata/string_misc.star b/src/test/java/net/starlark/java/eval/testdata/string_misc.star
index 1d294da5d5c6f5..9f1408cabe70e3 100644
--- a/src/test/java/net/starlark/java/eval/testdata/string_misc.star
+++ b/src/test/java/net/starlark/java/eval/testdata/string_misc.star
@@ -177,3 +177,23 @@ assert_eq("abc" * 0, "")
assert_eq("abc" * -1, "")
assert_fails(lambda: "abc" * (1 << 35), "got 34359738368 for repeat, want value in signed 32-bit range")
assert_fails(lambda: "abc" * (1 << 30), "excessive repeat \\(3 \\* 1073741824 characters\\)")
+
+# removeprefix
+assert_eq("Apricot".removeprefix("Apr"), "icot")
+assert_eq("Apricot".removeprefix("apr"), "Apricot")
+assert_eq("Apricot".removeprefix("A"), "pricot")
+assert_eq("a".removeprefix(""), "a")
+assert_eq("".removeprefix(""), "")
+assert_eq("".removeprefix("a"), "")
+assert_eq("Apricot".removeprefix("pr"), "Apricot")
+assert_eq("AprApricot".removeprefix("Apr"), "Apricot")
+
+# removesuffix
+assert_eq("Apricot".removesuffix("cot"), "Apri")
+assert_eq("Apricot".removesuffix("Cot"), "Apricot")
+assert_eq("Apricot".removesuffix("t"), "Aprico")
+assert_eq("a".removesuffix(""), "a")
+assert_eq("".removesuffix(""), "")
+assert_eq("".removesuffix("a"), "")
+assert_eq("Apricot".removesuffix("co"), "Apricot")
+assert_eq("Apricotcot".removesuffix("cot"), "Apricot")