supabase/apps/studio/components/interfaces/ConnectSheet/useMcpUrl.ts

import { IS_PLATFORM } from 'lib/constants'
import { useMemo } from 'react'
import { FEATURE_GROUPS_NON_PLATFORM, FEATURE_GROUPS_PLATFORM } from 'ui-patterns/McpUrlBuilder'

import { StepContentProps } from './Connect.types'

export function useMcpUrl(
  state: StepContentProps['state'],
  projectKeys: StepContentProps['projectKeys']
): string {
  const readonly = Boolean(state.mcpReadonly)
  const baseUrl = IS_PLATFORM ? 'https://mcp.supabase.com' : (projectKeys.apiUrl ?? '')

  return useMemo(() => {
    const params = new URLSearchParams()
    if (readonly) params.set('readonly', 'true')

    const selectedFeatures = Array.isArray(state.mcpFeatures) ? state.mcpFeatures : []
    const supportedFeatures = IS_PLATFORM ? FEATURE_GROUPS_PLATFORM : FEATURE_GROUPS_NON_PLATFORM
    const validFeatures = selectedFeatures.filter((f) =>
      supportedFeatures.some((group) => group.id === f)
    )
    if (validFeatures.length > 0) {
      params.set('features', validFeatures.join(','))
    }

    const queryString = params.toString()
    return queryString ? `${baseUrl}?${queryString}` : baseUrl
  }, [baseUrl, readonly, state.mcpFeatures])
}