Add connect enhancer

migueloller · Jun 27, 2016 · 1c8d940 · 1c8d940
1 parent 8b2843a
commit 1c8d940
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/src/components/connect.js b/src/components/connect.js
@@ -32,7 +32,17 @@ function tryCatch(fn, ctx) {
 // Helps track hot reloading.
 let nextVersion = 0
 
-export default function connect(mapStateToProps, mapDispatchToProps, mergeProps, options = {}) {
+export default function connect(...args) {
+  const [ mapStateToProps, mapDispatchToProps, mergeProps, { enhancer, ...options } = {} ] = args
+
+  if (typeof enhancer !== 'undefined') {
+    if (typeof enhancer !== 'function') {
+      throw new Error('Expected the enhancer to be a function.')
+    }
+
+    return enhancer(connect)(...args.slice(0, args.length - 1), options)
+  }
+
   const shouldSubscribe = Boolean(mapStateToProps)
   const mapState = mapStateToProps || defaultMapStateToProps