36 lines
1.6 KiB
TypeScript
36 lines
1.6 KiB
TypeScript
|
import {
|
||
|
|
useCallback,
|
||
|
|
useEffect,
|
||
|
|
useState,
|
||
|
|
} from 'react'
|
||
|
|
import type { Plugin } from '@/app/components/plugins/types'
|
||
|
|
import type { MarketplaceCollection } from '@/app/components/plugins/marketplace/types'
|
||
|
|
|
||
|
|
export const useMarketplace = () => {
|
||
|
|
const [marketplaceCollections, setMarketplaceCollections] = useState<MarketplaceCollection[]>([])
|
||
|
|
const [marketplaceCollectionPluginsMap, setMarketplaceCollectionPluginsMap] = useState<Record<string, Plugin[]>>({})
|
||
|
|
const getMarketplaceCollections = useCallback(async () => {
|
||
|
|
const marketplaceCollectionsData = await globalThis.fetch('https://marketplace.dify.dev/api/v1/collections')
|
||
|
|
const marketplaceCollectionsDataJson = await marketplaceCollectionsData.json()
|
||
|
|
const marketplaceCollections = marketplaceCollectionsDataJson.data.collections
|
||
|
|
const marketplaceCollectionPluginsMap = {} as Record<string, Plugin[]>
|
||
|
|
await Promise.all(marketplaceCollections.map(async (collection: MarketplaceCollection) => {
|
||
|
|
const marketplaceCollectionPluginsData = await globalThis.fetch(`https://marketplace.dify.dev/api/v1/collections/${collection.name}/plugins`)
|
||
|
|
const marketplaceCollectionPluginsDataJson = await marketplaceCollectionPluginsData.json()
|
||
|
|
const plugins = marketplaceCollectionPluginsDataJson.data.plugins
|
||
|
|
|
||
|
|
marketplaceCollectionPluginsMap[collection.name] = plugins
|
||
|
|
}))
|
||
|
|
setMarketplaceCollections(marketplaceCollections)
|
||
|
|
setMarketplaceCollectionPluginsMap(marketplaceCollectionPluginsMap)
|
||
|
|
}, [])
|
||
|
|
useEffect(() => {
|
||
|
|
getMarketplaceCollections()
|
||
|
|
}, [getMarketplaceCollections])
|
||
|
|
|
||
|
|
return {
|
||
|
|
marketplaceCollections,
|
||
|
|
marketplaceCollectionPluginsMap,
|
||
|
|
}
|
||
|
|
}
|