Fix for default labels

10up · Jun 25, 2024 · aa6e141 · aa6e141
1 parent 2510b9d
commit aa6e141
Showing 1 changed file with 11 additions and 8 deletions.
diff --git a/components/media-toolbar/index.tsx b/components/media-toolbar/index.tsx
@@ -36,6 +36,12 @@ interface MediaToolbarProps {
 	};
 }
 
+const DEFAULT_LABELS = {
+	add: __('Add Image', '10up-block-components'),
+	remove: __('Remove Image', '10up-block-components'),
+	replace: __('Replace Image', '10up-block-components'),
+};
+
 /*
  * MediaToolbar
  *
@@ -49,14 +55,11 @@ export const MediaToolbar: React.FC<MediaToolbarProps> = ({
 	onRemove,
 	isOptional = false,
 	id,
-	labels = {
-		add: __('Add Image', '10up-block-components'),
-		remove: __('Remove Image', '10up-block-components'),
-		replace: __('Replace Image', '10up-block-components'),
-	},
+	labels = {},
 }) => {
 	const hasImage = !!id;
 	const { media } = useMedia(id);
+	const mergedLabels = { ...DEFAULT_LABELS, ...labels };
 
 	return (
 		<ToolbarGroup>
@@ -65,11 +68,11 @@ export const MediaToolbar: React.FC<MediaToolbarProps> = ({
 					<MediaReplaceFlow
 						mediaUrl={media?.source_url}
 						onSelect={onSelect}
-						name={labels.replace}
+						name={mergedLabels.replace}
 					/>
 					{!!isOptional && (
 						<ToolbarButton onClick={onRemove}>
-							{labels.remove}
+							{mergedLabels.remove}
 						</ToolbarButton>
 					)}
 				</>
@@ -79,7 +82,7 @@ export const MediaToolbar: React.FC<MediaToolbarProps> = ({
 						onSelect={onSelect}
 						render={({ open }) => (
 							<ToolbarButton onClick={open}>
-								{labels.add}
+								{mergedLabels.add}
 							</ToolbarButton>
 						)}
 					/>