auto-inject.test.ts

import { describe, expect, it } from 'vitest'
import { getAllProxyConfigs } from '../../src/first-party/proxy-configs'
import { applyAutoInject } from '../../src/first-party/setup'
import { normalizeRegistryConfig } from '../../src/normalize'

function makeRuntimeConfig(scripts: Record<string, any> = {}) {
  return { public: { scripts } }
}

/**
 * Apply auto-inject for all proxy configs that have autoInject defined,
 * mimicking what finalizeFirstParty does (normalize → inject).
 */
function autoInjectAll(registry: any, rt: any, proxyPrefix: string) {
  normalizeRegistryConfig(registry)
  const configs = getAllProxyConfigs(proxyPrefix)
  for (const [key, config] of Object.entries(configs)) {
    if (config.autoInject && registry[key] !== undefined) {
      applyAutoInject(registry, rt, proxyPrefix, key, config.autoInject)
    }
  }
}

describe('autoInject via proxy configs', () => {
  describe('object entries', () => {
    it('injects apiHost for posthog config object', () => {
      const registry: any = { posthog: { apiKey: 'phc_test' } }
      const rt = makeRuntimeConfig({ posthog: { apiKey: 'phc_test' } })

      autoInjectAll(registry, rt, '/_proxy')

      // After normalization, object entries become [input]
      expect(registry.posthog[0].apiHost).toBe('/_proxy/ph')
      expect(rt.public.scripts.posthog.apiHost).toBe('/_proxy/ph')
    })

    it('injects endpoint for plausible config object', () => {
      const registry: any = { plausibleAnalytics: { domain: 'example.com' } }
      const rt = makeRuntimeConfig({ plausibleAnalytics: { domain: 'example.com' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(registry.plausibleAnalytics[0].endpoint).toBe('/_proxy/plausible/api/event')
      expect(rt.public.scripts.plausibleAnalytics.endpoint).toBe('/_proxy/plausible/api/event')
    })

    it('does not override existing config field', () => {
      const registry: any = { posthog: { apiKey: 'phc_test', apiHost: 'https://custom.host' } }
      const rt = makeRuntimeConfig({ posthog: { apiKey: 'phc_test', apiHost: 'https://custom.host' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(registry.posthog[0].apiHost).toBe('https://custom.host')
      expect(rt.public.scripts.posthog.apiHost).toBe('https://custom.host')
    })

    it('uses EU prefix when posthog region is eu', () => {
      const registry: any = { posthog: { apiKey: 'phc_test', region: 'eu' } }
      const rt = makeRuntimeConfig({ posthog: { apiKey: 'phc_test', region: 'eu' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(registry.posthog[0].apiHost).toBe('/_proxy/ph-eu')
    })
  })

  describe('array entries', () => {
    it('injects into first element of array entry', () => {
      const registry: any = { posthog: [{ apiKey: 'phc_test' }, { partytown: true }] }
      const rt = makeRuntimeConfig({ posthog: { apiKey: 'phc_test' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(registry.posthog[0].apiHost).toBe('/_proxy/ph')
      expect(rt.public.scripts.posthog.apiHost).toBe('/_proxy/ph')
    })

    it('skips empty array entries', () => {
      const registry: any = { posthog: [] }
      const rt = makeRuntimeConfig({ posthog: { apiKey: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.posthog.apiHost).toBeUndefined()
    })
  })

  describe('empty object entries (env var driven)', () => {
    it('injects into runtimeConfig for posthog: {}', () => {
      const registry: any = { posthog: {} }
      const rt = makeRuntimeConfig({ posthog: { apiKey: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      // After normalization, {} becomes [{}] — both input and runtimeConfig get the value
      expect(registry.posthog[0].apiHost).toBe('/_proxy/ph')
      expect(rt.public.scripts.posthog.apiHost).toBe('/_proxy/ph')
    })

    it('uses EU prefix for posthog: {} when runtime region is eu', () => {
      const registry: any = { posthog: {} }
      const rt = makeRuntimeConfig({ posthog: { apiKey: '', region: 'eu' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.posthog.apiHost).toBe('/_proxy/ph-eu')
    })

    it('injects into runtimeConfig for plausibleAnalytics: {}', () => {
      const registry: any = { plausibleAnalytics: {} }
      const rt = makeRuntimeConfig({ plausibleAnalytics: { domain: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.plausibleAnalytics.endpoint).toBe('/_proxy/plausible/api/event')
    })

    it('injects into runtimeConfig for umamiAnalytics: {}', () => {
      const registry: any = { umamiAnalytics: {} }
      const rt = makeRuntimeConfig({ umamiAnalytics: { websiteId: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.umamiAnalytics.hostUrl).toBe('/_proxy/umami')
    })

    it('injects into runtimeConfig for rybbitAnalytics: {}', () => {
      const registry: any = { rybbitAnalytics: {} }
      const rt = makeRuntimeConfig({ rybbitAnalytics: { siteId: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.rybbitAnalytics.analyticsHost).toBe('/_proxy/rybbit/api')
    })

    it('injects into runtimeConfig for databuddyAnalytics: {}', () => {
      const registry: any = { databuddyAnalytics: {} }
      const rt = makeRuntimeConfig({ databuddyAnalytics: { clientId: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.databuddyAnalytics.apiUrl).toBe('/_proxy/databuddy-api')
    })
  })

  describe('mock entries', () => {
    it('injects into runtimeConfig for posthog: mock', () => {
      const registry: any = { posthog: 'mock' }
      const rt = makeRuntimeConfig({ posthog: { apiKey: '' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.posthog.apiHost).toBe('/_proxy/ph')
    })
  })

  describe('skipped entries', () => {
    it('skips scripts without autoInject config', () => {
      const registry: any = { googleAnalytics: { id: 'G-TEST' } }
      const rt = makeRuntimeConfig({ googleAnalytics: { id: 'G-TEST' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.googleAnalytics).toEqual({ id: 'G-TEST' })
    })

    it('skips falsy entries', () => {
      const registry: any = { posthog: false }
      const rt = makeRuntimeConfig()

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.posthog).toBeUndefined()
    })

    it('does not modify existing runtimeConfig for falsy entry', () => {
      const registry: any = { posthog: false }
      const rt = makeRuntimeConfig({ posthog: { existing: 'value' } })

      autoInjectAll(registry, rt, '/_proxy')

      expect(rt.public.scripts.posthog).toEqual({ existing: 'value' })
    })
  })

  describe('custom proxyPrefix', () => {
    it('uses custom prefix in computed values', () => {
      const registry: any = { posthog: {} }
      const rt = makeRuntimeConfig({ posthog: { apiKey: '' } })

      autoInjectAll(registry, rt, '/_analytics')

      expect(rt.public.scripts.posthog.apiHost).toBe('/_analytics/ph')
    })
  })
})