Merge pull request #8 from browngraphicslab/server_database_merge

Server database merge

1 files changed, 81 insertions, 0 deletions

diff --git a/src/server/database.ts b/src/server/database.ts
new file mode 100644
index 000000000..07c5819ab
--- /dev/null
+++ b/src/server/database.ts
@@ -0,0 +1,81 @@
+import { action, configure } from 'mobx';
+import * as mongodb from 'mongodb';
+import { ObjectID } from 'mongodb';
+import { Transferable } from './Message';
+import { Utils } from '../Utils';
+
+export class Database {
+    public static Instance = new Database()
+    private MongoClient = mongodb.MongoClient;
+    private url = 'mongodb://localhost:27017/Dash';
+    private db?: mongodb.Db;
+
+    constructor() {
+        this.MongoClient.connect(this.url, (err, client) => {
+            this.db = client.db()
+        })
+    }
+
+    public update(id: string, value: any) {
+        if (this.db) {
+            let collection = this.db.collection('documents');
+            collection.update({ _id: id }, { $set: value }, {
+                upsert: true
+            });
+        }
+    }
+
+    public delete(id: string) {
+        if (this.db) {
+            let collection = this.db.collection('documents');
+            collection.remove({ _id: id });
+        }
+    }
+
+    public deleteAll() {
+        if (this.db) {
+            let collection = this.db.collection('documents');
+            collection.deleteMany({});
+        }
+    }
+
+    public insert(kvpairs: any) {
+        if (this.db) {
+            let collection = this.db.collection('documents');
+            collection.insertOne(kvpairs, (err: any, res: any) => {
+                if (err) {
+                    // console.log(err)
+                    return
+                }
+            });
+        }
+    }
+
+    public getDocument(id: string, fn: (res: any) => void) {
+        var result: JSON;
+        if (this.db) {
+            let collection = this.db.collection('documents');
+            collection.findOne({ _id: id }, (err: any, res: any) => {
+                result = res
+                if (!result) {
+                    fn(undefined)
+                }
+                fn(result)
+            })
+        };
+    }
+
+    public getDocuments(ids: string[], fn: (res: any) => void) {
+        if (this.db) {
+            let collection = this.db.collection('documents');
+            let cursor = collection.find({ _id: { "$in": ids } })
+            cursor.toArray((err, docs) => {
+                fn(docs);
+            })
+        };
+    }
+
+    public print() {
+        console.log("db says hi!")
+    }
+}