add auth tokens as get params

seveibar · Oct 22, 2020 · dc9608d · dc9608d
1 parent e2699e4
commit dc9608d
Show file tree

Hide file tree

Showing 5 changed files with 1,668 additions and 9 deletions.
diff --git a/api/audio.js b/api/audio.js
@@ -15,6 +15,14 @@ const redisClient = redis.createClient(
 module.exports = async (req, res) => {
   const params = query(req)
 
+  if (process.env.GET_AUTH_TOKEN) {
+    if (process.env.GET_AUTH_TOKEN !== params.token) {
+      micro.send(res, 401, "you do not have auth header")
+    } else if (!params.token) {
+      micro.send(res, 401, "you need to supply auth token")
+    }
+  }
+
   if (!params.video_url) {
     micro.send(res, 400, 'You forgot the "video_url", you idiot!')
     return

diff --git a/api/list-cache.js b/api/list-cache.js
@@ -5,6 +5,16 @@ const redisClient = redis.createClient(
 )
 
 module.exports = async (req, res) => {
+  const params = query(req)
+
+  if (
+    process.env.GET_AUTH_TOKEN &&
+    process.env.GET_AUTH_TOKEN !== params.token
+  ) {
+    micro.send(res, 401, "you do not have auth header")
+    return
+  }
+
   const allKeys = await new Promise((resolve) => {
     redisClient.keys("*", (err, keys) => {
       if (err) {

diff --git a/api/video.js b/api/video.js
@@ -16,6 +16,14 @@ module.exports = async (req, res) => {
   const params = query(req)
   const qualityLabel = params.quality || "480p"
 
+  if (
+    process.env.GET_AUTH_TOKEN &&
+    process.env.GET_AUTH_TOKEN !== params.token
+  ) {
+    micro.send(res, 401, "you do not have auth header")
+    return
+  }
+
   if (!params.video_url) {
     micro.send(res, 400, 'You forgot the "video_url", you idiot!')
     return

diff --git a/package.json b/package.json
@@ -9,11 +9,14 @@
     "start:prod": "micro app.js"
   },
   "dependencies": {
+    "ava": "^3.13.0",
+    "axios": "^0.20.0",
     "micro": "^9.3.4",
     "micro-dev": "^3.0.0",
     "micro-query": "^0.3.0",
     "redis": "^3.0.2",
     "rimraf": "^3.0.2",
+    "test-listen": "^1.1.0",
     "tmp": "^0.2.1",
     "ytdl-core": "^3.4.1"
   }