Defaults+Subscripts.swift 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859
  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. // DefaultsKey
  26. public extension UserDefaults {
  27. subscript<T: DefaultsSerializable>(key: DefaultsKey<T?>) -> T.T? {
  28. get {
  29. if let value = T._defaults.get(key: key._key, userDefaults: self) {
  30. return value
  31. } else if let defaultValue = key.defaultValue as? T.T {
  32. return defaultValue
  33. } else {
  34. return nil
  35. }
  36. }
  37. set {
  38. T._defaults.save(key: key._key, value: newValue, userDefaults: self)
  39. }
  40. }
  41. subscript<T: DefaultsSerializable>(key: DefaultsKey<T>) -> T.T where T.T == T {
  42. get {
  43. if let value = T._defaults.get(key: key._key, userDefaults: self) {
  44. return value
  45. } else if let defaultValue = key.defaultValue {
  46. return defaultValue
  47. } else {
  48. fatalError("Shouldn't happen, please report!")
  49. }
  50. }
  51. set {
  52. T._defaults.save(key: key._key, value: newValue, userDefaults: self)
  53. }
  54. }
  55. }