-
Notifications
You must be signed in to change notification settings - Fork 7
/
Copy pathserver.js
140 lines (116 loc) · 3.36 KB
/
server.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
const express = require("express");
const cors = require("cors");
const fs = require("fs");
const app = express();
const port = 3002;
app.use(express.json());
app.use(cors({ origin: "*" }));
const dest = "uploads/";
app.get("/", (req, res) => {
res.send("Hello World!");
});
app.post("/", function (req, res) {
res.send("Got a POST request");
});
app.get("/upload/status", (req, res) => {
const uniqueFileId = String(req.headers["x-file-name"]);
const fileSize = parseInt(String(req.headers["file-size"]), 10);
if (!fs.existsSync(dest)) {
fs.mkdirSync(dest);
}
if (!fileSize) {
res.status(400).send("No file-size header found");
res.end(400);
return;
}
if (!uniqueFileId) {
res.status(400).send("No x-file-name header found");
res.end(400);
return;
}
if (uniqueFileId) {
try {
const stats = fs.statSync(dest + uniqueFileId);
if (stats.isFile()) {
if (fileSize === stats.size) {
res.send({
status: "ALREADY_UPLOADED_FILE",
uploaded: stats.size,
});
return;
}
if (!uploads[uniqueFileId]) uploads[uniqueFileId] = {};
uploads[uniqueFileId]["bytesReceived"] = stats.size;
res.send({ uploaded: stats.size });
}
} catch (err) {
const upload = uploads[uniqueFileId];
if (upload)
res.send({ uploaded: upload.bytesReceived, status: "RESUMED_FILE" });
else res.send({ uploaded: 0, status: "NEW_FILE" });
}
}
});
let uploads = {};
app.post("/upload/files", (req, res) => {
const uniqueFileId = String(req.headers["x-file-name"]);
const match = req.headers["content-range"].match(/(\d+)-(\d+)\/(\d+)/);
const start = parseInt(match[1]);
const fileSize = parseInt(String(req.headers["file-size"]), 10);
if (!fs.existsSync(dest)) {
fs.mkdirSync(dest);
}
if (
uploads[uniqueFileId] &&
fileSize === uploads[uniqueFileId].bytesReceived
) {
res.status(200).send("File already uploaded");
res.end();
return;
}
if (!uniqueFileId) {
res.status(400).send("No x-file-name header found");
res.end(400);
return;
}
if (!uploads[uniqueFileId]) uploads[uniqueFileId] = {};
const upload = uploads[uniqueFileId];
let fileStream;
if (!start) {
upload.bytesReceived = 0;
fileStream = fs.createWriteStream(`./uploads/${uniqueFileId}`, {
flags: "w",
});
} else {
if (upload.bytesReceived != start) {
res.writeHead(400, "Wrong start byte");
res.end(upload.bytesReceived);
return;
}
// append to existing file
fileStream = fs.createWriteStream(`./uploads/${uniqueFileId}`, {
flags: "a",
});
}
req.on("data", function (data) {
upload.bytesReceived += data.length;
});
req.pipe(fileStream);
// when the request is finished, and all its data is written
fileStream.on("close", function () {
res.status(201).send({ status: "UPLOAD_COMPLETE" });
});
// in case of I/O error - finish the request
fileStream.on("error", function (_err) {
res.status(500).send("File error");
res.end();
});
});
app.post("/upload/complete", (req, res) => {
const uniqueFileId = String(req.headers["x-file-name"]);
delete uploads[uniqueFileId];
res.status(201).send({ status: "SUCCESSFULLY_UPLOADED" });
});
app.listen(port, () => {
console.log(`Example app listening at http://localhost:${port}`);
});