Briven/apps/studio/components/interfaces/Platform/Webhooks/PlatformWebhooks.store.ts

import { useEffect, useState } from 'react'

import { PLATFORM_WEBHOOKS_MOCK_DATA } from './PlatformWebhooks.mock'
import type {
  PlatformWebhooksState,
  UpsertWebhookEndpointInput,
  WebhookDelivery,
  WebhookEndpoint,
  WebhookScope,
} from './PlatformWebhooks.types'
import { getWebhookEndpointDisplayName } from './PlatformWebhooks.utils'

interface CreateEndpointOptions {
  now?: string
  endpointId?: string
  createdBy?: string
  signingSecret?: string
}

interface UpdateEndpointOptions {
  headerIdFactory?: () => string
}

interface RetryDeliveryOptions {
  now?: string
}

const secureRandomHex = (length: number) => {
  if (length <= 0) return ''

  const cryptoApi = globalThis.crypto
  if (!cryptoApi?.getRandomValues) {
    throw new Error('Web Crypto API is not available')
  }

  const byteCount = Math.ceil(length / 2)
  const bytes = new Uint8Array(byteCount)
  cryptoApi.getRandomValues(bytes)

  return Array.from(bytes, (byte) => byte.toString(16).padStart(2, '0'))
    .join('')
    .slice(0, length)
}

const randomId = (prefix: string) => `${prefix}-${secureRandomHex(8)}`
const randomUuid = () => {
  const cryptoApi = globalThis.crypto
  if (cryptoApi?.randomUUID) return cryptoApi.randomUUID()

  return [
    secureRandomHex(8),
    secureRandomHex(4),
    `4${secureRandomHex(3)}`,
    `${((parseInt(secureRandomHex(2), 16) & 0x3f) | 0x80).toString(16)}${secureRandomHex(2)}`,
    secureRandomHex(12),
  ].join('-')
}

const generateSigningSecret = () => `whsec_${secureRandomHex(16)}`

const deepClone = <T>(value: T): T => JSON.parse(JSON.stringify(value))

const toHeaders = (
  headers: UpsertWebhookEndpointInput['customHeaders'],
  options?: UpdateEndpointOptions
) => {
  const headerIdFactory = options?.headerIdFactory ?? (() => randomId('header'))
  return headers
    .map((header) => ({
      id: headerIdFactory(),
      key: header.key.trim(),
      value: header.value.trim(),
    }))
    .filter((header) => header.key.length > 0 && header.value.length > 0)
}

const normalizeSearch = (value: string) => value.trim().toLowerCase()

export const createInitialPlatformWebhooksState = (scope: WebhookScope): PlatformWebhooksState => {
  const seed = PLATFORM_WEBHOOKS_MOCK_DATA[scope]
  return {
    endpoints: deepClone(seed.endpoints),
    deliveries: deepClone(seed.deliveries),
  }
}

const persistedMockStateByScope: Partial<Record<WebhookScope, PlatformWebhooksState>> = {}

const getPersistedMockState = (scope: WebhookScope) => {
  const persistedState = persistedMockStateByScope[scope]
  if (persistedState) return persistedState

  const initialState = createInitialPlatformWebhooksState(scope)
  persistedMockStateByScope[scope] = initialState
  return initialState
}

// Test-only helper to avoid cross-test state leakage in hook tests.
export const resetPlatformWebhooksMockStateForTests = (scope?: WebhookScope) => {
  if (scope) {
    delete persistedMockStateByScope[scope]
    return
  }

  for (const key of Object.keys(persistedMockStateByScope) as WebhookScope[]) {
    delete persistedMockStateByScope[key]
  }
}

export const createWebhookEndpoint = (
  state: PlatformWebhooksState,
  input: UpsertWebhookEndpointInput,
  options?: CreateEndpointOptions
): { state: PlatformWebhooksState; endpoint: WebhookEndpoint; signingSecret: string } => {
  const endpointId = options?.endpointId ?? randomUuid()
  const signingSecret = options?.signingSecret ?? generateSigningSecret()
  const endpoint: WebhookEndpoint = {
    id: endpointId,
    name: input.name.trim(),
    url: input.url.trim(),
    description: input.description.trim(),
    enabled: input.enabled,
    eventTypes: input.eventTypes.length > 0 ? input.eventTypes : ['*'],
    customHeaders: toHeaders(input.customHeaders),
    createdBy: options?.createdBy ?? 'mock-user@supabase.io',
    createdAt: options?.now ?? new Date().toISOString(),
  }

  return {
    endpoint,
    signingSecret,
    state: {
      ...state,
      endpoints: [endpoint, ...state.endpoints],
    },
  }
}

export const updateWebhookEndpoint = (
  state: PlatformWebhooksState,
  endpointId: string,
  input: UpsertWebhookEndpointInput,
  options?: UpdateEndpointOptions
) => {
  return {
    ...state,
    endpoints: state.endpoints.map((endpoint) =>
      endpoint.id === endpointId
        ? {
            ...endpoint,
            name: input.name.trim(),
            url: input.url.trim(),
            description: input.description.trim(),
            enabled: input.enabled,
            eventTypes: input.eventTypes.length > 0 ? input.eventTypes : ['*'],
            customHeaders: toHeaders(input.customHeaders, options),
          }
        : endpoint
    ),
  }
}

export const deleteWebhookEndpoint = (state: PlatformWebhooksState, endpointId: string) => {
  return {
    endpoints: state.endpoints.filter((endpoint) => endpoint.id !== endpointId),
    deliveries: state.deliveries.filter((delivery) => delivery.endpointId !== endpointId),
  }
}

export const toggleWebhookEndpoint = (
  state: PlatformWebhooksState,
  endpointId: string,
  enabled?: boolean
) => {
  return {
    ...state,
    endpoints: state.endpoints.map((endpoint) =>
      endpoint.id === endpointId
        ? { ...endpoint, enabled: enabled === undefined ? !endpoint.enabled : enabled }
        : endpoint
    ),
  }
}

export const regenerateWebhookEndpointSecret = (
  state: PlatformWebhooksState,
  endpointId: string,
  secret?: string
) => {
  const endpointExists = state.endpoints.some((endpoint) => endpoint.id === endpointId)
  if (!endpointExists) return { state, signingSecret: null }

  return {
    state: { ...state },
    signingSecret: secret ?? generateSigningSecret(),
  }
}

export const retryWebhookDelivery = (
  state: PlatformWebhooksState,
  deliveryId: string,
  options?: RetryDeliveryOptions
) => {
  const delivery = state.deliveries.find((item) => item.id === deliveryId)
  if (!delivery || delivery.status === 'success') return state

  const now = options?.now ?? new Date().toISOString()

  return {
    ...state,
    deliveries: state.deliveries.map<WebhookDelivery>((delivery) => {
      if (delivery.id !== deliveryId) return delivery

      return {
        ...delivery,
        status: 'pending',
        responseCode: undefined,
        attemptAt: now,
      }
    }),
  }
}

export const filterWebhookEndpoints = (endpoints: WebhookEndpoint[], search: string) => {
  const normalizedSearch = normalizeSearch(search)
  if (normalizedSearch.length === 0) return endpoints

  return endpoints.filter((endpoint) => {
    const haystack =
      `${getWebhookEndpointDisplayName(endpoint)} ${endpoint.url} ${endpoint.description}`.toLowerCase()
    return haystack.includes(normalizedSearch)
  })
}

export const filterWebhookDeliveries = (
  deliveries: WebhookDelivery[],
  endpointId: string,
  search: string
) => {
  const normalizedSearch = normalizeSearch(search)

  return deliveries
    .filter((delivery) => delivery.endpointId === endpointId)
    .filter((delivery) => {
      if (normalizedSearch.length === 0) return true
      const haystack =
        `${delivery.eventType} ${delivery.status} ${delivery.responseCode ?? ''}`.toLowerCase()
      return haystack.includes(normalizedSearch)
    })
    .sort((a, b) => new Date(b.attemptAt).getTime() - new Date(a.attemptAt).getTime())
}

export const usePlatformWebhooksMockStore = (scope: WebhookScope) => {
  const [state, setState] = useState<PlatformWebhooksState>(() =>
    deepClone(getPersistedMockState(scope))
  )

  useEffect(() => {
    setState(deepClone(getPersistedMockState(scope)))
  }, [scope])

  const applyStateUpdate = (
    updater: (previous: PlatformWebhooksState) => PlatformWebhooksState
  ) => {
    setState((previous) => {
      const next = updater(previous)
      persistedMockStateByScope[scope] = next
      return next
    })
  }

  return {
    ...state,
    createEndpoint: (input: UpsertWebhookEndpointInput) => {
      const endpointId = randomUuid()
      const now = new Date().toISOString()
      const signingSecret = generateSigningSecret()
      const createdBy = 'mock-user@supabase.io'
      let createdSecret = signingSecret
      applyStateUpdate((prev) => {
        const next = createWebhookEndpoint(prev, input, {
          endpointId,
          now,
          signingSecret,
          createdBy,
        })
        createdSecret = next.signingSecret
        return next.state
      })
      return { endpointId, signingSecret: createdSecret }
    },
    updateEndpoint: (endpointId: string, input: UpsertWebhookEndpointInput) => {
      applyStateUpdate((prev) => updateWebhookEndpoint(prev, endpointId, input))
    },
    deleteEndpoint: (endpointId: string) => {
      applyStateUpdate((prev) => deleteWebhookEndpoint(prev, endpointId))
    },
    toggleEndpoint: (endpointId: string, enabled?: boolean) => {
      applyStateUpdate((prev) => toggleWebhookEndpoint(prev, endpointId, enabled))
    },
    regenerateSecret: (endpointId: string) => {
      const currentState = persistedMockStateByScope[scope] ?? state
      const next = regenerateWebhookEndpointSecret(currentState, endpointId)
      if (!next.signingSecret) return null

      applyStateUpdate(() => next.state)
      return next.signingSecret
    },
    retryDelivery: (deliveryId: string) => {
      applyStateUpdate((prev) => retryWebhookDelivery(prev, deliveryId))
    },
  }
}