From 8ade4242708f7ea8d1518126510dfd9c11219aa8 Mon Sep 17 00:00:00 2001
From: Joe Steele <joe@madewell.net>
Date: Thu, 28 Aug 2014 17:48:37 -0400
Subject: [PATCH] Save/Restore the FoldableLinearLayout state

---
 res/layout/account_setup_basics.xml           |  1 +
 .../fsck/k9/view/FoldableLinearLayout.java    | 56 +++++++++++++++++++
 2 files changed, 57 insertions(+)

diff --git a/res/layout/account_setup_basics.xml b/res/layout/account_setup_basics.xml
index 2521de031..7a3458649 100644
--- a/res/layout/account_setup_basics.xml
+++ b/res/layout/account_setup_basics.xml
@@ -48,6 +48,7 @@
         android:visibility="gone"
         />
     <com.fsck.k9.view.FoldableLinearLayout
+        android:id="@+id/foldable_advanced_options"
         android:layout_width="match_parent"
         android:layout_height="wrap_content"
         android:layout_marginTop="16dp"
diff --git a/src/com/fsck/k9/view/FoldableLinearLayout.java b/src/com/fsck/k9/view/FoldableLinearLayout.java
index 1203caea6..8b426cf7d 100644
--- a/src/com/fsck/k9/view/FoldableLinearLayout.java
+++ b/src/com/fsck/k9/view/FoldableLinearLayout.java
@@ -5,6 +5,8 @@ import com.fsck.k9.R;
 import android.content.Context;
 import android.content.res.Resources.Theme;
 import android.content.res.TypedArray;
+import android.os.Parcel;
+import android.os.Parcelable;
 import android.util.AttributeSet;
 import android.util.TypedValue;
 import android.view.LayoutInflater;
@@ -97,6 +99,60 @@ public class FoldableLinearLayout extends LinearLayout {
         super.onFinishInflate();
     }
 
+    @Override
+    protected Parcelable onSaveInstanceState() {
+        Parcelable superState = super.onSaveInstanceState();
+        SavedState savedState = new SavedState(superState);
+        savedState.mFolded = mIsFolded;
+        return savedState;
+    }
+
+    @Override
+    protected void onRestoreInstanceState(Parcelable state) {
+        if (state instanceof SavedState) {
+            SavedState savedState = (SavedState) state;
+            super.onRestoreInstanceState(savedState.getSuperState());
+            mIsFolded = savedState.mFolded;
+            updateFoldedState(mIsFolded, false);
+        } else {
+            super.onRestoreInstanceState(state);
+        }
+    }
+
+    static class SavedState extends BaseSavedState {
+
+        static final Parcelable.Creator<SavedState> CREATOR =
+                new Parcelable.Creator<FoldableLinearLayout.SavedState>() {
+
+            @Override
+            public SavedState createFromParcel(Parcel source) {
+                return new SavedState(source);
+            }
+
+            @Override
+            public SavedState[] newArray(int size) {
+                return new SavedState[size];
+            }
+        };
+
+        private boolean mFolded;
+
+        private SavedState(Parcel parcel) {
+            super(parcel);
+            mFolded = (parcel.readInt() == 1);
+        }
+
+        private SavedState(Parcelable superState) {
+            super(superState);
+        }
+
+        @Override
+        public void writeToParcel(Parcel dest, int flags) {
+            super.writeToParcel(dest, flags);
+            dest.writeInt(mFolded ? 1 : 0);
+        }
+    }
+
     /**
      * Migrates Child views as declared in xml to the inner foldableContainer
      */