NSNull check fix.

avaidyam · Feb 3, 2016 · 0a9d10a · 0a9d10a
1 parent e9dd7aa
commit 0a9d10a
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 12 deletions.
diff --git a/Hangouts/Messages.swift b/Hangouts/Messages.swift
@@ -726,7 +726,7 @@ public typealias GetSelfInfoResponse = GET_SELF_INFO_RESPONSE
 
 @objc(RESPONSE_HEADER)
 public class RESPONSE_HEADER : Message {
-    public var status: NSNumber? // ResponseStatus?
+    public var status: ResponseStatus?
     public var field1: AnyObject?
     public var field2: AnyObject?
     public var request_trace_id: NSString = ""
@@ -1068,10 +1068,10 @@ public class UserConversationState: Message {
 public class ConversationParticipantData: Message {
 	public var id: ParticipantId?
 	public var fallback_name: NSString?
-	public var invitation_status: NSNumber? // InvitationStatus?
+	public var invitation_status: InvitationStatus?
 	public var field4: AnyObject?
-	public var participant_type: NSNumber? // ParticipantType?
-	public var new_invitation_status: NSNumber? // InvitationStatus?
+	public var participant_type: ParticipantType?
+	public var new_invitation_status: InvitationStatus?
 }
 public typealias CONVERSATION_PARTICIPANT_DATA = ConversationParticipantData
 
@@ -1165,16 +1165,16 @@ public class Entity: Message {
 	public var properties: EntityProperties?
 	public var field11: AnyObject?
 	public var field12: AnyObject?
-	public var entity_type: AnyObject? // ParticipantType?
+	public var entity_type: ParticipantType?
 	public var field14: AnyObject?
 	public var field15: AnyObject?
-	public var had_past_hangout_state: AnyObject? // PastHangoutState?
+	public var had_past_hangout_state: PastHangoutState?
 }
 public typealias ENTITY = Entity
 
 @objc(EntityProperties)
 public class EntityProperties: Message {
-	public var type: NSNumber? // ProfileType?
+	public var type: ProfileType?
 	public var display_name: NSString?
 	public var first_name: NSString?
 	public var photo_url: NSString?
@@ -1184,8 +1184,8 @@ public class EntityProperties: Message {
 	public var field8: AnyObject?
 	public var field9: AnyObject?
 	public var in_users_domain: NSNumber?
-	public var gender: NSNumber? // Gender?
-	public var photo_url_status: NSNumber? // PhotoUrlStatus?
+	public var gender: Gender?
+	public var photo_url_status: PhotoUrlStatus?
 	public var field13: AnyObject?
 	public var field14: AnyObject?
 	public var canonical_email: NSString?

diff --git a/Hangouts/PBLite.swift b/Hangouts/PBLite.swift
@@ -200,9 +200,10 @@ public class PBLiteSerialization {
 
 				//  Unwrapping an optional enum
 			} else if let type = _unwrapOptionalType(property) as? Enum.Type {
-				print("TEST \(propertyName): \(type)")
-				let val: (AnyObject?) = type.init(value: (arr[i] as! NSNumber))
-				instance.setValue(val, forKey: propertyName)
+				if !(arr[i] is NSNull) {
+					let val: (AnyObject?) = type.init(value: (arr[i] as! NSNumber))
+					instance.setValue(val, forKey: propertyName)
+				}
 
 				//  Using a non-optional sub-struct
 			} else if let enumv = property as? Enum {