mirror of https://github.com/langgenius/dify.git
102 lines
3.2 KiB
TypeScript
102 lines
3.2 KiB
TypeScript
import {
|
|
useCallback,
|
|
useEffect,
|
|
useMemo,
|
|
useRef,
|
|
} from 'react'
|
|
import {
|
|
useMarketplaceCollectionsAndPlugins,
|
|
useMarketplacePlugins,
|
|
} from '@/app/components/plugins/marketplace/hooks'
|
|
import { SCROLL_BOTTOM_THRESHOLD } from '@/app/components/plugins/marketplace/constants'
|
|
import { PluginCategoryEnum } from '@/app/components/plugins/types'
|
|
import { getMarketplaceListCondition } from '@/app/components/plugins/marketplace/utils'
|
|
import { useAllToolProviders } from '@/service/use-tools'
|
|
|
|
export const useMarketplace = (searchPluginText: string, filterPluginTags: string[]) => {
|
|
const { data: toolProvidersData, isSuccess } = useAllToolProviders()
|
|
const exclude = useMemo(() => {
|
|
if (isSuccess)
|
|
return toolProvidersData?.filter(toolProvider => !!toolProvider.plugin_id).map(toolProvider => toolProvider.plugin_id!)
|
|
}, [isSuccess, toolProvidersData])
|
|
const {
|
|
isLoading,
|
|
marketplaceCollections,
|
|
marketplaceCollectionPluginsMap,
|
|
queryMarketplaceCollectionsAndPlugins,
|
|
} = useMarketplaceCollectionsAndPlugins()
|
|
const {
|
|
plugins,
|
|
resetPlugins,
|
|
queryPlugins,
|
|
queryPluginsWithDebounced,
|
|
isLoading: isPluginsLoading,
|
|
fetchNextPage,
|
|
hasNextPage,
|
|
page: pluginsPage,
|
|
} = useMarketplacePlugins()
|
|
const searchPluginTextRef = useRef(searchPluginText)
|
|
const filterPluginTagsRef = useRef(filterPluginTags)
|
|
|
|
useEffect(() => {
|
|
searchPluginTextRef.current = searchPluginText
|
|
filterPluginTagsRef.current = filterPluginTags
|
|
}, [searchPluginText, filterPluginTags])
|
|
useEffect(() => {
|
|
if ((searchPluginText || filterPluginTags.length) && isSuccess) {
|
|
if (searchPluginText) {
|
|
queryPluginsWithDebounced({
|
|
category: PluginCategoryEnum.tool,
|
|
query: searchPluginText,
|
|
tags: filterPluginTags,
|
|
exclude,
|
|
type: 'plugin',
|
|
})
|
|
return
|
|
}
|
|
queryPlugins({
|
|
category: PluginCategoryEnum.tool,
|
|
query: searchPluginText,
|
|
tags: filterPluginTags,
|
|
exclude,
|
|
type: 'plugin',
|
|
})
|
|
}
|
|
else {
|
|
if (isSuccess) {
|
|
queryMarketplaceCollectionsAndPlugins({
|
|
category: PluginCategoryEnum.tool,
|
|
condition: getMarketplaceListCondition(PluginCategoryEnum.tool),
|
|
exclude,
|
|
type: 'plugin',
|
|
})
|
|
resetPlugins()
|
|
}
|
|
}
|
|
}, [searchPluginText, filterPluginTags, queryPlugins, queryMarketplaceCollectionsAndPlugins, queryPluginsWithDebounced, resetPlugins, exclude, isSuccess])
|
|
|
|
const handleScroll = useCallback((e: Event) => {
|
|
const target = e.target as HTMLDivElement
|
|
const {
|
|
scrollTop,
|
|
scrollHeight,
|
|
clientHeight,
|
|
} = target
|
|
if (scrollTop + clientHeight >= scrollHeight - SCROLL_BOTTOM_THRESHOLD && scrollTop > 0) {
|
|
const searchPluginText = searchPluginTextRef.current
|
|
const filterPluginTags = filterPluginTagsRef.current
|
|
if (hasNextPage && (!!searchPluginText || !!filterPluginTags.length))
|
|
fetchNextPage()
|
|
}
|
|
}, [exclude, fetchNextPage, hasNextPage, plugins, queryPlugins])
|
|
|
|
return {
|
|
isLoading: isLoading || isPluginsLoading,
|
|
marketplaceCollections,
|
|
marketplaceCollectionPluginsMap,
|
|
plugins,
|
|
handleScroll,
|
|
page: Math.max(pluginsPage || 0, 1),
|
|
}
|
|
}
|