Check array items for surrogate methods (#93)

duckduckgo · May 2, 2022 · 80a5ce3 · 80a5ce3
1 parent e05c621
commit 80a5ce3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/Sources/BrowserServicesKit/ContentBlocking/UserScripts/SurrogatesUserScript.swift b/Sources/BrowserServicesKit/ContentBlocking/UserScripts/SurrogatesUserScript.swift
@@ -145,7 +145,10 @@ open class SurrogatesUserScript: NSObject, UserScript {
                 return ""
             }
             let instructionsRow = codeLines.removeFirst()
-            let pattern = instructionsRow.split(separator: " ")[0].split(separator: "/")[1]
+            guard let path = instructionsRow.split(separator: " ").first,
+                  let pattern = path.split(separator: "/").last else {
+                return ""
+            }
             let stringifiedFunction = codeLines.joined(separator: "\n")
             return "surrogates['\(pattern)'] = function () {\(stringifiedFunction)}"
         }