AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 22:16:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

authorizing

Browse Source
This commit is contained in:
jZonG 2025-05-23 15:59:32 +08:00
parent 62b4be9bb1
commit bbd0dbf29b
2 changed files with 35 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
web/app/components/tools/mcp/detail/content.tsx
View File

@ -20,6 +20,7 @@ import OperationDropdown from './operation-dropdown'
import ListLoading from './list-loading'
import ToolItem from './tool-item'
import {
useAuthorizeMCP,
useDeleteMCP,
useInvalidateMCPTools,
useMCPTools,
@ -44,14 +45,15 @@ const MCPDetailContent: FC<Props> = ({
const { data: toolList = [], isPending: isGettingTools } = useMCPTools(detail.is_team_authorization ? detail.id : '')
const invalidateMCPTools = useInvalidateMCPTools()
const { mutateAsync, isPending: isUpdating } = useUpdateMCPTools(detail.id)
const { mutateAsync: updateTools, isPending: isUpdating } = useUpdateMCPTools(detail.id)
const { mutateAsync: authorizeMcp, isPending: isAuthorizing } = useAuthorizeMCP()
const handleUpdateTools = useCallback(async () => {
if (!detail)
return
await mutateAsync()
await updateTools()
invalidateMCPTools(detail.id)
}, [detail, mutateAsync])
}, [detail, updateTools])
const { mutate: updateMCP } = useUpdateMCP({
onSuccess: onUpdate,
@ -75,6 +77,20 @@ const MCPDetailContent: FC<Props> = ({
setFalse: hideDeleting,
}] = useBoolean(false)
const handleAuthorize = useCallback(async () => {
if (!detail)
return
const res = await authorizeMcp({
provider_id: detail.id,
server_url: detail.server_url!,
})
// TODO
if ((res as any)?.result === 'success') {
hideUpdateModal()
onUpdate()
}
}, [detail, updateMCP, hideUpdateModal, onUpdate])
const handleUpdate = useCallback(async (data: any) => {
if (!detail)
return
@ -140,22 +156,20 @@ const MCPDetailContent: FC<Props> = ({
{t('tools.auth.authorized')}
</Button>
)}
{!detail.is_team_authorization && (
{!detail.is_team_authorization && !isAuthorizing && (
<Button
variant='primary'
className='w-full'
// onClick={() => setShowSettingAuth(true)}
onClick={handleAuthorize}
disabled={!isCurrentWorkspaceManager}
>
{t('tools.mcp.authorize')}
</Button>
)}
{/* TODO */}
{deleting && (
{isAuthorizing && (
<Button
variant='primary'
className='w-full'
// onClick={() => setShowSettingAuth(true)}
disabled
>
<RiLoader2Line className={cn('mr-1 h-4 w-4 animate-spin')} />
@ -215,8 +229,8 @@ const MCPDetailContent: FC<Props> = ({
{!detail.is_team_authorization && (
<div className='flex h-full w-full flex-col items-center justify-center'>
{!loading && <div className='system-md-medium mb-1 text-text-secondary'>{t('tools.mcp.authorizingRequired')}</div>}
{loading && <div className='system-md-medium mb-1 text-text-secondary'>{t('tools.mcp.authorizing')}</div>}
{!isAuthorizing && <div className='system-md-medium mb-1 text-text-secondary'>{t('tools.mcp.authorizingRequired')}</div>}
{isAuthorizing && <div className='system-md-medium mb-1 text-text-secondary'>{t('tools.mcp.authorizing')}</div>}
<div className='system-sm-regular text-text-tertiary'>{t('tools.mcp.authorizeTip')}</div>
</div>
)}

11
web/service/use-tools.ts
View File

@ -141,6 +141,17 @@ export const useDeleteMCP = ({
})
}
export const useAuthorizeMCP = () => {
return useMutation({
mutationKey: [NAME_SPACE, 'authorize-mcp'],
mutationFn: (payload: { provider_id: string; server_url: string }) => {
return post('/workspaces/current/tool-provider/mcp/auth', {
body: payload,
})
},
})
}
export const useMCPTools = (providerID: string) => {
return useQuery({
enabled: !!providerID,