Add an override content view property for temporary views

Author: TimOliver

TORoundedButton/TORoundedButton.h

@@ -49,6 +49,11 @@ IB_DESIGNABLE @interface TORoundedButton : UIControl
 /// this property to your own custom UIView subclass in order to more efficiently manage sizing and layout.
 @property (nonatomic, strong, null_resettable) UIView *contentView;
 
+/// An additional content view that will hide `contentView` and show in its place when set.
+/// When this is set back to `nil`, the original content view is restored.
+/// This is useful for temporarily showing an alternative UI, such as a loading or download spinner.
+@property (nonatomic, strong, nullable) UIView *overrideContentView;
+
 /// The amount of inset padding between the content view and the edges of the button.
 /// (Default value is 15 points inset from each edge).
 @property (nonatomic, assign) UIEdgeInsets contentInset;
@@ -60,6 +65,7 @@ IB_DESIGNABLE @interface TORoundedButton : UIControl
 @property (nonatomic, assign) UIBlurEffectStyle blurStyle;
 
 /// The text that is displayed in center of the button (Default is nil).
+/// This adds an internally controlled label view to the main content view.
 @property (nonatomic, copy, nullable) IBInspectable NSString *text;
 
 /// The attributed string used in the label of this button. 

TORoundedButton/TORoundedButton.m

@@ -178,23 +178,35 @@ - (UIView *)_makeBackgroundViewWithBlur:(BOOL)withBlur TOROUNDEDBUTTON_OBJC_DIRE
 - (void)layoutSubviews {
     [super layoutSubviews];
 
+    // Determine the content view's available maximum size accounting for the insetting
     const CGSize boundsSize = self.bounds.size;
-    _contentView.frame = (CGRect){
+    const CGRect contentBounds = (CGRect){
         .origin.x = _contentInset.left,
         .origin.y = _contentInset.top,
         .size.width = boundsSize.width - (_contentInset.left + _contentInset.right),
         .size.height = boundsSize.height - (_contentInset.top + _contentInset.bottom),
     };
 
-    // Configure the button text
-    if (_titleLabel) {
-        [_titleLabel sizeToFit];
-        _titleLabel.center = (CGPoint){
-            .x = CGRectGetMidX(_contentView.bounds),
-            .y = CGRectGetMidY(_contentView.bounds)
-        };
-        _titleLabel.frame = CGRectIntegral(_titleLabel.frame);
-    }
+    // Let the content view shrink itself to wrap its content if needed,
+    // and position it in the middle of the view.
+    UIView *const contentView = _overrideContentView ?: _contentView;
+    contentView.frame = ({
+        CGRect frame = contentBounds;
+        frame.size = [contentView sizeThatFits:contentBounds.size];
+        frame.origin.x = (boundsSize.width - frame.size.width) * 0.5f;
+        frame.origin.y = (boundsSize.height - frame.size.height) * 0.5f;
+        CGRectIntegral(frame);
+    });
+
+    // Lay out the title label
+    if (!_titleLabel) { return; }
+
+    [_titleLabel sizeToFit];
+    _titleLabel.center = (CGPoint){
+        .x = CGRectGetMidX(_contentView.bounds),
+        .y = CGRectGetMidY(_contentView.bounds)
+    };
+    _titleLabel.frame = CGRectIntegral(_titleLabel.frame);
 }
 
 - (void)sizeToFit { [super sizeToFit]; }
@@ -413,7 +425,24 @@ - (void)setContentView:(UIView *)contentView {
     _titleLabel = nil;
     [_contentView removeFromSuperview];
     _contentView = contentView ?: [UIView new];
-    [self addSubview:_contentView];
+    [_containerView addSubview:_contentView];
+    [self setNeedsLayout];
+}
+
+- (void)setOverrideContentView:(UIView *)overrideContentView {
+    if (_overrideContentView == overrideContentView) { return; }
+    if (_overrideContentView != nil) {
+        [_overrideContentView removeFromSuperview];
+    }
+
+    _overrideContentView = overrideContentView;
+
+    if (_overrideContentView != nil) {
+        _contentView.hidden = YES;
+        [_containerView addSubview:_overrideContentView];
+    } else {
+        _contentView.hidden = NO;
+    }
     [self setNeedsLayout];
 }