wehub-resource-sync
79 lines
2.3 KiB
JavaScript
79 lines
2.3 KiB
JavaScript
const path = require("path");
|
|
const fs = require("fs");
|
|
const { broadcast } = require("../broadcast");
|
|
|
|
const UPLOADS_DIR = path.join(process.env.HOME || "/home/agent", "uploads");
|
|
|
|
function registerUploadWebSocket(uploadWss) {
|
|
uploadWss.on("connection", (ws) => {
|
|
let currentUpload = null;
|
|
|
|
ws.on("message", (data, isBinary) => {
|
|
if (isBinary) {
|
|
if (!currentUpload) {
|
|
ws.send(JSON.stringify({ type: "upload_error", error: "No upload in progress" }));
|
|
return;
|
|
}
|
|
try {
|
|
fs.appendFileSync(currentUpload.path, data);
|
|
currentUpload.received++;
|
|
ws.send(JSON.stringify({ type: "chunk_ack", chunk: currentUpload.received }));
|
|
} catch (err) {
|
|
ws.send(JSON.stringify({ type: "upload_error", error: err.message }));
|
|
currentUpload = null;
|
|
}
|
|
return;
|
|
}
|
|
|
|
let msg;
|
|
try {
|
|
msg = JSON.parse(data.toString());
|
|
} catch {
|
|
return;
|
|
}
|
|
|
|
if (msg.type === "upload_start") {
|
|
try { fs.mkdirSync(UPLOADS_DIR, { recursive: true }); } catch {}
|
|
const safeName = path
|
|
.basename(msg.filename || "file")
|
|
.replace(/[^a-zA-Z0-9._-]/g, "_");
|
|
const destPath = path.join(UPLOADS_DIR, safeName);
|
|
try { fs.writeFileSync(destPath, ""); } catch {}
|
|
currentUpload = {
|
|
id: msg.id,
|
|
filename: safeName,
|
|
path: destPath,
|
|
size: msg.size,
|
|
totalChunks: msg.totalChunks,
|
|
received: 0,
|
|
};
|
|
console.log(
|
|
`[upload] Starting: ${safeName} (${msg.size} bytes, ${msg.totalChunks} chunks)`,
|
|
);
|
|
} else if (msg.type === "upload_end") {
|
|
if (currentUpload) {
|
|
console.log(`[upload] Complete: ${currentUpload.filename}`);
|
|
ws.send(
|
|
JSON.stringify({
|
|
type: "upload_complete",
|
|
filename: currentUpload.filename,
|
|
path: currentUpload.path,
|
|
}),
|
|
);
|
|
broadcast({
|
|
type: "agent_log",
|
|
content: `[upload] File received: ${currentUpload.filename} → ~/uploads/`,
|
|
});
|
|
currentUpload = null;
|
|
}
|
|
}
|
|
});
|
|
|
|
ws.on("close", () => {
|
|
currentUpload = null;
|
|
});
|
|
});
|
|
}
|
|
|
|
module.exports = { registerUploadWebSocket };
|