Initial

1 files changed, 132 insertions, 0 deletions

diff --git a/Firebase/Messaging/FIRMessagingCheckinService.m b/Firebase/Messaging/FIRMessagingCheckinService.m
new file mode 100644
index 0000000..9dad847
--- /dev/null
+++ b/Firebase/Messaging/FIRMessagingCheckinService.m
@@ -0,0 +1,132 @@
+/*
+ * Copyright 2017 Google
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#import "FIRMessagingCheckinService.h"
+
+#import "FIRMessagingUtilities.h"
+#import "NSError+FIRMessaging.h"
+
+@interface FIRMessagingCheckinService ()
+
+// This property is of type FIRInstanceIDCheckinPreferences, if InstanceID was directly linkable
+@property(nonatomic, readwrite, strong) id checkinPreferences;
+
+@end
+
+@implementation FIRMessagingCheckinService;
+
+#pragma mark - Reflection-Based Getter Functions
+
+// Encapsulates the -hasValidCheckinInfo method of FIRInstanceIDCheckinPreferences
+BOOL FIRMessagingCheckinService_hasValidCheckinInfo(id checkinPreferences) {
+  SEL hasValidCheckinInfoSelector = NSSelectorFromString(@"hasValidCheckinInfo");
+  if (![checkinPreferences respondsToSelector:hasValidCheckinInfoSelector]) {
+    // Can't check hasValidCheckinInfo
+    return NO;
+  }
+
+  // Since hasValidCheckinInfo returns a BOOL, use NSInvocation
+  NSMethodSignature *methodSignature =
+      [[checkinPreferences class] instanceMethodSignatureForSelector:hasValidCheckinInfoSelector];
+  NSInvocation *invocation = [NSInvocation invocationWithMethodSignature:methodSignature];
+  invocation.selector = hasValidCheckinInfoSelector;
+  invocation.target = checkinPreferences;
+  [invocation invoke];
+  BOOL returnValue;
+  [invocation getReturnValue:&returnValue];
+  return returnValue;
+}
+
+// Returns a non-scalar (id) object based on the property name
+id FIRMessagingCheckinService_propertyNamed(id checkinPreferences, NSString *propertyName) {
+  SEL propertyGetterSelector = NSSelectorFromString(propertyName);
+  if ([checkinPreferences respondsToSelector:propertyGetterSelector]) {
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
+    return [checkinPreferences performSelector:propertyGetterSelector];
+#pragma clang diagnostic pop
+  }
+  return nil;
+}
+
+#pragma mark - Methods
+
+- (BOOL)tryToLoadPrefetchedCheckinPreferences {
+  Class instanceIDClass = NSClassFromString(@"FIRInstanceID");
+  if (!instanceIDClass) {
+    // InstanceID is not linked
+    return NO;
+  }
+
+  // [FIRInstanceID instanceID]
+  SEL instanceIDSelector = NSSelectorFromString(@"instanceID");
+  if (![instanceIDClass respondsToSelector:instanceIDSelector]) {
+    return NO;
+  }
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
+  id instanceID = [instanceIDClass performSelector:instanceIDSelector];
+#pragma clang diagnostic pop
+  if (!instanceID) {
+    // Instance ID singleton not available
+    return NO;
+  }
+
+  // [[FIRInstanceID instanceID] cachedCheckinPreferences]
+  SEL cachedCheckinPrefsSelector = NSSelectorFromString(@"cachedCheckinPreferences");
+  if (![instanceID respondsToSelector:cachedCheckinPrefsSelector]) {
+    // cachedCheckinPreferences is not accessible
+    return NO;
+  }
+#pragma clang diagnostic push
+#pragma clang diagnostic ignored "-Warc-performSelector-leaks"
+  id checkinPreferences = [instanceID performSelector:cachedCheckinPrefsSelector];
+#pragma clang diagnostic pop
+  if (!checkinPreferences) {
+    // No cached checkin prefs
+    return NO;
+  }
+
+  BOOL hasValidInfo = FIRMessagingCheckinService_hasValidCheckinInfo(checkinPreferences);
+  if (hasValidInfo) {
+    self.checkinPreferences = checkinPreferences;
+  }
+  return hasValidInfo;
+}
+
+#pragma mark - API
+
+- (NSString *)deviceAuthID {
+  return FIRMessagingCheckinService_propertyNamed(self.checkinPreferences, @"deviceID");
+}
+
+- (NSString *)secretToken {
+  return FIRMessagingCheckinService_propertyNamed(self.checkinPreferences, @"secretToken");
+}
+
+- (NSString *)versionInfo {
+  return FIRMessagingCheckinService_propertyNamed(self.checkinPreferences, @"versionInfo");
+}
+
+- (NSString *)digest {
+  return FIRMessagingCheckinService_propertyNamed(self.checkinPreferences, @"digest");
+}
+
+- (BOOL)hasValidCheckinInfo {
+  return FIRMessagingCheckinService_hasValidCheckinInfo(self.checkinPreferences);
+}
+
+@end