feat(elasticsearch-plugin): Added mapQuery option

The discussion for the feature took place in vendure-ecommerce#364. This commit closes vendure-ecommerce#364.
Tyratox · Jun 8, 2020 · e2c443c · e2c443c
1 parent 96ea811
commit e2c443c
Show file tree

Hide file tree

Showing 2 changed files with 51 additions and 4 deletions.
diff --git a/packages/elasticsearch-plugin/src/build-elastic-body.ts b/packages/elasticsearch-plugin/src/build-elastic-body.ts
@@ -52,7 +52,7 @@ export function buildElasticBody(
         const operator = facetValueOperator === LogicalOperator.AND ? 'must' : 'should';
         query.bool.filter = query.bool.filter.concat([
             {
-                bool: { [operator]: facetValueIds.map(id => ({ term: { facetValueIds: id } })) },
+                bool: { [operator]: facetValueIds.map((id) => ({ term: { facetValueIds: id } })) },
             },
         ]);
     }
@@ -88,7 +88,9 @@ export function buildElasticBody(
         }
     }
     return {
-        query,
+        query: searchConfig.mapQuery
+            ? searchConfig.mapQuery(query, input, searchConfig, channelId, enabledOnly)
+            : query,
         sort: sortArray,
         from: skip || 0,
         size: take || 10,

diff --git a/packages/elasticsearch-plugin/src/options.ts b/packages/elasticsearch-plugin/src/options.ts
@@ -1,7 +1,7 @@
-import { DeepRequired, Product, ProductVariant } from '@vendure/core';
+import { DeepRequired, ID, Product, ProductVariant } from '@vendure/core';
 import deepmerge from 'deepmerge';
 
-import { CustomMapping } from './types';
+import { CustomMapping, ElasticSearchInput } from './types';
 
 /**
  * @description
@@ -188,6 +188,50 @@ export interface SearchConfig {
      * ```
      */
     priceRangeBucketInterval?: number;
+    /**
+     * @description
+     * This config option allows the the modification of the whole (already built) search query. This allows
+     * for e.g. wildcard / fuzzy searches on the index.
+     *
+     * @example
+     * ```TypeScript
+     * mapQuery: (query, input, searchConfig, channelId, enabledOnly){
+     *     if(query.bool.must){
+     *         delete query.bool.must;
+     *     }
+     *     query.bool.should = [
+     *         {
+     *             query_string: {
+     *                 query: "*" + term + "*",
+     *                 fields: [
+     *                     `productName^${searchConfig.boostFields.productName}`,
+     *                     `productVariantName^${searchConfig.boostFields.productVariantName}`,
+     *                 ]
+     *             }
+     *         },
+     *         {
+     *             multi_match: {
+     *                 query: term,
+     *                 type: searchConfig.multiMatchType,
+     *                 fields: [
+     *                     `description^${searchConfig.boostFields.description}`,
+     *                     `sku^${searchConfig.boostFields.sku}`,
+     *                 ],
+     *             },
+     *         },
+     *     ];
+     *
+     *     return query;
+     * }
+     * ```
+     */
+    mapQuery?: (
+        query: any,
+        input: ElasticSearchInput,
+        searchConfig: DeepRequired<SearchConfig>,
+        channelId: ID,
+        enabledOnly: boolean,
+    ) => any;
 }
 
 /**
@@ -246,6 +290,7 @@ export const defaultOptions: DeepRequired<ElasticsearchOptions> = {
             sku: 1,
         },
         priceRangeBucketInterval: 1000,
+        mapQuery: (query) => query,
     },
     customProductMappings: {},
     customProductVariantMappings: {},