Added categoryTree param to mount href with dataplane data

Author: renatoalmeidav

node/resolvers/search/category.ts

@@ -1,9 +1,9 @@
 import { compose, last, prop, split } from 'ramda'
 
-import { getCategoryInfo, logDegradedSearchError } from './utils'
 import { formatTranslatableProp, shouldTranslateToBinding } from '../../utils/i18n'
 import { Slugify } from '../../utils/slug'
 import { APP_NAME } from './constants'
+import { getCategoryInfo, logDegradedSearchError } from './utils'
 
 const lastSegment = compose<string, string[], string>(
   last,
@@ -14,6 +14,28 @@ function cleanUrl(url: string) {
   return url.replace(/https:\/\/[A-z0-9]+\.vtexcommercestable\.com\.br/, '').toLowerCase()
 }
 
+export async function mountHrefRewritter({ url, id }: any, _: unknown, ctx: Context) {
+    const settings: AppSettings = await ctx.clients.apps.getAppSettings(APP_NAME)
+
+    if (shouldTranslateToBinding(ctx)) {
+      try {
+        const rewriterUrl = await ctx.clients.rewriter.getRoute(id.toString(), 'anyCategoryEntity', ctx.vtex.binding!.id!)
+        if (rewriterUrl) {
+          url = rewriterUrl
+        }
+      } catch (e) {
+        logDegradedSearchError(ctx.vtex.logger, {
+          service: 'Rewriter getRoute',
+          error: `Rewriter getRoute query returned an error for category ${id}. Category href may be incorrect.`,
+          errorStack: e,
+        })
+      }
+    }
+    const pathname = cleanUrl(url)
+
+    return settings.slugifyLinks ? Slugify(pathname) : pathname
+  }
+
 /** This type has to be created because the Catlog API to get category by ID does not return the url or children for now.
  * These fields only come if you get the category from the categroy tree api.
  */
@@ -29,27 +51,7 @@ export const resolvers = {
 
     cacheId: prop('id'),
 
-    href: async ({ url, id }: SafeCategory, _: unknown, ctx: Context) => {
-      const settings: AppSettings = await ctx.clients.apps.getAppSettings(APP_NAME)
-
-      if (shouldTranslateToBinding(ctx)) {
-        try {
-          const rewriterUrl = await ctx.clients.rewriter.getRoute(id.toString(), 'anyCategoryEntity', ctx.vtex.binding!.id!)
-          if (rewriterUrl) {
-            url = rewriterUrl
-          }
-        } catch (e) {
-          logDegradedSearchError(ctx.vtex.logger, {
-            service: 'Rewriter getRoute',
-            error: `Rewriter getRoute query returned an error for category ${id}. Category href may be incorrect.`,
-            errorStack: e,
-          })
-        }
-      }
-      const pathname = cleanUrl(url)
-
-      return settings.slugifyLinks ? Slugify(pathname) : pathname
-    },
+    href: mountHrefRewritter,
 
     metaTagDescription: formatTranslatableProp<SafeCategory, 'MetaTagDescription', 'id'>(
       'MetaTagDescription',

node/resolvers/search/product.ts

@@ -1,12 +1,14 @@
-import { compose, last, omit, pathOr, split, flatten } from 'ramda'
+import { compose, flatten, last, omit, pathOr, split } from 'ramda'
 
+import { fetchAppSettings } from '../../services/settings'
 import {
   addContextToTranslatableString,
   formatTranslatableProp,
   shouldTranslateToBinding,
   shouldTranslateToUserLocale,
 } from '../../utils/i18n'
 import { getBenefits } from '../benefits'
+import { mountHrefRewritter } from './category'
 import { buildCategoryMap, logDegradedSearchError } from './utils'
 
 type DynamicKey<T> = Record<string, T>
@@ -121,18 +123,27 @@ const findMainTree = (categoriesIds: string[], prodCategoryId: string) => {
 }
 
 const productCategoriesToCategoryTree = async (
-  { categories, categoriesIds, categoryId: prodCategoryId }: SearchProduct,
+  // TODO categoryTree will be added
+  { categories, categoriesIds, categoryId: prodCategoryId, categoryTree }: SearchProduct & { categoryTree: Array<{ id: number, name: string, href: string }> },
   _: any,
-  { clients: { search }, vtex: { platform } }: Context
+  ctx: Context
 ) => {
+  const { clients: { search }, vtex: { platform } } = ctx
+  const { shouldUseNewPDPEndpoint } = await fetchAppSettings(ctx)
   if (!categories || !categoriesIds) {
     return []
   }
 
   const mainTreeIds = findMainTree(categoriesIds, prodCategoryId)
 
   if (platform === 'vtex') {
-    return mainTreeIds.map(categoryId => search.category(Number(categoryId)))
+    if (shouldUseNewPDPEndpoint) {
+      return categoryTree.map(category => ({ ...category, href: mountHrefRewritter(category, _, ctx) }))
+    }
+    return mainTreeIds.map(async categoryId => {
+      const category = await search.category(Number(categoryId))
+      return {...category,}
+    })
   }
   const categoriesTree = await search.categories(mainTreeIds.length)
   const categoryMap = buildCategoryMap(categoriesTree)

node/utils/i18n.ts

@@ -47,6 +47,7 @@ export const addContextToTranslatableString = (message: Message, ctx: Context) =
 
     const context = (originalContext || message.context)?.toString()
     const from = originalFrom || message.from || locale
+    // @ts-expect-error
     return formatTranslatableStringV2({ content, context, from, state })
   } catch (e) {
     logDegradedSearchError(ctx.vtex.logger, {