From ac80b2322bfa904b90e79c4d289f48220b9f9c01 Mon Sep 17 00:00:00 2001
From: Ajay Ramachandran <dev@ajay.app>
Date: Fri, 4 Sep 2020 12:16:13 -0400
Subject: [PATCH] Fix broken shadow ban check

---
 src/routes/getSkipSegments.js       | 4 ++--
 src/routes/getSkipSegmentsByHash.js | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/routes/getSkipSegments.js b/src/routes/getSkipSegments.js
index 4bf6a6a..81ea763 100644
--- a/src/routes/getSkipSegments.js
+++ b/src/routes/getSkipSegments.js
@@ -8,7 +8,7 @@ var logger = require('../utils/logger.js');
 var getHash = require('../utils/getHash.js');
 var getIP = require('../utils/getIP.js');
 
-function cleanGetSegments(videoID, categories) {
+function cleanGetSegments(req, videoID, categories) {
     let userHashedIP, shadowHiddenSegments;
 
     let segments = [];
@@ -160,7 +160,7 @@ function handleGetSegments(req, res) {
         ? [req.query.category]
         : ['sponsor'];
 
-    let segments = cleanGetSegments(videoID, categories);
+    let segments = cleanGetSegments(req, videoID, categories);
 
     if (segments === null || segments === undefined) {
         res.sendStatus(500);
diff --git a/src/routes/getSkipSegmentsByHash.js b/src/routes/getSkipSegmentsByHash.js
index 702a28d..23a9167 100644
--- a/src/routes/getSkipSegmentsByHash.js
+++ b/src/routes/getSkipSegmentsByHash.js
@@ -29,7 +29,7 @@ module.exports = async function (req, res) {
         return {
             videoID: video.videoID,
             hash: video.hashedVideoID,
-            segments: getSegments(video.videoID, categories)
+            segments: getSegments(req, video.videoID, categories)
         };
     });