You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

73 lines
2.5 KiB

  1. //
  2. // CodableTransform.swift
  3. // ObjectMapper
  4. //
  5. // Created by Jari Kalinainen on 10/10/2018.
  6. //
  7. // The MIT License (MIT)
  8. //
  9. // Copyright (c) 2014-2018 Tristan Himmelman
  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. /// Transforms JSON dictionary to Codable type T and back
  30. open class CodableTransform<T: Codable>: TransformType {
  31. public typealias Object = T
  32. public typealias JSON = Any
  33. public init() {}
  34. open func transformFromJSON(_ value: Any?) -> Object? {
  35. var _data: Data? = nil
  36. switch value {
  37. case let dict as [String : Any]:
  38. _data = try? JSONSerialization.data(withJSONObject: dict, options: [])
  39. case let array as [[String : Any]]:
  40. _data = try? JSONSerialization.data(withJSONObject: array, options: [])
  41. default:
  42. _data = nil
  43. }
  44. guard let data = _data else { return nil }
  45. do {
  46. let decoder = JSONDecoder()
  47. let item = try decoder.decode(T.self, from: data)
  48. return item
  49. } catch {
  50. return nil
  51. }
  52. }
  53. open func transformToJSON(_ value: T?) -> JSON? {
  54. guard let item = value else {
  55. return nil
  56. }
  57. do {
  58. let encoder = JSONEncoder()
  59. let data = try encoder.encode(item)
  60. let dictionary = try JSONSerialization.jsonObject(with: data, options: .allowFragments)
  61. return dictionary
  62. } catch {
  63. return nil
  64. }
  65. }
  66. }