@secretstache/wordpress-gutenberg 0.6.2 → 0.6.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@secretstache/wordpress-gutenberg",
-  "version": "0.6.2",
+  "version": "0.6.3",
   "description": "",
   "author": "Secret Stache",
   "license": "GPL-2.0-or-later",

package/src/utils/helpers.js

@@ -234,7 +234,25 @@ export const getFiltersByNamespace = (namespace) => {
 };
 
 /**
- * Unregisters a block type for a specific post type when editor loads.
+ * Allow a block for a specific post type
+ * @param {string} blockName - Name of the block to unregister
+ * @param {string} postType - Post type to check against
+ */
+export const setBlockForPostType = (blockName, postType) => {
+    const unsubscribe = subscribe(
+        () => {
+            const currentPostType = select('core/editor').getCurrentPostType();
+            if (currentPostType && currentPostType !== postType && getBlockType(blockName)) {
+                unregisterBlockType(blockName);
+                unsubscribe();
+            }
+        },
+        'core/editor',
+    );
+};
+
+/**
+ * Disallow a block for a specific post type
  * @param {string} blockName - Name of the block to unregister
  * @param {string} postType - Post type to check against
  */
@@ -242,7 +260,7 @@ export const unsetBlockForPostType = (blockName, postType) => {
     const unsubscribe = subscribe(
         () => {
             const currentPostType = select('core/editor').getCurrentPostType();
-            if (currentPostType === postType && getBlockType(blockName)) {
+            if (currentPostType && currentPostType === postType && getBlockType(blockName)) {
                 unregisterBlockType(blockName);
                 unsubscribe();
             }