@elementor/editor-props 0.3.0 → 0.4.0

package/src/utils/create-prop-utils.ts

@@ -1,7 +1,14 @@
-import { z, type ZodTypeAny } from '@elementor/schema';
+import { z, type ZodType, type ZodTypeAny } from '@elementor/schema';
+
+import { type PropValue } from '../types';
 
 type Updater< T > = ( prev?: T ) => T;
 
+export type CreateOptions = {
+	base?: unknown;
+	disabled?: boolean;
+};
+
 /**
  * Usage example:
  *
@@ -10,32 +17,46 @@ type Updater< T > = ( prev?: T ) => T;
  *
  * elementsPropUtils.isValid( element.props?.children );
  * elementsPropUtils.create( [ 'a', 'b' ] );
- * elementsPropUtils.create( ( prev = [] ) => [ ...prev, 'c' ], element.props?.children );
+ * elementsPropUtils.create( ( prev = [] ) => [ ...prev, 'c' ], { base: element.props?.children } );
+ * elementsPropUtils.create( ( prev = [] ) => [ ...prev, 'c' ], { disabled: true } );
+ * elementsPropUtils.extract( element.props?.children );
+ *
  * ```
  */
 
+export type PropTypeUtil< TKey extends string, TValue extends PropValue > = ReturnType<
+	typeof createPropUtils< TKey, ZodType< TValue > >
+>;
+
 export function createPropUtils< TKey extends string, TValue extends ZodTypeAny >( key: TKey, valueSchema: TValue ) {
-	const schema = z.object( {
+	const schema = z.strictObject( {
 		$$type: z.literal( key ),
 		value: valueSchema,
+		disabled: z.boolean().optional(),
 	} );
 
 	type Prop = z.infer< typeof schema >;
 
+	type TPropValue = z.infer< typeof valueSchema >;
+
 	function isValid( prop: unknown ): prop is Prop {
 		return schema.safeParse( prop ).success;
 	}
 
-	function create( value: Prop[ 'value' ] ): Prop;
-	function create( value: Updater< Prop[ 'value' ] >, base: unknown ): Prop;
-	function create( value: Prop[ 'value' ] | Updater< Prop[ 'value' ] >, base?: unknown ): Prop {
-		const fn = ( typeof value === 'function' ? value : () => value ) as Updater< Prop[ 'value' ] >;
+	function create( value: TPropValue ): Prop;
+	function create( value: TPropValue, createOptions?: CreateOptions ): Prop;
+	function create( value: Updater< TPropValue >, createOptions: CreateOptions ): Prop;
+	function create( value: TPropValue | Updater< TPropValue >, createOptions?: CreateOptions ): Prop {
+		const fn = ( typeof value === 'function' ? value : () => value ) as Updater< TPropValue >;
+
+		const { base, disabled } = createOptions || {};
 
 		if ( ! base ) {
 			return {
 				$$type: key,
 				value: fn(),
-			};
+				...( disabled && { disabled } ),
+			} as Prop;
 		}
 
 		if ( ! isValid( base ) ) {
@@ -45,12 +66,23 @@ export function createPropUtils< TKey extends string, TValue extends ZodTypeAny
 		return {
 			$$type: key,
 			value: fn( base.value ),
-		};
+			...( disabled && { disabled } ),
+		} as Prop;
+	}
+
+	function extract( prop: unknown ): TPropValue | null {
+		if ( ! isValid( prop ) ) {
+			return null;
+		}
+
+		return prop.value;
 	}
 
 	return {
+		extract,
 		isValid,
 		create,
-		schema,
+		// this type fails in build due to zod issue that does not recognize the schema as a ZodType.
+		schema: schema as ZodType< Prop >,
 	};
 }