feat: add new hoc withPlatform (#2431)

Author: addictional

src/components/mobile/index.ts

@@ -2,5 +2,6 @@ export * from './MobileContext';
 export * from './MobileProvider';
 export * from './useMobile';
 export * from './usePlatform';
+export * from './withPlatform';
 export * from './withMobile';
 export * from './constants';

src/components/mobile/withPlatform.tsx

@@ -0,0 +1,24 @@
+import * as React from 'react';
+
+import {getComponentName} from '../utils/getComponentName';
+
+import {MobileContext} from './MobileContext';
+import type {MobileContextProps} from './MobileContext';
+
+export type WithPlatformProps = {platform: MobileContextProps['platform']};
+
+export function withPlatform<T extends WithPlatformProps>(
+    WrappedComponent: React.ComponentType<T>,
+): React.ComponentType<Omit<T, keyof WithPlatformProps>> {
+    const componentName = getComponentName(WrappedComponent);
+
+    return class WithPlatformComponent extends React.Component<Omit<T, keyof WithPlatformProps>> {
+        static displayName = `withPlatform(${componentName})`;
+        static contextType = MobileContext;
+        declare context: React.ContextType<typeof MobileContext>;
+
+        render() {
+            return <WrappedComponent {...(this.props as T)} platform={this.context.platform} />;
+        }
+    };
+}