kanshan/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-02-06 11:01:30 +08:00
Code Issues Actions 10 Packages Projects Releases Wiki Activity

fix(trigger): Prevent marketplace tool downloads from retriggering on tab change

Browse Source
This commit is contained in:
hjlarry 2025-11-12 13:40:56 +08:00
parent 044ee7ef54
commit a25f469bde
1 changed files with 21 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
web/app/components/workflow/block-selector/market-place-plugin/action.tsx
View File

@ -1,6 +1,6 @@
'use client'
import type { FC } from 'react'
import React, { useCallback, useEffect, useRef, useState } from 'react'
import React, { useCallback, useEffect, useMemo, useRef, useState } from 'react'
import { useTheme } from 'next-themes'
import { useTranslation } from 'react-i18next'
import { RiMoreFill } from '@remixicon/react'
@ -15,6 +15,7 @@ import cn from '@/utils/classnames'
import { useDownloadPlugin } from '@/service/use-plugins'
import { downloadFile } from '@/utils/format'
import { getMarketplaceUrl } from '@/utils/var'
import { useQueryClient } from '@tanstack/react-query'
type Props = {
open: boolean
@ -33,6 +34,7 @@ const OperationDropdown: FC<Props> = ({
}) => {
const { t } = useTranslation()
const { theme } = useTheme()
const queryClient = useQueryClient()
const openRef = useRef(open)
const setOpen = useCallback((v: boolean) => {
onOpenChange(v)
@ -44,23 +46,32 @@ const OperationDropdown: FC<Props> = ({
}, [setOpen])
const [needDownload, setNeedDownload] = useState(false)
const { data: blob, isLoading } = useDownloadPlugin({
const downloadInfo = useMemo(() => ({
organization: author,
pluginName: name,
version,
}, needDownload)
}), [author, name, version])
const { data: blob, isLoading } = useDownloadPlugin(downloadInfo, needDownload)
const handleDownload = useCallback(() => {
if (isLoading) return
queryClient.removeQueries({
queryKey: ['plugins', 'downloadPlugin', downloadInfo],
exact: true,
})
setNeedDownload(true)
}, [isLoading])
}, [downloadInfo, isLoading, queryClient])
useEffect(() => {
if (blob) {
const fileName = `${author}-${name}_${version}.zip`
downloadFile({ data: blob, fileName })
setNeedDownload(false)
}
}, [blob])
if (!needDownload || !blob)
return
const fileName = `${author}-${name}_${version}.zip`
downloadFile({ data: blob, fileName })
setNeedDownload(false)
queryClient.removeQueries({
queryKey: ['plugins', 'downloadPlugin', downloadInfo],
exact: true,
})
}, [author, blob, downloadInfo, name, needDownload, queryClient, version])
return (
<PortalToFollowElem
open={open}