@@ -11,12 +11,17 @@ import UIKit
1111class IMUIImageMessageCell : IMUIBaseMessageCell {
1212
1313 var imageView = UIImageView ( )
14-
14+ var myMessage : IMUIMessageModelProtocol ?
1515 override init ( frame: CGRect ) {
1616 super. init ( frame: frame)
1717 bubbleView. addSubview ( imageView)
18+ NotificationCenter . default. addObserver ( self , selector: #selector( clickDidCompletePic ( notification: ) ) , name: NSNotification . Name ( rawValue: " RNNeteaseimDidCompletePic " ) , object: nil )
1819 }
1920
21+ func clickDidCompletePic( notification: Notification ) {
22+ self . layoutImage ( image: UIImage ( contentsOfFile: ( self . myMessage? . mediaFilePath ( ) ) !) !)
23+ }
24+
2025 required init ? ( coder aDecoder: NSCoder ) {
2126 fatalError ( " init(coder:) has not been implemented " )
2227 }
@@ -29,14 +34,18 @@ class IMUIImageMessageCell: IMUIBaseMessageCell {
2934 super. presentCell ( with: message, viewCache: viewCache, delegate: delegate)
3035
3136 let layout = message. layout
37+ self . myMessage = message
3238 self . imageView. frame = UIEdgeInsetsInsetRect ( CGRect ( origin: CGPoint . zero, size: layout. bubbleFrame. size) , layout. bubbleContentInset)
3339 let image = UIImage ( contentsOfFile: message. mediaFilePath ( ) )
3440 if image != nil {
3541 self . layoutImage ( image: image!)
3642 }
37-
3843 }
3944
45+ deinit {
46+ NotificationCenter . default. removeObserver ( self )
47+ }
48+
4049 func layoutImage( image: UIImage ) {
4150 self . imageView. image = image
4251 }
0 commit comments