UIImage Extensions

UIImage+Extension.h

#import <UIKit/UIKit.h>

NS_ASSUME_NONNULL_BEGIN

@interface UIImage (Extension)

+ (instancetype)imageWithSize:(CGSize)size drawingHandler:(BOOL (^)(CGRect bounds))handler;

+ (instancetype)imageWithColor:(UIColor *)color;

@end

NS_ASSUME_NONNULL_END

UIImage+Extension.m

#import "UIImage+Extension.h"

@implementation UIImage (Extension)

+ (instancetype)imageWithSize:(CGSize)size drawingHandler:(BOOL (^)(CGRect bounds))handler {
    if (size.width <= 0 || size.height <= 0 || !handler) {
        return nil;
    }

    UIImage *image = nil;

    UIGraphicsBeginImageContextWithOptions(size, NO, 0);

    CGContextRef context = UIGraphicsGetCurrentContext();
    if (context) {
        CGRect bounds = CGContextGetClipBoundingBox(context);
        if (handler(bounds)) {
            image = UIGraphicsGetImageFromCurrentImageContext();
        }
    }

    UIGraphicsEndImageContext();

    return image;
}

+ (instancetype)imageWithColor:(UIColor *)color {
    if (!color) {
        return nil;
    }

    return [self imageWithSize:CGSizeMake(1, 1) drawingHandler:^BOOL(CGRect bounds) {
        [color setFill];
        UIRectFill(bounds);

        return YES;
    }];
}

@end