mirror of
https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git
synced 2025-08-12 06:29:03 +08:00
feat: xxo enhancement. (#5671)
This commit is contained in:
parent
2a13ef9ae0
commit
f0ea540b34
1
.gitignore
vendored
1
.gitignore
vendored
@ -139,6 +139,7 @@ web/.vscode/settings.json
|
|||||||
!.idea/icon.png
|
!.idea/icon.png
|
||||||
.ideaDataSources/
|
.ideaDataSources/
|
||||||
*.iml
|
*.iml
|
||||||
|
api/.idea
|
||||||
|
|
||||||
api/.env
|
api/.env
|
||||||
api/storage/*
|
api/storage/*
|
||||||
|
@ -1,85 +1,10 @@
|
|||||||
'use client'
|
'use client'
|
||||||
import React, { useEffect } from 'react'
|
import React from 'react'
|
||||||
import cn from 'classnames'
|
|
||||||
import EmbeddedChatbot from '@/app/components/base/chat/embedded-chatbot'
|
import EmbeddedChatbot from '@/app/components/base/chat/embedded-chatbot'
|
||||||
import Loading from '@/app/components/base/loading'
|
|
||||||
import { fetchSystemFeatures } from '@/service/share'
|
|
||||||
import LogoSite from '@/app/components/base/logo/logo-site'
|
|
||||||
|
|
||||||
const Chatbot = () => {
|
const Chatbot = () => {
|
||||||
const [isSSOEnforced, setIsSSOEnforced] = React.useState(true)
|
|
||||||
const [loading, setLoading] = React.useState(true)
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
fetchSystemFeatures().then((res) => {
|
|
||||||
setIsSSOEnforced(res.sso_enforced_for_web)
|
|
||||||
setLoading(false)
|
|
||||||
})
|
|
||||||
}, [])
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<>
|
<EmbeddedChatbot />
|
||||||
{
|
|
||||||
loading
|
|
||||||
? (
|
|
||||||
<div className="flex items-center justify-center h-full" >
|
|
||||||
<div className={
|
|
||||||
cn(
|
|
||||||
'flex flex-col items-center w-full grow items-center justify-center',
|
|
||||||
'px-6',
|
|
||||||
'md:px-[108px]',
|
|
||||||
)
|
|
||||||
}>
|
|
||||||
<Loading type='area' />
|
|
||||||
</div>
|
|
||||||
</div >
|
|
||||||
)
|
|
||||||
: (
|
|
||||||
<>
|
|
||||||
{isSSOEnforced
|
|
||||||
? (
|
|
||||||
<div className={cn(
|
|
||||||
'flex w-full min-h-screen',
|
|
||||||
'sm:p-4 lg:p-8',
|
|
||||||
'gap-x-20',
|
|
||||||
'justify-center lg:justify-start',
|
|
||||||
)}>
|
|
||||||
<div className={
|
|
||||||
cn(
|
|
||||||
'flex w-full flex-col bg-white shadow rounded-2xl shrink-0',
|
|
||||||
'space-between',
|
|
||||||
)
|
|
||||||
}>
|
|
||||||
<div className='flex items-center justify-between p-6 w-full'>
|
|
||||||
<LogoSite />
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className={
|
|
||||||
cn(
|
|
||||||
'flex flex-col items-center w-full grow items-center justify-center',
|
|
||||||
'px-6',
|
|
||||||
'md:px-[108px]',
|
|
||||||
)
|
|
||||||
}>
|
|
||||||
<div className='flex flex-col md:w-[400px]'>
|
|
||||||
<div className="w-full mx-auto">
|
|
||||||
<h2 className="text-[16px] font-bold text-gray-900">
|
|
||||||
Warning: Chatbot is not available
|
|
||||||
</h2>
|
|
||||||
<p className="text-[16px] text-gray-600 mt-2">
|
|
||||||
Because SSO is enforced. Please contact your administrator.
|
|
||||||
</p>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
: <EmbeddedChatbot />
|
|
||||||
}
|
|
||||||
</>
|
|
||||||
)}
|
|
||||||
</>
|
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -2,150 +2,99 @@
|
|||||||
import cn from 'classnames'
|
import cn from 'classnames'
|
||||||
import { useRouter, useSearchParams } from 'next/navigation'
|
import { useRouter, useSearchParams } from 'next/navigation'
|
||||||
import type { FC } from 'react'
|
import type { FC } from 'react'
|
||||||
import React, { useEffect, useState } from 'react'
|
import React, { useEffect } from 'react'
|
||||||
import { useTranslation } from 'react-i18next'
|
|
||||||
import Toast from '@/app/components/base/toast'
|
import Toast from '@/app/components/base/toast'
|
||||||
import Button from '@/app/components/base/button'
|
|
||||||
import { fetchSystemFeatures, fetchWebOAuth2SSOUrl, fetchWebOIDCSSOUrl, fetchWebSAMLSSOUrl } from '@/service/share'
|
import { fetchSystemFeatures, fetchWebOAuth2SSOUrl, fetchWebOIDCSSOUrl, fetchWebSAMLSSOUrl } from '@/service/share'
|
||||||
import LogoSite from '@/app/components/base/logo/logo-site'
|
|
||||||
import { setAccessToken } from '@/app/components/share/utils'
|
import { setAccessToken } from '@/app/components/share/utils'
|
||||||
|
import Loading from '@/app/components/base/loading'
|
||||||
|
|
||||||
const WebSSOForm: FC = () => {
|
const WebSSOForm: FC = () => {
|
||||||
const searchParams = useSearchParams()
|
const searchParams = useSearchParams()
|
||||||
|
const router = useRouter()
|
||||||
|
|
||||||
const redirectUrl = searchParams.get('redirect_url')
|
const redirectUrl = searchParams.get('redirect_url')
|
||||||
const tokenFromUrl = searchParams.get('web_sso_token')
|
const tokenFromUrl = searchParams.get('web_sso_token')
|
||||||
const message = searchParams.get('message')
|
const message = searchParams.get('message')
|
||||||
|
|
||||||
const router = useRouter()
|
const showErrorToast = (message: string) => {
|
||||||
const { t } = useTranslation()
|
Toast.notify({
|
||||||
|
type: 'error',
|
||||||
|
message,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
const [isLoading, setIsLoading] = useState(false)
|
const getAppCodeFromRedirectUrl = () => {
|
||||||
const [protocol, setProtocol] = useState('')
|
const appCode = redirectUrl?.split('/').pop()
|
||||||
|
if (!appCode)
|
||||||
|
return null
|
||||||
|
|
||||||
useEffect(() => {
|
return appCode
|
||||||
const fetchFeaturesAndSetToken = async () => {
|
}
|
||||||
await fetchSystemFeatures().then((res) => {
|
|
||||||
setProtocol(res.sso_enforced_for_web_protocol)
|
|
||||||
})
|
|
||||||
|
|
||||||
// Callback from SSO, process token and redirect
|
const processTokenAndRedirect = async () => {
|
||||||
if (tokenFromUrl && redirectUrl) {
|
const appCode = getAppCodeFromRedirectUrl()
|
||||||
const appCode = redirectUrl.split('/').pop()
|
if (!appCode || !tokenFromUrl || !redirectUrl) {
|
||||||
if (!appCode) {
|
showErrorToast('redirect url or app code or token is invalid.')
|
||||||
Toast.notify({
|
return
|
||||||
type: 'error',
|
}
|
||||||
message: 'redirect url is invalid. App code is not found.',
|
|
||||||
})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
await setAccessToken(appCode, tokenFromUrl)
|
await setAccessToken(appCode, tokenFromUrl)
|
||||||
router.push(redirectUrl)
|
router.push(redirectUrl)
|
||||||
|
}
|
||||||
|
|
||||||
|
const handleSSOLogin = async (protocol: string) => {
|
||||||
|
const appCode = getAppCodeFromRedirectUrl()
|
||||||
|
if (!appCode || !redirectUrl) {
|
||||||
|
showErrorToast('redirect url or app code is invalid.')
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
switch (protocol) {
|
||||||
|
case 'saml': {
|
||||||
|
const samlRes = await fetchWebSAMLSSOUrl(appCode, redirectUrl)
|
||||||
|
router.push(samlRes.url)
|
||||||
|
break
|
||||||
}
|
}
|
||||||
}
|
case 'oidc': {
|
||||||
|
const oidcRes = await fetchWebOIDCSSOUrl(appCode, redirectUrl)
|
||||||
fetchFeaturesAndSetToken()
|
router.push(oidcRes.url)
|
||||||
|
break
|
||||||
if (message) {
|
}
|
||||||
Toast.notify({
|
case 'oauth2': {
|
||||||
type: 'error',
|
const oauth2Res = await fetchWebOAuth2SSOUrl(appCode, redirectUrl)
|
||||||
message,
|
router.push(oauth2Res.url)
|
||||||
})
|
break
|
||||||
}
|
}
|
||||||
}, [])
|
default:
|
||||||
|
showErrorToast('SSO protocol is not supported.')
|
||||||
const handleSSOLogin = () => {
|
|
||||||
setIsLoading(true)
|
|
||||||
|
|
||||||
if (!redirectUrl) {
|
|
||||||
Toast.notify({
|
|
||||||
type: 'error',
|
|
||||||
message: 'redirect url is not found.',
|
|
||||||
})
|
|
||||||
setIsLoading(false)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
const appCode = redirectUrl.split('/').pop()
|
|
||||||
if (!appCode) {
|
|
||||||
Toast.notify({
|
|
||||||
type: 'error',
|
|
||||||
message: 'redirect url is invalid. App code is not found.',
|
|
||||||
})
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if (protocol === 'saml') {
|
|
||||||
fetchWebSAMLSSOUrl(appCode, redirectUrl).then((res) => {
|
|
||||||
router.push(res.url)
|
|
||||||
}).finally(() => {
|
|
||||||
setIsLoading(false)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
else if (protocol === 'oidc') {
|
|
||||||
fetchWebOIDCSSOUrl(appCode, redirectUrl).then((res) => {
|
|
||||||
router.push(res.url)
|
|
||||||
}).finally(() => {
|
|
||||||
setIsLoading(false)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
else if (protocol === 'oauth2') {
|
|
||||||
fetchWebOAuth2SSOUrl(appCode, redirectUrl).then((res) => {
|
|
||||||
router.push(res.url)
|
|
||||||
}).finally(() => {
|
|
||||||
setIsLoading(false)
|
|
||||||
})
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
Toast.notify({
|
|
||||||
type: 'error',
|
|
||||||
message: 'sso protocol is not supported.',
|
|
||||||
})
|
|
||||||
setIsLoading(false)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
return (
|
useEffect(() => {
|
||||||
<div className={cn(
|
const init = async () => {
|
||||||
'flex w-full min-h-screen',
|
const res = await fetchSystemFeatures()
|
||||||
'sm:p-4 lg:p-8',
|
const protocol = res.sso_enforced_for_web_protocol
|
||||||
'gap-x-20',
|
|
||||||
'justify-center lg:justify-start',
|
|
||||||
)}>
|
|
||||||
<div className={
|
|
||||||
cn(
|
|
||||||
'flex w-full flex-col bg-white shadow rounded-2xl shrink-0',
|
|
||||||
'space-between',
|
|
||||||
)
|
|
||||||
}>
|
|
||||||
<div className='flex items-center justify-between p-6 w-full'>
|
|
||||||
<LogoSite />
|
|
||||||
</div>
|
|
||||||
|
|
||||||
<div className={
|
if (message) {
|
||||||
cn(
|
showErrorToast(message)
|
||||||
'flex flex-col items-center w-full grow items-center justify-center',
|
return
|
||||||
'px-6',
|
}
|
||||||
'md:px-[108px]',
|
|
||||||
)
|
if (!tokenFromUrl) {
|
||||||
}>
|
await handleSSOLogin(protocol)
|
||||||
<div className='flex flex-col md:w-[400px]'>
|
return
|
||||||
<div className="w-full mx-auto">
|
}
|
||||||
<h2 className="text-[32px] font-bold text-gray-900">{t('login.pageTitle')}</h2>
|
|
||||||
</div>
|
await processTokenAndRedirect()
|
||||||
<div className="w-full mx-auto mt-10">
|
}
|
||||||
<Button
|
|
||||||
tabIndex={0}
|
init()
|
||||||
variant='primary'
|
}, [message, tokenFromUrl]) // Added dependencies to useEffect
|
||||||
onClick={() => { handleSSOLogin() }}
|
|
||||||
disabled={isLoading}
|
return (
|
||||||
className="w-full !text-sm"
|
<div className="flex items-center justify-center h-full">
|
||||||
>{t('login.sso')}
|
<div className={cn('flex flex-col items-center w-full grow justify-center', 'px-6', 'md:px-[108px]')}>
|
||||||
</Button>
|
<Loading type='area' />
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
</div>
|
</div>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user