posthog.ts

import posthog from 'posthog-js'

import { createPostHogBeforeSend } from '@comfyorg/shared-frontend-utils/piiUtil'

import type { Platform } from '@/composables/useDownloadUrl'

const POSTHOG_KEY =
  import.meta.env.PUBLIC_POSTHOG_KEY ??
  'phc_iKfK86id4xVYws9LybMje0h44eGtfwFgRPIBehmy8rO'
const POSTHOG_API_HOST =
  import.meta.env.PUBLIC_POSTHOG_API_HOST ?? 'https://t.comfy.org'
const POSTHOG_UI_HOST =
  import.meta.env.PUBLIC_POSTHOG_UI_HOST ?? 'https://us.posthog.com'

let initialized = false

export function initPostHog() {
  if (initialized || typeof window === 'undefined' || !POSTHOG_KEY) return
  try {
    posthog.init(POSTHOG_KEY, {
      api_host: POSTHOG_API_HOST,
      ui_host: POSTHOG_UI_HOST,
      capture_pageview: false,
      capture_pageleave: true,
      person_profiles: 'identified_only',
      // cookie_domain omitted — see PostHogTelemetryProvider.ts note + posthog-js#3578
      before_send: createPostHogBeforeSend()
    })
    initialized = true
  } catch (error) {
    console.error('PostHog init failed', error)
  }
}

export function capturePageview() {
  if (!initialized) return
  try {
    posthog.capture('$pageview')
  } catch (error) {
    console.error('PostHog pageview capture failed', error)
  }
}

export function captureDownloadClick(platform: Platform) {
  if (!initialized) return
  try {
    posthog.capture('website:download_button_clicked', { platform })
  } catch (error) {
    console.error('PostHog download click capture failed', error)
  }
}

export type CtaButton =
  | 'launch_cloud'
  | 'download_desktop'
  | 'pricing'
  | 'run_first_workflow'
  | 'products'
  | 'comfy_desktop'
  | 'comfy_cloud'

export type CtaLocation = 'nav' | 'hero' | 'products_section'

export function captureCtaClick(button: CtaButton, location: CtaLocation) {
  if (!initialized) return
  try {
    posthog.capture('website:cta_clicked', { button, location })
  } catch (error) {
    console.error('PostHog CTA click capture failed', error)
  }
}