Apps.tsx 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100
  1. 'use client'
  2. import { useCallback, useEffect, useRef, useState } from 'react'
  3. import { useRouter } from 'next/navigation'
  4. import useSWRInfinite from 'swr/infinite'
  5. import { useTranslation } from 'react-i18next'
  6. import AppCard from './AppCard'
  7. import NewAppCard from './NewAppCard'
  8. import type { AppListResponse } from '@/models/app'
  9. import { fetchAppList } from '@/service/apps'
  10. import { useAppContext } from '@/context/app-context'
  11. import { NEED_REFRESH_APP_LIST_KEY } from '@/config'
  12. import Confirm from '@/app/components/base/confirm/common'
  13. import {
  14. useAnthropicCheckPay,
  15. useSparkCheckQuota,
  16. } from '@/hooks/use-pay'
  17. const getKey = (pageIndex: number, previousPageData: AppListResponse) => {
  18. if (!pageIndex || previousPageData.has_more)
  19. return { url: 'apps', params: { page: pageIndex + 1, limit: 30 } }
  20. return null
  21. }
  22. const Apps = () => {
  23. const { t } = useTranslation()
  24. const { isCurrentWorkspaceManager } = useAppContext()
  25. const { data, isLoading, setSize, mutate } = useSWRInfinite(getKey, fetchAppList, { revalidateFirstPage: false })
  26. const anchorRef = useRef<HTMLDivElement>(null)
  27. const router = useRouter()
  28. const [showPayStatusModal, setShowPayStatusModal] = useState(true)
  29. const anthropicConfirmInfo = useAnthropicCheckPay()
  30. const sparkConfirmInfo = useSparkCheckQuota()
  31. const handleCancelShowPayStatusModal = useCallback(() => {
  32. setShowPayStatusModal(false)
  33. router.replace('/', { forceOptimisticNavigation: false })
  34. }, [router])
  35. useEffect(() => {
  36. document.title = `${t('app.title')} - Dify`
  37. if (localStorage.getItem(NEED_REFRESH_APP_LIST_KEY) === '1') {
  38. localStorage.removeItem(NEED_REFRESH_APP_LIST_KEY)
  39. mutate()
  40. }
  41. }, [mutate, t])
  42. useEffect(() => {
  43. let observer: IntersectionObserver | undefined
  44. if (anchorRef.current) {
  45. observer = new IntersectionObserver((entries) => {
  46. if (entries[0].isIntersecting)
  47. setSize(size => size + 1)
  48. }, { rootMargin: '100px' })
  49. observer.observe(anchorRef.current)
  50. }
  51. return () => observer?.disconnect()
  52. }, [isLoading, setSize, anchorRef, mutate])
  53. return (
  54. <><nav className='grid content-start grid-cols-1 gap-4 px-12 pt-8 sm:grid-cols-2 md:grid-cols-3 lg:grid-cols-4 grow shrink-0'>
  55. { isCurrentWorkspaceManager
  56. && <NewAppCard onSuccess={mutate} />}
  57. {data?.map(({ data: apps }) => apps.map(app => (
  58. <AppCard key={app.id} app={app} onRefresh={mutate} />
  59. )))}
  60. {
  61. showPayStatusModal && anthropicConfirmInfo && (
  62. <Confirm
  63. isShow
  64. onCancel={handleCancelShowPayStatusModal}
  65. onConfirm={handleCancelShowPayStatusModal}
  66. type={anthropicConfirmInfo.type}
  67. title={anthropicConfirmInfo.title}
  68. showOperateCancel={false}
  69. confirmText={(anthropicConfirmInfo.type === 'danger' && t('common.operation.ok')) || ''}
  70. />
  71. )
  72. }
  73. {
  74. showPayStatusModal && sparkConfirmInfo && (
  75. <Confirm
  76. isShow
  77. onCancel={handleCancelShowPayStatusModal}
  78. onConfirm={handleCancelShowPayStatusModal}
  79. type={sparkConfirmInfo.type}
  80. title={sparkConfirmInfo.title}
  81. desc={sparkConfirmInfo.desc}
  82. showOperateCancel={false}
  83. confirmText={(sparkConfirmInfo.type === 'danger' && t('common.operation.ok')) || ''}
  84. />
  85. )
  86. }
  87. </nav>
  88. <div ref={anchorRef} className='h-0'> </div>
  89. </>
  90. )
  91. }
  92. export default Apps