@sendbird/uikit-react-native/src/hooks/usePushTokenRegistration.ts

Sendbird UIKit for React Native: A feature-rich and customizable chat UI kit with messaging, channel management, and user authentication.

import { useRef } from 'react';
import { Platform } from 'react-native';

import { Logger, useFreshCallback, useIIFE } from '@sendbird/uikit-utils';

import { usePlatformService, useSendbirdChat } from './useContext';

const usePushTokenRegistration = () => {
  const { sdk } = useSendbirdChat();
  const { notificationService } = usePlatformService();

  const refreshListener = useRef<() => void>();
  const [registerToken, unregisterToken, getToken] = useIIFE(() => {
    return [
      Platform.select({
        ios: (token: string) => sdk.registerAPNSPushTokenForCurrentUser(token),
        default: (token: string) => sdk.registerFCMPushTokenForCurrentUser(token),
      }),
      Platform.select({
        ios: (token: string) => sdk.unregisterAPNSPushTokenForCurrentUser(token),
        default: (token: string) => sdk.unregisterFCMPushTokenForCurrentUser(token),
      }),
      Platform.select({
        ios: () => notificationService.getAPNSToken(),
        default: () => notificationService.getFCMToken(),
      }),
    ];
  });

  const registerPushTokenForCurrentUser = useFreshCallback(async () => {
    // Check and request push permission
    const hasPermission = await notificationService.hasPushPermission();
    if (!hasPermission) {
      const pushPermission = await notificationService.requestPushPermission();
      if (!pushPermission) {
        Logger.warn('[usePushTokenRegistration]', 'Not granted push permission');
        return;
      }
    }

    // Register token refresh listener
    refreshListener.current = notificationService.onTokenRefresh(registerToken);

    // Register token
    const token = await getToken();
    if (token) {
      try {
        await registerToken(token);
        Logger.log('[usePushTokenRegistration]', 'registered token:', token);
      } catch (error) {
        Logger.error('[usePushTokenRegistration]', 'failed to register token:', error);
      }
    }

    // Remove listener
    refreshListener.current = notificationService.onTokenRefresh(async (token) => {
      try {
        await registerToken(token);
        Logger.log('[usePushTokenRegistration]', 'registered token:', token);
      } catch (error) {
        Logger.error('[usePushTokenRegistration]', 'failed to register token:', error);
      }
    });
  });

  const unregisterPushTokenForCurrentUser = useFreshCallback(async () => {
    // Unregister token refresh listener
    refreshListener.current?.();

    // Unregister token
    const token = await getToken();
    if (token) {
      try {
        await unregisterToken(token);
        Logger.log('[usePushTokenRegistration]', 'unregistered token:', token);
      } catch (error) {
        Logger.error('[usePushTokenRegistration]', 'failed to unregister token:', error);
      }
    }
  });

  return { registerPushTokenForCurrentUser, unregisterPushTokenForCurrentUser };
};

export default usePushTokenRegistration;