CaffGeek/MBACNationals

window.wp = window.wp || {};

(function( $, wp ) {

    wp.updates = {};

    /**
     * Decrement update counts throughout the various menus
     *
     * @param {string} updateType
     */
    wp.updates.decrementCount = function( upgradeType ) {
        var count, pluginCount, $elem;

        $elem = $( '#wp-admin-bar-updates .ab-label' );
        count = $elem.text();
        count = parseInt( count, 10 ) - 1;
        if ( count < 0 ) {
            return;
        }
        $( '#wp-admin-bar-updates .ab-item' ).removeAttr( 'title' );
        $elem.text( count );

        $elem = $( 'a[href="update-core.php"] .update-plugins' );
        $elem.each( function( index, elem ) {
            elem.className = elem.className.replace( /count-\d+/, 'count-' + count );
        } );
        $elem.removeAttr( 'title' );
        $elem.find( '.update-count' ).text( count );

        if ( 'plugin' === upgradeType ) {
            $elem = $( '#menu-plugins' );
            pluginCount = $elem.find( '.plugin-count' ).eq(0).text();
            pluginCount = parseInt( pluginCount, 10 ) - 1;
            if ( count < 0 ) {
                return;
            }
            $elem.find( '.plugin-count' ).text( pluginCount );
            $elem.find( '.update-plugins' ).each( function( index, elem ) {
                elem.className = elem.className.replace( /count-\d+/, 'count-' + pluginCount );
            } );
        }
    };

    $( window ).on( 'message', function( e ) {
        var event = e.originalEvent,
            message,
            loc = document.location,
            expectedOrigin = loc.protocol + '//' + loc.hostname;

        if ( event.origin !== expectedOrigin ) {
            return;
        }

        message = $.parseJSON( event.data );

        if ( typeof message.action === 'undefined' || message.action !== 'decrementUpdateCount' ) {
            return;
        }

        wp.updates.decrementCount( message.upgradeType );

    } );

})( jQuery, window.wp );