server.js

const express = require("express");
const fs = require("fs");
const path = require("path");

const app = express();
const PORT = 8000;
const ROOT_DIR = path.resolve(__dirname, "sharing");

/* ───────────── LOGGING ───────────── */
app.use((req, res, next) => {
    res.on("finish", () => {
        const ip = req.headers["x-forwarded-for"] || req.socket.remoteAddress;
        const ua = req.headers["user-agent"] || "Unknown";
        console.log(`(${ip})(${ua})(${req.method})(${req.path})(${res.statusCode})`);
    });
    next();
});

/* ───────────── FILE TYPES ───────────── */
function mediaType(file) {
    const ext = path.extname(file).toLowerCase();
    return {
        image: [".png", ".jpg", ".jpeg", ".gif", ".webp"].includes(ext),
        audio: [".mp3", ".wav", ".ogg"].includes(ext),
        video: [".mp4", ".webm", ".mov"].includes(ext),
        html: [".html", ".htm"].includes(ext),
        text: [".txt", ".md", ".json", ".js", ".css"].includes(ext),
    };
}

/* ───────────── MIME ───────────── */
function mimeType(file) {
    const ext = path.extname(file).toLowerCase();
    return {
        ".mp3": "audio/mpeg",
        ".wav": "audio/wav",
        ".ogg": "audio/ogg",
        ".mp4": "video/mp4",
        ".webm": "video/webm",
        ".mov": "video/quicktime",
        ".html": "text/html",
        ".htm": "text/html",
    }[ext] || "application/octet-stream";
}

/* ───────────── RAW STREAM ───────────── */
app.get(/^\/raw\/(.+)/, (req, res) => {
    const rel = "/" + req.params[0];
    const abs = path.resolve(ROOT_DIR, "." + rel);

    if (!abs.startsWith(ROOT_DIR) || !fs.existsSync(abs)) return res.sendStatus(404);

    const stat = fs.statSync(abs);
    const range = req.headers.range;
    const type = mimeType(abs);

    if (range) {
        const [startStr, endStr] = range.replace("bytes=", "").split("-");
        const start = parseInt(startStr, 10);
        const end = endStr ? parseInt(endStr, 10) : stat.size - 1;

        res.writeHead(206, {
            "Content-Range": `bytes ${start}-${end}/${stat.size}`,
            "Accept-Ranges": "bytes",
            "Content-Length": end - start + 1,
            "Content-Type": type,
        });

        return fs.createReadStream(abs, { start, end }).pipe(res);
    }

    res.writeHead(200, {
        "Content-Length": stat.size,
        "Accept-Ranges": "bytes",
        "Content-Type": type,
    });

    fs.createReadStream(abs).pipe(res);
});

/* ───────────── MAIN ROUTE ───────────── */
app.use((req, res) => {
    try {
        if (req.path === "/favicon.ico") return res.sendStatus(204);

        const decoded = decodeURIComponent(req.path);
        const abs = path.resolve(ROOT_DIR, "." + decoded);

        if (!abs.startsWith(ROOT_DIR) || !fs.existsSync(abs)) {
            return res.status(404).send("Not Found");
        }

        const stat = fs.statSync(abs);
        const media = mediaType(abs);

        /* 📁 DIRECTORY */
        if (stat.isDirectory()) {
            const items = fs.readdirSync(abs);
            const list = items.map(name => {
                const link = encodeURI(path.posix.join(decoded, name));
                return `<li><a href="${link}">${name}</a></li>`;
            }).join("");

            return res.send(directoryPage(decoded, list));
        }

        const raw = "/raw" + encodeURI(decoded);
        const download = encodeURI(decoded) + "?download=1";
        const preview = encodeURI(decoded) + "?preview=1";

        if (req.query.download === "1") {
            return res.download(abs);
        }

        /* 🖼 IMAGE */
        if (media.image) {
            return res.send(previewPage(`<img src="${raw}">`, download));
        }

        /* 🎥 VIDEO */
        if (media.video) {
            return res.send(previewPage(`<video controls playsinline src="${raw}"></video>`, download));
        }

        /* 🔊 AUDIO */
        if (media.audio) {
            return res.send(previewPage(`<audio controls src="${raw}"></audio>`, download));
        }

        /* 🌐 HTML */
        if (media.html && req.query.preview === "1") {
            return res.sendFile(abs);
        }

        /* 📄 TEXT / HTML AS TEXT */
        if (media.text || media.html) {
            const text = fs.readFileSync(abs, "utf8")
                .replace(/</g, "&lt;")
                .replace(/>/g, "&gt;");

            const extraBtn = media.html
                ? `<a href="${preview}" class="btn">👁 Preview</a>`
                : "";

            return res.send(previewPage(`<pre>${text}</pre>${extraBtn}`, download));
        }

        res.sendFile(abs);

    } catch (e) {
        console.error(e);
        res.status(500).send("Server Error");
    }
});

/* ───────────── UI TEMPLATES ───────────── */
function directoryPage(pathName, list) {
    return `
<!DOCTYPE html>
<html>
<head>
<meta name="viewport" content="width=device-width">
<style>
body{background:#0f172a;color:#e5e7eb;font-family:system-ui;padding:16px}
h2{margin-bottom:10px}
ul{list-style:none;padding:0}
li{background:#020617;margin:8px 0;padding:12px;border-radius:12px}
a{color:#4cff7a;text-decoration:none;font-weight:500}
</style>
</head>
<body>
<h2>📂 ${pathName}</h2>
<ul>${list}</ul>
</body>
</html>`;
}

function previewPage(content, download) {
    return `
<!DOCTYPE html>
<html>
<head>
<meta name="viewport" content="width=device-width">
<style>
body{background:#020617;color:#e5e7eb;font-family:system-ui;padding:16px;text-align:center}
img,video{max-width:100%;border-radius:14px}
audio{width:100%}
pre{background:#020617;padding:14px;border-radius:12px;overflow:auto;text-align:left}
.top{display:flex;gap:10px;justify-content:center;margin-bottom:14px}
.btn{padding:10px 16px;border-radius:14px;background:#4cff7a;color:black;font-weight:700;text-decoration:none}
.back{color:#4cff7a;text-decoration:none;font-weight:600}
</style>
</head>
<body>
<div class="top">
<a class="back" href="javascript:history.back()">⬅ Back</a>
<a class="btn" href="${download}">⬇ Download</a>
</div>
${content}
</body>
</html>`;
}

app.listen(PORT, "0.0.0.0", () => {
    console.log(`✅ Server running at http://localhost:${PORT}`);
});