chore: Adding extension to strip first prefix from string (#397)

smithy-lang · Jan 20, 2022 · 0faded8 · 0faded8
1 parent 567abc3
commit 0faded8
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/Packages/ClientRuntime/Sources/PrimitiveTypeExtensions/String+Extensions.swift b/Packages/ClientRuntime/Sources/PrimitiveTypeExtensions/String+Extensions.swift
@@ -113,3 +113,14 @@ extension String {
         return ByteBuffer(data: self.data(using: .utf8) ?? Data())
     }
 }
+
+public extension String {
+    func stripFirstMatching(prefixes: [String]) -> String {
+        for prefixToStrip in prefixes {
+            if self.hasPrefix(prefixToStrip) {
+                return self.removePrefix(prefixToStrip)
+            }
+        }
+        return self
+    }
+}