brock/torrentio
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

rate limit only stream requests

Browse Source
This commit is contained in:
TheBeastLT
2022-03-23 09:41:06 +01:00
parent f864f95ca0
commit 137f56bfd1
2 changed files with 9 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
addon/index.js
View File

@@ -1,10 +1,19 @@
const express = require('express'); const express = require('express');
const rateLimit = require("express-rate-limit");
const requestIp = require("request-ip");
const serverless = require('./serverless'); const serverless = require('./serverless');
const { initBestTrackers } = require('./lib/magnetHelper'); const { initBestTrackers } = require('./lib/magnetHelper');
const app = express(); const app = express();
const limiter = rateLimit({
windowMs: 60 * 60 * 1000, // 1 hours
max: 300, // limit each IP to 300 requests per windowMs
headers: false,
keyGenerator: (req) => requestIp.getClientIp(req)
});
app.use(express.static('static', { maxAge: '1y' })); app.use(express.static('static', { maxAge: '1y' }));
app.use(/^\/.*stream\/.+/, limiter);
app.use((req, res, next) => serverless(req, res, next)); app.use((req, res, next) => serverless(req, res, next));
app.listen(process.env.PORT || 7000, () => { app.listen(process.env.PORT || 7000, () => {
initBestTrackers() initBestTrackers()

9
addon/serverless.js
View File

@@ -1,4 +1,3 @@
const rateLimit = require('express-rate-limit');
const { getRouter } = require('stremio-addon-sdk'); const { getRouter } = require('stremio-addon-sdk');
const requestIp = require('request-ip'); const requestIp = require('request-ip');
const userAgentParser = require('ua-parser-js'); const userAgentParser = require('ua-parser-js');
@@ -10,14 +9,6 @@ const landingTemplate = require('./lib/landingTemplate');
const moch = require('./moch/moch'); const moch = require('./moch/moch');
const router = getRouter({ ...addonInterface, manifest: manifest() }); const router = getRouter({ ...addonInterface, manifest: manifest() });
const limiter = rateLimit({
windowMs: 60 * 60 * 1000, // 1 hours
max: 300, // limit each IP to 300 requests per windowMs
headers: false,
keyGenerator: (req) => requestIp.getClientIp(req)
});
router.use(limiter);
router.get('/', (_, res) => { router.get('/', (_, res) => {
res.redirect('/configure') res.redirect('/configure')