79 lines
1.9 KiB
JavaScript
79 lines
1.9 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);
|
|
Object.defineProperty(this, "state", {
|
|
configurable: true,
|
|
enumerable: true,
|
|
writable: true,
|
|
value: {}
|
|
});
|
|
Object.defineProperty(this, "unsubscribeId", {
|
|
configurable: true,
|
|
enumerable: true,
|
|
writable: true,
|
|
value: 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; |