Bpistle/src/components/ImageGallery/index.tsx
2025-03-09 11:22:41 +08:00

112 lines
3.1 KiB
TypeScript

import { cn } from '@/lib/utils'
import { useScreenSize } from '@/providers/ScreenSizeProvider'
import { TImageInfo } from '@/types'
import { ReactNode, useState } from 'react'
import { createPortal } from 'react-dom'
import Lightbox from 'yet-another-react-lightbox'
import Zoom from 'yet-another-react-lightbox/plugins/zoom'
import Image from '../Image'
import NsfwOverlay from '../NsfwOverlay'
export default function ImageGallery({
className,
images,
isNsfw = false,
size = 'normal'
}: {
className?: string
images: TImageInfo[]
isNsfw?: boolean
size?: 'normal' | 'small'
}) {
const { isSmallScreen } = useScreenSize()
const [index, setIndex] = useState(-1)
const handlePhotoClick = (event: React.MouseEvent, current: number) => {
event.stopPropagation()
event.preventDefault()
setIndex(current)
}
let imageContent: ReactNode | null = null
if (images.length === 1) {
imageContent = (
<Image
key={0}
className={cn('rounded-lg', size === 'small' ? 'max-h-[15vh]' : 'max-h-[30vh]')}
classNames={{
errorPlaceholder: cn('aspect-square', size === 'small' ? 'h-[15vh]' : 'h-[30vh]')
}}
image={images[0]}
onClick={(e) => handlePhotoClick(e, 0)}
/>
)
} else if (size === 'small') {
imageContent = (
<div className="grid grid-cols-4 gap-2">
{images.map((image, i) => (
<Image
key={i}
className={cn('aspect-square w-full rounded-lg')}
image={image}
onClick={(e) => handlePhotoClick(e, i)}
/>
))}
</div>
)
} else if (isSmallScreen && (images.length === 2 || images.length === 4)) {
imageContent = (
<div className="grid grid-cols-2 gap-2">
{images.map((image, i) => (
<Image
key={i}
className={cn('aspect-square w-full rounded-lg')}
image={image}
onClick={(e) => handlePhotoClick(e, i)}
/>
))}
</div>
)
} else {
imageContent = (
<div className="grid grid-cols-3 gap-2 w-full">
{images.map((image, i) => (
<Image
key={i}
className={cn('aspect-square w-full rounded-lg')}
image={image}
onClick={(e) => handlePhotoClick(e, i)}
/>
))}
</div>
)
}
return (
<div className={cn('relative', images.length === 1 ? 'w-fit max-w-full' : 'w-full', className)}>
{imageContent}
{index >= 0 &&
createPortal(
<div onClick={(e) => e.stopPropagation()}>
<Lightbox
index={index}
slides={images.map(({ url }) => ({ src: url }))}
plugins={[Zoom]}
open={index >= 0}
close={() => setIndex(-1)}
controller={{
closeOnBackdropClick: true,
closeOnPullUp: true,
closeOnPullDown: true
}}
styles={{
toolbar: { paddingTop: '2.25rem' }
}}
/>
</div>,
document.body
)}
{isNsfw && <NsfwOverlay className="rounded-lg" />}
</div>
)
}