improve logic for revert button

woocommerce · May 9, 2023 · 7730ef0 · 7730ef0
1 parent 954335b
commit 7730ef0
Showing 1 changed file with 16 additions and 4 deletions.
diff --git a/assets/js/templates/revert-button/index.tsx b/assets/js/templates/revert-button/index.tsx
@@ -12,7 +12,11 @@ import { store as blockEditorStore } from '@wordpress/block-editor';
 
 // @ts-expect-error: @wordpress/plugin is typed in the newer versions
 // eslint-disable-next-line @woocommerce/dependency-group
-import { registerPlugin, unregisterPlugin } from '@wordpress/plugins';
+import {
+	registerPlugin,
+	unregisterPlugin,
+	getPlugin,
+} from '@wordpress/plugins';
 
 /**
  * Internal dependencies
@@ -150,10 +154,18 @@ subscribe( () => {
 		PluginTemplateSettingPanel !== undefined;
 
 	if ( isWooTemplate && hasSupportForPluginTemplateSettingPanel ) {
-		registerPlugin( REVERT_BUTTON_PLUGIN_NAME, {
+		if ( getPlugin( REVERT_BUTTON_PLUGIN_NAME ) ) {
+			return;
+		}
+
+		return registerPlugin( REVERT_BUTTON_PLUGIN_NAME, {
 			render: RevertClassicTemplateButton,
 		} );
-	} else {
-		unregisterPlugin( REVERT_BUTTON_PLUGIN_NAME );
 	}
+
+	if ( getPlugin( REVERT_BUTTON_PLUGIN_NAME ) === undefined ) {
+		return;
+	}
+
+	unregisterPlugin( REVERT_BUTTON_PLUGIN_NAME );
 }, 'core/edit-site' );