From 18755e8e64f5b155d834be18d353606dabba40a0 Mon Sep 17 00:00:00 2001
From: nielsandriesse <andriesseniels@gmail.com>
Date: Fri, 6 Nov 2020 15:53:08 +1100
Subject: [PATCH] Implement profile

---
 .../VisibleMessage+Profile.swift              | 44 ++++++++++++++++++-
 1 file changed, 42 insertions(+), 2 deletions(-)

diff --git a/SessionMessagingKit/Messages/Visible Message/VisibleMessage+Profile.swift b/SessionMessagingKit/Messages/Visible Message/VisibleMessage+Profile.swift
index 618b4c4ea..68708d1b2 100644
--- a/SessionMessagingKit/Messages/Visible Message/VisibleMessage+Profile.swift	
+++ b/SessionMessagingKit/Messages/Visible Message/VisibleMessage+Profile.swift	
@@ -1,15 +1,55 @@
+import SessionUtilities
 
 public extension VisibleMessage {
 
     @objc(SNProfile)
     class Profile : NSObject, NSCoding {
+        public var displayName: String?
+        public var profileKey: Data?
+        public var profilePictureURL: String?
+
+        internal init(displayName: String, profileKey: Data? = nil, profilePictureURL: String? = nil) {
+            self.displayName = displayName
+            self.profileKey = profileKey
+            self.profilePictureURL = profilePictureURL
+        }
 
         public required init?(coder: NSCoder) {
-            fatalError("Not implemented.")
+            if let displayName = coder.decodeObject(forKey: "displayName") as! String? { self.displayName = displayName }
+            if let profileKey = coder.decodeObject(forKey: "profileKey") as! Data? { self.profileKey = profileKey }
+            if let profilePictureURL = coder.decodeObject(forKey: "profilePictureURL") as! String? { self.profilePictureURL = profilePictureURL }
         }
 
         public func encode(with coder: NSCoder) {
-            fatalError("Not implemented.")
+            coder.encode(displayName, forKey: "displayName")
+            coder.encode(profileKey, forKey: "profileKey")
+            coder.encode(profilePictureURL, forKey: "profilePictureURL")
+        }
+
+        public static func fromProto(_ proto: SNProtoDataMessage) -> Profile? {
+            guard let profileProto = proto.profile, let displayName = profileProto.displayName else { return nil }
+            let profileKey = proto.profileKey
+            let profilePictureURL = profileProto.profilePicture
+            if let profileKey = profileKey, let profilePictureURL = profilePictureURL {
+                return Profile(displayName: displayName, profileKey: profileKey, profilePictureURL: profilePictureURL)
+            } else {
+                return Profile(displayName: displayName)
+            }
+        }
+
+        public func toProto() -> SNProtoDataMessageLokiProfile? {
+            guard let displayName = displayName else {
+                SNLog("Couldn't construct profile proto from: \(self).")
+                return nil
+            }
+            let profileProto = SNProtoDataMessageLokiProfile.builder()
+            profileProto.setDisplayName(displayName)
+            do {
+                return try profileProto.build()
+            } catch {
+                SNLog("Couldn't construct profile proto from: \(self).")
+                return nil
+            }
         }
     }
 }