draw.swift

import UIKit

public func draw(size: CGSize, drawBlock: (CGContext) -> ()) -> UIImage? {
	UIGraphicsBeginImageContextWithOptions(size, true, 0.0)
	defer {
		UIGraphicsEndImageContext()
	}

	guard let context = UIGraphicsGetCurrentContext() else {
		return nil
	}

	drawBlock(context)

	return UIGraphicsGetImageFromCurrentImageContext()
}