Merge pull request #130 from browngraphicslab/new_search

New search

1 files changed, 45 insertions, 0 deletions

diff --git a/src/server/Search.ts b/src/server/Search.ts
new file mode 100644
index 000000000..1d8cfdcf0
--- /dev/null
+++ b/src/server/Search.ts
@@ -0,0 +1,45 @@
+import * as rp from 'request-promise';
+import { Database } from './database';
+import { thisExpression } from 'babel-types';
+
+export class Search {
+    public static Instance = new Search();
+    private url = 'http://localhost:8983/solr/';
+
+    public async updateDocument(document: any) {
+        try {
+            return rp.post(this.url + "dash/update", {
+                headers: { 'content-type': 'application/json' },
+                body: JSON.stringify([document])
+            });
+        } catch { }
+    }
+
+    public async search(query: string) {
+        try {
+            const searchResults = JSON.parse(await rp.get(this.url + "dash/select", {
+                qs: {
+                    q: query
+                }
+            }));
+            const fields = searchResults.response.docs;
+            const ids = fields.map((field: any) => field.id);
+            return ids;
+        } catch {
+            return [];
+        }
+    }
+
+    public async clear() {
+        try {
+            return rp.post(this.url + "dash/update", {
+                body: {
+                    delete: {
+                        query: "*:*"
+                    }
+                },
+                json: true
+            });
+        } catch { }
+    }
+}
\ No newline at end of file