dify/web/utils/create-app-tracking.ts

import Cookies from 'js-cookie'
import { flushEvents, trackEvent } from '@/app/components/base/amplitude/utils'
import { AppModeEnum } from '@/types/app'

const CREATE_APP_EXTERNAL_ATTRIBUTION_STORAGE_KEY = 'create_app_external_attribution'

// const EXTERNAL_UTM_SOURCE_MAP = {
//   'blog': 'blog',
//   'dify_blog': 'blog',
//   'linkedin': 'linkedin',
//   'newsletter': 'blog',
//   'twitter': 'twitter/x',
//   'twitter/x': 'twitter/x',
//   'x': 'twitter/x',
// } as const

type SearchParamReader = {
  get: (name: string) => string | null
}

type OriginalCreateAppMode = 'workflow' | 'chatflow' | 'agent'

type CreateAppSource =
  | 'external'
  | 'explore_template_list'
  | 'explore_template_preview'
  | 'studio_blank'
  | 'studio_template_list'
  | 'studio_template_preview'
  | 'studio_upload'

export type TrackCreateAppParams = {
  source: CreateAppSource
  appMode: string
  templateId?: string
}

type ExternalCreateAppAttribution = {
  utmSource: string
  slug?: string
}

const normalizeString = (value?: string | null) => {
  const trimmed = value?.trim()
  return trimmed || undefined
}

const getObjectStringValue = (value: unknown) => {
  return typeof value === 'string' ? normalizeString(value) : undefined
}

const getSearchParamValue = (searchParams?: SearchParamReader | null, key?: string) => {
  if (!searchParams || !key) return undefined
  return normalizeString(searchParams.get(key))
}

const parseJSONRecord = (value?: string | null): Record<string, unknown> | null => {
  if (!value) return null

  try {
    const parsed = JSON.parse(value)
    return parsed && typeof parsed === 'object' ? (parsed as Record<string, unknown>) : null
  } catch {
    return null
  }
}

const getCookieUtmInfo = () => {
  return parseJSONRecord(Cookies.get('utm_info'))
}

const padTimeValue = (value: number) => String(value).padStart(2, '0')

const formatCreateAppTime = (date: Date) => {
  return `${padTimeValue(date.getMonth() + 1)}-${padTimeValue(date.getDate())}-${padTimeValue(date.getHours())}:${padTimeValue(date.getMinutes())}:${padTimeValue(date.getSeconds())}`
}

const mapOriginalCreateAppMode = (appMode: string): OriginalCreateAppMode => {
  if (appMode === AppModeEnum.WORKFLOW) return 'workflow'

  if (appMode === AppModeEnum.AGENT_CHAT || appMode === 'agent') return 'agent'

  return 'chatflow'
}

export const extractExternalCreateAppAttribution = ({
  searchParams,
  utmInfo,
}: {
  searchParams?: SearchParamReader | null
  utmInfo?: Record<string, unknown> | null
}) => {
  const rawSource =
    getSearchParamValue(searchParams, 'utm_source') ?? getObjectStringValue(utmInfo?.utm_source)

  if (!rawSource) return null

  const slug =
    getSearchParamValue(searchParams, 'slug') ??
    getSearchParamValue(searchParams, 'utm_campaign') ??
    getObjectStringValue(utmInfo?.slug) ??
    getObjectStringValue(utmInfo?.utm_campaign)

  return {
    utmSource: rawSource,
    ...(slug ? { slug } : {}),
  } satisfies ExternalCreateAppAttribution
}

const readRememberedExternalCreateAppAttribution = (): ExternalCreateAppAttribution | null => {
  const attribution = parseJSONRecord(
    window.sessionStorage.getItem(CREATE_APP_EXTERNAL_ATTRIBUTION_STORAGE_KEY),
  )
  const rawSource =
    getObjectStringValue(attribution?.utmSource) ?? getObjectStringValue(attribution?.utm_source)

  if (!rawSource) return null

  const slug =
    getObjectStringValue(attribution?.slug) ??
    getObjectStringValue(attribution?.utmCampaign) ??
    getObjectStringValue(attribution?.utm_campaign)

  return {
    utmSource: rawSource,
    ...(slug ? { slug } : {}),
  }
}

const writeRememberedExternalCreateAppAttribution = (attribution: ExternalCreateAppAttribution) => {
  window.sessionStorage.setItem(
    CREATE_APP_EXTERNAL_ATTRIBUTION_STORAGE_KEY,
    JSON.stringify(attribution),
  )
}

const clearRememberedExternalCreateAppAttribution = () => {
  window.sessionStorage.removeItem(CREATE_APP_EXTERNAL_ATTRIBUTION_STORAGE_KEY)
}

export const rememberCreateAppExternalAttribution = ({
  searchParams,
  utmInfo,
}: {
  searchParams?: SearchParamReader | null
  utmInfo?: Record<string, unknown> | null
} = {}) => {
  const attribution = extractExternalCreateAppAttribution({
    searchParams,
    utmInfo: utmInfo ?? getCookieUtmInfo(),
  })

  if (attribution && typeof window !== 'undefined')
    writeRememberedExternalCreateAppAttribution(attribution)

  return attribution
}

const resolveCurrentExternalCreateAppAttribution = () => {
  if (typeof window === 'undefined') return null

  return readRememberedExternalCreateAppAttribution()
}

export const buildCreateAppEventPayload = (
  params: TrackCreateAppParams,
  externalAttribution?: ExternalCreateAppAttribution | null,
  currentTime = new Date(),
) => {
  const source = externalAttribution ? 'external' : params.source

  if (source === 'external' && !externalAttribution) return null

  return {
    source,
    app_mode: mapOriginalCreateAppMode(params.appMode),
    time: formatCreateAppTime(currentTime),
    ...(params.templateId ? { template_id: params.templateId } : {}),
    ...(externalAttribution
      ? {
          utm_source: externalAttribution.utmSource,
          ...(externalAttribution.slug ? { slug: externalAttribution.slug } : {}),
        }
      : {}),
  } satisfies Record<string, string>
}

export const trackCreateApp = (params: TrackCreateAppParams): Promise<void> | undefined => {
  const externalAttribution = resolveCurrentExternalCreateAppAttribution()
  const payload = buildCreateAppEventPayload(params, externalAttribution)

  if (!payload) return

  if (externalAttribution) clearRememberedExternalCreateAppAttribution()

  const trackingResult = trackEvent('create_app', payload)

  if (!trackingResult) return

  const flushResult = flushEvents()

  const completionPromise = flushResult
    ? Promise.all([trackingResult.promise, flushResult.promise]).then(() => undefined)
    : trackingResult.promise.then(() => undefined)

  return completionPromise.catch(() => undefined)
}