From eacdb6103dbcabe0804f9e7c028f1a656adeb41d Mon Sep 17 00:00:00 2001
From: kean <grebenyuk.alexander@gmail.com>
Date: Thu, 18 Apr 2024 13:03:54 -0400
Subject: [PATCH] Fix TCP connection issues due to too many URLSession
 instances being created

---
 Sources/CoreAPI/WordPressComRestApi.swift | 38 ++++++++++-------------
 1 file changed, 16 insertions(+), 22 deletions(-)

diff --git a/Sources/CoreAPI/WordPressComRestApi.swift b/Sources/CoreAPI/WordPressComRestApi.swift
index 6983e8c8..9e27bd07 100644
--- a/Sources/CoreAPI/WordPressComRestApi.swift
+++ b/Sources/CoreAPI/WordPressComRestApi.swift
@@ -152,12 +152,6 @@ open class WordPressComRestApi: NSObject {
         super.init()
     }
 
-    deinit {
-        for session in [urlSession, uploadURLSession] {
-            session.finishTasksAndInvalidate()
-        }
-    }
-
     /// Cancels all outgoing tasks asynchronously without invalidating the session.
     public func cancelTasks() {
         for session in [urlSession, uploadURLSession] {
@@ -321,6 +315,12 @@ open class WordPressComRestApi: NSObject {
             builder = builder.query(defaults: [URLQueryItem(name: localeKey, value: preferredLanguageIdentifier)])
         }
 
+        if let oAuthToken {
+            builder = builder.header(name: "Authorization", value: "Bearer \(oAuthToken)")
+        }
+        if let userAgent {
+            builder = builder.header(name: "User-Agent", value: userAgent)
+        }
         return builder
     }
 
@@ -333,34 +333,28 @@ open class WordPressComRestApi: NSObject {
 
     // MARK: - Async
 
-    private lazy var urlSession: URLSession = {
-        URLSession(configuration: sessionConfiguration(background: false))
-    }()
+    private var urlSession: URLSession { WordPressComRestApi.urlSession }
 
     private lazy var uploadURLSession: URLSession = {
-        let configuration = sessionConfiguration(background: backgroundUploads)
-        configuration.sharedContainerIdentifier = self.sharedContainerIdentifier
-        if configuration.identifier != nil {
-            return URLSession.backgroundSession(configuration: configuration)
-        } else {
-            return URLSession(configuration: configuration)
+        if backgroundUploads {
+            return makeBackgroundSession()
         }
+        return WordPressComRestApi.urlSession
     }()
 
-    private func sessionConfiguration(background: Bool) -> URLSessionConfiguration {
-        let configuration = background ? URLSessionConfiguration.background(withIdentifier: self.backgroundSessionIdentifier) : URLSessionConfiguration.default
+    private static var urlSession = URLSession(configuration: .default)
 
+    private func makeBackgroundSession() -> URLSession {
+        let configuration = URLSessionConfiguration.background(withIdentifier: self.backgroundSessionIdentifier)
         var additionalHeaders: [String: AnyObject] = [:]
-        if let oAuthToken = self.oAuthToken {
+        if let oAuthToken {
             additionalHeaders["Authorization"] = "Bearer \(oAuthToken)" as AnyObject
         }
-        if let userAgent = self.userAgent {
+        if let userAgent {
             additionalHeaders["User-Agent"] = userAgent as AnyObject
         }
-
         configuration.httpAdditionalHeaders = additionalHeaders
-
-        return configuration
+        return URLSession.backgroundSession(configuration: configuration)
     }
 
     func perform(