import { NativeModules, DeviceEventEmitter, Platform } from 'react-native'; import { Base } from './../base'; const FirebaseMessaging = NativeModules.RNFirebaseMessaging; export const EVENT_TYPE = { RefreshToken: 'FCMTokenRefreshed', Notification: 'FCMNotificationReceived', }; export const NOTIFICATION_TYPE = { Remote: 'remote_notification', NotificationResponse: 'notification_response', WillPresent: 'will_present_notification', Local: 'local_notification', }; export const REMOTE_NOTIFICATION_RESULT = { NewData: 'UIBackgroundFetchResultNewData', NoData: 'UIBackgroundFetchResultNoData', ResultFailed: 'UIBackgroundFetchResultFailed', }; export const WILL_PRESENT_RESULT = { All: 'UNNotificationPresentationOptionAll', None: 'UNNotificationPresentationOptionNone', }; type RemoteMessage = { id: string, type: string, ttl?: number, sender: string, collapseKey?: string, data: Object, }; /** * IOS only finish function * @param data */ function finish(data) { if (Platform.OS !== 'ios') { return; } if (!this._finishCalled && this._completionHandlerId) { let result = Object.assign({}, data); this._finishCalled = true; switch (this._notificationType) { case NOTIFICATION_TYPE.Remote: result = result || REMOTE_NOTIFICATION_RESULT.NoData; if (!Object.values(REMOTE_NOTIFICATION_RESULT).includes(result)) { throw new Error('Invalid REMOTE_NOTIFICATION_RESULT value, use messaging().REMOTE_NOTIFICATION_RESULT'); } FirebaseMessaging.finishRemoteNotification(this._completionHandlerId, result); return; case NOTIFICATION_TYPE.NotificationResponse: FirebaseMessaging.finishNotificationResponse(this._completionHandlerId); return; case NOTIFICATION_TYPE.WillPresent: result = result || (this.show_in_foreground ? WILL_PRESENT_RESULT.All : WILL_PRESENT_RESULT.None); if (!Object.values(WILL_PRESENT_RESULT).includes(result)) { throw new Error('Invalid WILL_PRESENT_RESULT value, use messaging().WILL_PRESENT_RESULT'); } FirebaseMessaging.finishWillPresentNotification(this._completionHandlerId, result); return; default: return; } } } /** * @class Messaging */ export default class Messaging extends Base { constructor(firebase, options = {}) { super(firebase, options); this.namespace = 'firebase:messaging'; } get EVENT_TYPE() { return EVENT_TYPE; } get NOTIFICATION_TYPE() { return NOTIFICATION_TYPE; } get REMOTE_NOTIFICATION_RESULT() { return REMOTE_NOTIFICATION_RESULT; } get WILL_PRESENT_RESULT() { return WILL_PRESENT_RESULT; } /** * Returns the notification that triggered application open * @returns {*} */ getInitialNotification() { return FirebaseMessaging.getInitialNotification(); } /** * Returns the fcm token for the current device * @returns {*|Promise.} */ getToken() { return FirebaseMessaging.getToken(); } /** * Create and display a local notification * @param notification * @returns {*} */ createLocalNotification(notification: Object) { const _notification = Object.assign({}, notification); _notification.id = _notification.id || new Date().getTime().toString(); _notification.local_notification = true; return FirebaseMessaging.createLocalNotification(_notification); } /** * * @param notification * @returns {*} */ scheduleLocalNotification(notification: Object) { const _notification = Object.assign({}, notification); if (!notification.id) return Promise.reject(new Error('An id is required to schedule a local notification.')); _notification.local_notification = true; return FirebaseMessaging.scheduleLocalNotification(_notification); } /** * Returns an array of all scheduled notifications * @returns {Promise.} */ getScheduledLocalNotifications() { return FirebaseMessaging.getScheduledLocalNotifications(); } /** * Cancel a local notification by id - using '*' will cancel * all local notifications. * @param id * @returns {*} */ cancelLocalNotification(id: string) { if (!id) return null; if (id === '*') return FirebaseMessaging.cancelAllLocalNotifications(); return FirebaseMessaging.cancelLocalNotification(id); } /** * Remove a delivered notification - using '*' will remove * all delivered notifications. * @param id * @returns {*} */ removeDeliveredNotification(id: string) { if (!id) return null; if (id === '*') return FirebaseMessaging.removeAllDeliveredNotifications(); return FirebaseMessaging.removeDeliveredNotification(id); } /** * Request notification permission * @platforms ios * @returns {*|Promise.<*>} */ requestPermissions() { return FirebaseMessaging.requestPermissions(); } /** * Set notification count badge number * @param n */ setBadgeNumber(n: number) { FirebaseMessaging.setBadgeNumber(n); } /** * set notification count badge number * @returns {Promise.} */ getBadgeNumber() { return FirebaseMessaging.getBadgeNumber(); } /** * Subscribe to messages / notifications * @param listener * @returns {*} */ onMessage(listener: Function) { return DeviceEventEmitter.addListener(EVENT_TYPE.Notification, async(event) => { const data = Object.assign({}, event); data.finish = finish; await listener(data); if (!data._finishCalled) { data.finish(); } }); } /** * Subscribe to token refresh events * @param listener * @returns {*} */ onTokenRefresh(listener: Function) { return DeviceEventEmitter.addListener(EVENT_TYPE.RefreshToken, listener); } /** * Subscribe to a topic * @param topic */ subscribeToTopic(topic: String) { FirebaseMessaging.subscribeToTopic(topic); } /** * Unsubscribe from a topic * @param topic */ unsubscribeFromTopic(topic: String) { FirebaseMessaging.unsubscribeFromTopic(topic); } }