add this.actions support to react components

Author: mariusandra

src/kea/index.js

@@ -3,10 +3,14 @@ import { connect } from 'react-redux'
 import { convertInputToLogic, convertPartialDynamicInput, clearLogicCache } from '../logic/index'
 
 export function kea (input) {
-  const wrapper = connect(
-    mapStateToPropsCreator(input),
-    mapDispatchToPropsCreator(input)
-  )
+  const wrapper = (Klass) => {
+    injectActionsIntoClass(Klass)
+
+    return connect(
+      mapStateToPropsCreator(input),
+      mapDispatchToPropsCreator(input)
+    )(Klass)
+  }
 
   wrapper._isKeaFunction = true
   wrapper._isKeaSingleton = !input.key
@@ -46,6 +50,27 @@ const mapDispatchToPropsCreator = input => (dispatch, ownProps) => {
   }
 }
 
+function isStateless (Component) {
+  return (
+    typeof Component === 'function' && // can be various things
+    !(Component.prototype && Component.prototype.isReactComponent) // native arrows don't have prototypes // special property
+  )
+}
+
+function injectActionsIntoClass (Klass) {
+  if (!isStateless(Klass)) {
+    // inject to the component something that
+    // converts this.props.actions to this.actions
+    if (!Object.getOwnPropertyDescriptor(Klass.prototype, 'actions')) {
+      Object.defineProperty(Klass.prototype, 'actions', {
+        get: function actions () {
+          return this.props.actions
+        }
+      })
+    }
+  }
+}
+
 export function resetKeaLogicCache () {
   clearLogicCache()
 }