feat(medusa-plugin-algolia): Revamp Algolia search plugin (#3510)

packages/medusa-plugin-algolia/src/loaders/index.js

@@ -1,13 +0,0 @@
-export default async (container, options) => {
-  try {
-    const algoliaService = container.resolve("algoliaService")
-
-    await Promise.all(
-      Object.entries(options.settings).map(([key, value]) =>
-        algoliaService.updateSettings(key, value)
-      )
-    )
-  } catch (err) {
-    console.log(err)
-  }
-}

packages/medusa-plugin-algolia/src/loaders/index.ts

@@ -0,0 +1,24 @@
+import { Logger, MedusaContainer } from "@medusajs/modules-sdk"
+import AlgoliaService from "../services/algolia"
+import { AlgoliaPluginOptions } from "../types"
+
+export default async (
+  container: MedusaContainer,
+  options: AlgoliaPluginOptions
+) => {
+  const logger: Logger = container.resolve("logger")
+  try {
+    const algoliaService: AlgoliaService = container.resolve("algoliaService")
+
+    const { settings } = options
+
+    await Promise.all(
+      Object.entries(settings || {}).map(async ([indexName, value]) => {
+        return await algoliaService.updateSettings(indexName, value)
+      })
+    )
+  } catch (err) {
+    // ignore
+    logger.warn(err)
+  }
+}

packages/medusa-plugin-algolia/src/services/algolia.ts

@@ -1,24 +1,31 @@
-import algoliasearch from "algoliasearch"
-import { indexTypes } from "medusa-core-utils"
-import { SearchService } from "medusa-interfaces"
-import { transformProduct } from "../utils/transform-product"
+import { SearchTypes } from "@medusajs/types"
+import { SearchUtils } from "@medusajs/utils"
+import Algolia, { SearchClient } from "algoliasearch"
+import { AlgoliaPluginOptions, SearchOptions } from "../types"
+import { transformProduct } from "../utils/transformer"
 
-class AlgoliaService extends SearchService {
-  constructor(container, options) {
-    super()
+class AlgoliaService extends SearchUtils.AbstractSearchService {
+  isDefault = false
 
-    this.options_ = options
-    const { application_id, admin_api_key } = this.options_
+  protected readonly config_: AlgoliaPluginOptions
+  protected readonly client_: SearchClient
 
-    if (!application_id) {
+  constructor(_, options: AlgoliaPluginOptions) {
+    super(_, options)
+
+    this.config_ = options
+
+    const { applicationId, adminApiKey } = options
+
+    if (!applicationId) {
       throw new Error("Please provide a valid Application ID")
     }
 
-    if (!admin_api_key) {
+    if (!adminApiKey) {
       throw new Error("Please provide a valid Admin Api Key")
     }
 
-    this.client_ = algoliasearch(application_id, admin_api_key)
+    this.client_ = Algolia(applicationId, adminApiKey)
   }
 
   /**
@@ -27,7 +34,7 @@ class AlgoliaService extends SearchService {
    * @param {*} options - not required just to match the schema we are used it
    * @return {*}
    */
-  createIndex(indexName, options = {}) {
+  createIndex(indexName: string, options: Record<string, unknown> = {}) {
     return this.client_.initIndex(indexName)
   }
 
@@ -36,8 +43,9 @@ class AlgoliaService extends SearchService {
    * @param {string} indexName  - the index name.
    * @return {Promise<{object}>} - returns response from search engine provider
    */
-  async getIndex(indexName) {
-    let hits = []
+  async getIndex(indexName: string) {
+    let hits: Record<string, unknown>[] = []
+
     return await this.client_
       .initIndex(indexName)
       .browseObjects({
@@ -56,8 +64,12 @@ class AlgoliaService extends SearchService {
    * @param {*} type
    * @return {*}
    */
-  async addDocuments(indexName, documents, type) {
-    const transformedDocuments = this.getTransformedDocuments(type, documents)
+  async addDocuments(indexName: string, documents: any, type: string) {
+    const transformedDocuments = await this.getTransformedDocuments(
+      type,
+      documents
+    )
+
     return await this.client_
       .initIndex(indexName)
       .saveObjects(transformedDocuments)
@@ -70,8 +82,11 @@ class AlgoliaService extends SearchService {
    * @param {Array.<Object>} type  - type of documents to be replaced (e.g: products, regions, orders, etc)
    * @return {Promise<{object}>} - returns response from search engine provider
    */
-  async replaceDocuments(indexName, documents, type) {
-    const transformedDocuments = this.getTransformedDocuments(type, documents)
+  async replaceDocuments(indexName: string, documents: any, type: string) {
+    const transformedDocuments = await this.getTransformedDocuments(
+      type,
+      documents
+    )
     return await this.client_
       .initIndex(indexName)
       .replaceAllObjects(transformedDocuments)
@@ -80,11 +95,11 @@ class AlgoliaService extends SearchService {
   /**
    * Used to delete document
    * @param {string} indexName  - the index name
-   * @param {string} document_id  - the id of the document
+   * @param {string} documentId  - the id of the document
    * @return {Promise<{object}>} - returns response from search engine provider
    */
-  async deleteDocument(indexName, document_id) {
-    return await this.client_.initIndex(indexName).deleteObject(document_id)
+  async deleteDocument(indexName: string, documentId: string) {
+    return await this.client_.initIndex(indexName).deleteObject(documentId)
   }
 
   /**
@@ -92,7 +107,7 @@ class AlgoliaService extends SearchService {
    * @param {string} indexName  - the index name
    * @return {Promise<{object}>} - returns response from search engine provider
    */
-  async deleteAllDocuments(indexName) {
+  async deleteAllDocuments(indexName: string) {
     return await this.client_.initIndex(indexName).delete()
   }
 
@@ -105,9 +120,15 @@ class AlgoliaService extends SearchService {
    * - additionalOptions contain any provider specific options
    * @return {*} - returns response from search engine provider
    */
-  async search(indexName, query, options) {
+  async search(
+    indexName: string,
+    query: string,
+    options: SearchOptions & Record<string, unknown>
+  ) {
     const { paginationOptions, filter, additionalOptions } = options
-    if ("limit" in paginationOptions) {
+
+    // fit our pagination options to what Algolia expects
+    if ("limit" in paginationOptions && paginationOptions.limit != null) {
       paginationOptions["length"] = paginationOptions.limit
       delete paginationOptions.limit
     }
@@ -125,25 +146,32 @@ class AlgoliaService extends SearchService {
    * @param {object} settings  - settings object
    * @return {Promise<{object}>} - returns response from search engine provider
    */
-  async updateSettings(indexName, settings) {
-    return await this.client_.initIndex(indexName).setSettings(settings)
+  async updateSettings(
+    indexName: string,
+    settings: SearchTypes.IndexSettings & Record<string, unknown>
+  ) {
+    // backward compatibility
+    const indexSettings = settings.indexSettings ?? settings ?? {}
+
+    return await this.client_.initIndex(indexName).setSettings(indexSettings)
   }
 
-  getTransformedDocuments(type, documents) {
+  async getTransformedDocuments(type: string, documents: any[]) {
+    if (!documents?.length) {
+      return []
+    }
+
     switch (type) {
-      case indexTypes.products:
-        return this.transformProducts(documents)
+      case SearchTypes.indexTypes.PRODUCTS:
+        const productsTransformer =
+          this.config_.settings?.[SearchTypes.indexTypes.PRODUCTS]
+            ?.transformer ?? transformProduct
+
+        return documents.map(productsTransformer)
       default:
         return documents
     }
   }
-
-  transformProducts(products) {
-    if (!products) {
-      return []
-    }
-    return products.map(transformProduct)
-  }
 }
 
 export default AlgoliaService

packages/medusa-plugin-algolia/src/types.ts

@@ -0,0 +1,18 @@
+import { SearchTypes } from "@medusajs/types"
+
+export type SearchOptions = {
+  paginationOptions: Record<string, unknown>
+  filter: string
+  additionalOptions: Record<string, unknown>
+}
+
+export type AlgoliaPluginOptions = {
+  applicationId: string
+  adminApiKey: string
+  /**
+   * Index settings
+   */
+  settings?: {
+    [key: string]: SearchTypes.IndexSettings
+  }
+}

packages/medusa-plugin-algolia/src/utils/transform-product.js

@@ -1,42 +0,0 @@
-const variantKeys = [
-  "sku",
-  "title",
-  "upc",
-  "ean",
-  "mid_code",
-  "hs_code",
-  "options",
-]
-const prefix = `variant`
-
-export const transformProduct = (product) => {
-  const initialObj = variantKeys.reduce((obj, key) => {
-    obj[`${prefix}_${key}`] = []
-    return obj
-  }, {})
-  initialObj[`${prefix}_options_value`] = []
-
-  const flattenedVariantFields = product.variants.reduce((obj, variant) => {
-    variantKeys.forEach((k) => {
-      if (k === "options" && variant[k]) {
-        const values = variant[k].map((option) => option.value)
-        obj[`${prefix}_options_value`] =
-          obj[`${prefix}_options_value`].concat(values)
-        return
-      }
-      return variant[k] && obj[`${prefix}_${k}`].push(variant[k])
-    })
-    return obj
-  }, initialObj)
-
-  product.objectID = product.id
-  product.type_value = product.type && product.type.value
-  product.collection_title = product.collection && product.collection.title
-  product.collection_handle = product.collection && product.collection.handle
-  product.tags_value = product.tags ? product.tags.map((t) => t.value) : []
-
-  return {
-    ...product,
-    ...flattenedVariantFields,
-  }
-}

packages/medusa-plugin-algolia/src/utils/transformer.ts

@@ -0,0 +1,44 @@
+import { variantKeys } from "@medusajs/types"
+
+const prefix = `variant`
+
+export const transformProduct = (product: any) => {
+  let transformedProduct = { ...product } as Record<string, unknown>
+
+  const initialObj = variantKeys.reduce((obj, key) => {
+    obj[`${prefix}_${key}`] = []
+    return obj
+  }, {})
+  initialObj[`${prefix}_options_value`] = []
+
+  const flattenedVariantFields = product.variants.reduce((obj, variant) => {
+    variantKeys.forEach((k) => {
+      if (k === "options" && variant[k]) {
+        const values = variant[k].map((option) => option.value)
+        obj[`${prefix}_options_value`] =
+          obj[`${prefix}_options_value`].concat(values)
+        return
+      }
+      return variant[k] && obj[`${prefix}_${k}`].push(variant[k])
+    })
+    return obj
+  }, initialObj)
+
+  transformedProduct.objectID = product.id
+  transformedProduct.type_value = product.type && product.type.value
+  transformedProduct.collection_title =
+    product.collection && product.collection.title
+  transformedProduct.collection_handle =
+    product.collection && product.collection.handle
+  transformedProduct.tags_value = product.tags
+    ? product.tags.map((t) => t.value)
+    : []
+  transformedProduct.categories = (product?.categories || []).map((c) => c.name)
+
+  const prod = {
+    ...transformedProduct,
+    ...flattenedVariantFields,
+  }
+
+  return prod
+}