From 3afb611c516ef5a5f82bb9c9f263dca99f4ecd7e Mon Sep 17 00:00:00 2001 From: Mustafa Bharmal Date: Tue, 17 Feb 2026 08:58:27 +0530 Subject: [PATCH] Media Utils: Auto-select uploaded files in media modal experiment --- .../src/components/media-upload-modal/index.tsx | 15 ++++++++++++++- 1 file changed, 14 insertions(+), 1 deletion(-) diff --git a/packages/media-utils/src/components/media-upload-modal/index.tsx b/packages/media-utils/src/components/media-upload-modal/index.tsx index 13b694a01726f8..5a8953a38f8176 100644 --- a/packages/media-utils/src/components/media-upload-modal/index.tsx +++ b/packages/media-utils/src/components/media-upload-modal/index.tsx @@ -373,6 +373,19 @@ export function MediaUploadModal( { } ); + // Auto-select the newly uploaded items + const uploadedIds = attachments + .map( ( attachment ) => String( attachment.id ) ) + .filter( Boolean ); + + if ( multiple ) { + // In multiple mode, add to existing selection + setSelection( ( prev ) => [ ...prev, ...uploadedIds ] ); + } else { + // In single mode, replace selection with the first uploaded item + setSelection( uploadedIds.slice( 0, 1 ) ); + } + // Invalidate the entity records resolution to refresh the view invalidateResolution( 'getEntityRecords', [ 'postType', @@ -381,7 +394,7 @@ export function MediaUploadModal( { ] ); } }, - [ createSuccessNotice, invalidateResolution, queryArgs ] + [ createSuccessNotice, invalidateResolution, queryArgs, multiple ] ); // Shared upload error handler