WIP: Messaging

auser · auser · commit 710d5ad61ee5 · 2016-09-15T23:16:31.000-07:00
diff --git a/ios/Firestack/Firestack.h b/ios/Firestack/Firestack.h
@@ -15,6 +15,7 @@
 }
 
 + (void) registerForNotification:(NSString *) typeStr andToken:(NSData *)deviceToken;
++ (void) setup:(UIApplication *) application;
 
 @property (nonatomic) BOOL debug;
 @property (atomic) BOOL configured;
diff --git a/ios/Firestack/Firestack.m b/ios/Firestack/Firestack.m
@@ -12,6 +12,7 @@
 @import Firebase;
 
 @implementation Firestack
+typedef void (^UserWithTokenResponse)(NSDictionary *, NSError *);
 
 - (void)dealloc
 {
@@ -25,7 +26,10 @@ - (void)dealloc
     }];
 }
 
-typedef void (^UserWithTokenResponse)(NSDictionary *, NSError *);
++ (void) setup:(UIApplication *) application
+{
+    [FirestackCloudMessaging setup:application];
+}
 
 RCT_EXPORT_MODULE(Firestack);
 
@@ -277,11 +281,6 @@ - (void)dealloc
 
 #pragma mark Messaging
 
-+ (void) registerForNotification:(NSString *) typeStr andToken:(NSData *)deviceToken
-{
-    [FirestackCloudMessaging registerForNotification:typeStr andToken:deviceToken];
-}
-
 #pragma mark Helpers
 
 - (NSDictionary *) getConfig
diff --git a/ios/Firestack/FirestackCloudMessaging.h b/ios/Firestack/FirestackCloudMessaging.h
@@ -6,12 +6,17 @@
 //  Copyright © 2016 Facebook. All rights reserved.
 //
 
+#import "Firebase.h"
+#import "RCTEventEmitter.h"
 #import "RCTBridgeModule.h"
+#import "RCTUtils.h"
 
-@interface FirestackCloudMessaging : NSObject <RCTBridgeModule> {
+@import FirebaseMessaging;
+
+@interface FirestackCloudMessaging : RCTEventEmitter <RCTBridgeModule> {
     
 }
 
-+ (void) registerForNotification:(NSString *) typeStr andToken:(NSData *)deviceToken;
++ (void) setup;
 
 @end
diff --git a/ios/Firestack/FirestackCloudMessaging.m b/ios/Firestack/FirestackCloudMessaging.m
@@ -6,43 +6,243 @@
 //  Copyright © 2016 Facebook. All rights reserved.
 //
 
-#import "FirestackCloudMessaging.h"
+#import <Foundation/Foundation.h>
 #import <NotificationCenter/NotificationCenter.h>
+#import "FirestackCloudMessaging.h"
+#import "FirestackEvents.h"
+#import "RCTConvert.h"
 
 @import FirebaseInstanceID;
 @import FirebaseMessaging;
 
+// https://github.com/facebook/react-native/blob/master/Libraries/PushNotificationIOS/RCTPushNotificationManager.m
+@implementation RCTConvert (UILocalNotification)
+
++ (UILocalNotification *)UILocalNotification:(id)json
+{
+  NSDictionary<NSString *, id> *details = [self NSDictionary:json];
+  UILocalNotification *notification = [UILocalNotification new];
+  notification.fireDate = [RCTConvert NSDate:details[@"fireDate"]] ?: [NSDate date];
+  notification.alertBody = [RCTConvert NSString:details[@"alertBody"]];
+  notification.alertAction = [RCTConvert NSString:details[@"alertAction"]];
+  notification.soundName = [RCTConvert NSString:details[@"soundName"]] ?: UILocalNotificationDefaultSoundName;
+  notification.userInfo = [RCTConvert NSDictionary:details[@"userInfo"]];
+  notification.category = [RCTConvert NSString:details[@"category"]];
+  if (details[@"applicationIconBadgeNumber"]) {
+    notification.applicationIconBadgeNumber = [RCTConvert NSInteger:details[@"applicationIconBadgeNumber"]];
+  }
+  return notification;
+}
+
+@end
+
 @implementation FirestackCloudMessaging
 
-+ (void) registerForNotification:(NSString *) typeStr andToken:(NSData *)deviceToken
+// https://github.com/facebook/react-native/blob/master/Libraries/PushNotificationIOS/RCTPushNotificationManager.m
+static NSDictionary *RCTFormatLocalNotification(UILocalNotification *notification)
+{
+  NSMutableDictionary *formattedLocalNotification = [NSMutableDictionary dictionary];
+  if (notification.fireDate) {
+    NSDateFormatter *formatter = [NSDateFormatter new];
+    [formatter setDateFormat:@"yyyy-MM-dd'T'HH:mm:ss.SSSZZZZZ"];
+    NSString *fireDateString = [formatter stringFromDate:notification.fireDate];
+    formattedLocalNotification[@"fireDate"] = fireDateString;
+  }
+  formattedLocalNotification[@"alertAction"] = RCTNullIfNil(notification.alertAction);
+  formattedLocalNotification[@"alertBody"] = RCTNullIfNil(notification.alertBody);
+  formattedLocalNotification[@"applicationIconBadgeNumber"] = @(notification.applicationIconBadgeNumber);
+  formattedLocalNotification[@"category"] = RCTNullIfNil(notification.category);
+  formattedLocalNotification[@"soundName"] = RCTNullIfNil(notification.soundName);
+  formattedLocalNotification[@"userInfo"] = RCTNullIfNil(RCTJSONClean(notification.userInfo));
+  formattedLocalNotification[@"remote"] = @NO;
+  return formattedLocalNotification;
+}
+
+- (void) dealloc
 {
+  [[NSNotificationCenter defaultCenter] removeObserver: self];
+}
+
++ (void) setup:(UIApplication *) application
+{
+  [[NSNotificationCenter defaultCenter] addObserver:self
+    selector:@selector(connectToFirebase)
+    name: UIApplicationDidEnterBackgroundNotification
+    object: nil];
+
+  [[NSNotificationCenter defaultCenter] addObserver:self
+    selector:@selector(disconnectFromFirebase)
+    name: UIApplicationDidBecomeActiveNotification
+    object: nil];
+
+  [[NSNotificationCenter defaultCenter] addObserver:self
+    selector:@selector(handleRemoteNotificationReceived:)
+    name:MESSAGING_MESSAGE_RECEIVED_REMOTE
+    object: nil];
+
+  [[NSNotificationCenter defaultCenter] addObserver:self
+    selector:@selector(handleLocalNotificationReceived:)
+    name:MESSAGING_MESSAGE_RECEIVED_LOCAL
+    object: nil];
+
+  [[NSNotificationCenter defaultCenter] addObserver:self
+    selector:@selector(handleTokenRefresh)
+    name:kFIRInstanceIDTokenRefreshNotification
+    object: nil];
+
+  if (SYSTEM_VERSION_LESS_THAN_OR_EQUAL_TO(@"9.0")) {
   UIUserNotificationType allNotificationTypes =
   (UIUserNotificationTypeSound | UIUserNotificationTypeAlert | UIUserNotificationTypeBadge);
   UIUserNotificationSettings *settings =
   [UIUserNotificationSettings settingsForTypes:allNotificationTypes categories:nil];
   [[UIApplication sharedApplication] registerUserNotificationSettings:settings];
-  [[UIApplication sharedApplication] registerForRemoteNotifications];
+} else {
+  // iOS 10 or later
+  #if defined(__IPHONE_10_0) && __IPHONE_OS_VERSION_MAX_ALLOWED >= __IPHONE_10_0
+  UNAuthorizationOptions authOptions =
+      UNAuthorizationOptionAlert
+      | UNAuthorizationOptionSound
+      | UNAuthorizationOptionBadge;
+  [[UNUserNotificationCenter currentNotificationCenter]
+      requestAuthorizationWithOptions:authOptions
+      completionHandler:^(BOOL granted, NSError * _Nullable error) {
+      }
+   ];
+
+  // For iOS 10 display notification (sent via APNS)
+  [[UNUserNotificationCenter currentNotificationCenter] setDelegate:self];
+  // For iOS 10 data message (sent via FCM)
+  [[FIRMessaging messaging] setRemoteMessageDelegate:self];
+  #endif
+}
+
+[[UIApplication sharedApplication] registerForRemoteNotifications];
+}
+
+#pragma mark callbacks
+- (void) connectToFirebase
+{
+  [[FIRMessaging messaging] connectWithCompletion:^(NSError *error) {
+    NSDictionary *evt;
+    NSString *evtName;
+    if (error != nil) {
+      NSLog(@"Error connecting: %@", [error debugDescription]);
+      evtName = MESSAGING_SUBSYSTEM_ERROR;
+      evt = @{
+              @"eventName": MESSAGING_SUBSYSTEM_ERROR,
+              @"msg": [error debugDescription]
+              };
+    } else {
+      NSLog(@"Connected to Firebase messaging");
+      evtName = MESSAGING_SUBSYSTEM_EVENT;
+      evt = @{
+            @"result": @"connected"
+            };
+    [self
+     sendJSEvent:evtName
+     props: evt];
     
-    [[FIRMessaging messaging] connectWithCompletion:^(NSError * _Nullable error) {
-        NSLog(@"Connected: %@", [error debugDescription]);
-    }];
+    }
+  }];
 }
 
-RCT_EXPORT_MODULE(FirestackCloudMessaging);
+- (void) disconnectFromFirebase
+{
+  [[FIRMessaging messaging] disconnect];
+  NSLog(@"Disconnect from Firebase");
+  [self
+     sendJSEvent:MESSAGING_SUBSYSTEM_EVENT
+     props: @{
+      @"status": @"disconnected"
+     }];
+}
 
-+ (void) getToken:(NSString *) typeStr andToken:(NSData *)deviceToken
+- (void) handleRemoteNotificationReceived:(NSNotification *) n
 {
+  NSMutableDictionary *props = [[NSMutableDictionary alloc] initWithDictionary: n.userInfo];
+  [self sendJSEvent:MESSAGING_MESSAGE_RECEIVED_REMOTE props: props];
 }
 
+- (void) handleLocalNotificationReceived:(NSNotification *) n
+{
+  NSMutableDictionary *props = [[NSMutableDictionary alloc] initWithDictionary: n.userInfo];
+  [self sendJSEvent:MESSAGING_MESSAGE_RECEIVED_LOCAL props: props];
+}
+
+- (void) handleTokenRefresh
+{
+  NSDictionary *props = @{
+    @"status": @"token_refreshed",
+    @"token": [[FIRInstanceID instanceID] token]
+  };
+  [self sendJSEvent:MESSAGING_TOKEN_REFRESH props: props];
+}
+
+RCT_EXPORT_MODULE(FirestackCloudMessaging);
+
+
 RCT_EXPORT_METHOD(getToken:(RCTResponseSenderBlock)callback)
-{}
+{
+  NSString *token = [[FIRInstanceID instanceID] token];
+  callback(@[[NSNull null], @{
+    @"status": @"success",
+    @"token": token
+  }]);
+}
+
+RCT_EXPORT_METHOD(sendLocal:(UILocalNotification *)notification
+  callback:(RCTResponseSenderBlock) callback)
+{
+  NSLog(@"sendLocal called with notification: %@", notification);
+  [RCTSharedApplication() presentLocalNotificationNow:notification];
+}
+RCT_EXPORT_METHOD(scheduleLocal:(UILocalNotification *)notification
+  callback:(RCTResponseSenderBlock) callback)
+{
+  [RCTSharedApplication() scheduleLocalNotification:notification];
+}
+
+RCT_EXPORT_METHOD(cancelAllLocalNotifications)
+{
+  [RCTSharedApplication() cancelAllLocalNotifications];
+}
+
+RCT_EXPORT_METHOD(cancelLocalNotifications:(NSDictionary<NSString *, id> *)userInfo)
+{
+  for (UILocalNotification *notification in [UIApplication sharedApplication].scheduledLocalNotifications) {
+    __block BOOL matchesAll = YES;
+    NSDictionary<NSString *, id> *notificationInfo = notification.userInfo;
+    // Note: we do this with a loop instead of just `isEqualToDictionary:`
+    // because we only require that all specified userInfo values match the
+    // notificationInfo values - notificationInfo may contain additional values
+    // which we don't care about.
+    [userInfo enumerateKeysAndObjectsUsingBlock:^(NSString *key, id obj, BOOL *stop) {
+      if (![notificationInfo[key] isEqual:obj]) {
+        matchesAll = NO;
+        *stop = YES;
+      }
+    }];
+    if (matchesAll) {
+      [[UIApplication sharedApplication] cancelLocalNotification:notification];
+    }
+  }
+}
+
+RCT_EXPORT_METHOD(sendRemote:(UILocalNotification *)notification
+  callback:(RCTResponseSenderBlock) callback)
+{
+  
+}
+
 
 RCT_EXPORT_METHOD(send:(NSString *) senderId
   messageId:(NSString *) messageId
   messageType:(NSString *) messageType
   msg: (NSString *) msg
   callback:(RCTResponseSenderBlock)callback)
-{}
+{
+
+}
 
 RCT_EXPORT_METHOD(listenForTokenRefresh:(RCTResponseSenderBlock)callback)
 {}
@@ -52,11 +252,41 @@ + (void) getToken:(NSString *) typeStr andToken:(NSData *)deviceToken
 
 RCT_EXPORT_METHOD(subscribeToTopic:(NSString *) topic
   callback:(RCTResponseSenderBlock)callback)
-{}
+{
+  [[FIRMessaging messaging] subscribeToTopic:topic];
+  callback(@[[NSNull null], @{
+    @"result": @"success",
+    @"topic": topic
+  }]);
+}
 
 RCT_EXPORT_METHOD(unsubscribeFromTopic:(NSString *) topic
   callback: (RCTResponseSenderBlock)callback)
-{}
+{
+  [[FIRMessaging messaging] unsubscribeFromTopic:topic];
+  callback(@[[NSNull null], @{
+    @"result": @"success",
+    @"topic": topic
+  }]);
+}
+
+RCT_EXPORT_METHOD(setBadge:(NSInteger) number
+  callback:(RCTResponseSenderBlock) callback)
+{
+  RCTSharedApplication().applicationIconBadgeNumber = number;
+  callback(@[[NSNull null], @{
+      @"result": @"success",
+      @"number": @(number)
+    }]);
+}
+
+RCT_EXPORT_METHOD(getBadge:(RCTResponseSenderBlock) callback)
+{
+  callback(@[[NSNull null], @{
+    @"result": @"success",
+    @"number": @(RCTSharedApplication().applicationIconBadgeNumber)
+  }]);
+}
 
 RCT_EXPORT_METHOD(listenForReceiveNotification:(RCTResponseSenderBlock)callback)
 {}
@@ -70,4 +300,30 @@ + (void) getToken:(NSString *) typeStr andToken:(NSData *)deviceToken
 RCT_EXPORT_METHOD(unlistenForReceiveUpstreamSend:(RCTResponseSenderBlock)callback)
 {}
 
+// Not sure how to get away from this... yet
+- (NSArray<NSString *> *)supportedEvents {
+    return @[
+      MESSAGING_SUBSYSTEM_EVENT, 
+      MESSAGING_SUBSYSTEM_ERROR,
+      MESSAGING_TOKEN_REFRESH,
+      MESSAGING_MESSAGE_RECEIVED_LOCAL,
+      MESSAGING_MESSAGE_RECEIVED_REMOTE];
+}
+
+- (void) sendJSEvent:(NSString *)title
+               props:(NSDictionary *)props
+{
+    @try {
+        [self sendEventWithName:title
+                           body:@{
+                            @"eventName": title,
+                            @"body": props
+                        }];
+    }
+    @catch (NSException *err) {
+        NSLog(@"An error occurred in sendJSEvent: %@", [err debugDescription]);
+        NSLog(@"Tried to send: %@ with %@", title, props);
+    }
+}
+
 @end
diff --git a/ios/Firestack/FirestackEvents.h b/ios/Firestack/FirestackEvents.h
diff --git a/lib/modules/cloudmessaging.js b/lib/modules/cloudmessaging.js

Original file line number	Diff line number	Diff line change
`@@ -15,6 +15,7 @@`
`15`	`15`	`}`
`16`	`16`
`17`	`17`	`+ (void) registerForNotification:(NSString ) typeStr andToken:(NSData )deviceToken;`
	`18`	`++ (void) setup:(UIApplication *) application;`
`18`	`19`
`19`	`20`	`@property (nonatomic) BOOL debug;`
`20`	`21`	`@property (atomic) BOOL configured;`