fix: edge case prefix04

torusresearch · Jun 3, 2024 · 44ad391 · 44ad391
1 parent f20a23b
commit 44ad391
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/Sources/TorusUtils/Extensions/String+Extension.swift b/Sources/TorusUtils/Extensions/String+Extension.swift
@@ -24,15 +24,15 @@ extension String {
         return hasPrefix("04")
     }
 
-    func add04Prefix() -> String {
-        if !hasPrefix("04") {
+    func add04Prefix(targetLength: Int = 128) -> String {
+        if !hasPrefix("04") && self.count == targetLength{
             return "04" + self
         }
         return self
     }
 
-    func strip04Prefix() -> String {
-        if hasPrefix("04") {
+    func strip04Prefix(targetLength: Int = 130) -> String {
+        if hasPrefix("04") && self.count == targetLength {
             let indexStart = index(startIndex, offsetBy: 2)
             return String(self[indexStart...])
         }