添加功能，可自定义资源下载

rztime · Aug 27, 2024 · db9fa45 · db9fa45
1 parent 977be32
commit db9fa45
Show file tree

Hide file tree

Showing 8 changed files with 90 additions and 20 deletions.
diff --git a/Example/RZRichTextView/HTML2AttrViewController.swift b/Example/RZRichTextView/HTML2AttrViewController.swift
@@ -69,13 +69,8 @@ class HTML2AttrViewController: UIViewController {
         }, disposebag: btn)
 
         self.navigationItem.rightBarButtonItem = UIBarButtonItem.init(customView: btn)
-
-        var html = try? String.init(contentsOfFile: "/Users/rztime/Desktop/test.html")
-        html = """
-<p style=\"margin:0.0px 0.0px 0.0px 0.0px;font-size:16.0px;\"><span style=\"color:#191919;font-size:16.0px;\">I yu</span></p><br/><p style=\"margin:0.0px 0.0px 0.0px 0.0px;font-size:16.0px;\"><span style=\"color:#191919;font-size:16.0px;\"><a href=\"https\">百度一下</a></span><span style=\"color:#191919;font-size:16.0px;\"></span></p><p style=\"margin:0.0px 0.0px 0.0px 0.0px;font-size:16.0px;\"><span style=\"color:#191919;font-size:16.0px;\">生活的</span></p>
-"""
+        let html = (try? String.init(contentsOfFile: "/Users/rztime/Desktop/test.html"))
         let t = "<body style=\"font-size:16px;color:#110000;\">\(html ?? "")</body>"
         textView.html2Attributedstring(html: t)
-
     }
 }
diff --git a/Example/RZRichTextView/HowToUseDemo.swift b/Example/RZRichTextView/HowToUseDemo.swift
@@ -14,12 +14,39 @@ import Kingfisher
 
 /// 使用时，直接将此代码复制到项目中，并完成相关FIXME的地方即可
 public extension RZRichTextViewModel {
+    /// 如果有需要自定义实现资源下载，可以放开代码，并实现sync_imageBy、async_imageBy方法
+    static var configure: RZRichTextViewConfigure = {
+        /// 同步获取图片
+        RZRichTextViewConfigure.shared.sync_imageBy = { source in
+            print("sync source:\(source ?? "")")
+            let imgView = UIImageView()
+            imgView.kf.setImage(with: source?.qtoURL)
+            return imgView.image
+        }
+        /// 异步获取图片
+        RZRichTextViewConfigure.shared.async_imageBy = { source, complete in
+            print("async source:\(source ?? "")")
+            let comp = complete
+            let s = source
+            let imgView = UIImageView()
+            imgView.kf.setImage(with: source?.qtoURL) { result in
+                let image = try? result.get().image
+                comp?(s, image)
+            }
+        }
+        return RZRichTextViewConfigure.shared
+    }()
     class func shared(edit: Bool = true) -> RZRichTextViewModel {
         /// 自定义遮罩view 默认RZAttachmentInfoLayerView
 //        RZAttachmentOption.register(attachmentLayer: RZAttachmentInfoLayerView.self)
 
+        /// 如果有需要自定义实现资源下载，可以放开代码，并实现sync_imageBy、async_imageBy方法
+//        _ = RZRichTextViewModel.configure
+
         let viewModel = RZRichTextViewModel.init()
         viewModel.canEdit = edit
+        /// 链接颜色
+        viewModel.defaultLinkTypingAttributes = [.foregroundColor: UIColor.qhex(0x307bf6), .underlineColor: UIColor.qhex(0x307bf6), .underlineStyle: NSUnderlineStyle.styleSingle.rawValue]
         /// 显示音频文件名字
 //        viewModel.showAudioName = false
         /// 音频高度

diff --git a/Example/RZRichTextView/LabelLoadHtmlViewController.swift b/Example/RZRichTextView/LabelLoadHtmlViewController.swift
@@ -31,7 +31,7 @@ class LabelLoadHtmlViewController: UIViewController {
             cell.reload = { [weak self] indexPath in
                 self?.tableView.reloadRows(at: [indexPath], with: .automatic)
             }
-            cell.html = try? String.init(contentsOfFile: "/Users/rztime/Desktop/test1.html")
+            cell.html = try? String.init(contentsOfFile: "/Users/rztime/Desktop/test.html")
             return cell
         }
     }

diff --git a/Example/RZRichTextView/ViewController.swift b/Example/RZRichTextView/ViewController.swift
@@ -60,7 +60,6 @@ class ViewController: UIViewController {
                 make.edges.equalToSuperview()
             })
         ])
-
     }
 
     override func didReceiveMemoryWarning() {

diff --git a/RZRichTextView/Classes/LabelRZRich.swift b/RZRichTextView/Classes/LabelRZRich.swift
@@ -72,10 +72,19 @@ public extension UILabel {
                 switch info.type {
                 case .image, .video:
                     let url = info.poster?.qtoURL ?? info.src?.qtoURL
-                    UIImage.asyncImageBy(url?.absoluteString) {[weak info]  image in
-                        info?.image = image
-                        info?.image = UILabel.creatAttachmentInfoView(info, width: max)
-                        fix(attachment: at.0, range: at.1)
+                    if let c = RZRichTextViewConfigure.shared.async_imageBy {
+                        let complete: ((String?, UIImage?) -> Void)? = { [weak info] source, image in
+                            info?.image = image
+                            info?.image = UILabel.creatAttachmentInfoView(info, width: max)
+                            fix(attachment: at.0, range: at.1)
+                        }
+                        c(url?.absoluteString, complete)
+                    } else {
+                        UIImage.asyncImageBy(url?.absoluteString) { [weak info] image in
+                            info?.image = image
+                            info?.image = UILabel.creatAttachmentInfoView(info, width: max)
+                            fix(attachment: at.0, range: at.1)
+                        }
                     }
                 case .audio:
                     info.image = UILabel.creatAttachmentInfoView(info, width: max)

diff --git a/RZRichTextView/Classes/RZAttachmentInfoLayerView.swift b/RZRichTextView/Classes/RZAttachmentInfoLayerView.swift
@@ -59,9 +59,17 @@ open class RZAttachmentInfoLayerView: UIView, RZAttachmentInfoLayerProtocol {
                         }
                     }
                 } else if let url = info.src {
-                    self.imageView.kf.setImage(with: url.qtoURL, completionHandler: { [weak self] _ in
-                        self?.updateImageViewSize()
-                    })
+                    if let c = RZRichTextViewConfigure.shared.async_imageBy {
+                        let complete: ((String?, UIImage?) -> Void)? = { [weak self] source, image in
+                            self?.imageView.image = image
+                            self?.updateImageViewSize()
+                        }
+                        c(url, complete)
+                    } else {
+                        self.imageView.kf.setImage(with: url.qtoURL, completionHandler: { [weak self] _ in
+                            self?.updateImageViewSize()
+                        })
+                    } 
                 } else {
                     info.imagePublish.subscribe({ [weak self] value in
                         guard let self = self else { return }

diff --git a/RZRichTextView/Classes/RZHtml.swift b/RZRichTextView/Classes/RZHtml.swift
@@ -80,9 +80,17 @@ public extension RZRichTextView {
                 case .image, .video:
                     let url = info.poster?.qtoURL ?? info.src?.qtoURL
                     if let url = url?.absoluteString {
-                        UIImage.asyncImageBy(url) { [weak info] image in
-                            info?.image = image
-                            fix(attachment: at.0, range: at.1)
+                        if let c = RZRichTextViewConfigure.shared.async_imageBy {
+                            let complete: ((String?, UIImage?) -> Void)? = { [weak info] source, image in
+                                info?.image = image
+                                fix(attachment: at.0, range: at.1)
+                            }
+                            c(url, complete)
+                        } else {
+                            UIImage.asyncImageBy(url) { [weak info] image in
+                                info?.image = image
+                                fix(attachment: at.0, range: at.1)
+                            }
                         }
                     }
                 case .audio:
@@ -400,10 +408,18 @@ public extension String {
                 at.rzattachmentInfo = att
                 switch att.type {
                 case .image:
-                    att.image = UIImage.syncImageBy(att.src)
+                    if let c = RZRichTextViewConfigure.shared.sync_imageBy {
+                        att.image = c(att.src)
+                    } else {
+                        att.image = UIImage.syncImageBy(att.src)
+                    }
                 case .video:
                     let src = ((att.poster.qisEmpty ? att.src : att.poster) ?? "")
-                    att.image = UIImage.syncImageBy(src)
+                    if let c = RZRichTextViewConfigure.shared.sync_imageBy {
+                        att.image = c(att.src)
+                    } else {
+                        att.image = UIImage.syncImageBy(src)
+                    }
                 case .audio:
                     break
                 }

diff --git a/RZRichTextView/Classes/RZRichTextViewConfigure.swift b/RZRichTextView/Classes/RZRichTextViewConfigure.swift
@@ -0,0 +1,16 @@
+//
+//  RZRichTextViewConfigure.swift
+//  RZRichTextView
+//
+//  Created by rztime on 2024/8/23.
+//
+
+import UIKit
+
+open class RZRichTextViewConfigure: NSObject {
+    public static var shared: RZRichTextViewConfigure = .init()
+    /// 同步获取图片
+    public var sync_imageBy: ((_ source: String?) -> UIImage?)?
+    /// 异步获取图片，complete里的source需要于图片对应
+    public var async_imageBy: ((_ source: String?, _ complete: ((_ source: String?, _ image: UIImage?) -> Void)?) -> Void)?
+}
-Original file line number
+Diff line change
@@ Expand Up / @@ -60,7 +60,6 @@ class ViewController: UIViewController { @@
                     make.edges.equalToSuperview()
                 })
             ])
         }
         override func didReceiveMemoryWarning() {
@@ Expand Down @@