From a1a619471793b78938ffd3849948463b91e6eb6a Mon Sep 17 00:00:00 2001
From: adueck <adueck@users.noreply.github.com>
Date: Sun, 29 Jan 2023 22:46:27 +0500
Subject: [PATCH] add allWords database to account

---
 account/src/lib/dictionary.ts            | 38 +++++++++++++++++++-----
 account/src/routers/dictionary-router.ts | 14 +++++++++
 2 files changed, 45 insertions(+), 7 deletions(-)

diff --git a/account/src/lib/dictionary.ts b/account/src/lib/dictionary.ts
index 41883bc..ade8b8c 100644
--- a/account/src/lib/dictionary.ts
+++ b/account/src/lib/dictionary.ts
@@ -2,6 +2,7 @@ import loki, { Collection, LokiMemoryAdapter } from "lokijs";
 import fetch from "node-fetch";
 import { CronJob } from "cron";
 const collectionName = "ps-dictionary";
+const allWordsCollectionName = "all-words";
 import {
     readDictionary,
     readDictionaryInfo,
@@ -11,7 +12,8 @@ import {
     standardizePashto,
 } from "@lingdocs/inflect"
 
-export let collection: Collection<any> | undefined = undefined;
+export let collection: Collection<T.DictionaryEntry> | undefined = undefined;
+export let allWordsCollection: Collection<T.PsString> | undefined = undefined;
 const adapter = new LokiMemoryAdapter();
 const lokidb = new loki("", {
     adapter,
@@ -30,6 +32,11 @@ async function fetchDictionary(): Promise<T.Dictionary> {
     return readDictionary(buffer as Uint8Array);
 }
 
+async function fetchAllWords(): Promise<T.AllWordsWithInflections> {
+    const res = await fetch(process.env.LINGDOCS_DICTIONARY_URL?.slice(0, -4) + "all-words.json");
+    return await res.json();
+}
+
 async function fetchDictionaryInfo(): Promise<T.DictionaryInfo> {
     const res = await fetch(process.env.LINGDOCS_DICTIONARY_URL + "-info" || "");
     const buffer = await res.arrayBuffer();
@@ -54,10 +61,22 @@ function getOneByTs(ts: number): T.DictionaryEntry {
         throw new Error("dictionary not initialized");
     }
     const r = collection.by("ts", ts);
+    // @ts-ignore
     const { $loki, meta, ...entry } = r;
     return entry;
 }
 
+export function findInAllWords(p: string | RegExp): T.PsWord[] | undefined {
+    if (!allWordsCollection) {
+        throw new Error("allWords not initialized");
+    }
+    return allWordsCollection.find({
+        p: typeof p === "string"
+            ? p
+            : { $regex: p },
+    });
+}
+
 export async function getEntries(ids: (number | string)[]): Promise<{
     results: (T.DictionaryEntry | T.VerbEntry)[],
     notFound: (number | string)[],
@@ -100,15 +119,20 @@ export async function getEntries(ids: (number | string)[]): Promise<{
 
 lokidb.loadDatabase({}, (err: Error) => {
     lokidb.removeCollection(collectionName);
-    collection = lokidb.addCollection(collectionName, {
-        // TODO: THIS ISN'T WORKING!
-        disableMeta: true,
-        indices: ["i", "p"],
-        unique: ["ts"],
-    });
+    lokidb.removeCollection(allWordsCollectionName);
     fetchDictionary().then((dictionary) => {
+        collection = lokidb.addCollection(collectionName, {
+            indices: ["i", "p"],
+            unique: ["ts"],
+        });
         version = dictionary.info.release;
         collection?.insert(dictionary.entries);
         updateJob.start();
     }).catch(console.error);
+    fetchAllWords().then((allWords) => {
+        allWordsCollection = lokidb.addCollection(allWordsCollectionName, {
+            indices: ["p"],
+        });
+        allWordsCollection?.insert(allWords.words);
+    });
 });
diff --git a/account/src/routers/dictionary-router.ts b/account/src/routers/dictionary-router.ts
index 29b2aaf..aa75cf2 100644
--- a/account/src/routers/dictionary-router.ts
+++ b/account/src/routers/dictionary-router.ts
@@ -1,6 +1,8 @@
 import express from "express";
 import {
+    allWordsCollection,
     collection,
+    findInAllWords,
     getEntries,
     updateDictionary,    
 } from "../lib/dictionary";
@@ -12,6 +14,18 @@ dictionaryRouter.post("/update", async (req, res, next) => {
     res.send({ ok: true, result });
 });
 
+dictionaryRouter.post("/all-words", async (req, res, next) => {
+    if (!allWordsCollection) {
+        return res.send({ ok: false, message: "allWords not ready" });
+    }
+    const word = req.body.word as string;
+    if (!word) {
+        return res.status(400).send({ ok: false, error: "invalid query" });
+    }
+    const results = await findInAllWords(word);
+    res.send(results);
+})
+
 dictionaryRouter.post("/entries", async (req, res, next) => {
     if (!collection) {
         return res.send({ ok: false, message: "dictionary not ready" });