URLTransform.swift 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465
  1. //
  2. // URLTransform.swift
  3. // ObjectMapper
  4. //
  5. // Created by Tristan Himmelman on 2014-10-27.
  6. //
  7. // The MIT License (MIT)
  8. //
  9. // Copyright (c) 2014-2016 Hearst
  10. //
  11. // Permission is hereby granted, free of charge, to any person obtaining a copy
  12. // of this software and associated documentation files (the "Software"), to deal
  13. // in the Software without restriction, including without limitation the rights
  14. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  15. // copies of the Software, and to permit persons to whom the Software is
  16. // furnished to do so, subject to the following conditions:
  17. //
  18. // The above copyright notice and this permission notice shall be included in
  19. // all copies or substantial portions of the Software.
  20. //
  21. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  22. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  23. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  24. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  25. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  26. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  27. // THE SOFTWARE.
  28. import Foundation
  29. open class URLTransform: TransformType {
  30. public typealias Object = URL
  31. public typealias JSON = String
  32. private let shouldEncodeURLString: Bool
  33. /**
  34. Initializes the URLTransform with an option to encode URL strings before converting them to an NSURL
  35. - parameter shouldEncodeUrlString: when true (the default) the string is encoded before passing
  36. to `NSURL(string:)`
  37. - returns: an initialized transformer
  38. */
  39. public init(shouldEncodeURLString: Bool = true) {
  40. self.shouldEncodeURLString = shouldEncodeURLString
  41. }
  42. open func transformFromJSON(_ value: Any?) -> URL? {
  43. guard let URLString = value as? String else { return nil }
  44. if !shouldEncodeURLString {
  45. return URL(string: URLString)
  46. }
  47. guard let escapedURLString = URLString.addingPercentEncoding(withAllowedCharacters: CharacterSet.urlQueryAllowed) else {
  48. return nil
  49. }
  50. return URL(string: escapedURLString)
  51. }
  52. open func transformToJSON(_ value: URL?) -> String? {
  53. if let URL = value {
  54. return URL.absoluteString
  55. }
  56. return nil
  57. }
  58. }