metatext/Views/UIKit/ImagePastableTextView.swift

// Copyright © 2021 Metabolist. All rights reserved.

import Combine
import UIKit

final class ImagePastableTextView: UITextView {
    var canPasteImage = true
    private(set) lazy var pastedItemProviders: AnyPublisher<NSItemProvider, Never> =
        pastedItemProvidersSubject.eraseToAnyPublisher()

    private let pastedItemProvidersSubject = PassthroughSubject<NSItemProvider, Never>()

    override func canPerformAction(_ action: Selector, withSender sender: Any?) -> Bool {
        if action == #selector(paste(_:)) {
            return UIPasteboard.general.hasStrings || (UIPasteboard.general.hasImages && canPasteImage)
        }

        return super.canPerformAction(action, withSender: sender)
    }

    override func paste(_ sender: Any?) {
        if UIPasteboard.general.hasImages, let itemProvider = UIPasteboard.general.itemProviders.first {
            pastedItemProvidersSubject.send(itemProvider)
        } else {
            super.paste(sender)
        }
    }
}