Merge "Fix DND conditional null state" into nyc-dev

Author: Monkopedia

src/com/android/settings/dashboard/conditional/DndCondition.java

@@ -22,14 +22,17 @@
 import android.content.Context;
 import android.content.Intent;
 import android.graphics.drawable.Icon;
+import android.os.PersistableBundle;
 import android.provider.Settings;
+import android.provider.Settings.Global;
 import android.service.notification.ZenModeConfig;
 import com.android.internal.logging.MetricsProto.MetricsEvent;
 import com.android.settings.R;
 
 public class DndCondition extends Condition {
 
     private static final String TAG = "DndCondition";
+    private static final String KEY_STATE = "state";
 
     private int mZen;
     private ZenModeConfig mConfig;
@@ -52,6 +55,18 @@ public void refreshState() {
         setActive(zenModeEnabled);
     }
 
+    @Override
+    boolean saveState(PersistableBundle bundle) {
+        bundle.putInt(KEY_STATE, mZen);
+        return super.saveState(bundle);
+    }
+
+    @Override
+    void restoreState(PersistableBundle bundle) {
+        super.restoreState(bundle);
+        mZen = bundle.getInt(KEY_STATE, Global.ZEN_MODE_OFF);
+    }
+
     @Override
     protected Class<?> getReceiverClass() {
         return Receiver.class;