-
Notifications
You must be signed in to change notification settings - Fork 4.2k
/
with-registry-provider.js
55 lines (49 loc) · 1.25 KB
/
with-registry-provider.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
/**
* WordPress dependencies
*/
import { useState, useEffect } from '@wordpress/element';
import {
withRegistry,
createRegistry,
RegistryProvider,
} from '@wordpress/data';
import { createHigherOrderComponent } from '@wordpress/compose';
import { storeConfig as blockEditorStoreConfig } from '@wordpress/block-editor';
/**
* Internal dependencies
*/
import { storeConfig } from '../../store';
const withRegistryProvider = createHigherOrderComponent(
( WrappedComponent ) =>
withRegistry( ( props ) => {
const {
useSubRegistry = true,
registry,
...additionalProps
} = props;
if ( ! useSubRegistry ) {
return <WrappedComponent { ...additionalProps } />;
}
const [ subRegistry, setSubRegistry ] = useState( null );
useEffect( () => {
const newRegistry = createRegistry(
{
'core/block-editor': blockEditorStoreConfig,
},
registry
);
newRegistry.registerStore( 'core/editor', storeConfig );
setSubRegistry( newRegistry );
}, [ registry ] );
if ( ! subRegistry ) {
return null;
}
return (
<RegistryProvider value={ subRegistry }>
<WrappedComponent { ...additionalProps } />
</RegistryProvider>
);
} ),
'withRegistryProvider'
);
export default withRegistryProvider;