后端: 添加文件预览API
This commit is contained in:
parent
1b345233ee
commit
9e0d6850e1
@ -4,6 +4,7 @@ const authRoutes = require('./routes/auth');
|
|||||||
const fileRoutes = require('./routes/files');
|
const fileRoutes = require('./routes/files');
|
||||||
const shareRoutes = require('./routes/share');
|
const shareRoutes = require('./routes/share');
|
||||||
const syncRoutes = require('./routes/sync');
|
const syncRoutes = require('./routes/sync');
|
||||||
|
const previewRoutes = require('./routes/preview');
|
||||||
const errorHandler = require('./middleware/errorHandler');
|
const errorHandler = require('./middleware/errorHandler');
|
||||||
|
|
||||||
const app = express();
|
const app = express();
|
||||||
@ -17,6 +18,7 @@ app.use('/api/auth', authRoutes);
|
|||||||
app.use('/api/files', fileRoutes);
|
app.use('/api/files', fileRoutes);
|
||||||
app.use('/api/share', shareRoutes);
|
app.use('/api/share', shareRoutes);
|
||||||
app.use('/api/sync', syncRoutes);
|
app.use('/api/sync', syncRoutes);
|
||||||
|
app.use('/api/preview', previewRoutes);
|
||||||
|
|
||||||
// Error handler
|
// Error handler
|
||||||
app.use(errorHandler);
|
app.use(errorHandler);
|
||||||
|
|||||||
84
backend/src/routes/preview.js
Normal file
84
backend/src/routes/preview.js
Normal file
@ -0,0 +1,84 @@
|
|||||||
|
const express = require('express');
|
||||||
|
const path = require('path');
|
||||||
|
const fs = require('fs');
|
||||||
|
const db = require('../db');
|
||||||
|
|
||||||
|
const router = express.Router();
|
||||||
|
|
||||||
|
// Get preview info
|
||||||
|
router.get('/:id', (req, res) => {
|
||||||
|
const token = req.headers.authorization?.replace('Bearer ', '');
|
||||||
|
if (!token) return res.status(401).json({ error: 'No token' });
|
||||||
|
|
||||||
|
try {
|
||||||
|
const jwt = require('jsonwebtoken');
|
||||||
|
const decoded = jwt.verify(token, process.env.JWT_SECRET || 'clouddisk-secret-key');
|
||||||
|
|
||||||
|
const file = db.query(
|
||||||
|
'SELECT * FROM files WHERE id = ? AND user_id = ? AND is_folder = 0',
|
||||||
|
[req.params.id, decoded.userId]
|
||||||
|
);
|
||||||
|
|
||||||
|
if (file.length === 0) {
|
||||||
|
return res.status(404).json({ error: 'File not found' });
|
||||||
|
}
|
||||||
|
|
||||||
|
const f = file[0];
|
||||||
|
|
||||||
|
// Determine if file is previewable
|
||||||
|
const previewableTypes = ['jpg', 'jpeg', 'png', 'gif', 'pdf', 'txt', 'md'];
|
||||||
|
const ext = f.name.split('.').pop().toLowerCase();
|
||||||
|
const isPreviewable = previewableTypes.includes(ext);
|
||||||
|
|
||||||
|
res.json({
|
||||||
|
fileId: f.id,
|
||||||
|
filename: f.name,
|
||||||
|
size: f.size,
|
||||||
|
type: ext,
|
||||||
|
isPreviewable,
|
||||||
|
canDownload: true
|
||||||
|
});
|
||||||
|
} catch (error) {
|
||||||
|
res.status(500).json({ error: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
// Get preview content (for text files)
|
||||||
|
router.get('/:id/content', (req, res) => {
|
||||||
|
const token = req.headers.authorization?.replace('Bearer ', '');
|
||||||
|
if (!token) return res.status(401).json({ error: 'No token' });
|
||||||
|
|
||||||
|
try {
|
||||||
|
const jwt = require('jsonwebtoken');
|
||||||
|
const decoded = jwt.verify(token, process.env.JWT_SECRET || 'clouddisk-secret-key');
|
||||||
|
|
||||||
|
const file = db.query(
|
||||||
|
'SELECT * FROM files WHERE id = ? AND user_id = ? AND is_folder = 0',
|
||||||
|
[req.params.id, decoded.userId]
|
||||||
|
);
|
||||||
|
|
||||||
|
if (file.length === 0) {
|
||||||
|
return res.status(404).json({ error: 'File not found' });
|
||||||
|
}
|
||||||
|
|
||||||
|
const f = file[0];
|
||||||
|
const ext = f.name.split('.').pop().toLowerCase();
|
||||||
|
|
||||||
|
// Only allow text files
|
||||||
|
if (!['txt', 'md', 'json', 'js', 'ts', 'html', 'css'].includes(ext)) {
|
||||||
|
return res.status(400).json({ error: 'File type not supported for preview' });
|
||||||
|
}
|
||||||
|
|
||||||
|
// Limit preview size (100KB)
|
||||||
|
if (f.size > 100 * 1024) {
|
||||||
|
return res.status(400).json({ error: 'File too large for preview' });
|
||||||
|
}
|
||||||
|
|
||||||
|
const content = fs.readFileSync(f.path, 'utf8');
|
||||||
|
res.json({ content });
|
||||||
|
} catch (error) {
|
||||||
|
res.status(500).json({ error: error.message });
|
||||||
|
}
|
||||||
|
});
|
||||||
|
|
||||||
|
module.exports = router;
|
||||||
Loading…
Reference in New Issue
Block a user