mostly working version with latest libraries

1 files changed, 71 insertions, 60 deletions

diff --git a/src/server/database.ts b/src/server/database.ts
index 37bc00a85..0893bfd35 100644
--- a/src/server/database.ts
+++ b/src/server/database.ts
@@ -7,11 +7,14 @@ import { DocumentsCollection, IDatabase } from './IDatabase';
 import { MemoryDatabase } from './MemoryDatabase';
 import { Transferable } from './Message';
 import { Upload } from './SharedMediaTypes';
-import { ObjectId } from 'mongodb';
 
 export namespace Database {
-
     export let disconnect: Function;
+
+    class DocSchema implements mongodb.BSON.Document {
+        _id!: string;
+        id!: string;
+    }
     const schema = 'Dash';
     const port = 27017;
     export const url = `mongodb://localhost:${port}/${schema}`;
@@ -36,11 +39,11 @@ export namespace Database {
                         resolve();
                     });
                     mongoose.connect(url, {
-                        //useNewUrlParser: true, 
+                        //useNewUrlParser: true,
                         dbName: schema,
                         // reconnectTries: Number.MAX_VALUE,
                         // reconnectInterval: 1000,
-                    }); 
+                    });
                 });
             }
         } catch (e) {
@@ -60,10 +63,10 @@ export namespace Database {
         async doConnect() {
             console.error(`\nConnecting to Mongo with URL : ${url}\n`);
             return new Promise<void>(resolve => {
-                this.MongoClient.connect(url, { connectTimeoutMS: 30000, socketTimeoutMS: 30000, }).then(client => {
-                    console.error("mongo connect response\n");
+                this.MongoClient.connect(url, { connectTimeoutMS: 30000, socketTimeoutMS: 30000 }).then(client => {
+                    console.error('mongo connect response\n');
                     if (!client) {
-                        console.error("\nMongo connect failed with the error:\n");
+                        console.error('\nMongo connect failed with the error:\n');
                         process.exit(0);
                     }
                     this.db = client.db();
@@ -75,18 +78,18 @@ export namespace Database {
 
         public async update(id: string, value: any, callback: (err: mongodb.MongoError, res: mongodb.UpdateResult) => void, upsert = true, collectionName = DocumentsCollection) {
             if (this.db) {
-                const collection = this.db.collection(collectionName);
+                const collection = this.db.collection<DocSchema>(collectionName);
                 const prom = this.currentWrites[id];
                 let newProm: Promise<void>;
                 const run = (): Promise<void> => {
                     return new Promise<void>(resolve => {
-                        collection.updateOne({ _id: new ObjectId(id) }, value, { upsert }).then(res => {
-                                if (this.currentWrites[id] === newProm) {
-                                    delete this.currentWrites[id];
-                                }
-                                resolve();
-                                callback(undefined as any, res);
-                            });
+                        collection.updateOne({ _id: id }, value, { upsert }).then(res => {
+                            if (this.currentWrites[id] === newProm) {
+                                delete this.currentWrites[id];
+                            }
+                            resolve();
+                            callback(undefined as any, res);
+                        });
                     });
                 };
                 newProm = prom ? prom.then(run) : run();
@@ -99,18 +102,18 @@ export namespace Database {
 
         public replace(id: string, value: any, callback: (err: mongodb.MongoError, res: mongodb.UpdateResult<mongodb.Document>) => void, upsert = true, collectionName = DocumentsCollection) {
             if (this.db) {
-                const collection = this.db.collection(collectionName);
+                const collection = this.db.collection<DocSchema>(collectionName);
                 const prom = this.currentWrites[id];
                 let newProm: Promise<void>;
                 const run = (): Promise<void> => {
                     return new Promise<void>(resolve => {
-                        collection.replaceOne({ _id: new ObjectId(id)}, value, { upsert }).then( res => {
-                                if (this.currentWrites[id] === newProm) {
-                                    delete this.currentWrites[id];
-                                }
-                                resolve();
-                                callback(undefined as any, res as any);
-                            });
+                        collection.replaceOne({ _id: id }, value, { upsert }).then(res => {
+                            if (this.currentWrites[id] === newProm) {
+                                delete this.currentWrites[id];
+                            }
+                            resolve();
+                            callback(undefined as any, res as any);
+                        });
                     });
                 };
                 newProm = prom ? prom.then(run) : run();
@@ -135,12 +138,17 @@ export namespace Database {
         public delete(query: any, collectionName?: string): Promise<mongodb.DeleteResult>;
         public delete(id: string, collectionName?: string): Promise<mongodb.DeleteResult>;
         public delete(id: any, collectionName = DocumentsCollection) {
-            if (typeof id === "string") {
-                id = { _id: new ObjectId(id) };
+            if (typeof id === 'string') {
+                id = { _id: id };
             }
             if (this.db) {
                 const db = this.db;
-                return new Promise(res => db.collection(collectionName).deleteMany(id).then(result => res(result)));
+                return new Promise(res =>
+                    db
+                        .collection(collectionName)
+                        .deleteMany(id)
+                        .then(result => res(result))
+                );
             } else {
                 return new Promise(res => this.onConnect.push(() => res(this.delete(id, collectionName))));
             }
@@ -167,12 +175,12 @@ export namespace Database {
 
         public async insert(value: any, collectionName = DocumentsCollection) {
             if (this.db && value !== null) {
-                if ("id" in value) {
+                if ('id' in value) {
                     value._id = value.id;
                     delete value.id;
                 }
                 const id = value._id;
-                const collection = this.db.collection(collectionName);
+                const collection = this.db.collection<DocSchema>(collectionName);
                 const prom = this.currentWrites[id];
                 let newProm: Promise<void>;
                 const run = (): Promise<void> => {
@@ -195,11 +203,12 @@ export namespace Database {
 
         public getDocument(id: string, fn: (result?: Transferable) => void, collectionName = DocumentsCollection) {
             if (this.db) {
-                this.db.collection(collectionName).findOne({ _id: new ObjectId(id) }).then(result => {
+                const collection = this.db.collection<DocSchema>(collectionName);
+                collection.findOne({ _id: id }).then(result => {
                     if (result) {
                         result.id = result._id;
                         //delete result._id;
-                        fn(result.id);
+                        fn(result as any);
                     } else {
                         fn(undefined);
                     }
@@ -209,19 +218,19 @@ export namespace Database {
             }
         }
 
-        public getDocuments(ids: string[], fn: (result: Transferable[]) => void, collectionName = DocumentsCollection) {
+        public async getDocuments(ids: string[], fn: (result: Transferable[]) => void, collectionName = DocumentsCollection) {
             if (this.db) {
-                this.db.collection(collectionName).find({ _id: { "$in": ids.map(id => new ObjectId(id)) } }).map(docs => {
-                    // if (err) {
-                    //     console.log(err.message);
-                    //     console.log(err.errmsg);
-                    // }
-                    fn(docs.map((doc:any) => {
+                const found = await this.db
+                    .collection<DocSchema>(collectionName)
+                    .find({ _id: { $in: ids } })
+                    .toArray();
+                fn(
+                    found.map((doc: any) => {
                         doc.id = doc._id;
                         delete doc._id;
                         return doc;
-                    }));
-                });
+                    })
+                );
             } else {
                 this.onConnect.push(() => this.getDocuments(ids, fn, collectionName));
             }
@@ -256,7 +265,7 @@ export namespace Database {
 
         public query(query: { [key: string]: any }, projection?: { [key: string]: 0 | 1 }, collectionName = DocumentsCollection): Promise<mongodb.FindCursor> {
             if (this.db) {
-                let cursor = this.db.collection(collectionName).find(query);
+                let cursor = this.db.collection<DocSchema>(collectionName).find(query);
                 if (projection) {
                     cursor = cursor.project(projection);
                 }
@@ -271,7 +280,12 @@ export namespace Database {
         public updateMany(query: any, update: any, collectionName = DocumentsCollection) {
             if (this.db) {
                 const db = this.db;
-                return new Promise<mongodb.UpdateResult>(res => db.collection(collectionName).updateMany(query, update).then(result => res(result)));
+                return new Promise<mongodb.UpdateResult>(res =>
+                    db
+                        .collection(collectionName)
+                        .updateMany(query, update)
+                        .then(result => res(result))
+                );
             } else {
                 return new Promise<mongodb.UpdateResult>(res => {
                     this.onConnect.push(() => this.updateMany(query, update, collectionName).then(res));
@@ -280,13 +294,13 @@ export namespace Database {
         }
 
         public print() {
-            console.log("db says hi!");
+            console.log('db says hi!');
         }
     }
 
     function getDatabase() {
         switch (process.env.DB) {
-            case "MEM":
+            case 'MEM':
                 return new MemoryDatabase();
             default:
                 return new Database();
@@ -301,13 +315,12 @@ export namespace Database {
      * or Dash-internal user data.
      */
     export namespace Auxiliary {
-
         /**
          * All the auxiliary MongoDB collections (schemas)
          */
         export enum AuxiliaryCollections {
-            GooglePhotosUploadHistory = "uploadedFromGooglePhotos",
-            GoogleAccess = "googleAuthentication",
+            GooglePhotosUploadHistory = 'uploadedFromGooglePhotos',
+            GoogleAccess = 'googleAuthentication',
         }
 
         /**
@@ -319,16 +332,18 @@ export namespace Database {
             const cursor = await Instance.query(query, undefined, collection);
             const results = await cursor.toArray();
             const slice = results.slice(0, Math.min(cap, results.length));
-            return removeId ? slice.map((result:any) => {
-                delete result._id;
-                return result;
-            }) : slice;
+            return removeId
+                ? slice.map((result: any) => {
+                      delete result._id;
+                      return result;
+                  })
+                : slice;
         };
 
         /**
          * Searches for the @param query in the specified @param collection,
          * and returns at most the first result. If @param removeId is true,
-         * as it is by default, each object will be stripped of its database id. 
+         * as it is by default, each object will be stripped of its database id.
          * Worth the special case since it converts the Array return type to a single
          * object of the specified type.
          */
@@ -338,7 +353,7 @@ export namespace Database {
         };
 
         /**
-         * Checks to see if an image with the given @param contentSize 
+         * Checks to see if an image with the given @param contentSize
          * already exists in the aux database, i.e. has already been downloaded from Google Photos.
          */
         export const QueryUploadHistory = async (contentSize: number) => {
@@ -352,7 +367,7 @@ export namespace Database {
         export const LogUpload = async (information: Upload.ImageInformation) => {
             const bundle = {
                 _id: Utils.GenerateDeterministicGuid(String(information.contentSize)),
-                ...information
+                ...information,
             };
             return Instance.insert(bundle, AuxiliaryCollections.GooglePhotosUploadHistory);
         };
@@ -362,7 +377,6 @@ export namespace Database {
          * facilitates interactions with all their APIs for a given account.
          */
         export namespace GoogleAccessToken {
-
             /**
              * Format stored in database.
              */
@@ -370,7 +384,7 @@ export namespace Database {
 
             /**
              * Retrieves the credentials associaed with @param userId
-             * and optionally removes their database id according to @param removeId. 
+             * and optionally removes their database id according to @param removeId.
              */
             export const Fetch = async (userId: string, removeId = true): Promise<Opt<StoredCredentials>> => {
                 return SanitizedSingletonQuery<StoredCredentials>({ userId }, AuxiliaryCollections.GoogleAccess, removeId);
@@ -378,7 +392,7 @@ export namespace Database {
 
             /**
              * Writes the @param enrichedCredentials to the database, associated
-             * with @param userId for later retrieval and updating. 
+             * with @param userId for later retrieval and updating.
              */
             export const Write = async (userId: string, enrichedCredentials: GoogleApiServerUtils.EnrichedCredentials) => {
                 return Instance.insert({ userId, canAccess: [], ...enrichedCredentials }, AuxiliaryCollections.GoogleAccess);
@@ -397,7 +411,7 @@ export namespace Database {
             };
 
             /**
-             * Revokes the credentials associated with @param userId. 
+             * Revokes the credentials associated with @param userId.
              */
             export const Revoke = async (userId: string) => {
                 const entry = await Fetch(userId, false);
@@ -405,9 +419,6 @@ export namespace Database {
                     Instance.delete({ _id: entry._id }, AuxiliaryCollections.GoogleAccess);
                 }
             };
-
         }
-
     }
-
 }