supabase/apps/studio/components/interfaces/Organization/PrivateApps/PrivateAppsContext.tsx

import type { components } from 'api-types'
import { createContext, PropsWithChildren, useContext, useMemo, useState } from 'react'

import type { Installation, PrivateApp } from './PrivateApps.types'
import { usePlatformAppInstallationsQuery } from '@/data/platform-apps/platform-app-installations-query'
import { usePlatformAppsQuery } from '@/data/platform-apps/platform-apps-query'
import { useSelectedOrganizationQuery } from '@/hooks/misc/useSelectedOrganization'

export type { Installation, PrivateApp } from './PrivateApps.types'

interface PrivateAppsContextValue {
  slug: string | undefined
  apps: PrivateApp[]
  isLoading: boolean
  isLoadingInstallations: boolean
  installations: Installation[]
  addInstallation: (
    data: components['schemas']['InstallPlatformAppResponse_Output'],
    projectScope: 'all' | string[]
  ) => void
  removeInstallation: (id: string) => void
  removeInstallationsByAppId: (appId: string) => void
}

const PrivateAppsContext = createContext<PrivateAppsContextValue | null>(null)

export function PrivateAppsProvider({ children }: PropsWithChildren) {
  const { data: org } = useSelectedOrganizationQuery()
  const slug = org?.slug

  const { data: appsData, isLoading } = usePlatformAppsQuery({ slug })
  const {
    data: installationsData,
    isLoading: isLoadingInstallations,
    isError: installationsError,
  } = usePlatformAppInstallationsQuery({ slug }, { retry: false })

  // Local state fallback when GET installations endpoint is unavailable
  const [localInstallations, setLocalInstallations] = useState<Installation[]>([])

  const installations = useMemo<Installation[]>(() => {
    if (!installationsError && installationsData?.installations) {
      return installationsData.installations.map((inst) => ({
        ...(inst as components['schemas']['InstallPlatformAppResponse_Output']),
        projectScope: 'all' as const,
      }))
    }
    return localInstallations
  }, [installationsData, installationsError, localInstallations])

  function addInstallation(
    data: components['schemas']['InstallPlatformAppResponse_Output'],
    projectScope: 'all' | string[]
  ) {
    setLocalInstallations((prev) => [...prev, { ...data, projectScope }])
  }

  function removeInstallation(id: string) {
    setLocalInstallations((prev) => prev.filter((i) => i.id !== id))
  }

  function removeInstallationsByAppId(appId: string) {
    setLocalInstallations((prev) => prev.filter((i) => i.app_id !== appId))
  }

  return (
    <PrivateAppsContext.Provider
      value={{
        slug,
        apps: appsData?.apps ?? [],
        isLoading: isLoading || !slug,
        isLoadingInstallations: isLoadingInstallations || !slug,
        installations,
        addInstallation,
        removeInstallation,
        removeInstallationsByAppId,
      }}
    >
      {children}
    </PrivateAppsContext.Provider>
  )
}

export function usePrivateApps() {
  const ctx = useContext(PrivateAppsContext)
  if (!ctx) throw new Error('usePrivateApps must be used within PrivateAppsProvider')
  return ctx
}