From 36f871817a2708d638501abdfa547f4708931ac3 Mon Sep 17 00:00:00 2001
From: iluz <iluz@ciandt.com>
Date: Wed, 6 Apr 2016 16:32:00 -0300
Subject: [PATCH] Add position param to onCreateHeaderViewHolder, so we can
 create different view holders

---
 .../StickyRecyclerHeadersAdapter.java                          | 3 ++-
 .../stickyheadersrecyclerview/caching/HeaderViewCache.java     | 2 +-
 .../timehop/stickyheadersrecyclerview/sample/MainActivity.java | 2 +-
 3 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/library/src/main/java/com/timehop/stickyheadersrecyclerview/StickyRecyclerHeadersAdapter.java b/library/src/main/java/com/timehop/stickyheadersrecyclerview/StickyRecyclerHeadersAdapter.java
index 23990fa..89ef868 100644
--- a/library/src/main/java/com/timehop/stickyheadersrecyclerview/StickyRecyclerHeadersAdapter.java
+++ b/library/src/main/java/com/timehop/stickyheadersrecyclerview/StickyRecyclerHeadersAdapter.java
@@ -20,9 +20,10 @@ public interface StickyRecyclerHeadersAdapter<VH extends RecyclerView.ViewHolder
    * to inflate the layout for the header.
    *
    * @param parent the view to create a header view holder for
+   * @param position the position of the item to create the view holder for.
    * @return the view holder
    */
-  VH onCreateHeaderViewHolder(ViewGroup parent);
+  VH onCreateHeaderViewHolder(ViewGroup parent, int position);
 
   /**
    * Binds an existing ViewHolder to the specified adapter position.
diff --git a/library/src/main/java/com/timehop/stickyheadersrecyclerview/caching/HeaderViewCache.java b/library/src/main/java/com/timehop/stickyheadersrecyclerview/caching/HeaderViewCache.java
index 94247fd..bdbce03 100644
--- a/library/src/main/java/com/timehop/stickyheadersrecyclerview/caching/HeaderViewCache.java
+++ b/library/src/main/java/com/timehop/stickyheadersrecyclerview/caching/HeaderViewCache.java
@@ -31,7 +31,7 @@ public View getHeader(RecyclerView parent, int position) {
     View header = mHeaderViews.get(headerId);
     if (header == null) {
       //TODO - recycle views
-      RecyclerView.ViewHolder viewHolder = mAdapter.onCreateHeaderViewHolder(parent);
+      RecyclerView.ViewHolder viewHolder = mAdapter.onCreateHeaderViewHolder(parent, position);
       mAdapter.onBindHeaderViewHolder(viewHolder, position);
       header = viewHolder.itemView;
       if (header.getLayoutParams() == null) {
diff --git a/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/MainActivity.java b/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/MainActivity.java
index 591bd10..50edb92 100644
--- a/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/MainActivity.java
+++ b/sample/src/main/java/com/timehop/stickyheadersrecyclerview/sample/MainActivity.java
@@ -142,7 +142,7 @@ public long getHeaderId(int position) {
     }
 
     @Override
-    public RecyclerView.ViewHolder onCreateHeaderViewHolder(ViewGroup parent) {
+    public RecyclerView.ViewHolder onCreateHeaderViewHolder(ViewGroup parent, int position) {
       View view = LayoutInflater.from(parent.getContext())
           .inflate(R.layout.view_header, parent, false);
       return new RecyclerView.ViewHolder(view) {