Spaces:
Running
Running
File size: 2,828 Bytes
c237e22 |
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 |
import { Router } from "express";
import { searchTorrents } from "./torrent/search.js";
import {
getFile,
getOrAddTorrent,
getStats,
getTorrentInfo,
streamClosed,
streamOpened,
} from "./torrent/webtorrent.js";
import { getStreamingMimeType } from "./utils/file.js";
export const router = Router();
router.get("/stats", (req, res) => {
const stats = getStats();
res.json(stats);
});
router.get("/torrents/:query", async (req, res) => {
const { query } = req.params;
const torrents = await searchTorrents(query);
res.json(torrents);
});
router.post("/torrents/:query", async (req, res) => {
const { query } = req.params;
const options = req.body;
const torrents = await searchTorrents(query, options);
res.json(torrents);
});
router.get("/torrent/:torrentUri", async (req, res) => {
const { torrentUri } = req.params;
const torrent = await getTorrentInfo(torrentUri);
if (!torrent) return res.status(500).send("Failed to get torrent");
torrent.files.forEach((file) => {
file.url = [
`${req.protocol}://${req.get("host")}`,
"stream",
encodeURIComponent(torrentUri),
encodeURIComponent(file.path),
].join("/");
});
res.json(torrent);
});
router.get("/stream/:torrentUri/:filePath", async (req, res) => {
const { torrentUri, filePath } = req.params;
const torrent = await getOrAddTorrent(torrentUri);
if (!torrent) return res.status(500).send("Failed to add torrent");
const file = getFile(torrent, filePath);
if (!file) return res.status(404).send("File not found");
const { range } = req.headers;
const positions = (range || "").replace(/bytes=/, "").split("-");
const start = Number(positions[0]);
const end = Number(positions[1]) || file.length - 1;
if (start >= file.length || end >= file.length) {
res.writeHead(416, {
"Content-Range": `bytes */${file.length}`,
});
return res.end();
}
const headers = {
"Content-Range": `bytes ${start}-${end}/${file.length}`,
"Accept-Ranges": "bytes",
"Content-Length": end - start + 1,
"Content-Type": getStreamingMimeType(file.name),
};
res.writeHead(206, headers);
try {
const noDataTimeout = setTimeout(() => {
res.status(500).end();
}, 10000);
const noReadTimeout = setTimeout(() => {
res.status(200).end();
}, 60000);
const videoStream = file.createReadStream({ start, end });
videoStream.on("data", () => {
clearTimeout(noDataTimeout);
});
videoStream.on("readable", () => {
noReadTimeout.refresh();
});
videoStream.on("error", (error) => {});
videoStream.pipe(res);
streamOpened(torrent.infoHash, file.name);
res.on("close", () => {
streamClosed(torrent.infoHash, file.name);
});
} catch (error) {
res.status(500).end();
}
});
|