ai-chatbot/apps/editor/app/viewer/[id]/thumbnail-generator.tsx
2026-02-17 12:07:32 +09:00

123 lines
4.1 KiB
TypeScript

'use client'
import { emitter } from '@pascal-app/core'
import { useThree } from '@react-three/fiber'
import { useEffect, useRef } from 'react'
import * as THREE from 'three'
import { uploadPropertyThumbnail } from '@/features/community/lib/properties/actions'
const THUMBNAIL_WIDTH = 1920
const THUMBNAIL_HEIGHT = 1080
interface ThumbnailGeneratorProps {
propertyId?: string
}
export const ThumbnailGenerator = ({ propertyId: propPropertyId }: ThumbnailGeneratorProps) => {
const gl = useThree((state) => state.gl)
const scene = useThree((state) => state.scene)
const camera = useThree((state) => state.camera)
const isGenerating = useRef(false)
// Use prop propertyId (from URL)
const fallbackPropertyId = propPropertyId
useEffect(() => {
const handleGenerateThumbnail = async (event: { propertyId: string }) => {
if (isGenerating.current) {
console.log('⏸️ Thumbnail generation already in progress')
return
}
// Prioritize prop propertyId over event propertyId (URL has priority over session)
const propertyId = fallbackPropertyId || event.propertyId
if (!propertyId) {
console.error('❌ No property ID provided')
return
}
isGenerating.current = true
console.log('📸 Generating thumbnail for property:', propertyId)
console.log('📝 Property ID from URL/prop:', fallbackPropertyId)
console.log('📝 Property ID from event:', event.propertyId)
console.log('✅ Using property ID:', propertyId, fallbackPropertyId ? '(from URL)' : '(from event)')
try {
// Save current renderer state
const currentSize = gl.getSize(new THREE.Vector2())
const currentPixelRatio = gl.getPixelRatio()
// Temporarily resize renderer to thumbnail size
gl.setPixelRatio(1)
gl.setSize(THUMBNAIL_WIDTH, THUMBNAIL_HEIGHT)
// Update camera aspect ratio if it's a perspective camera
if (camera instanceof THREE.PerspectiveCamera) {
camera.aspect = THUMBNAIL_WIDTH / THUMBNAIL_HEIGHT
camera.updateProjectionMatrix()
}
// Render the scene
gl.render(scene, camera)
// Wait a frame to ensure render is complete
await new Promise((resolve) => requestAnimationFrame(resolve))
// Capture canvas as blob
const canvas = gl.domElement
canvas.toBlob(async (blob) => {
if (blob) {
// Upload to Supabase Storage
console.log('☁️ Uploading thumbnail to storage...')
const result = await uploadPropertyThumbnail(propertyId, blob)
if (result.success) {
console.log('✅ Thumbnail uploaded successfully!')
console.log('🔗 URL:', result.data.thumbnail_url)
} else {
console.error('❌ Failed to upload thumbnail:', result.error)
}
} else {
console.error('❌ Failed to create blob from canvas')
}
// Restore renderer size and camera
gl.setPixelRatio(currentPixelRatio)
gl.setSize(currentSize.x, currentSize.y)
if (camera instanceof THREE.PerspectiveCamera) {
camera.aspect = currentSize.x / currentSize.y
camera.updateProjectionMatrix()
}
isGenerating.current = false
}, 'image/png')
} catch (error) {
console.error('❌ Failed to generate thumbnail:', error)
// Make sure to restore size even on error
const currentSize = gl.getSize(new THREE.Vector2())
const currentPixelRatio = gl.getPixelRatio()
gl.setPixelRatio(currentPixelRatio)
gl.setSize(currentSize.x, currentSize.y)
if (camera instanceof THREE.PerspectiveCamera) {
camera.aspect = currentSize.x / currentSize.y
camera.updateProjectionMatrix()
}
isGenerating.current = false
}
}
emitter.on('camera-controls:generate-thumbnail', handleGenerateThumbnail)
return () => {
emitter.off('camera-controls:generate-thumbnail', handleGenerateThumbnail)
}
}, [gl, scene, camera, fallbackPropertyId])
return null
}