Skeleton style that adds different cell styles

Author: BeatsKitano

Somo.podspec

@@ -2,7 +2,7 @@
 Pod::Spec.new do |s|
 
   s.name         = "Somo"
-  s.version      = "0.6.5"
+  s.version      = "0.7.0"
   s.summary      = "Somo is a Skeleton-style animation library that's simple enough."
 
   s.homepage     = "https://github.com/xorshine/Somo"

SomoDemo/SomoDemo.xcodeproj/project.xcworkspace/xcuserdata/xorshine.xcuserdatad/UserInterfaceState.xcuserstate

@@ -10,6 +10,10 @@
 
 #import <Foundation/Foundation.h>
 #import <UIKit/UIKit.h>
+@protocol SomoSkeletonLayoutProtocol;
+
+typedef UITableViewCell<SomoSkeletonLayoutProtocol> *(^SomoTableViewCellBlock)(UITableView *tableview, NSIndexPath *indexPath);
+typedef UICollectionViewCell<SomoSkeletonLayoutProtocol> *(^SomoCollectionViewCellBlock)(UICollectionView *tableview, NSIndexPath *indexPath);
 
 @interface SomoDataSourceProvider : NSObject<UITableViewDataSource,UITableViewDelegate,UICollectionViewDelegate,UICollectionViewDataSource>
 
@@ -18,7 +22,16 @@
  */
 @property (assign, nonatomic) NSInteger numberOfRowsInSection;
 
+/**
+ If you have only one cell style, use the following method
+ */
 - (instancetype)initWithCellReuseIdentifier:(NSString *)reuseIdentifier;
 + (instancetype)dataSourceProviderWithCellReuseIdentifier:(NSString *)reuseIdentifier;
 
+/**
+ If you have different cell styles, use the following method
+ */
+- (instancetype)initWithTableViewCellBlock:(SomoTableViewCellBlock)block;
+- (instancetype)initWithCollectionViewCellBlock:(SomoCollectionViewCellBlock)block;
+
 @end

SomoDemo/SomoDemo/Somo/SomoDataSourceProvider.h

@@ -14,6 +14,8 @@
 @implementation SomoDataSourceProvider
 {
 	NSString * _reuseIdentifier;
+	SomoTableViewCellBlock _tableViewCellBlock;
+	SomoCollectionViewCellBlock _collectionViewCellBlock;
 }
 
 - (instancetype)initWithCellReuseIdentifier:(NSString *)reuseIdentifier{
@@ -28,13 +30,32 @@ + (instancetype)dataSourceProviderWithCellReuseIdentifier:(NSString *)reuseIdent
 	return [[[self class] alloc] initWithCellReuseIdentifier:reuseIdentifier];
 }
 
+- (instancetype)initWithTableViewCellBlock:(SomoTableViewCellBlock)block{
+	if (self = [super init]) {
+		_tableViewCellBlock = block;
+		_numberOfRowsInSection = 30;
+	}
+	return self;
+}
+
+- (instancetype)initWithCollectionViewCellBlock:(SomoCollectionViewCellBlock)block{
+	if (self = [super init]) {
+		_collectionViewCellBlock = block;
+		_numberOfRowsInSection = 30;
+	}
+	return self;
+}
+
 - (NSInteger)tableView:(UITableView *)tableView numberOfRowsInSection:(NSInteger)section{
 	return self.numberOfRowsInSection;
 }
 
 - (UITableViewCell *)tableView:(UITableView *)tableView cellForRowAtIndexPath:(NSIndexPath *)indexPath{
-	UITableViewCell *cell = [tableView dequeueReusableCellWithIdentifier:_reuseIdentifier forIndexPath:indexPath];
-	return cell;
+	if (_reuseIdentifier) {
+		return [tableView dequeueReusableCellWithIdentifier:_reuseIdentifier forIndexPath:indexPath];
+	}else{
+		return _tableViewCellBlock(tableView,indexPath);
+	}
 }
 
 - (void)tableView:(UITableView *)tableView willDisplayCell:(UITableViewCell *)cell forRowAtIndexPath:(NSIndexPath *)indexPath{
@@ -48,8 +69,11 @@ - (NSInteger)collectionView:(UICollectionView *)collectionView numberOfItemsInSe
 }
 
 - (__kindof UICollectionViewCell *)collectionView:(UICollectionView *)collectionView cellForItemAtIndexPath:(NSIndexPath *)indexPath{
-	UICollectionViewCell *cell = [collectionView dequeueReusableCellWithReuseIdentifier:_reuseIdentifier forIndexPath:indexPath];
-	return cell;
+	if (_reuseIdentifier) {
+		return [collectionView dequeueReusableCellWithReuseIdentifier:_reuseIdentifier forIndexPath:indexPath];
+	}else{
+		return _collectionViewCellBlock(collectionView, indexPath);
+	}
 }
 
 - (void)collectionView:(UICollectionView *)collectionView willDisplayCell:(UICollectionViewCell *)cell forItemAtIndexPath:(NSIndexPath *)indexPath{