//
//  NightNight.swift
//  Pods
//
//  Created by Draveness on 6/30/16.
//
//

import UIKit
import ObjectiveC

public let NightNightThemeChangeNotification = "NightNightThemeChangeNotification"
private let NightNightThemeKey = "NightNightThemeKey"

public class NightNight {
    private static var currentTheme = Theme(rawValue: NSUserDefaults.standardUserDefaults().integerForKey(NightNightThemeKey)) ?? Theme.NORMAL

    public enum Theme: Int {
        case NORMAL
        case NIGHT
    }

    public static var theme: Theme {
        get { return currentTheme }
        set {
            currentTheme = newValue
            NSUserDefaults.standardUserDefaults().setInteger(currentTheme.rawValue, forKey: NightNightThemeKey)
            NSNotificationCenter.defaultCenter().postNotificationName(NightNightThemeChangeNotification, object: nil)
        }
    }
}

//
//  NightNight.swift
//  Pods
//
//  Created by Draveness on 6/30/16.
//
//

import UIKit
import ObjectiveC

public let NightNightThemeChangeNotification = "NightNightThemeChangeNotification"

public class NightNight {
    private static var currentTheme = Theme.NORMAL

    public enum Theme {
        case NORMAL
        case NIGHT
    }

    public static var theme: Theme {
        get { return currentTheme }
        set {
            currentTheme = newValue
            NSNotificationCenter.defaultCenter().postNotificationName(NightNightThemeChangeNotification, object: nil)
        }
    }
}

--- 
+++ 
@@ -10,11 +10,12 @@
 import ObjectiveC
 
 public let NightNightThemeChangeNotification = "NightNightThemeChangeNotification"
+private let NightNightThemeKey = "NightNightThemeKey"
 
 public class NightNight {
-    private static var currentTheme = Theme.NORMAL
+    private static var currentTheme = Theme(rawValue: NSUserDefaults.standardUserDefaults().integerForKey(NightNightThemeKey)) ?? Theme.NORMAL
 
-    public enum Theme {
+    public enum Theme: Int {
         case NORMAL
         case NIGHT
     }
@@ -23,6 +24,7 @@
         get { return currentTheme }
         set {
             currentTheme = newValue
+            NSUserDefaults.standardUserDefaults().setInteger(currentTheme.rawValue, forKey: NightNightThemeKey)
             NSNotificationCenter.defaultCenter().postNotificationName(NightNightThemeChangeNotification, object: nil)
         }
     }
Save current theme in user defatuls
