RequestModifier.swift 1.9 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. //
  2. // RequestModifier.swift
  3. // Kingfisher
  4. //
  5. // Created by Wei Wang on 2016/09/05.
  6. //
  7. // Copyright (c) 2018 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. /// Request modifier of image downloader.
  28. public protocol ImageDownloadRequestModifier {
  29. func modified(for request: URLRequest) -> URLRequest?
  30. }
  31. struct NoModifier: ImageDownloadRequestModifier {
  32. static let `default` = NoModifier()
  33. private init() {}
  34. func modified(for request: URLRequest) -> URLRequest? {
  35. return request
  36. }
  37. }
  38. public struct AnyModifier: ImageDownloadRequestModifier {
  39. let block: (URLRequest) -> URLRequest?
  40. public func modified(for request: URLRequest) -> URLRequest? {
  41. return block(request)
  42. }
  43. public init(modify: @escaping (URLRequest) -> URLRequest? ) {
  44. block = modify
  45. }
  46. }