dify/web/app/components/base/amplitude/utils.ts

import * as amplitude from '@amplitude/analytics-browser'

/**
 * Track custom event
 * @param eventName Event name
 * @param eventProperties Event properties (optional)
 */
export const trackEvent = (eventName: string, eventProperties?: Record<string, any>) => {
  amplitude.track(eventName, eventProperties)
}

/**
 * Set user ID
 * @param userId User ID
 */
export const setUserId = (userId: string) => {
  if (!userId) {
    console.warn('[Amplitude] ⚠️ Cannot set empty user ID')
    return
  }

  if (process.env.NODE_ENV === 'development')
    console.log('[Amplitude] 👤 Setting User ID:', userId)

  amplitude.setUserId(userId)
}

/**
 * Set user properties
 * @param properties User properties
 */
export const setUserProperties = (properties: Record<string, any>) => {
  // Filter out undefined and null values
  const validProperties = Object.entries(properties).reduce((acc, [key, value]) => {
    if (value !== undefined && value !== null)
      acc[key] = value

    return acc
  }, {} as Record<string, any>)

  if (Object.keys(validProperties).length === 0) {
    if (process.env.NODE_ENV === 'development')
      console.warn('[Amplitude] ⚠️ No valid properties to set')
    return
  }

  if (process.env.NODE_ENV === 'development')
    console.log('[Amplitude] 📊 Setting user properties:', validProperties)

  const identifyEvent = new amplitude.Identify()
  Object.entries(validProperties).forEach(([key, value]) => {
    identifyEvent.set(key, value)
  })

  const result = amplitude.identify(identifyEvent)

  // Log the result in development
  result.promise.then(() => {
    if (process.env.NODE_ENV === 'development')
      console.log('[Amplitude] ✅ User properties set successfully')
  }).catch((err) => {
    console.error('[Amplitude] ❌ Failed to set user properties:', err)
  })
}

/**
 * Reset user (e.g., when user logs out)
 */
export const resetUser = () => {
  amplitude.reset()
}