Example of a simple model decorator

NotificationItemDecorator.swift

class NotificationItemDecorator {
    
    var notification: Notification?
    
    func with(_ notification: Notification) -> NotificationItemDecorator {
        self.notification = notification
        return self
    }
    
    func add(confirm: UIButton?) {
        if let confirm = confirm, let currentNotification = notification {
            if let data = currentNotification.data {
                confirm.isHidden = data.isConfirmed
            } else {
                confirm.isHidden = true
            }
        }
    }
    
    func decorate(image: UIImageView?, title: UILabel?, message: UILabel?, hour: UILabel?, date: UILabel? = nil) -> NotificationItemDecorator {
        if let notification = self.notification {
            
            if let data = notification.data {
                self.decorate(with: data, type: notification.type, image: image, title: title)
            } else {
                title?.text = notification.title
            }
            
            self.commonDecorate(message: message, hour: hour, date: date)
        }
        return self
    }
    
    private func commonDecorate(message: UILabel?, hour: UILabel?, date: UILabel? = nil) {
        if let message = message {
            message.text = self.notification?.message
        }
        
        if let hour = hour, let dateString = self.notification?.date {
            hour.setup(date: dateString)
            
            if let date = date {
                date.setup(date: dateString, type: .verbose)
            }
        }
    }
    
    private func decorate(with data: NotificationData, type: NotificationType, image: UIImageView?, title: UILabel?) {
        if let image = image, let imageUrl = data.imageUrl {
            image.loadImageFromUrl(imageUrl, cacheIdentifier: CacheConstants.UserProfileImage.rawValue, circular: true, bordered: false, rounded: false)
        }
        
        if let title = title {
            title.text = type == .fall ? NotificationsHelper.fall(withName: data.elderName) : NotificationsHelper.cameOut(withName: data.elderName)
        }
    }
}

Uses:

NotificationItemDecorator()
                .with(notification)
                .decorate(
                    image: mImageView,
                    title: titleLabel,
                    message: messageLabel,
                    hour: hourLabel,
                    date: dateLabel)
                .add(confirm: confirmButton)

or

let _ = NotificationItemDecorator()
            .with(notification)
            .decorate(
                image: cell.mImageView,
                title: cell.titleLabel,
                message: cell.messageLabel,
                hour: cell.hourLabel
            )