From 7a645489beb4cfd21de74864e594e7afa9a89605 Mon Sep 17 00:00:00 2001
From: Jackson Harper <jacksonh@gmail.com>
Date: Thu, 2 Jun 2022 10:48:25 -0700
Subject: [PATCH] Normalize URLs and match on URL rather than id when saving
 items locally

---
 .../Services/DataService/DataService.swift    | 54 +++++++++++++++++--
 1 file changed, 50 insertions(+), 4 deletions(-)

diff --git a/apple/OmnivoreKit/Sources/Services/DataService/DataService.swift b/apple/OmnivoreKit/Sources/Services/DataService/DataService.swift
index 752155036..0f8078d4b 100644
--- a/apple/OmnivoreKit/Sources/Services/DataService/DataService.swift
+++ b/apple/OmnivoreKit/Sources/Services/DataService/DataService.swift
@@ -10,6 +10,16 @@ import Utils
 
 let logger = Logger(subsystem: "app.omnivore", category: "data-service")
 
+private extension String {
+  func replacingRegex(pattern: String, replaceWith: String = "") -> String {
+    do {
+      let regex = try NSRegularExpression(pattern: pattern, options: [.caseInsensitive, .anchorsMatchLines])
+      let range = NSRange(location: 0, length: utf16.count)
+      return regex.stringByReplacingMatches(in: self, options: [], range: range, withTemplate: replaceWith)
+    } catch { return self }
+  }
+}
+
 public final class DataService: ObservableObject {
   public static var registerIntercomUser: ((String) -> Void)?
   public static var showIntercomMessenger: (() -> Void)?
@@ -109,19 +119,55 @@ public final class DataService: ObservableObject {
     return isFirstRunOfVersion || isFirstRunWithBuildNumber
   }
 
+  // based losesly on the normalize-url npm package which we use on the backend
+  func normalizeURL(_ dirtyURL: String) -> String {
+    var urlString = dirtyURL
+
+    urlString = urlString.trimmingCharacters(in: .whitespacesAndNewlines)
+
+    if var urlObject = URLComponents(string: urlString) {
+      // Remove auth
+      if /* options.stripAuthentication */ true {
+        urlObject.user = nil
+        urlObject.password = nil
+      }
+
+      // Remove hash
+      if /* options.stripHash */ true {
+        urlObject.fragment = nil
+      }
+
+      urlObject.queryItems = urlObject.queryItems?.filter { item in
+        item.name.starts(with: "utm_")
+      }
+
+      if /* options.removeTrailingSlash */ true {
+        urlObject.path = urlObject.path.replacingRegex(pattern: "/$", replaceWith: "")
+      }
+
+      if let finalUrl = urlObject.url {
+        return finalUrl.absoluteString
+      }
+    }
+
+    return dirtyURL
+  }
+
   public func persistPageScrapePayload(_ pageScrape: PageScrapePayload, requestId: String) async throws {
+    let normalizedURL = normalizeURL(pageScrape.url)
+
     try await backgroundContext.perform { [weak self] in
       guard let self = self else { return }
       let fetchRequest: NSFetchRequest<Models.LinkedItem> = LinkedItem.fetchRequest()
-      fetchRequest.predicate = NSPredicate(format: "id == %@", requestId)
+      fetchRequest.predicate = NSPredicate(format: "pageURLString = %@", normalizedURL)
 
       let currentTime = Date()
       let existingItem = try? self.backgroundContext.fetch(fetchRequest).first
       let linkedItem = existingItem ?? LinkedItem(entity: LinkedItem.entity(), insertInto: self.backgroundContext)
 
-      linkedItem.id = requestId
-      linkedItem.title = pageScrape.url
-      linkedItem.pageURLString = pageScrape.url
+      linkedItem.id = existingItem?.unwrappedID ?? requestId
+      linkedItem.title = normalizedURL
+      linkedItem.pageURLString = normalizedURL
       linkedItem.serverSyncStatus = Int64(ServerSyncStatus.needsCreation.rawValue)
       linkedItem.savedAt = currentTime
       linkedItem.createdAt = currentTime