89 lines
2.4 KiB
TypeScript
89 lines
2.4 KiB
TypeScript
'use client'
|
|
import { useRouter, useSearchParams } from 'next/navigation'
|
|
import type { FC } from 'react'
|
|
import { useCallback } from 'react'
|
|
import { useState } from 'react'
|
|
import { useTranslation } from 'react-i18next'
|
|
import { Lock01 } from '@/app/components/base/icons/src/vender/solid/security'
|
|
import Toast from '@/app/components/base/toast'
|
|
import Button from '@/app/components/base/button'
|
|
import { SSOProtocol } from '@/types/feature'
|
|
import { fetchMembersOAuth2SSOUrl, fetchMembersOIDCSSOUrl, fetchMembersSAMLSSOUrl } from '@/service/share'
|
|
|
|
type SSOAuthProps = {
|
|
protocol: SSOProtocol | ''
|
|
}
|
|
|
|
const SSOAuth: FC<SSOAuthProps> = ({
|
|
protocol,
|
|
}) => {
|
|
const router = useRouter()
|
|
const { t } = useTranslation()
|
|
const searchParams = useSearchParams()
|
|
|
|
const redirectUrl = searchParams.get('redirect_url')
|
|
const getAppCodeFromRedirectUrl = useCallback(() => {
|
|
const appCode = redirectUrl?.split('/').pop()
|
|
if (!appCode)
|
|
return null
|
|
|
|
return appCode
|
|
}, [redirectUrl])
|
|
|
|
const [isLoading, setIsLoading] = useState(false)
|
|
|
|
const handleSSOLogin = () => {
|
|
const appCode = getAppCodeFromRedirectUrl()
|
|
if (!redirectUrl || !appCode) {
|
|
Toast.notify({
|
|
type: 'error',
|
|
message: 'invalid redirect URL or app code',
|
|
})
|
|
return
|
|
}
|
|
setIsLoading(true)
|
|
if (protocol === SSOProtocol.SAML) {
|
|
fetchMembersSAMLSSOUrl(appCode, redirectUrl).then((res) => {
|
|
router.push(res.url)
|
|
}).finally(() => {
|
|
setIsLoading(false)
|
|
})
|
|
}
|
|
else if (protocol === SSOProtocol.OIDC) {
|
|
fetchMembersOIDCSSOUrl(appCode, redirectUrl).then((res) => {
|
|
router.push(res.url)
|
|
}).finally(() => {
|
|
setIsLoading(false)
|
|
})
|
|
}
|
|
else if (protocol === SSOProtocol.OAuth2) {
|
|
fetchMembersOAuth2SSOUrl(appCode, redirectUrl).then((res) => {
|
|
router.push(res.url)
|
|
}).finally(() => {
|
|
setIsLoading(false)
|
|
})
|
|
}
|
|
else {
|
|
Toast.notify({
|
|
type: 'error',
|
|
message: 'invalid SSO protocol',
|
|
})
|
|
setIsLoading(false)
|
|
}
|
|
}
|
|
|
|
return (
|
|
<Button
|
|
tabIndex={0}
|
|
onClick={() => { handleSSOLogin() }}
|
|
disabled={isLoading}
|
|
className="w-full"
|
|
>
|
|
<Lock01 className='mr-2 h-5 w-5 text-text-accent-light-mode-only' />
|
|
<span className="truncate">{t('login.withSSO')}</span>
|
|
</Button>
|
|
)
|
|
}
|
|
|
|
export default SSOAuth
|