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

import type { Plugin, PluginManifestInMarket } from '../../types'
import type { SystemFeatures } from '@/types/feature'
import { useSuspenseQuery } from '@tanstack/react-query'
import { systemFeaturesQueryOptions } from '@/service/system-features'
import { InstallationScope } from '@/types/feature'

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

export function pluginInstallLimit(plugin: PluginProps, systemFeatures: SystemFeatures) {
  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) {
  const { data: systemFeatures } = useSuspenseQuery(systemFeaturesQueryOptions())
  return pluginInstallLimit(plugin, systemFeatures)
}