dify/web/app/components/plugins/install-plugin/hooks/use-install-plugin-limit.tsx

import type { GetSystemFeaturesResponse } from '@dify/contracts/api/console/system-features/types.gen'
import type { Plugin, PluginManifestInMarket } from '../../types'
import { useQuery } from '@tanstack/react-query'
import { systemFeaturesQueryOptions } from '@/features/system-features/client'
import { InstallationScope } from '@/features/system-features/constants'

type PluginProps = (Plugin | PluginManifestInMarket) & { from: 'github' | 'marketplace' | 'package' }
type PluginInstallLimitResult = {
  canInstall: boolean
  isLoading: boolean
}

export function pluginInstallLimit(plugin: PluginProps, systemFeatures: GetSystemFeaturesResponse) {
  if (systemFeatures.plugin_installation_permission.restrict_to_marketplace_only) {
    if (plugin.from === 'github' || plugin.from === 'package')
      return { canInstall: false }
  }

  if (systemFeatures.plugin_installation_permission.plugin_installation_scope === InstallationScope.ALL) {
    return {
      canInstall: true,
    }
  }
  if (systemFeatures.plugin_installation_permission.plugin_installation_scope === InstallationScope.NONE) {
    return {
      canInstall: false,
    }
  }
  const verification = plugin.verification || {}
  if (!plugin.verification || !plugin.verification.authorized_category)
    verification.authorized_category = 'langgenius'

  if (systemFeatures.plugin_installation_permission.plugin_installation_scope === InstallationScope.OFFICIAL_ONLY) {
    return {
      canInstall: verification.authorized_category === 'langgenius',
    }
  }
  if (systemFeatures.plugin_installation_permission.plugin_installation_scope === InstallationScope.OFFICIAL_AND_PARTNER) {
    return {
      canInstall: verification.authorized_category === 'langgenius' || verification.authorized_category === 'partner',
    }
  }
  return {
    canInstall: true,
  }
}

export default function usePluginInstallLimit(plugin: PluginProps): PluginInstallLimitResult {
  const {
    data: systemFeatures,
    isPending,
  } = useQuery(systemFeaturesQueryOptions())
  if (!systemFeatures) {
    return {
      canInstall: false,
      isLoading: isPending,
    }
  }

  return {
    ...pluginInstallLimit(plugin, systemFeatures),
    isLoading: false,
  }
}