Update temp.js

temp.js

@@ -1,48 +1,66 @@
+const express = require('express');
+const { v4: uuidv4 } = require('uuid');
+const fs = require('fs');
+const path = require('path');
 
-   const express = require('express');
-   const { v4: uuidv4 } = require('uuid');
-   const path = require('path');
-   const fs = require('fs');
-
-   const app = express();
-
-   app.put('/:filename', (req, res) => {
-       const shortId = uuidv4().slice(0, 5);
-       const filename = req.params.filename;
-       const filepath = path.join(__dirname, `/uploads/${shortId}-${filename}`);
-       const fileStream = fs.createWriteStream(filepath);
-
-       req.pipe(fileStream);
-
-       req.on('end', () => {
-           const fileUrl = `http://localhost:7860/${shortId}/${filename}`;
-           res.send(`Uploaded file to ${fileUrl}`);
-
-           // Hapus file setelah 24 jam
-           setTimeout(() => {
-               fs.unlink(filepath, (err) => {
-                   if (err) console.error(`Error deleting file: ${err}`);
-               });
-           }, 24 * 60 * 60 * 1000); // 24 jam dalam milidetik
-       });
-
-       req.on('error', (err) => {
-           console.error(`Error receiving file: ${err}`);
-           res.status(500).send('Error uploading file.');
-       });
-   });
-
-   app.get('/:id/:filename', (req, res) => {
-       const filepath = path.join(__dirname, `/uploads/${req.params.id}-${req.params.filename}`);
-       res.download(filepath, req.params.filename, (err) => {
-           if (err) {
-               console.error(`Error downloading file: ${err}`);
-               res.status(404).send('File not found.');
-           }
-       });
-   });
-
-   app.listen(7860, () => {
-       console.log('Server is running on port 7860');
-   });
-   
+const app = express();
+const storageDir = path.join(__dirname, 'uploads');
+
+// Ensure the storage directory exists
+if (!fs.existsSync(storageDir)) {
+    fs.mkdirSync(storageDir);
+}
+
+app.put('/:filename', (req, res) => {
+    const shortId = uuidv4().slice(0, 5); // Generate a short ID
+    const filename = req.params.filename;
+    const filepath = path.join(storageDir, `${shortId}-${filename}`);
+    const chunks = [];
+
+    req.on('data', chunk => {
+        chunks.push(chunk);
+    });
+
+    req.on('end', () => {
+        const fileBuffer = Buffer.concat(chunks);
+
+        fs.writeFile(filepath, fileBuffer, (err) => {
+            if (err) {
+                console.error(`Error writing file: ${err}`);
+                return res.status(500).send('Error uploading file.');
+            }
+
+            const fileUrl = `https://zhofang-temp-storage.hf.space/${shortId}/${filename}`;
+            res.send(`Uploaded 1 file, ${fileBuffer.length} bytes\n\nwget ${fileUrl}\n`);
+
+            // Delete the file after 24 hours
+            setTimeout(() => {
+                fs.unlink(filepath, (err) => {
+                    if (err) console.error(`Error deleting file: ${err}`);
+                });
+            }, 24 * 60 * 60 * 1000); // 24 hours in milliseconds
+        });
+    });
+
+    req.on('error', (err) => {
+        console.error(`Error receiving file: ${err}`);
+        res.status(500).send('Error uploading file.');
+    });
+});
+
+app.get('/:id/:filename', (req, res) => {
+    const filepath = path.join(storageDir, `${req.params.id}-${req.params.filename}`);
+    fs.readFile(filepath, (err, fileBuffer) => {
+        if (err) {
+            console.error(`Error downloading file: ${err}`);
+            return res.status(404).send('File not found.');
+        }
+
+        res.setHeader('Content-Disposition', `attachment; filename="${req.params.filename}"`);
+        res.send(fileBuffer);
+    });
+});
+
+app.listen(7860, () => {
+    console.log('Server is running on port 7860');
+});