RichTextView.swift

//
//  RichTextView.swift
//  Proton
//
//  Created by Rajdeep Kwatra on 4/1/20.
//  Copyright © 2020 Rajdeep Kwatra. All rights reserved.
//
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
//

import Foundation
import UIKit
import ProtonCore

class RichTextView: AutogrowingTextView {
    
    /// Equivalent, strongly-typed alternative to `textStorage`
    private let richTextStorage = PRTextStorage()
    private let _undoManager: UndoManager?
    static let defaultListLineFormatting = LineFormatting(indentation: 25, spacingBefore: 0)

    weak var richTextViewDelegate: RichTextViewDelegate?
    weak var richTextViewListDelegate: RichTextViewListDelegate?
    weak var richTextScrollViewDelegate: UIScrollViewDelegate?
    weak var lineNumberProvider: LineNumberProvider?

    private var delegateOverrides = [GestureRecognizerDelegateOverride]()

    var preserveBlockAttachmentNewline: PreserveBlockAttachmentNewline = .none {
        didSet {
            richTextStorage.preserveNewlineBeforeBlock = false
            richTextStorage.preserveNewlineAfterBlock = false

            if preserveBlockAttachmentNewline.contains(.before) {
                richTextStorage.preserveNewlineBeforeBlock = true
            }

            if preserveBlockAttachmentNewline.contains(.after) {
                richTextStorage.preserveNewlineAfterBlock = true
            }
        }
    }

    weak var defaultTextFormattingProvider: DefaultTextFormattingProviding?
    {
        get { richTextStorage.defaultTextFormattingProvider }
        set {
            richTextStorage.defaultTextFormattingProvider = newValue
//            typingAttributes[.font] = defaultTypingAttributes[.font]
//            typingAttributes[.paragraphStyle] = defaultTypingAttributes[.paragraphStyle]
//            typingAttributes[.foregroundColor] = defaultTypingAttributes[.foregroundColor]
        }
    }

    private lazy var placeholderLabel: UILabel = {
        let placeholderLabel = UILabel()
        placeholderLabel.accessibilityIdentifier = "RichTextView.placeholderLabel"
        placeholderLabel.translatesAutoresizingMaskIntoConstraints = false
        placeholderLabel.numberOfLines = 0
        placeholderLabel.lineBreakMode = .byTruncatingTail
        return placeholderLabel
    }()

    var placeholderText: NSAttributedString? {
        didSet {
            placeholderLabel.attributedText = placeholderText
            updatePlaceholderVisibility()
        }
    }

    var editorView: EditorView? {
        return superview as? EditorView
    }

    var defaultTypingAttributes: RichTextAttributes {
        return [
            .font: defaultTextFormattingProvider?.font ?? richTextStorage.defaultFont,
            .paragraphStyle: defaultTextFormattingProvider?.paragraphStyle ?? richTextStorage.defaultParagraphStyle,
            .foregroundColor: defaultTextFormattingProvider?.textColor ?? richTextStorage.defaultTextColor
        ]
    }
    var defaultFont: UIFont { richTextStorage.defaultFont }
    var defaultTextColor: UIColor { richTextStorage.defaultTextColor }
    var defaultBackgroundColor: UIColor {
        return .systemBackground
    }

    override var attributedText: NSAttributedString! {
        willSet {
            // Remove all attachment subviews else we may run into PRTextStorage "in middle of editing" crash
            subviews.filter { $0 is AttachmentContentView }.forEach { $0.removeFromSuperview() }
        }
    }

    var lineNumberFormatting = LineNumberFormatting.default {
        didSet {
            let gutterOffset = lineNumberFormatting.gutter.width + lineNumberFormatting.gutter.lineWidth
            let adjustedLeftInset = isLineNumbersEnabled ? (gutterOffset + textContainerInset.left - oldValue.gutter.width): nil

            textContainerInset = UIEdgeInsets(
                top: textContainerInset.top,
                left: adjustedLeftInset ?? textContainerInset.left,
                bottom: textContainerInset.bottom,
                right: textContainerInset.right
            )
            setNeedsDisplay()
        }
    }

    var isLineNumbersEnabled = false {
        didSet {
            let gutterOffset = lineNumberFormatting.gutter.width + lineNumberFormatting.gutter.lineWidth

            let adjustedLeftInset: CGFloat
            switch (oldValue, isLineNumbersEnabled) {
            case (false, true):
                adjustedLeftInset = gutterOffset + textContainerInset.left
            case (true, false):
                adjustedLeftInset = textContainerInset.left - gutterOffset
            default:
                adjustedLeftInset = textContainerInset.left
            }

            textContainerInset = UIEdgeInsets(
                top: textContainerInset.top,
                left: adjustedLeftInset,
                bottom: textContainerInset.bottom,
                right: textContainerInset.right
            )
            setNeedsDisplay()
        }
    }

    override func draw(_ rect: CGRect) {
        guard isLineNumbersEnabled,
              let currentCGContext = UIGraphicsGetCurrentContext() else {
            super.draw(rect)
            return
        }

        let height = max(contentSize.height, bounds.height)
        let rect = CGRect(x: 0, y: 0, width: lineNumberFormatting.gutter.width, height: height)
        let rectanglePath = UIBezierPath(rect: rect)

        currentCGContext.saveGState()
        currentCGContext.addPath(rectanglePath.cgPath)

        if let lineColor = lineNumberFormatting.gutter.lineColor {
            currentCGContext.setStrokeColor(lineColor.cgColor)
            currentCGContext.setLineWidth(lineNumberFormatting.gutter.lineWidth)
            currentCGContext.drawPath(using: .stroke)
        }

        currentCGContext.setFillColor(lineNumberFormatting.gutter.backgroundColor.cgColor)
        currentCGContext.fill(rect)

        // Draw line number if textView is empty
        if let layoutManager = layoutManager as? LayoutManager,
           attributedText.length == 0 {
            let lineNumberToDisplay = lineNumberString(for: 1) ?? "1"
            let width = lineNumberFormatting.gutter.width
            let height = defaultFont.lineHeight
            layoutManager.drawLineNumber(lineNumber: lineNumberToDisplay, rect: CGRect(origin: .zero, size: CGSize(width: width, height: height)), lineNumberFormatting: lineNumberFormatting, currentCGContext: currentCGContext)
        }

        currentCGContext.restoreGState()

        super.draw(rect)
    }

    func drawDefaultLineNumberIfRequired() {
        guard isLineNumbersEnabled else { return }
        draw(CGRect(origin: .zero, size: contentSize))
    }

    override var selectedTextRange: UITextRange? {
        didSet{
            let old = oldValue?.toNSRange(in: self)
            let new = selectedTextRange?.toNSRange(in: self)

            if let range = adjustedTextBlockRangeOnSelectionChange(oldRange: old, newRange: new) {
                selectedRange = range
            }
            richTextViewDelegate?.richTextView(self, selectedRangeChangedFrom: old, to: selectedTextRange?.toNSRange(in: self))
        }
    }

    override var selectedRange: NSRange {
        didSet{
            if let range = adjustedTextBlockRangeOnSelectionChange(oldRange: oldValue, newRange: selectedRange) {
                selectedRange = range
                richTextViewDelegate?.richTextView(self, selectedRangeChangedFrom: oldValue, to: selectedRange)
            }
        }
    }

    override var contentInset: UIEdgeInsets {
        didSet {
            updatePlaceholderVisibility()
        }
    }

    override var textContainerInset: UIEdgeInsets {
        didSet {
            updatePlaceholderVisibility()
        }
    }

    var nestedTextViews: [RichTextView] {
        getNestedEditors(for: self)
    }

    override public func pressesBegan(_ presses: Set<UIPress>, with event: UIPressesEvent?) {
        guard let key = presses.first?.key else {
            super.pressesBegan(presses, with: event)
            return
        }

        var handled: Bool = false

        guard let editorKey = EditorKey(key) else { return }
        richTextViewDelegate?.richTextView(self, shouldHandle: editorKey, modifierFlags: key.modifierFlags, at: self.selectedRange, handled: &handled)

        if handled == false {
            super.pressesBegan(presses, with: event)
        }
    }

    private func getNestedEditors(for containerView: UIView) -> [RichTextView] {
        var textViews = [RichTextView]()
        for view in containerView.subviews {
            if let textView = view as? RichTextView {
                textViews.append(textView)
            }
            textViews.append(contentsOf: getNestedEditors(for: view))
        }

        return textViews
    }

    private func adjustedTextBlockRangeOnSelectionChange(oldRange: NSRange?, newRange: NSRange?) -> NSRange? {
        guard let old = oldRange,
              let new = newRange,
              old != new
        else { return nil }

        let isReverseTraversal = (new.location < old.location) || (new.endLocation < old.endLocation)

        guard new.length > 0 else {
            if let textBlockRange = attributedText.rangeOf(attribute: .textBlock, at: new.location),
               textBlockRange.location != new.location {
                let location = isReverseTraversal ? textBlockRange.location : textBlockRange.endLocation
                return NSRange(location: location, length: 0)
            }
            return nil
        }

        let isLocationChanged = new.location != old.location
        let location = isLocationChanged ? new.location : max(0, new.endLocation - 1)

        guard let textBlockRange = attributedText.rangeOf(attribute: .textBlock, at: location),
              textBlockRange.contains(location)
        else { return nil }

        // return textblock range if new range is entirely contained within textblock range
        if textBlockRange.intersection(new) == new {
            return textBlockRange
        }

        if isReverseTraversal {
            return adjustedTextBlockRangeReverse(new: new, old: old, textBlockRange: textBlockRange)
        } else {
            return adjustedTextBlockRangeForward(new: new, old: old, textBlockRange: textBlockRange)
        }
    }

    private func adjustedTextBlockRangeReverse(new: NSRange, old: NSRange, textBlockRange: NSRange) -> NSRange {
        let range: NSRange
        if textBlockRange.union(new) == textBlockRange, new.endLocation == old.endLocation, textBlockRange.contains(new.location) == false {
            range = NSRange(location: textBlockRange.location, length: old.endLocation - textBlockRange.endLocation)
        } else if new.endLocation < textBlockRange.endLocation && new.endLocation > textBlockRange.location {
            range = NSRange(location: new.location, length: textBlockRange.location - new.location)
        } else {
            range = textBlockRange.union(new)
        }
        return range
    }

    private func adjustedTextBlockRangeForward(new: NSRange, old: NSRange, textBlockRange: NSRange) -> NSRange {
        let range: NSRange
        let isLocationChanged = new.location != old.location
        if (new.contains(textBlockRange.location) && new.contains(textBlockRange.endLocation - 1)
                || (textBlockRange.union(new) == textBlockRange && new.length > 0 && isLocationChanged == false)
                || isLocationChanged == false) {
            range = new.union(textBlockRange)
        } else {
            range = NSRange(location: textBlockRange.endLocation, length: new.endLocation - textBlockRange.endLocation)
        }
        return range
    }

    private func adjustRangeOnNonFocus(oldRange: UITextRange?) {
        guard let currentRange = selectedTextRange?.toNSRange(in: self),
              let previousRange = oldRange?.toNSRange(in: self)
        else { return }

        var rangeToSet: NSRange?
        let isReverseTraversal = currentRange.location < previousRange.location
        var rangeToTraverse = NSRange(location: currentRange.location, length: attributedText.length - (currentRange.location + currentRange.length))

        if isReverseTraversal {
            rangeToTraverse = NSRange(location: 0, length: currentRange.location)
            attributedText.enumerateAttribute(.textBlock, in: rangeToTraverse, options: [.longestEffectiveRangeNotRequired, .reverse]) { val, range, stop in
                if (val as? Bool != true), rangeToSet == nil {
                    rangeToSet = NSRange(location: range.location + range.length, length: 0)
                    stop.pointee = true
                }
            }
        }  else {
            attributedText.enumerateAttribute(.textBlock, in: rangeToTraverse, options: [.longestEffectiveRangeNotRequired]) { val, range, stop in
                if (val as? Bool != true), rangeToSet == nil {
                    rangeToSet = NSRange(location: range.location, length: 0)
                    stop.pointee = true
                }
            }
        }

        selectedTextRange = rangeToSet?.toTextRange(textInput: self) ?? oldRange
    }

    var context: RichTextViewContext? {
        delegate as? RichTextViewContext
    }

    init(
        frame: CGRect = .zero,
        context: RichTextViewContext,
        allowAutogrowing: Bool = false,
        undoManager: UndoManager? = nil) {
        let textContainer = TextContainer()
        let layoutManager = LayoutManager()

        layoutManager.addTextContainer(textContainer)
        richTextStorage.addLayoutManager(layoutManager)

        _undoManager = undoManager
        super.init(frame: frame, textContainer: textContainer, allowAutogrowing: allowAutogrowing)
        delegateOverrides = [GestureRecognizerDelegateOverride]()
        layoutManager.delegate = self
        layoutManager.layoutManagerDelegate = self
        textContainer.textView = self
        textContainer.heightTracksTextView = true
        textContainer.widthTracksTextView = true
        self.delegate = context
        richTextStorage.textStorageDelegate = self

        self.backgroundColor = defaultBackgroundColor
        self.textColor = defaultTextColor
//        self.typingAttributes = defaultTypingAttributes
    }

    override var undoManager: UndoManager? {
        _undoManager ?? super.undoManager
    }

    var contentLength: Int {
        return textStorage.length
    }

    weak var textProcessor: TextProcessor? {
        didSet {
            richTextStorage.delegate = textProcessor
        }
    }

    var textEndRange: NSRange {
        return NSRange(location: contentLength, length: 0)
    }

    var currentLineRange: NSRange? {
        return lineRange(from: selectedRange.location)
    }

    var viewport: CGRect {
        return bounds.inset(by: textContainerInset)
    }

    var visibleRange: NSRange? {
        return rangeForRect(viewport)
    }

    func contentLinesInRange(_ range: NSRange) -> [EditorLine] {
        guard range.isValidIn(self) else { return [] }
        var lines = [EditorLine]()

        var startingRange = NSRange(location: range.location, length: 0)
        let endLocation = max(startingRange.location, range.location + range.length - 1)

        while startingRange.location <= endLocation {
            let paraRange = rangeOfParagraph(at: startingRange.location)
            let text = self.attributedText.attributedSubstring(from: paraRange)
            let editorLine = EditorLine(text: text, range: paraRange)
            lines.append(editorLine)
            startingRange = NSRange(location: paraRange.length + paraRange.location + 1, length: 0)
        }

        return lines
    }

    func rangeOfParagraph(at location: Int) -> NSRange {
        guard let position = self.position(from: beginningOfDocument, offset: location),
              let paraRange = tokenizer.rangeEnclosingPosition(position, with: .paragraph, inDirection: UITextDirection(rawValue: UITextStorageDirection.backward.rawValue)),
              let range = paraRange.toNSRange(in: self)
        else {
            return NSRange(location: location, length: 0)
        }
        return range
    }

    func previousContentLine(from location: Int) -> EditorLine? {
        let currentLineRange = rangeOfParagraph(at: location)
        guard let position = self.position(from: beginningOfDocument, offset: currentLineRange.location - 1),
              let paraRange = tokenizer.rangeEnclosingPosition(position, with: .paragraph, inDirection: UITextDirection(rawValue: UITextStorageDirection.backward.rawValue)),
              let range = paraRange.toNSRange(in: self)
        else { return nil }
        
        return EditorLine(text: attributedText.attributedSubstring(from: range), range: range)
    }

    func nextContentLine(from location: Int) -> EditorLine? {
        let currentLineRange = rangeOfParagraph(at: location)
        guard let position = self.position(from: beginningOfDocument, offset: currentLineRange.endLocation + 1),
              let paraRange = tokenizer.rangeEnclosingPosition(position, with: .paragraph, inDirection: UITextDirection(rawValue: UITextStorageDirection.forward.rawValue)),
              let range = paraRange.toNSRange(in: self)
        else { return nil }
        
        return EditorLine(text: attributedText.attributedSubstring(from: range), range: range)
    }

    override var keyCommands: [UIKeyCommand]? {
        let tab = "\t"
        let enter = "\r"

        return [
            UIKeyCommand(input: tab, modifierFlags: [], action: #selector(handleKeyCommand(command:))),
            UIKeyCommand(input: tab, modifierFlags: .shift, action: #selector(handleKeyCommand(command:))),
            UIKeyCommand(input: enter, modifierFlags: .shift, action: #selector(handleKeyCommand(command:))),
            UIKeyCommand(input: enter, modifierFlags: .control, action: #selector(handleKeyCommand(command:))),
            UIKeyCommand(input: enter, modifierFlags: .alternate, action: #selector(handleKeyCommand(command:))),
        ]
    }

    @available(*, unavailable, message: "init(coder:) unavailable, use init")
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    #if targetEnvironment(macCatalyst)
    @objc(_focusRingType)
    var focusRingType: UInt {
        return 1 //NSFocusRingTypeNone
    }
    #endif

    @objc
    func handleKeyCommand(command: UIKeyCommand) {
        guard let input = command.input,
              let key = EditorKey(input)
        else { return }
        
        let modifierFlags = command.modifierFlags

        richTextViewDelegate?.richTextView(self, didReceive: key, modifierFlags: modifierFlags, at: selectedRange)
    }

    private func setupPlaceholder() {
        placeholderLabel.removeFromSuperview()
        addSubview(placeholderLabel)
        placeholderLabel.attributedText = placeholderText

        let placeholderLabelWidthConstraint = placeholderLabel.widthAnchor.constraint(equalTo: self.widthAnchor, constant: -textContainer.lineFragmentPadding - textContainerInset.right)
        placeholderLabelWidthConstraint.priority = .init(UILayoutPriority.required.rawValue - 1)

        NSLayoutConstraint.activate([
            placeholderLabel.topAnchor.constraint(equalTo: self.topAnchor, constant: textContainerInset.top),
            placeholderLabel.bottomAnchor.constraint(equalTo: self.bottomAnchor, constant: -textContainerInset.bottom),
            placeholderLabel.leadingAnchor.constraint(equalTo: self.leadingAnchor, constant: textContainer.lineFragmentPadding + textContainerInset.left),
            placeholderLabel.trailingAnchor.constraint(equalTo: self.trailingAnchor, constant: -textContainer.lineFragmentPadding - textContainerInset.right),
            placeholderLabelWidthConstraint
        ])
    }

    func wordAt(_ location: Int) -> NSAttributedString? {
        guard let position = self.position(from: beginningOfDocument, offset: location),
              let wordRange = tokenizer.rangeEnclosingPosition(position, with: .word, inDirection: UITextDirection(rawValue: UITextStorageDirection.backward.rawValue)),
              let range = wordRange.toNSRange(in: self)
        else { return nil }
        
        return attributedText.attributedSubstring(from: range)
    }

    func lineRange(from location: Int) -> NSRange? {
        var currentLocation = location
        guard contentLength > 0 else { return .zero }
        var range = NSRange()
        // In case this is called before layout has completed, e.g. from TextProcessor, the last entered glyph
        // will not have been laid out by layoutManager but would be present in TextStorage. It can also happen
        // when deleting multiple characters where layout is pending in the same case. Following logic finds the
        // last valid glyph that is already laid out.
        while currentLocation > 0 && layoutManager.isValidGlyphIndex(currentLocation) == false {
            currentLocation -= 1
        }
        guard layoutManager.isValidGlyphIndex(currentLocation) else { return NSRange(location: 0, length: 1) }
        layoutManager.lineFragmentUsedRect(forGlyphAt: currentLocation, effectiveRange: &range)
        guard range.location != NSNotFound else { return nil }
        // As mentioned above, in case of this getting called before layout is completed,
        // we need to account for the range that has been changed. storage.changeInLength provides
        // the change that might not have been laid already
        return NSRange(location: range.location, length: range.length + textStorage.changeInLength)
    }

    func invalidateLayout(for range: NSRange) {
        layoutManager.invalidateLayout(forCharacterRange: range.clamped(upperBound: textStorage.length), actualCharacterRange: nil)
    }

    func invalidateDisplay(for range: NSRange) {
        // invalidateDisplay does not need clamping like required by invalidateLayout
        layoutManager.invalidateDisplay(forCharacterRange: range)
    }

    func resetTypingAttributes() {
        self.typingAttributes = defaultTypingAttributes
    }

    override func deleteBackward() {
        defer {
            if contentLength == 0 {
                resetTypingAttributes()
            }
            richTextViewDelegate?.richTextView(self, didReceive: .backspace, modifierFlags: [], at: selectedRange)
        }

        guard contentLength > 0 else { return }
        let proposedRange = NSRange(location: max(0, selectedRange.location - 1), length: 0)

        let attributedText: NSAttributedString = self.attributedText // single allocation
        let attributeExists = (attributedText.attribute(.textBlock, at: proposedRange.location, effectiveRange: nil)) != nil

        guard attributeExists,
              let textRange = adjustedTextBlockRangeOnSelectionChange(oldRange: selectedRange, newRange: proposedRange)
        else {
            // if the character getting deleted is a list item spacer, do a double delete
            let textToBeDeleted = attributedText.substring(from: NSRange(location: proposedRange.location, length: 1))
            if textToBeDeleted == ListTextProcessor.blankLineFiller {
                super.deleteBackward()
            }
            super.deleteBackward()
            return
        }

        let rangeToDelete = NSRange(location: textRange.location, length: selectedRange.location - textRange.location)
        replaceCharacters(in: rangeToDelete, with: NSAttributedString())
        selectedRange = NSRange(location: textRange.location, length: 0)
    }

    func insertAttachment(in range: NSRange, attachment: Attachment) {
        richTextStorage.insertAttachment(in: range, attachment: attachment, withSpacer: attachment.spacer)
        // TODO: Temporary workaround to get around the issue of adding content type to attachments
        // This needs to be done outside PRTextStorage from ProtonCore as it can no longer depend on Proton framework
        // Ideally, attachment.string should be used - possibly consider using richTextStorage.replaceCharacters
        richTextStorage.addAttributes(attachment.attributes, range: NSRange(location: range.location, length: 1))
        if let rangeInContainer = attachment.rangeInContainer() {
            edited(range: rangeInContainer)
        }
//        scrollRangeToVisible(NSRange(location: range.location, length: 1))
    }

    func edited(range: NSRange) {
        textStorage.beginEditing()
        textStorage.edited([.editedCharacters, .editedAttributes], range: range, changeInLength: 0)
        textStorage.endEditing()
    }

    func transformContents<T: EditorContentEncoding>(in range: NSRange? = nil, using transformer: T) -> [T.EncodedType] {
        return contents(in: range).compactMap(transformer.encode)
    }

    func replaceCharacters(in range: NSRange, with attrString: NSAttributedString) {
        textStorage.replaceCharacters(in: range, with: attrString)
        ensuringValidSelectedRange()
    }

    func replaceCharacters(in range: NSRange, with string: String) {
        // Delegate to function with attrString so that default attributes are automatically applied
        textStorage.replaceCharacters(in: range, with: NSAttributedString(string: string))
    }

    @discardableResult
    func ensuringValidSelectedRange() -> NSRange {
        let clamped = selectedRange.clamped(upperBound: attributedText.length)
        if clamped != selectedRange {
            selectedRange = clamped
        }
        return selectedRange
    }

    private func updatePlaceholderVisibility() {
        guard (placeholderText ?? NSAttributedString(string: "")).string.count > 0,
            self.attributedText.length == 0 else {
            if placeholderLabel.superview != nil {
                placeholderLabel.removeFromSuperview()
            }
            return
        }
        setupPlaceholder()
        if isLineNumbersEnabled {
            //TODO: else use default
            contentMode = .redraw
        }
    }

    func attributeValue(at location: CGPoint, for attribute: NSAttributedString.Key) -> Any? {
        let characterIndex = layoutManager.characterIndex(for: location, in: textContainer, fractionOfDistanceBetweenInsertionPoints: nil)

        guard characterIndex < textStorage.length else {
            return nil
        }

        let attributes = textStorage.attributes(at: characterIndex, longestEffectiveRange: nil, in: textStorage.fullRange)
        return attributes[attribute]
    }

    func glyphRange(forCharacterRange range: NSRange) -> NSRange {
        return layoutManager.glyphRange(forCharacterRange: range, actualCharacterRange: nil)
    }

    func rects(for range: NSRange) -> [CGRect] {
        guard let textRange = range.toTextRange(textInput: self) else { return [] }
        let rects = selectionRects(for: textRange)
        return rects.map { $0.rect }
    }

    func boundingRect(forGlyphRange range: NSRange) -> CGRect {
        return layoutManager.boundingRect(forGlyphRange: range, in: textContainer)
    }

    func rangeForRect(_ rect: CGRect, performingLayout: Bool = false) -> NSRange? {
        let range = performingLayout
        ? layoutManager.glyphRange(forBoundingRect: rect, in: textContainer)
        : layoutManager.glyphRange(forBoundingRectWithoutAdditionalLayout: rect, in: textContainer)

        return range == .zero ? nil : range
    }

    func contents(in range: NSRange? = nil) -> AnySequence<EditorContent> {
        return self.attributedText.enumerateContents(in: range)
    }

    func setAttributes(_ attrs: [NSAttributedString.Key: Any], range: NSRange) {
        textStorage.setAttributes(attrs, range: range)
    }

    func addAttributes(_ attrs: [NSAttributedString.Key: Any], range: NSRange) {
        textStorage.addAttributes(attrs, range: range)
    }

    func removeAttributes(_ attrs: [NSAttributedString.Key], range: NSRange) {
        richTextStorage.removeAttributes(attrs, range: range)
    }

    func enumerateAttribute(_ attrName: NSAttributedString.Key, in enumerationRange: NSRange, options opts: NSAttributedString.EnumerationOptions = [], using block: (Any?, NSRange, UnsafeMutablePointer<ObjCBool>) -> Void) {
        textStorage.enumerateAttribute(attrName, in: enumerationRange, options: opts, using: block)
    }

    func rangeOfCharacter(at point: CGPoint) -> NSRange? {
        return characterRange(at: point)?.toNSRange(in: self)
    }

    func didTap(at location: CGPoint) {
        context?.selectedTextView = self
        let characterRange = rangeOfCharacter(at: location)
        richTextViewDelegate?.richTextView(self, didTapAtLocation: location, characterRange: characterRange)
    }

    override func touchesBegan(_ touches: Set<UITouch>, with event: UIEvent?) {
        if let touch = touches.first {
            let position = touch.location(in: self)
            didTap(at: position)
        }
    }

    // When a user enables `Use keyboard navigation to move focus between controls` it enables the focus system in the app.
    // It means focused item also becomes the first responder. UITextView is focusable by default, but if isEditable is set to false, it cannot be focussed anymore.
    // This leads to an issue where if a user selects text in non-editable text view and right clicks the text view loses the first responder to the focused menu, and therefore no actions are provided, and it is not possible to copy.
    // Returning true will make it focusable regardless if it is editable, and it will not be losing responder because it will stay focused.
    // It is not perfect, as a user can focus this view with Tab and make no actions on it.
    override var canBecomeFocused: Bool {
        return true
    }

    override func canPerformAction(_ action: Selector, withSender sender: Any?) -> Bool {
        // Check if the range is valid, as in some cases, if the range happens to be invalid, it
        // may result in a crash with [NSBigMutableString substringWithRange:]: Range out of bounds
        guard selectedRange.isValidIn(self) else { return false }
        guard editorView?.canPerformMenuAction(action, withSender: sender) == true else {
            return false
        }

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

    override func copy(_ sender: Any?) {
        if editorView?.responds(to: #selector(copy(_:))) ?? false {
            editorView?.copy(sender)
        } else {
            super.copy(sender)
        }
    }

    override func paste(_ sender: Any?) {
        if editorView?.responds(to: #selector(paste(_:))) ?? false {
            editorView?.paste(sender)
        } else {
            super.paste(sender)
        }
    }

    override func cut(_ sender: Any?) {
        if editorView?.responds(to: #selector(cut)) ?? false {
            editorView?.cut(sender)
        } else {
            super.cut(sender)
        }
    }

    override func select(_ sender: Any?) {
        if editorView?.responds(to: #selector(select)) ?? false {
            editorView?.select(sender)
        } else {
            super.select(sender)
        }
    }

    override func selectAll(_ sender: Any?) {
        if editorView?.responds(to: #selector(selectAll)) ?? false {
            editorView?.selectAll(sender)
        } else {
            super.selectAll(sender)
        }
    }

    override func toggleUnderline(_ sender: Any?) {
        if editorView?.responds(to: #selector(toggleUnderline)) ?? false {
            editorView?.toggleUnderline(sender)
        } else {
            super.toggleUnderline(sender)
        }
    }

    override func toggleItalics(_ sender: Any?) {
        if editorView?.responds(to: #selector(toggleItalics)) ?? false {
            editorView?.toggleItalics(sender)
        } else {
            super.toggleItalics(sender)
        }
    }

    override func toggleBoldface(_ sender: Any?) {
        if editorView?.responds(to: #selector(toggleBoldface)) ?? false {
            editorView?.toggleBoldface(sender)
        } else {
            super.toggleBoldface(sender)
        }
    }

    override func caretRect(for position: UITextPosition) -> CGRect {
        guard isEditable else {
            return super.caretRect(for: position)
        }

        let location = offset(from: beginningOfDocument, to: position)
        let lineRect = layoutManager.boundingRect(forGlyphRange: NSRange(location: location, length: 0), in: textContainer)

        var caretRect = super.caretRect(for: position)
        caretRect.origin.y = lineRect.minY + textContainerInset.top
        caretRect.size.height = lineRect.height
        return caretRect
    }

    override func addGestureRecognizer(_ gestureRecognizer: UIGestureRecognizer) {
        if gestureRecognizer.delegate is UIScrollView {
            super.addGestureRecognizer(gestureRecognizer)
        } else {
            // Add an override for gesture recognizer to handle long-press in attachments.
            // In absence of override, ling press on attachment will cause attachment to be lifted up
            // instead of allowing to select text in editor
            if let delegate = gestureRecognizer.delegate {
                let delegateOverride = GestureRecognizerDelegateOverride(baseDelegate: delegate)
                gestureRecognizer.delegate = delegateOverride
                delegateOverrides.append(delegateOverride)
            }
            super.addGestureRecognizer(gestureRecognizer)
        }
    }

    override func selectionRects(for range: UITextRange) -> [UITextSelectionRect] {
        let firstCharacterRect = caretRect(for: range.start)
        let lastCharacterRect = caretRect(for: range.end)

        return super.selectionRects(for: range).map { selectionRect -> UITextSelectionRect in
            if selectionRect.containsStart {
                return TextSelectionRect(selection: selectionRect, caretRect: firstCharacterRect)
            } else if selectionRect.containsEnd {
                return TextSelectionRect(selection: selectionRect, caretRect: lastCharacterRect)
            } else {
                return selectionRect
            }
        }
    }
}

extension RichTextView: NSLayoutManagerDelegate {
    func layoutManager(_ layoutManager: NSLayoutManager, didCompleteLayoutFor textContainer: NSTextContainer?, atEnd layoutFinishedFlag: Bool) {
        richTextViewDelegate?.richTextView(self, didFinishLayout: layoutFinishedFlag)
    }

    func layoutManager(_ layoutManager: NSLayoutManager, shouldBreakLineByHyphenatingBeforeCharacterAt charIndex: Int) -> Bool {
        // Disable hyphenation for backgroundStyle as otherwise, in some cases, with hyphenations,
        // layoutManager.glyphRange(forCharacterRange:) returns range with hyphenation and textstorage has
        // corresponding ranges without hyphenation
        // This will likely be revised in a future version to enable back hyphenation
        return false
    }

}

extension RichTextView: TextStorageDelegate {
    func textStorage(_ textStorage: PRTextStorage, didDelete attachment: NSTextAttachment) {
        guard let attachment = attachment as? Attachment else {
            return
        }
        attachment.removeFromSuperview()
    }
    
    func textStorage(_ textStorage: PRTextStorage, will deleteText: NSAttributedString, insertText insertedText: NSAttributedString, in range: NSRange) {
        textProcessor?.textStorage(textStorage, willProcessDeletedText: deleteText, insertedText: insertedText, range: range)
    }
    
    func textStorage(_ textStorage: PRTextStorage, edited actions: NSTextStorage.EditActions, in editedRange: NSRange, changeInLength delta: Int) {
        updatePlaceholderVisibility()
        drawDefaultLineNumberIfRequired()
    }
}

extension RichTextView: LayoutManagerDelegate {
    var lineNumberWrappingMarker: String? {
        lineNumberProvider?.lineNumberWrappingMarker
    }

    var listLineFormatting: LineFormatting {
        return richTextViewListDelegate?.listLineFormatting ?? RichTextView.defaultListLineFormatting
    }

    var paragraphStyle: NSMutableParagraphStyle? {
        return defaultTextFormattingProvider?.paragraphStyle
    }

    func lineNumberString(for index: Int) -> String? {
        lineNumberProvider?.lineNumberString(for: index)
    }

    func listMarkerForItem(at index: Int, level: Int, previousLevel: Int, attributeValue: Any?) -> ListLineMarker {
        let font = UIFont.preferredFont(forTextStyle: .body)
        let defaultValue = NSAttributedString(string: "*", attributes: [.font: font])
        return richTextViewListDelegate?.richTextView(self, listMarkerForItemAt: index, level: level, previousLevel: previousLevel, attributeValue: attributeValue) ?? .string(defaultValue)
    }
}

private final class TextSelectionRect: UITextSelectionRect {
    override var rect: CGRect { _rect }
    override var writingDirection: NSWritingDirection { _writingDirection }
    override var containsStart: Bool { _containsStart }
    override var containsEnd: Bool { _containsEnd }
    override var isVertical: Bool { _isVertical }

    private let _rect: CGRect
    private let _writingDirection: NSWritingDirection
    private let _containsStart: Bool
    private let _containsEnd: Bool
    private let _isVertical: Bool

    init(selection: UITextSelectionRect, caretRect: CGRect) {
        self._rect = .init(x: selection.rect.minX, y: caretRect.minY, width: selection.rect.width, height: caretRect.height)
        self._writingDirection = selection.writingDirection
        self._containsStart = selection.containsStart
        self._containsEnd = selection.containsEnd
        self._isVertical = selection.isVertical
    }
}