From 15508606a41057857cce0d2ae16e389432c76960 Mon Sep 17 00:00:00 2001
From: Nicolai Ort <info@nicolai-ort.com>
Date: Thu, 12 Aug 2021 19:33:59 +0200
Subject: [PATCH] Implemented short code generation via nanoid

---
 package.json  |  1 +
 src/server.js | 41 ++++++++++++++++++++++++++++++++++++++---
 2 files changed, 39 insertions(+), 3 deletions(-)

diff --git a/package.json b/package.json
index 2c74f66..8f22606 100644
--- a/package.json
+++ b/package.json
@@ -12,6 +12,7 @@
     "fastify": "^3.20.1",
     "knex": "^0.21.21",
     "level": "^7.0.0",
+    "nanoid": "^3.1.25",
     "sqlite3": "^5.0.2"
   },
   "devDependencies": {
diff --git a/src/server.js b/src/server.js
index 82b5052..a4da015 100644
--- a/src/server.js
+++ b/src/server.js
@@ -1,9 +1,10 @@
 const fastify = require('fastify')({ logger: true })
 const level = require('level')
+const { nanoid } = require('nanoid')
 
 const db = level('./db', { valueEncoding: 'json' })
 
-// Declare a route
+//Basic home route
 fastify.get('/', async (request, reply) => {
     return { hello: 'world' }
 })
@@ -18,6 +19,7 @@ fastify.get('/yt/:id', async (req, res) => {
     res.redirect(302, `https://youtu.be/${req.params.id}`)
 })
 
+//Normal shorturls
 fastify.get('/:id', async (req, res) => {
     try {
         const target = await db.get(req.params.id);
@@ -28,9 +30,42 @@ fastify.get('/:id', async (req, res) => {
     }
 })
 
+//Create new urls
 fastify.post('/new', async (req, res) => {
-    console.log(req.body);
-    return "";
+    const target = req.body?.target;
+    let shortcode = req.body?.code;
+    if (!req.body?.target) {
+        res.statusCode = 400;
+        return "Missing target";
+    }
+    if (!shortcode) {
+        shortcode = nanoid(12);
+    }
+    else{
+        try {
+            const exists = await db.get(shortcode);
+            if(exists){
+                res.statusCode = 400;
+                return "Shortcode already exists, please choose another code";
+            }
+        } catch (error) {
+            if(!error.notFound){
+                res.statusCode=500;
+                return "Internal Server Error."
+            }
+        }
+    }
+    try {
+        await db.put(shortcode, target);
+    } catch (error) {
+        res.statusCode = 500;
+        return "DB ERROR";
+    }
+    return {
+        shortcode,
+        url: `http://localhost:3000/${shortcode}`,
+        target
+    };
 })
 
 // Run the server!