1 files changed, 12 insertions, 25 deletions

diff --git a/src/server/Websocket/Websocket.ts b/src/server/Websocket/Websocket.ts
index be895c4bc..844535056 100644
--- a/src/server/Websocket/Websocket.ts
+++ b/src/server/Websocket/Websocket.ts
@@ -12,6 +12,7 @@ import { timeMap } from "../ApiManagers/UserManager";
 import { green } from "colors";
 import { networkInterfaces } from "os";
 import executeImport from "../../scraping/buxton/final/BuxtonImporter";
+import { DocumentsCollection } from "../IDatabase";
 
 export namespace WebSocket {
 
@@ -96,7 +97,7 @@ export namespace WebSocket {
             Utils.AddServerHandlerCallback(socket, MessageStore.GetField, getField);
             Utils.AddServerHandlerCallback(socket, MessageStore.GetFields, getFields);
             if (isRelease) {
-                Utils.AddServerHandler(socket, MessageStore.DeleteAll, deleteFields);
+                Utils.AddServerHandler(socket, MessageStore.DeleteAll, () => doDelete(false));
             }
 
             Utils.AddServerHandler(socket, MessageStore.CreateField, CreateField);
@@ -163,24 +164,10 @@ export namespace WebSocket {
         }
     }
 
-    export async function deleteFields() {
-        await Database.Instance.deleteAll();
-        if (process.env.DISABLE_SEARCH !== "true") {
-            await Search.clear();
-        }
-        await Database.Instance.deleteAll('newDocuments');
-    }
-
-    // export async function deleteUserDocuments() {
-    //     await Database.Instance.deleteAll();
-    //     await Database.Instance.deleteAll('newDocuments');
-    // }
-
-    export async function deleteAll() {
-        await Database.Instance.deleteAll();
-        await Database.Instance.deleteAll('newDocuments');
-        await Database.Instance.deleteAll('sessions');
-        await Database.Instance.deleteAll('users');
+    export async function doDelete(onlyFields = true) {
+        const target: string[] = [];
+        onlyFields && target.push(DocumentsCollection);
+        await Database.Instance.dropSchema(...target);
         if (process.env.DISABLE_SEARCH !== "true") {
             await Search.clear();
         }
@@ -210,11 +197,11 @@ export namespace WebSocket {
     }
 
     function GetRefField([id, callback]: [string, (result?: Transferable) => void]) {
-        Database.Instance.getDocument(id, callback, "newDocuments");
+        Database.Instance.getDocument(id, callback);
     }
 
     function GetRefFields([ids, callback]: [string[], (result?: Transferable[]) => void]) {
-        Database.Instance.getDocuments(ids, callback, "newDocuments");
+        Database.Instance.getDocuments(ids, callback);
     }
 
     const suffixMap: { [type: string]: (string | [string, string | ((json: any) => any)]) } = {
@@ -271,7 +258,7 @@ export namespace WebSocket {
 
     function UpdateField(socket: Socket, diff: Diff) {
         Database.Instance.update(diff.id, diff.diff,
-            () => socket.broadcast.emit(MessageStore.UpdateField.Message, diff), false, "newDocuments");
+            () => socket.broadcast.emit(MessageStore.UpdateField.Message, diff), false);
         const docfield = diff.diff.$set || diff.diff.$unset;
         if (!docfield) {
             return;
@@ -296,7 +283,7 @@ export namespace WebSocket {
     }
 
     function DeleteField(socket: Socket, id: string) {
-        Database.Instance.delete({ _id: id }, "newDocuments").then(() => {
+        Database.Instance.delete({ _id: id }).then(() => {
             socket.broadcast.emit(MessageStore.DeleteField.Message, id);
         });
 
@@ -304,14 +291,14 @@ export namespace WebSocket {
     }
 
     function DeleteFields(socket: Socket, ids: string[]) {
-        Database.Instance.delete({ _id: { $in: ids } }, "newDocuments").then(() => {
+        Database.Instance.delete({ _id: { $in: ids } }).then(() => {
             socket.broadcast.emit(MessageStore.DeleteFields.Message, ids);
         });
         Search.deleteDocuments(ids);
     }
 
     function CreateField(newValue: any) {
-        Database.Instance.insert(newValue, "newDocuments");
+        Database.Instance.insert(newValue);
     }
 
 }