dify/web/app/components/plugins/install-plugin/hooks/use-plugin-install-permission.ts

import type { StoreApi } from 'zustand'
import { createContext, use } from 'react'
import { useStore } from 'zustand'
import { createStore } from 'zustand/vanilla'

export type PluginInstallPermissionState = {
  canInstallPlugin: boolean
  canUpdatePlugin: boolean
  currentDifyVersion?: string
}

type PluginInstallPermissionAction = {
  setPluginInstallPermission: (state: PluginInstallPermissionState) => void
}

export type PluginInstallPermissionStoreState = PluginInstallPermissionState & PluginInstallPermissionAction

export type PluginInstallPermissionStore = StoreApi<PluginInstallPermissionStoreState>

const defaultPluginInstallPermissionState: PluginInstallPermissionState = {
  canInstallPlugin: true,
  canUpdatePlugin: true,
}

export const createPluginInstallPermissionStore = (initProps?: Partial<PluginInstallPermissionState>) => {
  const canInstallPlugin = initProps?.canInstallPlugin ?? defaultPluginInstallPermissionState.canInstallPlugin

  return createStore<PluginInstallPermissionStoreState>()(set => ({
    ...defaultPluginInstallPermissionState,
    ...initProps,
    canUpdatePlugin: initProps?.canUpdatePlugin ?? canInstallPlugin,
    setPluginInstallPermission: state => set(() => state),
  }))
}

const fallbackPluginInstallPermissionStore = createPluginInstallPermissionStore()

export const PluginInstallPermissionContext = createContext<PluginInstallPermissionStore | null>(null)

export const usePluginInstallPermissionStore = <T>(selector: (state: PluginInstallPermissionStoreState) => T): T => {
  const store = use(PluginInstallPermissionContext)
  if (!store)
    throw new Error('Missing PluginInstallPermissionProvider in the tree')

  return useStore(store, selector)
}

const usePluginInstallPermission = () => {
  const canInstallPlugin = usePluginInstallPermissionStore(state => state.canInstallPlugin)
  const canUpdatePlugin = usePluginInstallPermissionStore(state => state.canUpdatePlugin)
  const currentDifyVersion = usePluginInstallPermissionStore(state => state.currentDifyVersion)

  return {
    canInstallPlugin,
    canUpdatePlugin,
    currentDifyVersion,
  }
}

export const useOptionalPluginInstallPermission = () => {
  const store = use(PluginInstallPermissionContext) ?? fallbackPluginInstallPermissionStore
  const canInstallPlugin = useStore(store, state => state.canInstallPlugin)
  const canUpdatePlugin = useStore(store, state => state.canUpdatePlugin)
  const currentDifyVersion = useStore(store, state => state.currentDifyVersion)

  return {
    canInstallPlugin,
    canUpdatePlugin,
    currentDifyVersion,
  }
}

export default usePluginInstallPermission