feat: add faccToHoc()

Author: streamich

src/BatterySensor/__story__/story.ts

@@ -13,7 +13,7 @@ const Print = (props) => h('pre', {
   JSON.stringify(props, null, 4)
 );
 
-const PrintBattery = withBattery(Print);
+const PrintBattery = withBattery(Print, 'overwritten');
 
 @withBattery
 class Printer extends Component<any, any> {

src/BatterySensor/index.ts

@@ -1,6 +1,6 @@
 import {Component} from 'react';
 import {h, on, off, isClient} from '../util';
-import addClassDecoratorSupport from '../util/addClassDecoratorSupport';
+import faccToHoc from '../util/faccToHoc';
 
 export interface IBatterySensorProps {
   children?: (INetworkState) => React.ReactElement<any>;
@@ -70,14 +70,4 @@ export class BatterySensor extends Component<IBatterySensorProps, IBatterySensor
   }
 }
 
-export const withBattery = (Comp) => {
-  const Enhanced = (props) =>
-    h(BatterySensor, null, (battery) =>
-      h(Comp, {
-        ...props,
-        battery
-      })
-    );
-
-  return addClassDecoratorSupport(Enhanced);
-};
+export const withBattery = faccToHoc(BatterySensor, 'battery');

src/util/faccToHoc.ts

@@ -0,0 +1,18 @@
+import {h} from '../util';
+import addClassDecoratorSupport from './addClassDecoratorSupport';
+
+const faccToHoc = (Facc, prop?: string) =>
+  (Comp, propName: string = prop) => {
+    const Enhanced = (props) =>
+      h(Facc, null, (state) =>
+        h(Comp,
+          propName ?
+            {[propName]: state, ...props} :
+            {...state, ...props}
+        )
+      );
+
+    return addClassDecoratorSupport(Enhanced);
+  };
+
+export default faccToHoc;