add KeyHolder and set key for custom viewgroups

README.md

@@ -103,6 +103,8 @@ public class MainActivity extends AppCompatActivity implements StateChanger {
         View view = LayoutInflater.from(this).inflate(newKey.layout(), root, false);
         BackstackHolder backstackHolder = (BackstackHolder)view;
         backstackHolder.setBackstack(backstack); // view currently doesn't implicitly receive the backstack
+        KeyHolder keyHolder = (KeyHolder)view;
+        keyHolder.setKey(newKey); // view currently doesn't implicitly receive the key
         root.addView(view);
         completionCallback.stateChangeComplete();
     }
@@ -117,6 +119,6 @@ Scheduling a state change while a state change is already in progress throws an
 
 ## Demo limitations
 
-In the demo, keys associated with the given custom views are currently not manually set for the custom views, therefore it is not accessible at the moment.
+In the demo, keys associated with the given custom views are manually set for the custom views, therefore it is set via an interface, but is not accessible for any child views of the custom viewgroup.
 
-In the demo, the backstack is manually set to the custom viewgroup using a `BackStackHolder` interface, instead of implicitly providing it via the Context.
+In the demo, the backstack is also manually set to the custom viewgroup using a `BackStackHolder` interface, instead of implicitly providing it via the Context.

demo-stack-example/src/main/java/com/zhuinden/simplestackdemoexample/MainActivity.java

@@ -13,6 +13,7 @@
 import com.zhuinden.simplestackdemoexample.demo.FirstKey;
 import com.zhuinden.simplestackdemoexample.demo.Key;
 import com.zhuinden.simplestackdemoexample.R;
+import com.zhuinden.simplestackdemoexample.demo.KeyHolder;
 
 import java.util.ArrayList;
 import butterknife.BindView;
@@ -82,6 +83,8 @@ public void handleStateChange(StateChange stateChange, Callback completionCallba
         View view = LayoutInflater.from(this).inflate(newKey.layout(), root, false);
         BackstackHolder backstackHolder = (BackstackHolder)view;
         backstackHolder.setBackstack(backstack);
+        KeyHolder keyHolder = (KeyHolder)view;
+        keyHolder.setKey(newKey);
         root.addView(view);
         completionCallback.stateChangeComplete();
     }

demo-stack-example/src/main/java/com/zhuinden/simplestackdemoexample/demo/FirstView.java

@@ -13,7 +13,7 @@
  * Created by Owner on 2017. 01. 12..
  */
 
-public class FirstView extends RelativeLayout implements BackstackHolder {
+public class FirstView extends RelativeLayout implements BackstackHolder, KeyHolder {
     public FirstView(Context context) {
         super(context);
     }
@@ -33,6 +33,8 @@ public FirstView(Context context, AttributeSet attrs, int defStyleAttr, int defS
 
     Backstack backstack;
 
+    FirstKey firstKey;
+
     @OnClick(R.id.first_button)
     public void clickButton(View view) {
         backstack.goTo(new SecondKey());
@@ -48,4 +50,9 @@ protected void onFinishInflate() {
     public void setBackstack(Backstack backstack) {
         this.backstack = backstack;
     }
+
+    @Override
+    public void setKey(Key key) {
+        this.firstKey = (FirstKey)key;
+    }
 }

demo-stack-example/src/main/java/com/zhuinden/simplestackdemoexample/demo/KeyHolder.java

@@ -0,0 +1,9 @@
+package com.zhuinden.simplestackdemoexample.demo;
+
+/**
+ * Created by Owner on 2017. 01. 13..
+ */
+
+public interface KeyHolder {
+    void setKey(Key key);
+}

demo-stack-example/src/main/java/com/zhuinden/simplestackdemoexample/demo/SecondView.java

@@ -10,8 +10,7 @@
  * Created by Owner on 2017. 01. 12..
  */
 
-public class SecondView extends RelativeLayout implements BackstackHolder {
-
+public class SecondView extends RelativeLayout implements BackstackHolder, KeyHolder {
     public SecondView(Context context) {
         super(context);
     }
@@ -31,8 +30,15 @@ public SecondView(Context context, AttributeSet attrs, int defStyleAttr, int def
 
     Backstack backstack;
 
+    SecondKey secondKey;
+
     @Override
     public void setBackstack(Backstack backstack) {
         this.backstack = backstack;
     }
+
+    @Override
+    public void setKey(Key key) {
+        this.secondKey = (SecondKey)key;
+    }
 }