dify/web/app/reset-password/page.tsx

'use client'
import { Button } from '@langgenius/dify-ui/button'
import { toast } from '@langgenius/dify-ui/toast'
import { RiArrowLeftLine, RiLockPasswordLine } from '@remixicon/react'
import { noop } from 'es-toolkit/function'
import { useState } from 'react'
import { useTranslation } from 'react-i18next'
import Input from '@/app/components/base/input'
import { emailRegex } from '@/config'
import { useLocale } from '@/context/i18n'
import useDocumentTitle from '@/hooks/use-document-title'
import Link from '@/next/link'
import { useRouter, useSearchParams } from '@/next/navigation'
import { sendResetPasswordCode } from '@/service/common'
import { COUNT_DOWN_TIME_MS, useSetCountdownLeftTime } from '../components/signin/storage'

export default function CheckCode() {
  const { t } = useTranslation()
  useDocumentTitle('')
  const searchParams = useSearchParams()
  const router = useRouter()
  const [email, setEmail] = useState('')
  const [loading, setIsLoading] = useState(false)
  const locale = useLocale()
  const setCountdownLeftTime = useSetCountdownLeftTime()

  const handleGetEMailVerificationCode = async () => {
    try {
      if (!email) {
        toast.error(t(($) => $['error.emailEmpty'], { ns: 'login' }))
        return
      }

      if (!emailRegex.test(email)) {
        toast.error(t(($) => $['error.emailInValid'], { ns: 'login' }))
        return
      }
      setIsLoading(true)
      const res = await sendResetPasswordCode(email, locale)
      if (res.result === 'success') {
        setCountdownLeftTime(`${COUNT_DOWN_TIME_MS}`)
        const params = new URLSearchParams(searchParams)
        params.set('token', encodeURIComponent(res.data))
        params.set('email', encodeURIComponent(email))
        router.push(`/reset-password/check-code?${params.toString()}`)
      } else {
        toast.error(res.data)
      }
    } catch (error) {
      console.error(error)
    } finally {
      setIsLoading(false)
    }
  }

  return (
    <div className="flex flex-col gap-3">
      <div className="inline-flex size-14 items-center justify-center rounded-2xl border border-components-panel-border-subtle bg-background-default-dodge shadow-lg">
        <RiLockPasswordLine className="size-6 text-2xl text-text-accent-light-mode-only" />
      </div>
      <div className="pt-2 pb-4">
        <h2 className="title-4xl-semi-bold text-text-primary">
          {t(($) => $.resetPassword, { ns: 'login' })}
        </h2>
        <p className="mt-2 body-md-regular text-text-secondary">
          {t(($) => $.resetPasswordDesc, { ns: 'login' })}
        </p>
      </div>

      <form onSubmit={noop}>
        <input type="text" className="hidden" />
        <div className="mb-2">
          <label htmlFor="email" className="my-2 system-md-semibold text-text-secondary">
            {t(($) => $.email, { ns: 'login' })}
          </label>
          <div className="mt-1">
            <Input
              id="email"
              type="email"
              disabled={loading}
              value={email}
              placeholder={t(($) => $.emailPlaceholder, { ns: 'login' }) as string}
              onChange={(e) => setEmail(e.target.value)}
            />
          </div>
          <div className="mt-3">
            <Button
              loading={loading}
              disabled={loading}
              variant="primary"
              className="w-full"
              onClick={handleGetEMailVerificationCode}
            >
              {t(($) => $.sendVerificationCode, { ns: 'login' })}
            </Button>
          </div>
        </div>
      </form>
      <div className="py-2">
        <div className="h-px bg-linear-to-r from-background-gradient-mask-transparent via-divider-regular to-background-gradient-mask-transparent"></div>
      </div>
      <Link
        href={`/signin?${searchParams.toString()}`}
        className="flex h-9 items-center justify-center text-text-tertiary hover:text-text-primary"
      >
        <div className="inline-block rounded-full bg-background-default-dimmed p-1">
          <RiArrowLeftLine size={12} />
        </div>
        <span className="ml-2 system-xs-regular">{t(($) => $.backToLogin, { ns: 'login' })}</span>
      </Link>
    </div>
  )
}