feat: Algolia plugin for medusa (#718)

Co-authored-by: Rolwin Monterio <rolwin@yellow.ai>
Co-authored-by: olivermrbl <oliver@mrbltech.com>

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

@@ -0,0 +1,13 @@
+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/services/algolia.js

@@ -0,0 +1,147 @@
+import algoliasearch from "algoliasearch"
+import { indexTypes } from "medusa-core-utils"
+import { SearchService } from "medusa-interfaces"
+import { transformProduct } from "../utils/transform-product"
+
+class AlgoliaService extends SearchService {
+  constructor(container, options) {
+    super()
+
+    this.options_ = options
+    const { application_id, admin_api_key } = this.options_
+
+    if (!application_id) {
+      throw new Error("Please provide a valid Application ID")
+    }
+
+    if (!admin_api_key) {
+      throw new Error("Please provide a valid Admin Api Key")
+    }
+
+    this.client_ = algoliasearch(application_id, admin_api_key)
+  }
+
+  /**
+   * Add two numbers.
+   * @param {string} indexName - The name of the index
+   * @param {*} options - not required just to match the schema we are used it
+   * @return {*}
+   */
+  createIndex(indexName, options = {}) {
+    return this.client_.initIndex(indexName)
+  }
+
+  /**
+   * Used to get an index
+   * @param {string} indexName  - the index name.
+   * @return {Promise<{object}>} - returns response from search engine provider
+   */
+  getIndex(indexName) {
+    let hits = []
+    return this.client_
+      .initIndex(indexName)
+      .browseObjects({
+        query: indexName,
+        batch: (batch) => {
+          hits = hits.concat(batch)
+        },
+      })
+      .then(() => hits)
+  }
+
+  /**
+   *
+   * @param {string} indexName
+   * @param {Array} documents - products list array
+   * @param {*} type
+   * @return {*}
+   */
+  addDocuments(indexName, documents, type) {
+    const transformedDocuments = this.getTransformedDocuments(type, documents)
+    return this.client_.initIndex(indexName).saveObjects(transformedDocuments)
+  }
+
+  /**
+   * Used to replace documents
+   * @param {string} indexName  - the index name.
+   * @param {Object} documents  - array of document objects that will replace existing documents
+   * @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
+   */
+  replaceDocuments(indexName, documents, type) {
+    const transformedDocuments = this.getTransformedDocuments(type, documents)
+    return this.client_
+      .initIndex(indexName)
+      .replaceAllObjects(transformedDocuments)
+  }
+
+  /**
+   * Used to delete document
+   * @param {string} indexName  - the index name
+   * @param {string} document_id  - the id of the document
+   * @return {Promise<{object}>} - returns response from search engine provider
+   */
+  deleteDocument(indexName, document_id) {
+    return this.client_.initIndex(indexName).deleteObject(document_id)
+  }
+
+  /**
+   * Used to delete all documents
+   * @param {string} indexName  - the index name
+   * @return {Promise<{object}>} - returns response from search engine provider
+   */
+  deleteAllDocuments(indexName) {
+    return this.client_.initIndex(indexName).delete()
+  }
+
+  /**
+   * Used to search for a document in an index
+   * @param {string} indexName - the index name
+   * @param {string} query  - the search query
+   * @param {*} options
+   * - any options passed to the request object other than the query and indexName
+   * - additionalOptions contain any provider specific options
+   * @return {*} - returns response from search engine provider
+   */
+  search(indexName, query, options) {
+    const { paginationOptions, filter, additionalOptions } = options
+    if ("limit" in paginationOptions) {
+      paginationOptions["length"] = paginationOptions.limit
+      delete paginationOptions.limit 
+    }
+    
+    return this.client_.initIndex(indexName).search(query, {
+      filters: filter,
+      ...paginationOptions,
+      ...additionalOptions
+    })
+  }
+
+  /**
+   * Used to update the settings of an index
+   * @param  {string} indexName - the index name
+   * @param {object} settings  - settings object
+   * @return {Promise<{object}>} - returns response from search engine provider
+   */
+  updateSettings(indexName, settings) {
+    return this.client_.initIndex(indexName).setSettings(settings)
+  }
+
+  getTransformedDocuments(type, documents) {
+    switch (type) {
+      case indexTypes.products:
+        return this.transformProducts(documents)
+      default:
+        return documents
+    }
+  }
+
+  transformProducts(products) {
+    if (!products) {
+      return []
+    }
+    return products.map(transformProduct)
+  }
+}
+
+export default AlgoliaService

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

@@ -0,0 +1,42 @@
+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,
+  }
+}