Add Ribbon to Paid blocks in block picker

Automattic · Oct 10, 2019 · fbd5fa2 · fbd5fa2
1 parent fefcec9
commit fbd5fa2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/extensions/shared/register-jetpack-block.js b/extensions/shared/register-jetpack-block.js
@@ -1,8 +1,10 @@
 /**
  * External dependencies
  */
+import { __ } from '@wordpress/i18n';
 import { addFilter } from '@wordpress/hooks';
 import { registerBlockType } from '@wordpress/blocks';
+import { Ribbon } from '@automattic/calypso-ui';
 
 /**
  * Internal dependencies
@@ -48,6 +50,14 @@ export default function registerJetpackBlock( name, settings, childBlocks = [] )
 		...settings,
 		title: betaExtensions.includes( name ) ? `${ settings.title } (beta)` : settings.title,
 		edit: requiredPlan ? wrapPaidBlock( { requiredPlan } )( settings.edit ) : settings.edit,
+		icon: requiredPlan ? (
+			<>
+				<Ribbon>{ __( 'Paid' ) }</Ribbon>
+				{ settings.icon }
+			</>
+		) : (
+			settings.icon
+		),
 	} );
 
 	if ( requiredPlan ) {