Upload public.js

backend/src/routes/public.js

@@ -1147,4 +1147,88 @@ router.get('/direct-image/:userId/:pptId/:slideIndex?', async (req, res, next) =
   }
 });
 
+// 存储前端生成的图片数据
+const imageCache = new Map();
+
+// POST endpoint to receive frontend-generated images
+router.post('/save-image/:userId/:pptId/:slideIndex', express.json({ limit: '10mb' }), async (req, res, next) => {
+  try {
+    const { userId, pptId, slideIndex } = req.params;
+    const { imageData, format = 'jpeg', quality = 90 } = req.body;
+    
+    if (!imageData) {
+      return res.status(400).json({ error: 'Image data is required' });
+    }
+    
+    // 生成唯一的图片ID
+    const imageId = `${userId}-${pptId}-${slideIndex}-${Date.now()}`;
+    
+    // 存储图片数据到内存缓存
+    imageCache.set(imageId, {
+      data: imageData,
+      format,
+      quality,
+      userId,
+      pptId,
+      slideIndex,
+      timestamp: new Date().toISOString(),
+      contentType: `image/${format === 'jpg' ? 'jpeg' : format}`
+    });
+    
+    // 设置过期时间（1小时后自动清理）
+    setTimeout(() => {
+      imageCache.delete(imageId);
+    }, 3600000);
+    
+    // 返回图片访问URL
+    const baseUrl = req.protocol + '://' + req.get('host');
+    const imageUrl = `${baseUrl}/api/public/cached-image/${imageId}`;
+    
+    console.log(`✅ Image saved with ID: ${imageId}`);
+    
+    res.json({
+      success: true,
+      imageId,
+      imageUrl,
+      expiresAt: new Date(Date.now() + 3600000).toISOString()
+    });
+    
+  } catch (error) {
+    console.error('❌ Failed to save image:', error);
+    next(error);
+  }
+});
+
+// GET endpoint to serve cached images
+router.get('/cached-image/:imageId', async (req, res, next) => {
+  try {
+    const { imageId } = req.params;
+    
+    const cachedImage = imageCache.get(imageId);
+    if (!cachedImage) {
+      return res.status(404).json({ error: 'Image not found or expired' });
+    }
+    
+    // 设置响应头
+    res.setHeader('Content-Type', cachedImage.contentType);
+    res.setHeader('Cache-Control', 'public, max-age=3600');
+    res.setHeader('Access-Control-Allow-Origin', '*');
+    res.setHeader('X-Image-Source', 'frontend-generated');
+    res.setHeader('X-Generation-Time', cachedImage.timestamp);
+    
+    // 如果是base64数据，需要转换
+    if (cachedImage.data.startsWith('data:')) {
+      const base64Data = cachedImage.data.split(',')[1];
+      const buffer = Buffer.from(base64Data, 'base64');
+      res.send(buffer);
+    } else {
+      res.send(cachedImage.data);
+    }
+    
+  } catch (error) {
+    console.error('❌ Failed to serve cached image:', error);
+    next(error);
+  }
+});
+
 export default router;