Defaults+Subscripts.swift 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. //
  2. // SwiftyUserDefaults
  3. //
  4. // Copyright (c) 2015-present Radosław Pietruszewski, Łukasz Mróz
  5. //
  6. // Permission is hereby granted, free of charge, to any person obtaining a copy
  7. // of this software and associated documentation files (the "Software"), to deal
  8. // in the Software without restriction, including without limitation the rights
  9. // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
  10. // copies of the Software, and to permit persons to whom the Software is
  11. // furnished to do so, subject to the following conditions:
  12. //
  13. // The above copyright notice and this permission notice shall be included in all
  14. // copies or substantial portions of the Software.
  15. //
  16. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  17. // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  18. // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  19. // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  20. // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  21. // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
  22. // SOFTWARE.
  23. //
  24. import Foundation
  25. public extension DefaultsAdapter {
  26. subscript<T: DefaultsSerializable>(key key: DefaultsKey<T>) -> T.T where T: OptionalType, T.T == T {
  27. get {
  28. return defaults[key]
  29. }
  30. set {
  31. defaults[key] = newValue
  32. }
  33. }
  34. subscript<T: DefaultsSerializable>(key key: DefaultsKey<T>) -> T.T where T.T == T {
  35. get {
  36. return defaults[key]
  37. }
  38. set {
  39. defaults[key] = newValue
  40. }
  41. }
  42. subscript<T: DefaultsSerializable>(keyPath: KeyPath<KeyStore, DefaultsKey<T>>) -> T.T where T: OptionalType, T.T == T {
  43. get {
  44. return defaults[keyStore[keyPath: keyPath]]
  45. }
  46. set {
  47. defaults[keyStore[keyPath: keyPath]] = newValue
  48. }
  49. }
  50. subscript<T: DefaultsSerializable>(keyPath: KeyPath<KeyStore, DefaultsKey<T>>) -> T.T where T.T == T {
  51. get {
  52. return defaults[keyStore[keyPath: keyPath]]
  53. }
  54. set {
  55. defaults[keyStore[keyPath: keyPath]] = newValue
  56. }
  57. }
  58. // Weird flex, but needed these two for the dynamicMemberLookup :shrug:
  59. subscript<T: DefaultsSerializable>(dynamicMember keyPath: KeyPath<KeyStore, DefaultsKey<T>>) -> T.T where T: OptionalType, T.T == T {
  60. get {
  61. return self[keyPath]
  62. }
  63. set {
  64. self[keyPath] = newValue
  65. }
  66. }
  67. subscript<T: DefaultsSerializable>(dynamicMember keyPath: KeyPath<KeyStore, DefaultsKey<T>>) -> T.T where T.T == T {
  68. get {
  69. return self[keyPath]
  70. }
  71. set {
  72. self[keyPath] = newValue
  73. }
  74. }
  75. }
  76. public extension UserDefaults {
  77. subscript<T: DefaultsSerializable>(key: DefaultsKey<T>) -> T.T where T: OptionalType, T.T == T {
  78. get {
  79. if let value = T._defaults.get(key: key._key, userDefaults: self), let _value = value as? T.T.Wrapped {
  80. // swiftlint:disable:next force_cast
  81. return _value as! T
  82. } else if let defaultValue = key.defaultValue {
  83. return defaultValue
  84. } else {
  85. return T.T.empty
  86. }
  87. }
  88. set {
  89. T._defaults.save(key: key._key, value: newValue, userDefaults: self)
  90. }
  91. }
  92. subscript<T: DefaultsSerializable>(key: DefaultsKey<T>) -> T.T where T.T == T {
  93. get {
  94. if let value = T._defaults.get(key: key._key, userDefaults: self) {
  95. return value
  96. } else if let defaultValue = key.defaultValue {
  97. return defaultValue
  98. } else {
  99. fatalError("Shouldn't happen, please report!")
  100. }
  101. }
  102. set {
  103. T._defaults.save(key: key._key, value: newValue, userDefaults: self)
  104. }
  105. }
  106. }