pleroma-fe/src/lib/push_notifications_plugin.js

import { useInstanceStore } from 'src/stores/instance.js'
import { useInterfaceStore } from 'src/stores/interface.js'
import { useMergedConfigStore } from 'src/stores/merged_config.js'

export const piniaPushNotificationsPlugin = ({ store }) => {
  if (
    store.$id !== 'sync_config' &&
    store.$id !== 'instance' &&
    store.$id !== 'interface'
  )
    return

  store.$onAction(({ name: actionName, args }) => {
    if (
      store.$id === 'interface' &&
      actionName !== 'setNotificationPermission' &&
      actionName !== 'setLoginStatus'
    )
      return

    // Initial state
    let vapidPublicKey = useInstanceStore().vapidPublicKey
    let enabled = useMergedConfigStore().mergedConfig.webPushNotifications
    let permissionGranted =
      useInterfaceStore().notificationPermission === 'granted'
    let permissionPresent =
      useInterfaceStore().notificationPermission !== undefined
    let user = !!window.vuex.state.users.currentUser

    if (store.$id === 'instance') {
      if (actionName === 'set' && args[0].path === 'vapidPublicKey') {
        const { value } = args[0]
        vapidPublicKey = value
      }
    }

    if (!vapidPublicKey || !permissionPresent) {
      return
    }

    if (store.$id === 'interface') {
      if (actionName === 'setNotificationPermission') {
        permissionGranted = args[0] === 'granted'
      } else if (actionName === 'setLoginStatus') {
        user = args[0]
      } else {
        return
      }
    } else if (store.$id === 'sync_config') {
      if (
        actionName === 'setPreference' &&
        args[0].path === 'simple.webPushNotifications'
      ) {
        const { value } = args[0]
        enabled = value
      } else {
        return
      }
    }

    if (permissionGranted && enabled && user) {
      return window.vuex.dispatch('registerPushNotifications')
    } else {
      return window.vuex.dispatch('unregisterPushNotifications')
    }
  })
}

export const vuexPushNotificationsPlugin = (store) => {
  store.subscribe((mutation, state) => {
    // Initial state
    const vapidPublicKey = useInstanceStore().vapidPublicKey
    const enabled = useMergedConfigStore().mergedConfig.webPushNotifications
    const permissionGranted =
      useInterfaceStore().notificationPermission === 'granted'
    const permissionPresent =
      useInterfaceStore().notificationPermission !== undefined
    const user = state.users.currentUser

    if (!permissionPresent || !vapidPublicKey) return

    if (
      mutation.type === 'setCurrentUser' ||
      mutation.type === 'clearCurrentUser'
    ) {
      if (user && permissionGranted && enabled) {
        return store.dispatch('registerPushNotifications')
      } else {
        return store.dispatch('unregisterPushNotifications')
      }
    }
  })
}