From 06a5b7e561058582def53d4b54a665d3dd6941b3 Mon Sep 17 00:00:00 2001
From: jaketothepast <windlejacob12@gmail.com>
Date: Sun, 27 Jan 2019 09:06:39 -0500
Subject: [PATCH] Initial functions for dealing with the database

---
 index.js | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/index.js b/index.js
index 5550af8..03314bf 100644
--- a/index.js
+++ b/index.js
@@ -1,5 +1,8 @@
 var http = require('http');
 var winston = require("winston");
+var fs = require("fs");
+var sqlite3 = require('sqlite3').verbose();
+var db = new sqlite3.Database("./proxy.db");
 
 // Rudimentary logger.
 // TODO - Make this filename configurable.
@@ -51,4 +54,29 @@ function startServer() {
     return server;
 }
 
+/**
+ * Initialize our database.
+ */
+function initializeDatabase() {
+    db.serialize(() => {
+        db.run("create table hosts (hostname TEXT visitcount INTEGER)", (res, err) => {
+            if (err !== undefined || err !== null) {
+                console.log("Error creating database, it likely exists already.");
+            }
+        });
+    });
+    db.close();
+}
+
+/**
+ * Update the visit count for a host logged by the proxy
+ * @param {string} hostname Host logged by proxy
+ */
+function updateVisitCount(hostname) {
+    db.serialize(() => {
+        db.run("update hosts set visitcount = visitcount + 1 where hostname = (?)", [hostname]);
+    });
+}
+
+initializeDatabase();
 startServer();