feat: show schema suggestions

dendronhq · Jul 4, 2020 · 1313e79 · 1313e79
1 parent 102b997
commit 1313e79
Showing 1 changed file with 13 additions and 8 deletions.
diff --git a/packages/plugin-core/src/components/lookup/LookupProvider.ts b/packages/plugin-core/src/components/lookup/LookupProvider.ts
@@ -10,7 +10,12 @@ import { node2Uri } from "./utils";
 const L = createLogger("LookupProvider");
 
 class QueryStringUtils {
-  static getSchema(_qs: string, engineResp: Note[]): Schema[] {
+  /**
+   * Get all schema matches for current query
+   * @param _qs 
+   * @param engineResp 
+   */
+  static getAllSchemaAtLevel(_qs: string, engineResp: Note[]): Schema[] {
     const maybeSchema = engineResp[0]?.schema;
     return maybeSchema?.parent?.children as Schema[] || [];
   }
@@ -119,13 +124,13 @@ export class LookupProvider {
       }
 
 
-      // apply schemas to the results
-      // const schemas = QueryStringUtils.getSchema(querystring, updatedItems as Note[]);
-      // updatedItems = _.uniqBy(updatedItems.concat(schemas.map(schema => {
-      //   return Note.fromSchema(DNodeUtils.dirName(querystring), schema);
-      // })), (ent) => {
-      //   return ent.fname;
-      // });
+      // show schema suggestions
+      const schemas = QueryStringUtils.getAllSchemaAtLevel(querystring, updatedItems as Note[]);
+      updatedItems = _.uniqBy(updatedItems.concat(schemas.map(schema => {
+        return Note.fromSchema(DNodeUtils.dirName(querystring), schema);
+      })), (ent) => {
+        return ent.fname;
+      });
 
 
       // check if perfect match, remove @noActiveItem result if that's the case