feat(core): use postMap if available

nartc · Mar 6, 2021 · fb22792 · fb22792
1 parent 9552516
commit fb22792
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/packages/core/src/lib/create-mapper/create-mapper.ts b/packages/core/src/lib/create-mapper/create-mapper.ts
@@ -57,7 +57,7 @@ export function createMapper<TKey = unknown>({
         return sourceObj;
       }
 
-      const { preMap } = plugin;
+      const { preMap, postMap } = plugin;
 
       // run preMap if available
       const [sourceInstance] = preMap
@@ -79,13 +79,15 @@ export function createMapper<TKey = unknown>({
             'extraArguments' in destinationObjOrOptions)) ||
         destinationObjOrOptions == null
       ) {
-        return mapReturn(
+        const result = mapReturn(
           sourceInstance ?? sourceObj,
           mapping!,
           destinationObjOrOptions as MapOptions,
           this,
           errorHandler
         );
+
+        return postMap ? postMap.bind(plugin)(destination, result) : result;
       }
 
       mapMutate(
@@ -96,6 +98,9 @@ export function createMapper<TKey = unknown>({
         errorHandler,
         destinationObjOrOptions
       );
+      if (postMap) {
+        destinationObjOrOptions = postMap(destination, destinationObjOrOptions);
+      }
     },
     mapAsync(
       sourceObj: Record<string, unknown>,