chain-simple 1.2.1 → 1.2.2

package/built/index.d.ts

@@ -7,7 +7,9 @@ type TConfig = {
     getEntity?: string;
     extendProxed?: (propName: any) => {
         [k: string]: any;
-    };
+    } | ((item: any) => {
+        [k: string]: any;
+    });
 };
 /**
  * @example

package/built/index.js

@@ -68,7 +68,14 @@ function makePropertiesChainable(item, config) {
                 (0, sat_utils_1.isFunction)(config.extendProxed)) {
                 try {
                     const extension = config.extendProxed(p);
-                    Object.assign(item, extension);
+                    if ((0, sat_utils_1.isObject)(extension)) {
+                        Object.assign(item, extension);
+                    }
+                    else if ((0, sat_utils_1.isFunction)(extension)) {
+                        // @ts-ignore
+                        const result = extension(item);
+                        Object.assign(item, result);
+                    }
                 }
                 catch (error) {
                     console.error(error);

package/lib/index.ts

@@ -12,7 +12,10 @@ export type TChainable<T extends Record<string, TFn>> = {
   [K in keyof T]: TReplaceReturnType<T[K], ReturnType<T[K]> & TChainable<T>>;
 };
 
-type TConfig = { getEntity?: string; extendProxed?: (propName) => { [k: string]: any } };
+type TConfig = {
+  getEntity?: string;
+  extendProxed?: (propName) => { [k: string]: any } | ((item: any) => { [k: string]: any });
+};
 
 /**
  * @example
@@ -85,7 +88,13 @@ function makePropertiesChainable(item, config?: TConfig) {
       ) {
         try {
           const extension = config.extendProxed(p);
-          Object.assign(item, extension);
+          if (isObject(extension)) {
+            Object.assign(item, extension);
+          } else if (isFunction(extension)) {
+            // @ts-ignore
+            const result = extension(item);
+            Object.assign(item, result);
+          }
         } catch (error) {
           console.error(error);
         }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "chain-simple",
-  "version": "1.2.1",
+  "version": "1.2.2",
   "description": "Main purpose of this package is - provide simple way to build chain between any item methods",
   "main": "built/index.js",
   "directories": {