From 7cfcb54195eac7ba35b83b8679b61e36d572eb9b Mon Sep 17 00:00:00 2001
From: Nicolai Ort <info@nicolai-ort.com>
Date: Thu, 12 Aug 2021 20:24:26 +0200
Subject: [PATCH] Added config object

---
 src/server.js | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/server.js b/src/server.js
index 300c46f..3765416 100644
--- a/src/server.js
+++ b/src/server.js
@@ -1,6 +1,17 @@
 const fastify = require('fastify')({ logger: true })
 var uniqid = require('uniqid');
 
+let config = {
+    domain: process.env.DOMAIN || "localhost:3000",
+    https: false,
+    getBaseUrl(){
+        if(config.https){
+            return `https://${config.domain}`;
+        }
+        return `http://${config.domain}`;
+    }
+}
+
 const knex = require('knex')({
     client: 'sqlite3',
     connection: {
@@ -66,7 +77,7 @@ fastify.post('/new', { newUrlSchema }, async (req, res) => {
         if (exists.length != 0) {
             shortcode = exists[0].shortcode;
             return {
-                url: `http://localhost:3000/${shortcode}`,
+                url: `${config.getBaseUrl()}/${shortcode}`,
                 shortcode,
                 target
             }
@@ -86,7 +97,7 @@ fastify.post('/new', { newUrlSchema }, async (req, res) => {
     await knex('urls').insert({target, shortcode});
 
     return {
-        url: `http://localhost:3000${shortcode}`,
+        url: `${config.getBaseUrl()}/${shortcode}`,
         shortcode,
         target
     }