use proper session store

Author: devksingh4

.gitignore

@@ -103,4 +103,6 @@ dist
 # TernJS port file
 .tern-port
 
-keycloak.json
+keycloak.json
+sessions.sqlite3
+db.sqlite3

db.sqlite3

@@ -0,0 +1,14 @@
+services:
+  shorten:
+    build: .
+    restart: on-failure
+    environment:
+      - DB_VENDOR=sqlite
+      - DB_FILE=/data/db.sqlite3
+      - SESSION_DB_FILE_LOC=/data/
+      - SESSION_DB_FILE_NAME=sessions.sqlite3
+    ports:
+      - "9215:9215"
+    volumes:
+      - ${PWD}/db.sqlite3:/data/db.sqlite3
+      - ${PWD}/sessions.sqlite3/:/data/sessions.sqlite3

docker-compose.yml

@@ -2,17 +2,24 @@ const Keycloak = require('keycloak-connect');
 const hogan = require('hogan-express');
 const express = require('express');
 const session = require('express-session');
+const SQLiteStore = require('connect-sqlite3')(session);
 const fetch = require('node-fetch');
 const baseURL = process.env.baseURL || 'out.epochml.org';
 const favicon = require('serve-favicon');
 const sqlite3 = require('sqlite3')
-
-const db = new sqlite3.Database('./db.sqlite3', sqlite3.OPEN_READWRITE, (err) => {
-  if (err) {
-    console.error(err.message);
-    process.exit(1);
-  }
-})
+const dbVendor = process.env.DB_VENDOR;
+if (dbVendor === "postgresql") {
+  console.error("SQL support not yet implemented")
+  process.exit(2)
+} else {
+  var db = new sqlite3.Database(process.env.DB_FILE, sqlite3.OPEN_READWRITE, (err) => {
+    if (err) {
+      console.error(err.message);
+      process.exit(1);
+    }
+  })
+  var store = new SQLiteStore({'dir': process.env.SESSION_DB_FILE_LOC, 'db': process.env.SESSION_DB_FILE_NAME});
+}
 
 var app = express();
 var server = app.listen(9215, function () {
@@ -29,21 +36,21 @@ app.engine('html', hogan);
 // Create a session-store to be used by both the express-session
 // middleware and the keycloak middleware.
 
-var memoryStore = new session.MemoryStore();
-
 app.use(session({
   secret: 'mySecret',
   resave: false,
   saveUninitialized: true,
-  store: memoryStore
+  store: store
 }));
 
-app.use(favicon(__dirname + '/public/img/favicon.ico'));
-
 var keycloak = new Keycloak({
-  store: memoryStore
+  store: store
 });
 
+app.use(favicon(__dirname + '/public/img/favicon.ico'));
+
+
+
 async function getUserInfo(bearer_token) {
   const myHeaders = {
     'Content-Type': 'application/json',

index.js

@@ -4,13 +4,15 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "start": "forever start index.js",
+    "start": "SESSION_DB_FILE_LOC=${PWD} SESSION_DB_FILE_NAME=sessions.sqlite3 DB_VENDOR=sqlite DB_FILE=${PWD}/db.sqlite3 forever start index.js",
     "stop": "forever stop index.js",
-    "dev": "nodemon index.js"
+    "dev": "SESSION_DB_FILE_LOC=${PWD} SESSION_DB_FILE_NAME=sessions.sqlite3 DB_VENDOR=sqlite DB_FILE=${PWD}/db.sqlite3 nodemon index.js",
+    "container-start": "node index.js"
   },
   "author": "Dev Singh <[email protected]>",
   "license": "BSD-3-Clause",
   "dependencies": {
+    "connect-sqlite3": "^0.9.11",
     "express": "^4.17.1",
     "express-session": "^1.17.1",
     "forever": "^3.0.4",