Merge pull request ethanhua#22 from blazvantur/master

Added feature for multi items support on RecyclerView.

Author: ethanhua

library/src/main/java/com/ethanhua/skeleton/RecyclerViewSkeletonScreen.java

@@ -1,5 +1,6 @@
 package com.ethanhua.skeleton;
 
+import android.support.annotation.ArrayRes;
 import android.support.annotation.ColorRes;
 import android.support.annotation.IntRange;
 import android.support.annotation.LayoutRes;
@@ -23,6 +24,7 @@ private RecyclerViewSkeletonScreen(Builder builder) {
         mSkeletonAdapter = new SkeletonAdapter();
         mSkeletonAdapter.setItemCount(builder.mItemCount);
         mSkeletonAdapter.setLayoutReference(builder.mItemResID);
+        mSkeletonAdapter.setArrayOfLayoutReferences(builder.mItemsResIDArray);
         mSkeletonAdapter.shimmer(builder.mShimmer);
         mSkeletonAdapter.setShimmerColor(builder.mShimmerColor);
         mSkeletonAdapter.setShimmerAngle(builder.mShimmerAngle);
@@ -49,6 +51,7 @@ public static class Builder {
         private boolean mShimmer = true;
         private int mItemCount = 10;
         private int mItemResID = R.layout.layout_default_item_skeleton;
+        private int[] mItemsResIDArray;
         private int mShimmerColor;
         private int mShimmerDuration = 1000;
         private int mShimmerAngle = 20;
@@ -118,6 +121,14 @@ public Builder load(@LayoutRes int skeletonLayoutResID) {
             return this;
         }
 
+        /**
+         * @param skeletonLayoutResIDs the loading array of skeleton layoutResID
+         */
+        public Builder loadArrayOfLayouts(@ArrayRes int[] skeletonLayoutResIDs) {
+            this.mItemsResIDArray = skeletonLayoutResIDs;
+            return this;
+        }
+
         /**
          * @param frozen whether frozen recyclerView during skeleton showing
          * @return

library/src/main/java/com/ethanhua/skeleton/SkeletonAdapter.java

@@ -15,6 +15,7 @@ public class SkeletonAdapter extends RecyclerView.Adapter<RecyclerView.ViewHolde
 
     private int mItemCount;
     private int mLayoutReference;
+    private int[] mLayoutArrayReferences;
     private int mColor;
     private boolean mShimmer;
     private int mShimmerDuration;
@@ -23,9 +24,13 @@ public class SkeletonAdapter extends RecyclerView.Adapter<RecyclerView.ViewHolde
     @Override
     public RecyclerView.ViewHolder onCreateViewHolder(ViewGroup parent, int viewType) {
         LayoutInflater inflater = LayoutInflater.from(parent.getContext());
+        if (doesArrayOfLayoutsExist()) {
+            mLayoutReference = viewType;
+        }
         if (mShimmer) {
             return new ShimmerViewHolder(inflater, parent, mLayoutReference);
         }
+
         return new RecyclerView.ViewHolder(inflater.inflate(mLayoutReference, parent, false)) {
         };
     }
@@ -41,6 +46,19 @@ public void onBindViewHolder(RecyclerView.ViewHolder holder, int position) {
         }
     }
 
+    @Override
+    public int getItemViewType(int position) {
+        if(doesArrayOfLayoutsExist()) {
+            return getCorrectLayoutItem(position);
+        }
+        return super.getItemViewType(position);
+    }
+
+    @Override
+    public long getItemId(int position) {
+        return position;
+    }
+
     @Override
     public int getItemCount() {
         return mItemCount;
@@ -50,6 +68,10 @@ public void setLayoutReference(int layoutReference) {
         this.mLayoutReference = layoutReference;
     }
 
+    public void setArrayOfLayoutReferences(int[] layoutReferences) {
+        this.mLayoutArrayReferences = layoutReferences;
+    }
+
     public void setItemCount(int itemCount) {
         this.mItemCount = itemCount;
     }
@@ -69,4 +91,15 @@ public void setShimmerDuration(int shimmerDuration) {
     public void setShimmerAngle(@IntRange(from = 0, to = 30) int shimmerAngle) {
         this.mShimmerAngle = shimmerAngle;
     }
+
+    public int getCorrectLayoutItem(int position) {
+        if(doesArrayOfLayoutsExist()) {
+            return mLayoutArrayReferences[position % mLayoutArrayReferences.length];
+        }
+        return mLayoutReference;
+    }
+
+    private boolean doesArrayOfLayoutsExist() {
+        return mLayoutArrayReferences != null && mLayoutArrayReferences.length != 0;
+    }
 }