Merge branch 'feature/support-change-uid'

alibaba-archive · Apr 6, 2017 · ae1ebeb · ae1ebeb
2 parents 45d5a0b + de6a663
commit ae1ebeb
Show file tree

Hide file tree

Showing 6 changed files with 44 additions and 11 deletions.
diff --git a/MessageKit/BaseMessageModel.swift b/MessageKit/BaseMessageModel.swift
@@ -28,13 +28,18 @@ public protocol MessageModelProtocol: MessageItemProtocol {
 
 public protocol DecoratedMessageModelProtocol: MessageModelProtocol {
 
-    var messageModel: MessageModelProtocol { get }
+    var messageModel: MessageModelProtocol { get set }
 }
 
 public extension DecoratedMessageModelProtocol {
 
     var uid: String {
-        return self.messageModel.uid
+        get {
+            return self.messageModel.uid
+        }
+        set {
+            self.messageModel.uid = newValue
+        }
     }
 
     var senderId: String {

diff --git a/MessageKit/CustomMessageModel.swift b/MessageKit/CustomMessageModel.swift
@@ -15,9 +15,16 @@ public protocol CustomMessageModelProtocol: DecoratedMessageModelProtocol {
 
 open class CustomMessageModel: CustomMessageModelProtocol {
 
-    open let messageModel: MessageModelProtocol
+    open var messageModel: MessageModelProtocol
     open var customView: UIView
-    open var uid: String { return messageModel.uid }
+    open var uid: String {
+        get {
+            return messageModel.uid
+        }
+        set {
+            messageModel.uid = newValue
+        }
+    }
 
     public init(messageModel: MessageModelProtocol, customView: UIView) {
         self.messageModel = messageModel

diff --git a/MessageKit/FileMessageModel.swift b/MessageKit/FileMessageModel.swift
@@ -17,11 +17,18 @@ public protocol FileMessageModelProtocol: DecoratedMessageModelProtocol {
 
 open class FileMessageModel: FileMessageModelProtocol {
 
-    open let messageModel: MessageModelProtocol
+    open var messageModel: MessageModelProtocol
     open var fileName: String
     open var fileSize: String
     open var fileFolderColor: UIColor
-    open var uid: String { return messageModel.uid }
+    open var uid: String {
+        get {
+            return messageModel.uid
+        }
+        set {
+            messageModel.uid = newValue
+        }
+    }
 
     public init(messageModel: MessageModelProtocol, fileName: String, fileSize: String, fileFolderColor: UIColor) {
         self.messageModel = messageModel

diff --git a/MessageKit/MessageItemProtocol.swift b/MessageKit/MessageItemProtocol.swift
@@ -15,5 +15,5 @@ public protocol MessageItemProtocol: class, UniqueIdentificable {
 }
 
 public protocol UniqueIdentificable {
-    var uid: String { get }
+    var uid: String { get set }
 }
diff --git a/MessageKit/PhotoMessageModel.swift b/MessageKit/PhotoMessageModel.swift
@@ -18,10 +18,17 @@ public typealias PhotoClosure =  (_ imageview: UIImageView) -> ()
 
 open class PhotoMessageModel: PhotoMessageModelProtocol {
 
-    open let messageModel: MessageModelProtocol
+    open var messageModel: MessageModelProtocol
     open let imageSize: CGSize
     open let imageClosure: PhotoClosure
-    open var uid: String { return messageModel.uid }
+    open var uid: String {
+        get {
+            return messageModel.uid
+        }
+        set {
+            messageModel.uid = newValue
+        }
+    }
 
     public init(messageModel: MessageModelProtocol, imageSize: CGSize, imageClosure: @escaping PhotoClosure) {
         self.messageModel = messageModel

diff --git a/MessageKit/TextMessageModel.swift b/MessageKit/TextMessageModel.swift
@@ -15,9 +15,16 @@ public protocol TextMessageModelProtocol: DecoratedMessageModelProtocol {
 
 open class TextMessageModel: TextMessageModelProtocol {
 
-    open let messageModel: MessageModelProtocol
+    open var messageModel: MessageModelProtocol
     open let text: String
-    open var uid: String { return self.messageModel.uid }
+    open var uid: String {
+        get {
+            return messageModel.uid
+        }
+        set {
+            messageModel.uid = newValue
+        }
+    }
 
     public init(messageModel: MessageModelProtocol, text: String) {
         self.messageModel = messageModel