Hackathon_3_Security/db.js at master · code-untapped/Hackathon_3_Security · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
const { MongoClient, ObjectID } = require('mongodb');
const assert = require('assert');

let db,
    clientConn,
    msg="";

const dbName = 'test2';
const url = 'mongodb://localhost:27017';

//console.log(url);


module.exports = () => {
    return MongoClient
        .connect(url, {
            useNewUrlParser: true
        })
        .then((client) => {
            db = client.db(dbName);
            clientConn = client;
        });
};

module.exports.Article = {


    find(query, collection) {
        return new Promise(function(resolve, reject){
            db.collection(collection)
            .find(query)
            .toArray().then( function(docs) {
                resolve(docs);
                clientConn.close();
            }, function(error) {
                reject(error);
                clientConn.close();
            });
        });
    },

    create(data, collection) {
        return new Promise(function(resolve, reject){
            db.collection(collection).insertOne(data, { w: 1 }, function(err, docs){

                if (err) {
                    reject(Error("create: Unable to create record in collection: " + collection));
                    clientConn.close();
                } else {
                    //console.log(docs);
                    resolve(docs);
                    clientConn.close();
                }
            });
        });
    },


    deleteID(query, collection) {
        query = ObjectID(query);
        return new Promise(function(resolve, reject) {
            db.collection(collection).remove({ "_id" : query }, { w: 1 }, function(err, docs){
                if (err) {
                    reject(Error("deleteID: Unable to delete record in collection: " + collection));
                    clientConn.close();
                } else {
                    //console.log(docs);
                    resolve(docs);
                    clientConn.close();
                }
            });
        });
    },

    delete(query, collection) {
        return new Promise(function(resolve, reject) {
            db.collection(collection).remove(query, { w: 1 }, function(err, docs){
                if (err) {
                    reject(Error("delete: Unable to delete record in collection: " + collection));
                    clientConn.close();
                } else {
                    //console.log(docs);
                    resolve(docs);
                    clientConn.close();
                }
            });
        });
    },
};