feat(web): add asset CRUD and scanning API
Co-Authored-By: Claude Opus 4.7 <noreply@anthropic.com>
This commit is contained in:
117
web/server/routes/assets.ts
Normal file
117
web/server/routes/assets.ts
Normal file
@@ -0,0 +1,117 @@
|
||||
import { Router } from 'express';
|
||||
import { getDb } from '../db';
|
||||
import { randomUUID } from 'crypto';
|
||||
import fs from 'fs';
|
||||
import path from 'path';
|
||||
import { fileURLToPath } from 'url';
|
||||
|
||||
const __filename = fileURLToPath(import.meta.url);
|
||||
const __dirname = path.dirname(__filename);
|
||||
const PROJECT_ROOT = path.resolve(__dirname, '..', '..', '..');
|
||||
|
||||
export const assetsRouter = Router();
|
||||
|
||||
// List assets with optional filters
|
||||
assetsRouter.get('/', (req, res) => {
|
||||
const { accountId, type } = req.query;
|
||||
let sql = 'SELECT * FROM assets WHERE 1=1';
|
||||
const params: unknown[] = [];
|
||||
|
||||
if (accountId) { sql += ' AND account_id = ?'; params.push(accountId); }
|
||||
if (type) { sql += ' AND type = ?'; params.push(type); }
|
||||
|
||||
sql += ' ORDER BY created_at DESC LIMIT 200';
|
||||
const rows = getDb().prepare(sql).all(...params);
|
||||
res.json(rows);
|
||||
});
|
||||
|
||||
// Delete asset and its file
|
||||
assetsRouter.delete('/:id', (req, res) => {
|
||||
const asset = getDb().prepare('SELECT * FROM assets WHERE id = ?').get(req.params.id) as {
|
||||
file_path: string;
|
||||
} | undefined;
|
||||
|
||||
if (!asset) return res.status(404).json({ error: 'Asset not found' });
|
||||
|
||||
const filePath = path.join(PROJECT_ROOT, asset.file_path);
|
||||
if (fs.existsSync(filePath)) fs.unlinkSync(filePath);
|
||||
|
||||
getDb().prepare('DELETE FROM assets WHERE id = ?').run(req.params.id);
|
||||
res.status(204).send();
|
||||
});
|
||||
|
||||
// Scan output directories and index assets
|
||||
assetsRouter.post('/scan', (_req, res) => {
|
||||
const outputDir = path.join(PROJECT_ROOT, 'output');
|
||||
if (!fs.existsSync(outputDir)) return res.json({ indexed: 0 });
|
||||
|
||||
const dirs = fs.readdirSync(outputDir, { withFileTypes: true })
|
||||
.filter((d) => d.isDirectory());
|
||||
|
||||
let indexed = 0;
|
||||
for (const dir of dirs) {
|
||||
const manifestPath = path.join(outputDir, dir.name, 'manifest.json');
|
||||
if (!fs.existsSync(manifestPath)) continue;
|
||||
|
||||
let manifest;
|
||||
try {
|
||||
manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf-8'));
|
||||
} catch { continue; }
|
||||
|
||||
const accountId = manifest.account?.id || '';
|
||||
|
||||
// Scan images
|
||||
const imagesDir = path.join(outputDir, dir.name, 'images');
|
||||
if (fs.existsSync(imagesDir)) {
|
||||
const files = fs.readdirSync(imagesDir);
|
||||
for (const file of files) {
|
||||
if (!/\.(jpe?g|png|webp)$/i.test(file)) continue;
|
||||
const id = randomUUID();
|
||||
const relPath = path.relative(PROJECT_ROOT, path.join(imagesDir, file));
|
||||
const match = file.match(/scene_(\d+)/);
|
||||
const shotIndex = match ? parseInt(match[1]) : null;
|
||||
|
||||
const exists = getDb().prepare('SELECT id FROM assets WHERE file_path = ?').get(relPath);
|
||||
if (exists) continue;
|
||||
|
||||
getDb().prepare(
|
||||
'INSERT INTO assets (id, account_id, manifest_path, type, file_path, shot_index) VALUES (?,?,?,?,?,?)'
|
||||
).run(id, accountId, path.relative(PROJECT_ROOT, manifestPath), 'image', relPath, shotIndex);
|
||||
indexed++;
|
||||
}
|
||||
}
|
||||
|
||||
// Scan videos
|
||||
const videosDir = path.join(outputDir, dir.name, 'videos');
|
||||
if (fs.existsSync(videosDir)) {
|
||||
const files = fs.readdirSync(videosDir);
|
||||
for (const file of files) {
|
||||
if (!/\.mp4$/i.test(file)) continue;
|
||||
const id = randomUUID();
|
||||
const relPath = path.relative(PROJECT_ROOT, path.join(videosDir, file));
|
||||
const match = file.match(/scene_(\d+)/);
|
||||
const shotIndex = match ? parseInt(match[1]) : null;
|
||||
|
||||
const exists = getDb().prepare('SELECT id FROM assets WHERE file_path = ?').get(relPath);
|
||||
if (exists) continue;
|
||||
|
||||
getDb().prepare(
|
||||
'INSERT INTO assets (id, account_id, manifest_path, type, file_path, shot_index) VALUES (?,?,?,?,?,?)'
|
||||
).run(id, accountId, path.relative(PROJECT_ROOT, manifestPath), 'video', relPath, shotIndex);
|
||||
indexed++;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
res.json({ indexed });
|
||||
});
|
||||
|
||||
// Serve asset files
|
||||
assetsRouter.get('/file', (req, res) => {
|
||||
const filePath = req.query.path as string;
|
||||
if (!filePath) return res.status(400).send('Missing path');
|
||||
const fullPath = path.resolve(PROJECT_ROOT, filePath);
|
||||
if (!fullPath.startsWith(PROJECT_ROOT)) return res.status(403).send('Forbidden');
|
||||
if (!fs.existsSync(fullPath)) return res.status(404).send('Not found');
|
||||
res.sendFile(fullPath);
|
||||
});
|
||||
Reference in New Issue
Block a user