Resource.swift 4.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115
  1. //
  2. // Resource.swift
  3. // Kingfisher
  4. //
  5. // Created by Wei Wang on 15/4/6.
  6. //
  7. // Copyright (c) 2019 Wei Wang <onevcat@gmail.com>
  8. //
  9. // Permission is hereby granted, free of charge, to any person obtaining a copy
  10. // of this software and associated documentation files (the "Software"), to deal
  11. // in the Software without restriction, including without limitation the rights
  12. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  13. // copies of the Software, and to permit persons to whom the Software is
  14. // furnished to do so, subject to the following conditions:
  15. //
  16. // The above copyright notice and this permission notice shall be included in
  17. // all copies or substantial portions of the Software.
  18. //
  19. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  20. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  22. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  24. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  25. // THE SOFTWARE.
  26. import Foundation
  27. /// Represents an image resource at a certain url and a given cache key.
  28. /// Kingfisher will use a `Resource` to download a resource from network and cache it with the cache key when
  29. /// using `Source.network` as its image setting source.
  30. public protocol Resource {
  31. /// The key used in cache.
  32. var cacheKey: String { get }
  33. /// The target image URL.
  34. var downloadURL: URL { get }
  35. }
  36. extension Resource {
  37. /// Converts `self` to a valid `Source` based on its `downloadURL` scheme. A `.provider` with
  38. /// `LocalFileImageDataProvider` associated will be returned if the URL points to a local file. Otherwise,
  39. /// `.network` is returned.
  40. public func convertToSource(overrideCacheKey: String? = nil) -> Source {
  41. let key = overrideCacheKey ?? cacheKey
  42. return downloadURL.isFileURL ?
  43. .provider(LocalFileImageDataProvider(fileURL: downloadURL, cacheKey: key)) :
  44. .network(ImageResource(downloadURL: downloadURL, cacheKey: key))
  45. }
  46. }
  47. /// ImageResource is a simple combination of `downloadURL` and `cacheKey`.
  48. /// When passed to image view set methods, Kingfisher will try to download the target
  49. /// image from the `downloadURL`, and then store it with the `cacheKey` as the key in cache.
  50. public struct ImageResource: Resource {
  51. // MARK: - Initializers
  52. /// Creates an image resource.
  53. ///
  54. /// - Parameters:
  55. /// - downloadURL: The target image URL from where the image can be downloaded.
  56. /// - cacheKey: The cache key. If `nil`, Kingfisher will use the `absoluteString` of `downloadURL` as the key.
  57. /// Default is `nil`.
  58. public init(downloadURL: URL, cacheKey: String? = nil) {
  59. self.downloadURL = downloadURL
  60. self.cacheKey = cacheKey ?? downloadURL.cacheKey
  61. }
  62. // MARK: Protocol Conforming
  63. /// The key used in cache.
  64. public let cacheKey: String
  65. /// The target image URL.
  66. public let downloadURL: URL
  67. }
  68. /// URL conforms to `Resource` in Kingfisher.
  69. /// The `absoluteString` of this URL is used as `cacheKey`. And the URL itself will be used as `downloadURL`.
  70. /// If you need customize the url and/or cache key, use `ImageResource` instead.
  71. extension URL: Resource {
  72. public var cacheKey: String { return isFileURL ? localFileCacheKey : absoluteString }
  73. public var downloadURL: URL { return self }
  74. }
  75. extension URL {
  76. static let localFileCacheKeyPrefix = "kingfisher.local.cacheKey"
  77. // The special version of cache key for a local file on disk. Every time the app is reinstalled on the disk,
  78. // the system assigns a new container folder to hold the .app (and the extensions, .appex) folder. So the URL for
  79. // the same image in bundle might be different.
  80. //
  81. // This getter only uses the fixed part in the URL (until the bundle name folder) to provide a stable cache key
  82. // for the image under the same path inside the bundle.
  83. //
  84. // See #1825 (https://github.com/onevcat/Kingfisher/issues/1825)
  85. var localFileCacheKey: String {
  86. var validComponents: [String] = []
  87. for part in pathComponents.reversed() {
  88. validComponents.append(part)
  89. if part.hasSuffix(".app") || part.hasSuffix(".appex") {
  90. break
  91. }
  92. }
  93. let fixedPath = "\(Self.localFileCacheKeyPrefix)/\(validComponents.reversed().joined(separator: "/"))"
  94. if let q = query {
  95. return "\(fixedPath)?\(q)"
  96. } else {
  97. return fixedPath
  98. }
  99. }
  100. }