extend connectStyle

shoutem · Jul 26, 2017 · 180156f · 180156f
1 parent f9a20f1
commit 180156f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/.gitignore b/.gitignore
@@ -6,3 +6,6 @@ node_modules/
 
 # WebStorm
 .idea
+
+# Mac OS X folder attributes file
+.DS_Store
diff --git a/src/connectStyle.js b/src/connectStyle.js
@@ -49,6 +49,13 @@ export default (componentStyleName, componentStyle = {}, mapPropsToStyleNames, o
     return WrappedComponent.displayName || WrappedComponent.name || 'Component';
   }
 
+  function getBaseComponent(WrappedComponent) {
+    if (WrappedComponent.BaseComponent) {
+      return WrappedComponent.BaseComponent;
+    }
+    return WrappedComponent;
+  }
+
   return function wrapWithStyledComponent(WrappedComponent) {
     const componentDisplayName = getComponentDisplayName(WrappedComponent);
 
@@ -99,6 +106,7 @@ export default (componentStyleName, componentStyle = {}, mapPropsToStyleNames, o
 
       static displayName = `Styled(${componentDisplayName})`;
       static WrappedComponent = WrappedComponent;
+      static BaseComponent = getBaseComponent(WrappedComponent);
 
       constructor(props, context) {
         super(props, context);