feat: introduce flag for preserving the unit

WordPress · Jan 16, 2025 · 0c178dd · 0c178dd
1 parent 49eee39
commit 0c178dd
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 1 deletion.
diff --git a/packages/components/src/unit-control/index.tsx b/packages/components/src/unit-control/index.tsx
@@ -57,6 +57,7 @@ function UnforwardedUnitControl(
 		value: valueProp,
 		onFocus: onFocusProp,
 		__shouldNotWarnDeprecated36pxSize,
+		shouldPreserveUnit = false,
 		...props
 	} = useDeprecated36pxDefaultSizeProp( unitControlProps );
 
@@ -136,7 +137,13 @@ function UnforwardedUnitControl(
 			typeof nextQuantityValue === 'undefined' ||
 			nextQuantityValue === null
 		) {
-			onChangeProp?.( `0${ unit }`, changeProps );
+			// Add a check if we should preserve the unit when the quantity is empty.
+			if ( shouldPreserveUnit ) {
+				onChangeProp?.( `0${ unit }`, changeProps );
+				return;
+			}
+
+			onChangeProp?.( '', changeProps );
 			return;
 		}
 

diff --git a/packages/components/src/unit-control/types.ts b/packages/components/src/unit-control/types.ts
@@ -114,4 +114,10 @@ export type UnitControlProps = Pick< InputControlProps, 'size' > &
 		 * @ignore
 		 */
 		__shouldNotWarnDeprecated36pxSize?: boolean;
+		/**
+		 * Whether to preserve the unit when the value is empty.
+		 *
+		 * @default false
+		 */
+		shouldPreserveUnit?: boolean;
 	};