65 lines
1.7 KiB
JavaScript
65 lines
1.7 KiB
JavaScript
import _extends from 'babel-runtime/helpers/extends';
|
|
import React from 'react';
|
|
import hoistNonReactStatics from 'hoist-non-react-statics';
|
|
import wrapDisplayName from 'recompose/wrapDisplayName';
|
|
import createMuiTheme from './createMuiTheme';
|
|
import themeListener from './themeListener';
|
|
|
|
let defaultTheme;
|
|
|
|
function getDefaultTheme() {
|
|
if (defaultTheme) {
|
|
return defaultTheme;
|
|
}
|
|
|
|
defaultTheme = createMuiTheme();
|
|
return defaultTheme;
|
|
}
|
|
|
|
// Provide the theme object as a property to the input component.
|
|
const withTheme = () => Component => {
|
|
class WithTheme extends React.Component {
|
|
constructor(props, context) {
|
|
super(props, context);
|
|
this.state = {};
|
|
this.unsubscribeId = null;
|
|
this.state = {
|
|
// We use || as the function call is lazy evaluated.
|
|
theme: themeListener.initial(context) || getDefaultTheme()
|
|
};
|
|
}
|
|
|
|
componentDidMount() {
|
|
this.unsubscribeId = themeListener.subscribe(this.context, theme => {
|
|
this.setState({ theme });
|
|
});
|
|
}
|
|
|
|
componentWillUnmount() {
|
|
if (this.unsubscribeId !== null) {
|
|
themeListener.unsubscribe(this.context, this.unsubscribeId);
|
|
}
|
|
}
|
|
|
|
render() {
|
|
return React.createElement(Component, _extends({ theme: this.state.theme }, this.props));
|
|
}
|
|
}
|
|
|
|
WithTheme.contextTypes = themeListener.contextTypes;
|
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
WithTheme.displayName = wrapDisplayName(Component, 'WithTheme');
|
|
}
|
|
|
|
hoistNonReactStatics(WithTheme, Component);
|
|
|
|
if (process.env.NODE_ENV !== 'production') {
|
|
// Exposed for test purposes.
|
|
WithTheme.Naked = Component;
|
|
}
|
|
|
|
return WithTheme;
|
|
};
|
|
|
|
export default withTheme; |