make button toggle

WordPress · Oct 18, 2022 · 79caffe · 79caffe
1 parent e10a645
commit 79caffe
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/packages/block-library/src/image/image.js b/packages/block-library/src/image/image.js
@@ -333,14 +333,23 @@ export default function Image( {
 						onChange={ updateAlignment }
 					/>
 				) }
-				{ ! isContentLocked && ! caption && (
+				{ ! isContentLocked && (
 					<ToolbarButton
 						onClick={ () => {
 							setShowCaption( ! showCaption );
+							if (
+								showCaption &&
+								! RichText.isEmpty( caption )
+							) {
+								setAttributes( { caption: undefined } );
+							}
 						} }
 						icon={ captionIcon }
-						disabled={ showCaption }
-						label={ __( 'Add caption' ) }
+						label={
+							showCaption
+								? __( 'Remove caption' )
+								: __( 'Add caption' )
+						}
 					/>
 				) }
 				{ ! multiImageSelection && ! isEditingImage && (