jQuery( function( $ ) {
	var widgets_container, widgets_selector, flags;

	if ( 'undefined' !== typeof pll_widgets && pll_widgets.hasOwnProperty( 'flags' ) ) {
		flags = pll_widgets.flags;
	}

	/**
	 * Prepend widget titles with a flag once a language is selected.
	 * @param {object} widget The widget element.
	 * @return {void} Nothing.
	 */
	function add_flag( widget ) {
		if ( ! flags ) {
			return;
		}
		widget = $( widget );
		var title  = $( '.widget-top .widget-title h3', widget ),
			locale = $( '.pll-lang-choice option:selected', widget ).val(),
			icon   = ( locale && flags.hasOwnProperty( locale ) ) ? flags[ locale ] : null;

		if ( icon ) {
			icon += '   ';
			var current = $( '.pll-lang', title );
			if ( current.length ) {
				current.html( icon );
			} else {
				flag = '<span class="pll-lang">' + icon + '</span>';
				title.prepend( flag );
			}
		} else {
			$( '.pll-lang', title ).remove();
		}
	}

	if ( 'undefined' !== typeof wp.customize ) {

		widgets_container = $( '#customize-controls' );
		widgets_selector  = '.customize-control .widget';

		/**
		 * WP Customizer add control listener.
		 *
		 * @link https://wordpress.stackexchange.com/questions/256536/callback-after-wordpress-customizer-complete-loading
		 *
		 * @param {object} control The control type.
		 * @return {void} Nothing.
		 */
		function customize_add_flag( control ) {
			if ( ! control.extended( wp.customize.Widgets.WidgetControl ) ) {
				return;
			}

			/*
			 * Make sure the widget's contents are embedded; normally this is done
			 * when the control is expanded, for DOM performance reasons.
			 */
			control.embedWidgetContent();

			// Now we know for sure the widget is fully embedded.
			add_flag( control.container.find( '.widget' ) );
		}
		wp.customize.control.each( customize_add_flag );
		wp.customize.control.bind( 'add', customize_add_flag );

	} else {

		widgets_container = $( '#widgets-right' );
		widgets_selector  = '.widget';

	}

	// Add flags on load.
	$( widgets_selector, widgets_container ).each( function() {
		add_flag( this );
	} );

	// Update flags.
	widgets_container.on( 'change', '.pll-lang-choice', function() {
		add_flag( $( this ).parents( '.widget' ) );
	} );

} );