Update temp.js

temp.js

@@ -1,52 +1,63 @@
-   const express = require('express');
-   const { v4: uuidv4 } = require('uuid');
-   const MemoryFS = require('memory-fs');
-   const fs = require('fs');
-
-   const app = express();
-   const memFs = new MemoryFS();
-
-   app.put('/:filename', (req, res) => {
-       const shortId = uuidv4().slice(0, 5); // Menghasilkan ID pendek
-       const filename = req.params.filename;
-       const filepath = `/${shortId}-${filename}`;
-       const chunks = [];
-
-       req.on('data', chunk => {
-           chunks.push(chunk);
-       });
-
-       req.on('end', () => {
-           const fileBuffer = Buffer.concat(chunks);
-           memFs.writeFileSync(filepath, fileBuffer);
-
-           const fileUrl = `https://zhofang-temp-storage.hf.space/${shortId}/${filename}`;
-           res.send(`Uploaded 1 file, ${fileBuffer.length} bytes\n\nwget ${fileUrl}\n`);
-
-           // Hapus file setelah 24 jam
-           setTimeout(() => {
-               memFs.unlinkSync(filepath);
-           }, 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 = `/${req.params.id}-${req.params.filename}`;
-       try {
-           const fileBuffer = memFs.readFileSync(filepath);
-           res.setHeader('Content-Disposition', `attachment; filename="${req.params.filename}"`);
-           res.send(fileBuffer);
-       } catch (err) {
-           console.error(`Error downloading file: ${err}`);
-           res.status(404).send('File not found.');
-       }
-   });
-
-   app.listen(7860, () => {
-       console.log('Server is running on idk just guess it');
-   });
+const express = require('express');
+const { v4: uuidv4 } = require('uuid');
+const MemoryFS = require('memory-fs');
+const fs = require('fs');
+
+const app = express();
+const memFs = new MemoryFS();
+
+const port = process.env.PORT || 7860;
+
+app.put('/:filename', (req, res) => {
+    const shortId = uuidv4().slice(0, 5); // Menghasilkan ID pendek
+    const filename = req.params.filename;
+    const filepath = `/${shortId}-${filename}`;
+    const chunks = [];
+
+    req.on('data', chunk => {
+        chunks.push(chunk);
+    });
+
+    req.on('end', () => {
+        const fileBuffer = Buffer.concat(chunks);
+        memFs.writeFileSync(filepath, fileBuffer);
+
+      const protocol = req.protocol;
+      let host = req.get('host');
+      let port = protocol === 'https' ? 443 : 80;
+    
+        if (host.includes(':')) {
+            [host, port] = host.split(':');
+        }
+
+
+        const fileUrl = `${protocol}://${host}:${port}/${shortId}/${filename}`;
+        res.send(`Uploaded 1 file, ${fileBuffer.length} bytes\n\nwget ${fileUrl}\n`);
+
+        // Hapus file setelah 24 jam
+        setTimeout(() => {
+            memFs.unlinkSync(filepath);
+        }, 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 = `/${req.params.id}-${req.params.filename}`;
+    try {
+        const fileBuffer = memFs.readFileSync(filepath);
+        res.setHeader('Content-Disposition', `attachment; filename="${req.params.filename}"`);
+        res.send(fileBuffer);
+    } catch (err) {
+        console.error(`Error downloading file: ${err}`);
+        res.status(404).send('File not found.');
+    }
+});
+
+app.listen(port, () => {
+    console.log(`Server is running on port: ${port}`);
+});