UITextView+Extensions.swift

//
//  UITextViewExtensions.swift
//
//  Created by Daniel Tartaglia on 7/20/17.
//  Copyright © 2017 Daniel Tartaglia. MIT License.
//

import UIKit

extension UITextField {

    func set(text newText: String, preservingCursor: Bool) {
        if preservingCursor {
            let cursorPosition = offset(from: beginningOfDocument, to: selectedTextRange!.start) + newText.count - (text?.count ?? 0)
            text = newText
            if let newPosition = position(from: beginningOfDocument, offset: cursorPosition) {
                selectedTextRange = textRange(from: newPosition, to: newPosition)
            }
        }
        else {
            text = newText
        }
    }

    func set(attributedText newText: NSAttributedString, preservingCursor: Bool) {
        if preservingCursor {
            let cursorPosition = offset(from: beginningOfDocument, to: selectedTextRange!.start) + newText.string.count - (attributedText?.string.count ?? 0)
            attributedText = newText
            if let newPosition = position(from: beginningOfDocument, offset: cursorPosition) {
                selectedTextRange = textRange(from: newPosition, to: newPosition)
            }
        }
        else {
            attributedText = newText
        }
    }
}

extension UITextView {

    func set(text newText: String, preservingCursor: Bool) {
        if preservingCursor {
            let cursorPosition = offset(from: beginningOfDocument, to: selectedTextRange!.start) + newText.count - (text?.count ?? 0)
            text = newText
            if let newPosition = position(from: beginningOfDocument, offset: cursorPosition) {
                selectedTextRange = textRange(from: newPosition, to: newPosition)
            }
        }
        else {
            text = newText
        }
    }

    func set(attributedText newText: NSAttributedString, preservingCursor: Bool) {
        if preservingCursor {
            let cursorPosition = offset(from: beginningOfDocument, to: selectedTextRange!.start) + newText.string.count - (attributedText?.string.count ?? 0)
            attributedText = newText
            if let newPosition = position(from: beginningOfDocument, offset: cursorPosition) {
                selectedTextRange = textRange(from: newPosition, to: newPosition)
            }
        }
        else {
            attributedText = newText
        }
    }
}