From 75adbf73cfbf4fadf2cd632ecb687eff95749f52 Mon Sep 17 00:00:00 2001
From: Nicolai Ort <info@nicolai-ort.com>
Date: Sat, 25 Sep 2021 17:27:41 +0200
Subject: [PATCH] Now with working bad bot detection ref #2

---
 src/server.js | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/src/server.js b/src/server.js
index 59b2e7e..7011f95 100644
--- a/src/server.js
+++ b/src/server.js
@@ -112,6 +112,7 @@ fastify.get('/:shortcode', async (req, res) => {
         return 404
     }
 
+    console.log({ua: req.headers['user-agent'], bot: isBot(req.headers['user-agent']), prev: target[0].no_preview})
     if(isBot(req.headers['user-agent']) && target[0].no_preview){
         return "Bad Bot!"
     }
@@ -156,7 +157,7 @@ fastify.post('/api', { newUrlSchema }, async (req, res) => {
                 return response;
             }
         }
-        const exists = await knex.select('shortcode')
+        const exists = await knex.select('shortcode', 'no_preview')
             .from('urls')
             .where('target', '=', target)
             .limit(1);
@@ -165,7 +166,8 @@ fastify.post('/api', { newUrlSchema }, async (req, res) => {
             return {
                 url: `${config.getBaseUrl()}/${shortcode}`,
                 shortcode,
-                target
+                target,
+                no_preview: exists[0].no_preview
             }
         }
         shortcode = uniqid();
@@ -187,7 +189,7 @@ fastify.post('/api', { newUrlSchema }, async (req, res) => {
     }
 
     //Create a new db entry
-    await knex('urls').insert({ target, shortcode });
+    await knex('urls').insert({ target, shortcode, no_preview });
 
     return {
         url: `${config.getBaseUrl()}/${shortcode}`,