feat: multi-track file upload support (native images, code/text block formatting, and workspace binary uploads)

This commit is contained in:
2026-09-09 10:08:45 +08:00
parent cda806666f
commit 4401e9cea7
3 changed files with 301 additions and 29 deletions
+69 -1
View File
@@ -1,3 +1,6 @@
import fs from 'node:fs';
import path from 'node:path';
export const name = 'dsh-plugin-model-enhancer';
export const inject = ['webServer'];
@@ -33,12 +36,12 @@ async function getModelsDevData() {
}
export function apply(ctx) {
// 1. Models.dev Cache Proxy
ctx.effect(() => {
return ctx.webServer.register({
kind: 'exact',
path: '/api/models-dev/models',
handler: async (req, res) => {
// Add CORS headers so web client can always reach it seamlessly
res.setHeader('Access-Control-Allow-Origin', '*');
res.setHeader('Access-Control-Allow-Methods', 'GET, OPTIONS');
res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
@@ -70,4 +73,69 @@ export function apply(ctx) {
}
});
}, 'dsh-plugin-model-enhancer: webServer proxy route');
// 2. Binary / Large Document File Upload to Workspace Endpoint
ctx.effect(() => {
return ctx.webServer.register({
kind: 'exact',
path: '/api/model-enhancer/upload',
handler: async (req, res) => {
res.setHeader('Access-Control-Allow-Origin', '*');
res.setHeader('Access-Control-Allow-Methods', 'POST, OPTIONS');
res.setHeader('Access-Control-Allow-Headers', 'Content-Type, X-Filename');
if (req.method === 'OPTIONS') {
res.writeHead(204);
res.end();
return;
}
if (req.method !== 'POST') {
res.writeHead(405, { 'Content-Type': 'application/json' });
res.end(JSON.stringify({ error: 'Method Not Allowed' }));
return;
}
try {
const urlObj = new URL(req.url, 'http://127.0.0.1');
const rawName = req.headers['x-filename'] || urlObj.searchParams.get('filename') || `upload_${Date.now()}.bin`;
const decodedName = decodeURIComponent(rawName);
const safeName = path.basename(decodedName);
const uploadDir = path.join(process.cwd(), 'uploads');
fs.mkdirSync(uploadDir, { recursive: true });
const targetPath = path.join(uploadDir, safeName);
const writeStream = fs.createWriteStream(targetPath);
let totalBytes = 0;
req.on('data', (chunk) => {
totalBytes += chunk.length;
});
req.pipe(writeStream);
writeStream.on('finish', () => {
res.writeHead(200, { 'Content-Type': 'application/json' });
res.end(JSON.stringify({
ok: true,
filename: safeName,
path: path.join('uploads', safeName),
size: totalBytes
}));
});
writeStream.on('error', (err) => {
ctx.logger?.error?.('[model-enhancer] Upload stream error:', err);
res.writeHead(500, { 'Content-Type': 'application/json' });
res.end(JSON.stringify({ error: err.message || 'File write error' }));
});
} catch (err) {
ctx.logger?.error?.('[model-enhancer] Upload handler error:', err);
res.writeHead(500, { 'Content-Type': 'application/json' });
res.end(JSON.stringify({ error: err.message || 'Upload processing error' }));
}
}
});
}, 'dsh-plugin-model-enhancer: workspace file upload endpoint');
}