Merge pull request #7 from arangoml/expiry

Stores optional Email and expires and drops dbs when 2 weeks old

Author: rajivsam

index.js

@@ -6,13 +6,16 @@ const users = require("@arangodb/users");
 const router = createRouter();
 const joi = require('joi');
 
+const aisisInstances = "aisisInstances"
+
 module.context.use(router);
 
 router.post('/createDB', function (req,res) {
   const data = req.body;
   const dbName = data.dbName ? data.dbName : randomStringGenerator();
   const username = data.username ? data.username : randomStringGenerator();
   const password = data.password ? data.password : randomStringGenerator();
+  const email = data.email ? data.email : 'noreply@arangodb.com'
 
 
 // If user doesn't exist, create the user
@@ -29,12 +32,27 @@ router.post('/createDB', function (req,res) {
       res.send("Database already exists or invalid name supplied, please supply new dbName.").status(400);
     }
 
+
     users.save(username, password, true);
+    // Grants user access only to newly created database
+    users.grantDatabase(username, dbName, 'rw');
+
+
+
+    // @TODO: obtain hostname and port of external deployment if necessary.
+    // Returns hostname and port
     let hostname = req.hostname
     let port = req.port
 
-    // Grants user access only to newly created database
-    users.grantDatabase(username, dbName, 'rw');
+    let insertDoc = query`INSERT {
+      "dbName": ${dbName},
+      "username": ${username},
+      "hostname": ${hostname},
+      "port": ${port},
+      "email": ${email},
+      "timestamp": DATE_NOW()
+    } INTO aisisInstances`
+
     res.send({dbName, username, password, hostname, port});
   }
 })

manifest.json

@@ -4,6 +4,7 @@
   },
   "main": "index.js",
   "scripts": {
-    "setup": "scripts/setup.js"
+    "setup": "scripts/setup.js",
+    "expire":"scripts/expire.js"
   }
 }

scripts/expire.js

@@ -0,0 +1,25 @@
+'use strict';
+const { aql, query, db } = require("@arangodb");
+const users = require("@arangodb/users");
+
+const expirationTime = (336 * 60 * 60 * 1000)
+const expired = []
+
+let dbs = query`
+  FOR i IN aisisInstances
+  RETURN {key: i._key, dbName: i.dbName, timestamp: i.timestamp}`
+
+dbs.toArray().map((d) => {
+  (Date.now() - expirationTime ) > d.timestamp ? removeDatabase(d.dbName, d.key) : ''
+})
+
+query`
+FOR key IN ${expired}
+REMOVE { _key: key } IN aisisInstances
+`
+
+function removeDatabase(dbName, key) {
+  console.log(db._engineStats())
+  db._dropDatabase(dbName)
+  expired.push(key)
+}

scripts/setup.js

@@ -1 +1,20 @@
+'use strict';
+const db = require('@arangodb').db;
+const collectionName = 'aisisInstances';
 
+if (!db._collection(collectionName)) {
+  db._createDocumentCollection(collectionName);
+}
+const queues = require('@arangodb/foxx/queues');
+const queue = queues.create('expirationQueue');
+
+// Creates the expire job to check if databases should be expired and dropped.
+queue.push(
+  {mount: '/createDB', name: 'expire'},
+  {},
+  {
+    repeatTimes: Infinity,
+    repeatUntil: -1, // Forever
+    repeatDelay: (24 * 60 * 60 * 1000) // Daily
+  }
+);