Merge pull request #1237 from wordpress-mobile/issue/fix_color_attributes

Fix copy/paste of attributed string from app without support for dark/light mode

Author: SergioEstevao

Aztec/Classes/TextKit/TextView.swift

@@ -242,12 +242,20 @@ open class TextView: UITextView {
             .font: defaultFont,
             .paragraphStyle: defaultParagraphStyle,
         ]
-        if let color = textColor {
+        if let color = defaultTextColor {
             attributes[.foregroundColor] = color
         }
         return attributes
     }
-    
+
+    open lazy var defaultTextColor: UIColor? = {
+        if #available(iOS 13.0, *) {
+            return UIColor.label
+        } else {
+            return UIColor.darkText
+        }        
+    }()
+
     // MARK: - Plugin Loading
 
     var pluginManager: PluginManager {
@@ -637,7 +645,7 @@ open class TextView: UITextView {
         evaluateRemovalOfSingleLineParagraphAttributesAfterSelectionChange()
         ensureRemovalOfParagraphAttributesWhenPressingBackspaceAndEmptyingTheDocument()
         ensureCursorRedraw(afterEditing: deletedString.string)
-
+        recalculateTypingAttributes()
         notifyTextViewDidChange()
     }
 
@@ -1237,6 +1245,7 @@ open class TextView: UITextView {
     private func recalculateTypingAttributes(at location: Int) {
 
         guard storage.length > 0 else {
+            typingAttributes = defaultAttributes
             return
         }
 

Aztec/Classes/TextKit/TextViewPasteboardDelegate.swift

@@ -58,7 +58,7 @@ open class AztecTextViewPasteboardDelegate: TextViewPasteboardDelegate {
         guard let string = UIPasteboard.general.attributedString() else {
             return false
         }
-
+        string.loadLazyAttachments()
         let selectedRange = textView.selectedRange
         let storage = textView.storage
 
@@ -69,9 +69,9 @@ open class AztecTextViewPasteboardDelegate: TextViewPasteboardDelegate {
             textView?.undoTextReplacement(of: originalText, finalRange: finalRange)
         })
 
-        string.loadLazyAttachments()
+        let colorCorrectedString = fixColors(in: string, using: textView.defaultTextColor)
 
-        storage.replaceCharacters(in: selectedRange, with: string)
+        storage.replaceCharacters(in: selectedRange, with: colorCorrectedString)
         textView.notifyTextViewDidChange()
 
         let newSelectedRange = NSRange(location: selectedRange.location + string.length, length: 0)
@@ -80,6 +80,21 @@ open class AztecTextViewPasteboardDelegate: TextViewPasteboardDelegate {
         return true
     }
 
+    private func fixColors(in string: NSAttributedString, using baseColor: UIColor?) -> NSAttributedString {
+        guard #available(iOS 13.0, *) else {
+            return string
+        }
+        let colorToUse = baseColor ?? UIColor.label
+
+        let newString = NSMutableAttributedString(attributedString: string)
+        newString.enumerateAttributes(in: newString.rangeOfEntireString, options: []) { (attributes, range, stop) in
+            if attributes[.foregroundColor] == nil {
+                newString.setAttributes([.foregroundColor: colorToUse], range: range)
+            }
+        }
+        return newString
+    }
+
     /// Tries to paste raw text from the clipboard, replacing the selected range.
     ///
     /// - Returns: True if the paste succeeds, false if it does not.

AztecTests/TextKit/TextViewTests.swift

@@ -2078,4 +2078,32 @@ class TextViewTests: XCTestCase {
         XCTAssertEqual(UIPasteboard.general.html(), "<p><strong>bol</strong></p>")
     }
 
+    /// Check if pasting attributed text without color, the color is set to default color
+    func testPasteAttributedText() {
+        let sourceAttributedText = NSAttributedString(string: "Hello world")
+        var pasteItems = [String:Any]()
+        pasteItems[kUTTypePlainText as String] = try! sourceAttributedText.data(from: sourceAttributedText.rangeOfEntireString, documentAttributes: [.documentType: DocumentType.plain])
+        UIPasteboard.general.setItems([pasteItems], options: [:])
+        let textView = TextViewStub(withHTML: "")
+        textView.defaultTextColor = UIColor.red
+        textView.paste(nil)
+
+        let attributedString = textView.attributedText!
+
+        let attributes = attributedString.attributes(at: 0, effectiveRange: nil)
+
+        XCTAssertEqual(attributes[.foregroundColor] as! UIColor, UIColor.red)
+        XCTAssertEqual(textView.text, "Hello world")
+    }
+
+    /// Check that deleting all text resets the typying attributes to the default attributes
+    func testDeleteAllTextSetTypyingAttributesToDefault() {
+        let textView = TextViewStub(withHTML: "<p style=\"color:0xFF0000\">Hello world</p>")
+        textView.defaultTextColor = UIColor.green
+        textView.selectedRange = textView.attributedText!.rangeOfEntireString
+        textView.deleteBackward()
+        let attributes = textView.typingAttributes
+        XCTAssertEqual(attributes[.foregroundColor] as! UIColor,UIColor.green)
+    }
+
 }

Example/Example/EditorDemoController.swift

@@ -65,7 +65,7 @@ class EditorDemoController: UIViewController {
 
     private func setupHTMLTextView(_ textView: UITextView) {
         let accessibilityLabel = NSLocalizedString("HTML Content", comment: "Post HTML content")
-        self.configureDefaultProperties(for: textView, accessibilityLabel: accessibilityLabel)
+        self.configureDefaultProperties(htmlTextView: textView, accessibilityLabel: accessibilityLabel)
 
         textView.isHidden = true
         textView.delegate = self
@@ -325,7 +325,22 @@ class EditorDemoController: UIViewController {
     }
 
 
-    private func configureDefaultProperties(for textView: UITextView, accessibilityLabel: String) {
+    private func configureDefaultProperties(for textView: TextView, accessibilityLabel: String) {
+        textView.accessibilityLabel = accessibilityLabel
+        textView.font = Constants.defaultContentFont
+        textView.keyboardDismissMode = .interactive
+        if #available(iOS 13.0, *) {
+            textView.textColor = UIColor.label
+            textView.defaultTextColor = UIColor.label
+        } else {
+            // Fallback on earlier versions
+            textView.textColor = UIColor(red: 0x1A/255.0, green: 0x1A/255.0, blue: 0x1A/255.0, alpha: 1)
+            textView.defaultTextColor = UIColor(red: 0x1A/255.0, green: 0x1A/255.0, blue: 0x1A/255.0, alpha: 1)
+        }
+        textView.linkTextAttributes = [.foregroundColor: UIColor(red: 0x01 / 255.0, green: 0x60 / 255.0, blue: 0x87 / 255.0, alpha: 1), NSAttributedString.Key.underlineStyle: NSNumber(value: NSUnderlineStyle.single.rawValue)]
+    }
+
+    private func configureDefaultProperties(htmlTextView textView: UITextView, accessibilityLabel: String) {
         textView.accessibilityLabel = accessibilityLabel
         textView.font = Constants.defaultContentFont
         textView.keyboardDismissMode = .interactive