XuqmGroup-RNSDK/packages/push/src/PushSDK.ts

import { apiRequest, getConfig, getDeviceInfo, getUserId as getCommonUserId } from '@xuqm/rn-common'
import type { PushVendor } from '@xuqm/rn-common'

export type { PushVendor }

type PendingDeviceToken = {
  token: string
  vendor?: PushVendor
}

let currentUserId: string | null = null
let pendingToken: PendingDeviceToken | null = null

async function registerPendingToken(): Promise<void> {
  const userId = currentUserId ?? getCommonUserId()
  if (!userId || !pendingToken) return
  const device = await getDeviceInfo()
  const config = getConfig()
  await apiRequest('/api/push/register', {
    method: 'POST',
    params: {
      appId: config.appId,
      userId,
      vendor: pendingToken.vendor ?? device.pushVendor,
      token: pendingToken.token,
      platform: device.platform,
      deviceId: device.deviceId,
      brand: device.brand,
      model: device.model,
      osVersion: device.osVersion,
    },
  })
}

export const PushSDK = {
  async initialize(userId?: string): Promise<void> {
    currentUserId = userId ?? getCommonUserId()
    await registerPendingToken()
  },

  async setDeviceToken(token: string, vendor?: PushVendor): Promise<void> {
    pendingToken = { token, vendor }
    await registerPendingToken()
  },

  /**
   * Register a push device token for the given user.
   * If vendor is omitted, it is auto-detected from the device brand.
   *
   * @param userId  - The logged-in user's ID
   * @param token   - The device push token from the vendor SDK
   * @param vendor  - Optional; auto-detected when not provided
   */
  async registerToken(userId: string, token: string, vendor?: PushVendor): Promise<void> {
    currentUserId = userId
    pendingToken = { token, vendor }
    const config = getConfig()
    const device = await getDeviceInfo()
    await apiRequest('/api/push/register', {
      method: 'POST',
      params: {
        appId:     config.appId,
        userId,
        vendor:    vendor ?? device.pushVendor,
        token,
        platform:  device.platform,
        deviceId:  device.deviceId,
        brand:     device.brand,
        model:     device.model,
        osVersion: device.osVersion,
      },
    })
  },

  async unregisterToken(userId: string): Promise<void> {
    const config = getConfig()
    const { deviceId } = await getDeviceInfo()
    await apiRequest('/api/push/unregister', {
      method: 'DELETE',
      params: { appId: config.appId, userId, deviceId },
    })
    if (currentUserId === userId) currentUserId = null
    if (pendingToken && currentUserId === null) pendingToken = null
  },

  async logout(userId?: string): Promise<void> {
    const targetUserId = userId ?? currentUserId ?? getCommonUserId()
    if (!targetUserId) return
    await PushSDK.unregisterToken(targetUserId)
  },
}