Sora/Data/ImageCacheManager.swift


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38

import Combine
import SwiftUI

final class ImageCacheManager {
  // MARK: - Singleton
  static let shared = ImageCacheManager()

  // MARK: - Private Properties
  private let cache = URLCache(
    memoryCapacity: 100 * 1_024 * 1_024,  // 100 MB
    diskCapacity: 500 * 1_024 * 1_024,  // 500 MB
    directory: FileManager.default.urls(for: .cachesDirectory, in: .userDomainMask).first?
      .appendingPathComponent("SoraImageCache", isDirectory: true)
  )
  private var cancellables = Set<AnyCancellable>()
  private let downloadQueue = OperationQueue()

  // MARK: - Initialisation
  private init() {
    URLCache.shared = cache
    downloadQueue.maxConcurrentOperationCount = 5
  }

  // MARK: - Public Methods
  func preloadImages(_ urls: [URL]) {
    urls.forEach { url in
      downloadQueue.addOperation {
        URLSession.shared.dataTaskPublisher(for: url)
          .map { CachedURLResponse(response: $0.response, data: $0.data) }
          .sink(
            receiveCompletion: { _ in () },
            receiveValue: { [self] in cache.storeCachedResponse($0, for: URLRequest(url: url)) }
          )
          .store(in: &self.cancellables)
      }
    }
  }
}