feat: xxo enhancement. (#5671)

This commit is contained in:
Garfield Dai 2024-06-27 17:58:45 +08:00 committed by GitHub
parent 2a13ef9ae0
commit f0ea540b34
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 76 additions and 201 deletions

1
.gitignore vendored
View File

@ -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/*

View File

@ -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 />
}
</>
)}
</>
) )
} }

View File

@ -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>
) )