Fix: Align Hook: Impossible to set no alignment when a default align …

…exists
WordPress · Sep 21, 2018 · d36d40a · d36d40a
1 parent a8bc550
commit d36d40a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/packages/editor/src/hooks/align.js b/packages/editor/src/hooks/align.js
@@ -9,7 +9,7 @@ import { assign, get, includes } from 'lodash';
  */
 import { createHigherOrderComponent } from '@wordpress/compose';
 import { addFilter } from '@wordpress/hooks';
-import { hasBlockSupport, getBlockSupport } from '@wordpress/blocks';
+import { hasBlockSupport, getBlockSupport, getBlockType } from '@wordpress/blocks';
 
 /**
  * Internal dependencies
@@ -77,7 +77,17 @@ export const withToolbarControls = createHigherOrderComponent( ( BlockEdit ) =>
 	return ( props ) => {
 		const validAlignments = getBlockValidAlignments( props.name );
 
-		const updateAlignment = ( nextAlign ) => props.setAttributes( { align: nextAlign } );
+		const updateAlignment = ( nextAlign ) => {
+			if ( ! nextAlign ) {
+				const blockType = getBlockType( props.name );
+				const blockDefaultAlign = get( blockType, [ 'attributes', 'align', 'default' ] );
+				if ( blockDefaultAlign ) {
+					props.setAttributes( { align: 'none' } );
+					return;
+				}
+			}
+			props.setAttributes( { align: nextAlign } );
+		};
 
 		return [
 			validAlignments.length > 0 && props.isSelected && (