fix: Don't hard fail when block is missing

eea · Apr 22, 2024 · 4b38c0e · 4b38c0e
1 parent 00c4627
commit 4b38c0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/index.js b/src/index.js
@@ -21,7 +21,7 @@ import './styles.less';
 export const applyStyleWrapperToBlock = (blockConfig) => {
   const BaseEditComponent = blockConfig.edit;
   let EditComponent = BaseEditComponent;
-  if (!EditComponent._styleWrapped) {
+  if (EditComponent && !EditComponent._styleWrapped) {
     EditComponent = (props) => (
       <BlockStyleWrapperEdit {...props}>
         <BaseEditComponent {...props} />
@@ -33,7 +33,7 @@ export const applyStyleWrapperToBlock = (blockConfig) => {
 
   const BaseViewComponent = blockConfig.view;
   let ViewComponent = BaseViewComponent;
-  if (!ViewComponent._styleWrapped) {
+  if (ViewComponent && !ViewComponent._styleWrapped) {
     ViewComponent = (props) => (
       <BlockStyleWrapperView {...props}>
         <BaseViewComponent {...props} />