Add support for onInsertBlock with BlockChooser

eea · Apr 7, 2021 · 0e37fb3 · 0e37fb3
1 parent e7d8824
commit 0e37fb3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/src/components/manage/Blocks/Accordion/EditBlockWrapper.jsx b/src/components/manage/Blocks/Accordion/EditBlockWrapper.jsx
@@ -72,7 +72,9 @@ class EditBlockWrapper extends React.Component {
       block,
       data,
       onDeleteBlock,
+      onInsertBlock,
       onMutateBlock,
+      onSelectBlock,
       selected,
     } = blockProps;
     const type = data['@type'];
@@ -148,8 +150,12 @@ class EditBlockWrapper extends React.Component {
                   {this.state.addNewBlockOpened && (
                     <BlockChooser
                       onMutateBlock={(id, value) => {
-                        this.setState({ addNewBlockOpened: false });
                         onMutateBlock(id, value);
+                        this.setState({ addNewBlockOpened: false });
+                      }}
+                      onInsertBlock={(id, value) => {
+                        onSelectBlock(onInsertBlock(id, value));
+                        this.setState({ addNewBlockOpened: false });
                       }}
                       currentBlock={block}
                       allowedBlocks={allowedBlocks}