Spaces:

Zhofang
/

temp_storage

Running

File size: 9,326 Bytes

const express = require('express');
const fs = require('fs');
const path = require('path');
const multer = require('multer');

const app = express();
const uploadDir = path.join(__dirname, 'uploads');

// Ensure uploads directory exists
if (!fs.existsSync(uploadDir)) {
    fs.mkdirSync(uploadDir);
}

// Function to generate a short random ID (alphanumeric)
function generateShortId() {
    return Math.random().toString(36).substring(2, 7); // Generates a random string of length 5
}

// Configure multer for file storage
const storage = multer.diskStorage({
    destination: function (req, file, cb) {
        cb(null, uploadDir);
    },
    filename: function (req, file, cb) {
        const shortId = generateShortId();
        cb(null, `${shortId}-${file.originalname}`);
    }
});

const upload = multer({ storage: storage });

// Route for browser upload
app.get('/', (req, res) => {
    res.send(`
    <!DOCTYPE html>
    <html lang="en">
    <head>
        <meta charset="UTF-8">
        <meta name="viewport" content="width=device-width, initial-scale=1.0">
        <title>Enhanced File Upload</title>
        <style>
            body {
                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
                height: 100vh;
                margin: 0;
                display: flex;
                justify-content: center;
                align-items: center;
            }
            .container {
                background-color: rgba(255, 255, 255, 0.9);
                padding: 2rem;
                border-radius: 10px;
                box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
                text-align: center;
                width: 300px;
            }
            h1 {
                color: #4a5568;
                margin-bottom: 1.5rem;
            }
            form {
                display: flex;
                flex-direction: column;
                align-items: center;
            }
            input[type="file"] {
                display: none;
            }
            .file-label {
                background-color: #4a5568;
                color: white;
                padding: 0.5rem 1rem;
                border-radius: 5px;
                cursor: pointer;
                margin-bottom: 1rem;
                transition: background-color 0.3s ease;
            }
            .file-label:hover {
                background-color: #2d3748;
            }
            #file-name {
                margin-bottom: 1rem;
                word-break: break-all;
            }
            button {
                background-color: #4299e1;
                color: white;
                border: none;
                padding: 0.5rem 1rem;
                border-radius: 5px;
                cursor: pointer;
                transition: background-color 0.3s ease;
            }
            button:hover {
                background-color: #3182ce;
            }
            #upload-progress {
                width: 100%;
                background-color: #e2e8f0;
                border-radius: 5px;
                margin-top: 1rem;
                overflow: hidden;
                display: none;
            }
            #progress-bar {
                width: 0;
                height: 10px;
                background-color: #48bb78;
                transition: width 0.5s ease;
            }
        </style>
    </head>
    <body>
        <div class="container">
            <h1>Upload File</h1>
            <form id="upload-form" action="/upload" method="post" enctype="multipart/form-data">
                <label for="file-upload" class="file-label">Choose File</label>
                <input id="file-upload" type="file" name="file">
                <div id="file-name"></div>
                <button type="submit">Upload</button>
            </form>
            <div id="upload-progress">
                <div id="progress-bar"></div>
            </div>
        </div>

        <script>
            const fileUpload = document.getElementById('file-upload');
            const fileName = document.getElementById('file-name');
            const uploadForm = document.getElementById('upload-form');
            const uploadProgress = document.getElementById('upload-progress');
            const progressBar = document.getElementById('progress-bar');

            fileUpload.addEventListener('change', (e) => {
                if (e.target.files.length > 0) {
                    fileName.textContent = e.target.files[0].name;
                } else {
                    fileName.textContent = '';
                }
            });

            uploadForm.addEventListener('submit', (e) => {
                e.preventDefault();
                if (!fileUpload.files.length) {
                    alert('Please select a file to upload.');
                    return;
                }

                const formData = new FormData(uploadForm);
                const xhr = new XMLHttpRequest();

                xhr.open('POST', '/upload', true);

                xhr.upload.onprogress = (event) => {
                    if (event.lengthComputable) {
                        const percentComplete = (event.loaded / event.total) * 100;
                        uploadProgress.style.display = 'block';
                        progressBar.style.width = percentComplete + '%';
                    }
                };

                xhr.onload = function() {
                    if (xhr.status === 200) {
                        alert('Upload complete!');
                        document.body.innerHTML = xhr.responseText;
                    } else {
                        alert('Upload failed. Please try again.');
                    }
                    uploadProgress.style.display = 'none';
                    progressBar.style.width = '0';
                    fileName.textContent = '';
                    uploadForm.reset();
                };

                xhr.send(formData);
            });
        </script>
    </body>
    </html>
    `);
});

// Handler for browser upload
app.post('/upload', upload.single('file'), (req, res) => {
    if (!req.file) {
        return res.status(400).send('No file uploaded.');
    }

    const fileUrl = `https://zhofang-temp-storage.hf.space/${req.file.filename.split('-')[0]}/${req.file.originalname}`;
    res.send(`
    <html>
    <head>
        <title>File Uploaded</title>
        <style>
            body {
                font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
                background: linear-gradient(135deg, #667eea 0%, #764ba2 100%);
                height: 100vh;
                margin: 0;
                display: flex;
                justify-content: center;
                align-items: center;
                color: white;
            }
            .container {
                background-color: rgba(255, 255, 255, 0.1);
                padding: 2rem;
                border-radius: 10px;
                box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
                text-align: center;
            }
            a {
                color: #4299e1;
                text-decoration: none;
            }
            a:hover {
                text-decoration: underline;
            }
        </style>
    </head>
    <body>
        <div class="container">
            <h1>File Uploaded Successfully</h1>
            <p>Uploaded 1 file, ${req.file.size} bytes</p>
            <p>Download link: <a href="${fileUrl}">${fileUrl}</a></p>
            <p>wget command: <code>wget ${fileUrl}</code></p>
        </div>
    </body>
    </html>
    `);

    // Delete file after 24 hours
    setTimeout(() => {
        fs.unlink(req.file.path, (err) => {
            if (err) console.error(`Error deleting file: ${err}`);
        });
    }, 24 * 60 * 60 * 1000); // 24 hours in milliseconds
});

// Route for upload via PUT (like bashupload)
app.put('/:filename', (req, res) => {
    const shortId = generateShortId();
    const filename = req.params.filename;
    const filepath = path.join(uploadDir, `${shortId}-${filename}`);
    const fileStream = fs.createWriteStream(filepath);

    req.pipe(fileStream);

    fileStream.on('finish', () => {
        const fileUrl = `https://zhofang-temp-storage.hf.space/${shortId}/${filename}`;
        res.send(`Uploaded 1 file, ${req.headers['content-length']} bytes\n\nwget ${fileUrl}\n`);

        // Delete 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
    });

    fileStream.on('error', (err) => {
        console.error(`Error writing file: ${err}`);
        res.status(500).send('Error uploading file.');
    });
});

app.get('/:id/:filename', (req, res) => {
    const filepath = path.join(uploadDir, `${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 http://localhost:3000');
});