Database

.gitignore

@@ -59,3 +59,6 @@ typings/
 
 # next.js build output
 .next
+
+# Config files
+config.json

config/config.template.json

@@ -0,0 +1,12 @@
+{
+    "DATABASE": {
+        "USERNAME": "",
+        "DATABASE": "",
+        "PASSWORD": "",
+        "HOST": "",
+        "PORT": 3306
+    },
+    "CACHING": {
+        "DURATION": 15
+    }
+}

database/Tools/SQL/table.sql

@@ -0,0 +1,22 @@
+SET SQL_MODE = "NO_AUTO_VALUE_ON_ZERO";
+SET AUTOCOMMIT = 0;
+START TRANSACTION;
+SET time_zone = "+00:00";
+
+CREATE DATABASE IF NOT EXISTS `la1tv` DEFAULT CHARACTER SET utf8 COLLATE utf8_swedish_ci;
+USE `la1tv`;
+
+CREATE TABLE IF NOT EXISTS `api_keys` (
+  `ID` int(11) NOT NULL AUTO_INCREMENT,
+  `api_key` varchar(32) COLLATE utf8_swedish_ci NOT NULL,
+  `enabled` tinyint(1) NOT NULL,
+  `view_vod_uri` tinyint(1) NOT NULL,
+  `view_stream_uri` tinyint(1) NOT NULL,
+  `use_webhook` tinyint(1) NOT NULL,
+  `created_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `updated_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  `last_request_at` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP,
+  PRIMARY KEY (`ID`),
+  UNIQUE KEY `api_key` (`api_key`) USING BTREE
+) ENGINE=InnoDB DEFAULT CHARSET=utf8 COLLATE=utf8_swedish_ci;
+COMMIT;

database/api-key/create.js

@@ -0,0 +1,21 @@
+const mysql = require('../connect')
+const uuid = require('uuid/v4');
+
+// > Creates a new key based on parameters passed
+// > Returns the key that was created
+const createKey = ({ enabled = 1, view_vod_uri = 0, view_stream_uri = 0, use_webhook = 0 }) => {
+    return new Promise((resolve, reject) => {
+        const id = uuid().toString().replace(/-/g, '')
+        mysql.promise().execute('INSERT INTO `api_keys` (`ID`, `api_key`, `enabled`, `view_vod_uri`, `view_stream_uri`, `use_webhook`) VALUES (NULL, ?, ?, ?, ?, ?)', [id, enabled, view_vod_uri, view_stream_uri, use_webhook])
+            .then(([fields, rows]) => {
+                resolve(id)
+            })
+            .catch((err) => {
+                reject(err)
+            })
+    })
+}
+
+module.exports = {
+    createKey
+}

database/api-key/delete.js

@@ -0,0 +1,18 @@
+const mysql = require('../connect')
+
+// > Deletes an API Key
+const deleteKey = (api) => {
+    return new Promise((resolve, reject) => {
+        mysql.promise().execute('DELETE FROM `api_keys` WHERE api_key = ?;', [api])
+            .then(([fields, rows]) => {
+                resolve("success")
+            })
+            .catch(err => {
+                reject(err)
+            })
+    })
+}
+
+module.exports = {
+    deleteKey
+}

database/api-key/get.js

@@ -0,0 +1,38 @@
+const mysql = require('../connect')
+
+// > Returns a specified key
+const getSpecifiedKey = (api) => {
+    return new Promise((resolve, reject) => {
+        mysql.promise().execute('SELECT * FROM `api_keys` WHERE `api_key` = ?', [api])
+            .then(([rows, fields]) => {
+                resolve(JSON.stringify(rows[0]))
+            })
+            .catch(err => {
+                reject(err)
+            })
+    })
+}
+
+// > Returns all API keys
+const getAllKeys = ({ start = -1, quantity = 10 }) => {
+    var tempSQL = ""
+
+    if (start !== -1) {
+        tempSQL = " WHERE ID >= " + start + " LIMIT " + quantity
+    }
+
+    return new Promise((resolve, reject) => {
+        mysql.promise().execute('SELECT * FROM `api_keys`' + tempSQL + ';')
+            .then(([rows, fields]) => {
+                resolve(rows)
+            })
+            .catch(err => {
+                reject(err)
+            })
+    })
+}
+
+module.exports = {
+    getSpecifiedKey,
+    getAllKeys
+}

database/api-key/index.js

@@ -0,0 +1,17 @@
+const { getAllKeys, getSpecifiedKey } = require('./get')
+const { createKey } = require('./create')
+const { updateKey } = require('./update')
+const { deleteKey } = require('./delete')
+
+/*
+    > This file allows us to import all or just what we need from the SQL Queries for api-keys
+    > May add authorisation to this route but I may not
+*/
+
+module.exports = {
+    getAllKeys,
+    getSpecifiedKey,
+    createKey,
+    updateKey,
+    deleteKey
+}

database/api-key/update.js

@@ -0,0 +1,40 @@
+const mysql = require('../connect')
+
+const updateKey = (api, { enabled, view_vod_uri, view_stream_uri, use_webhook }) => {
+    return new Promise((resolve, reject) => {
+        let enabledSQL = enabled ? ('`enabled` = ' + enabled) : ('')
+        let view_vod_uri_SQL = view_vod_uri ? ('`view_vod_uri` = ' + view_vod_uri) : ('')
+        let view_stream_uri_SQL = view_stream_uri ? ('`view_stream_uri` = ' + view_stream_uri) : ('')
+        let use_webhook_SQL = use_webhook ? ('`use_webhook` = ' + use_webhook) : ('')
+
+        mysql.promise().execute('UPDATE api_keys SET ' + enabledSQL +
+                ' ' + view_vod_uri_SQL +
+                ' ' + view_stream_uri_SQL +
+                ' ' + use_webhook_SQL +
+                ' WHERE `api_key` = ?;', [api]).then(([fields, rows]) => {
+                resolve("success")
+            })
+            .catch(err => {
+                reject(err)
+            })
+    })
+}
+
+// > Updates the timestamp for the API key
+// > Is mainly used in the middleware RequireApi.js
+const updateTimeStamp = (api) => {
+    return new Promise((resolve, reject) => {
+        mysql.promise().execute('UPDATE api_keys SET last_request_at = CURRENT_TIMESTAMP WHERE `api_key` = ?;', [api])
+            .then(([fields, rows]) => {
+                resolve("success")
+            })
+            .catch(err => {
+                reject(err)
+            })
+    })
+}
+
+module.exports = {
+    updateKey,
+    updateTimeStamp
+}

database/builder.js

@@ -0,0 +1,14 @@
+const mysql = require("./connect")
+
+// > TODO: Generate database based off of SQL in table.
+
+function generateDatabase() {
+    // mysql.query(
+    //     "CREATE SCHEMA IF NOT EXISTS ${process.env.DATABASE};" +
+    //     "CREATE TABLE `la1tv`.`APIKEys` ( `ID` INT NOT NULL AUTO_INCREMENT , `Key` VARCHAR(32) NOT NULL , PRIMARY KEY (`ID`), UNIQUE (`Key`)) ENGINE = InnoDB;"
+    // )
+
+    console.log("In development")
+}
+
+module.exports = generateDatabase

database/connect.js

@@ -1,10 +1,16 @@
 const mysql = require('mysql2');
+const { DATABASE } = require('../config/config.json')
+
+// > TODO: Test the connection
+// > TODO: Create database creation scripts
+// > TODO: Create seeders
 
-// Create the connection pool. The pool-specific settings are the defaults
 const pool = mysql.createPool({
-    host: process.env.DATABASE_HOST,
-    user: process.env.DATABASE_USERNAME,
-    database: process.env.DATABASE_PASSWORD,
+    host: DATABASE.HOST | '127.0.0.1',
+    user: DATABASE.USERNAME,
+    password: DATABASE.PASSWORD,
+    database: DATABASE.DATABASE,
+    port: DATABASE.PORT | 3306,
     waitForConnections: true,
     connectionLimit: 10,
     queueLimit: 0

middleware/RequireApi.js

@@ -1,14 +1,34 @@
-const mysql = require('../database/connect')
+const { getSpecifiedKey } = require('../database/api-key/get')
+const { updateTimeStamp } = require('../database/api-key/update')
+
+// > TODO: Implement checking from database for valid API keys
+// > TODO: Add header to incoming request that filters for adding and / or removing
+// > TODO: Send a more suitable response
 
 const requireAPI = (req, res, next) => {
     if (!req.get("x-api-key")) {
-        console.log("No X-API-KEY sent");
-        res.send("Invalid or No API Key provided");
+        res.status(401).send("Invalid or No API Key provided");
 
         return;
     }
 
-    next();
+    getSpecifiedKey(req.get("x-api-key")).then((result) => {
+            if (JSON.parse(result).enabled === 1) {
+                updateTimeStamp(req.get("x-api-key")).then(data => {
+                    next()
+                }).catch(err => {
+                    console.log(err)
+                    res.status(401).send("Unable to update timestamp of API key provided")
+                    return
+                })
+            } else {
+                res.status(401).send("Invalid or No API Key provided")
+                return
+            }
+        })
+        .catch(err => {
+            res.status(500).send("Error retrieving API-Key information from database")
+        })
 }
 
 module.exports = requireAPI

middleware/cache.js

@@ -0,0 +1,28 @@
+const mcache = require('memory-cache')
+const { CACHING } = require('../config/config.json')
+
+const cache = (duration = CACHING.DURATION) => {
+    return (req, res, next) => {
+        if (req.type !== 'GET') {
+            next()
+            return
+        }
+
+        let key = '__express__' + req.originalUrl || req.url
+        let cachedObject = mcache.get(key)
+
+        if (cachedObject) {
+            res.status(200).send(cachedObject)
+            return
+        } else {
+            res.sendResponse = res.send
+            res.send = (body) => {
+                mcache.put(key, body, duration * 1000);
+                res.sendResponse(body)
+            }
+            next()
+        }
+    }
+}
+
+module.exports = cache