Ticket #561: 561.diff

wp-content/plugins/wc-post-types/wc-post-types.php

@@ -256 +256 @@
                         'posts_per_page' => -1,
                         'orderby' => 'date',
                         'order' => 'desc',
+                        'track' => 'all',
                 ), $attr );
 
                 $attr['show_avatars'] = $this->str_to_bool( $attr['show_avatars'] );
                 $attr['orderby'] = ( in_array( $attr['orderby'], array( 'date', 'title', 'rand' ) ) ) ? $attr['orderby'] : 'date';
                 $attr['order'] = ( in_array( $attr['order'], array( 'asc', 'desc') ) ) ? $attr['order'] : 'desc';
 
-                $speakers = new WP_Query( array(
+                $args = array(
                         'post_type' => 'wcb_speaker',
                         'posts_per_page' => intval( $attr['posts_per_page'] ),
                         'orderby' => $attr['orderby'],
                         'order' => $attr['order'],
-                ) );
+                );
 
+                // Fetch speakers from specified tracks only
+                if ( 'all' != $attr['track'] ) {
+                        $tracks = explode( ',', $attr['track'] );
+                        $args['meta_query'][] = array(
+                                'key' => '_wcpt_speaker_tracks',
+                                'value' => $tracks,
+                                'compare' => 'IN',
+                        );
+                }
+
+                $speakers = new WP_Query( $args );
+
                 if ( ! $speakers->have_posts() )
                         return '';
 
@@ -279 +292 @@
 
                         <?php while ( $speakers->have_posts() ) : $speakers->the_post(); ?>
 
-                                <div id="wcorg-speaker-<?php echo sanitize_html_class( $post->post_name ); ?>" class="wcorg-speaker">
+                                <?php
+                                // Get list of speaker tracks for div class
+                                $tracks = get_post_meta( get_the_ID(), '_wcpt_speaker_tracks', false );
+                                $track_class = '';
+                                foreach( $tracks as $track ) {
+                                        $track_class .= ' ' . sanitize_html_class( $track );
+                                }
+                                ?>
+
+                                <div id="wcorg-speaker-<?php echo sanitize_html_class( $post->post_name ); ?>" class="wcorg-speaker<?php echo $track_class; ?>">
                                         <h2><?php the_title(); ?></h2>
                                         <div class="wcorg-speaker-description">
                                                 <?php echo ( $attr['show_avatars'] ) ? get_avatar( get_post_meta( get_the_ID(), '_wcb_speaker_email', true ), absint( $attr['avatar_size'] ) ) : ''; ?>
@@ -1142 +1164 @@
         }
 
         /**
+         * Adds session tracks to all speakers
+         * @return void
+         */
+        function save_all_speaker_tracks() {
+                $args = array(
+                        'post_type' => 'wcb_speaker',
+                        'posts_per_page' => -1,
+                );
+
+                $speakers = get_posts( $args );
+
+                foreach( $speakers as $speaker ) {
+                        $this->save_single_speaker_tracks( $speaker->ID );
+                }
+        }
+
+        /**
+         * Adds session tracks to specified speaker
+         * @param  integer  $speaker_id   ID of speaker
+         * @return void
+         */
+        function save_single_speaker_tracks( $speaker_id = 0 ) {
+
+                if( ! $speaker_id ) return;
+
+                $args = array(
+                        'post_type' => 'wcb_session',
+                        'posts_per_page' => -1,
+                        'meta_query' => array(
+                                array(
+                                        'key' => '_wcpt_speaker_id',
+                                        'value' => $speaker_id,
+                                ),
+                        ),
+                );
+
+                $sessions = get_posts( $args );
+
+                // Delete all existing tracks from speaker
+                delete_post_meta( $speaker_id, '_wcpt_speaker_tracks' );
+
+                foreach( $sessions as $session ) {
+
+                        // Get all tracks for session
+                        $session_tracks = wp_get_post_terms( $session->ID, 'wcb_track' );
+
+                        if( ! is_wp_error( $session_tracks ) ) {
+
+                                // Add each track from session to speaker
+                                foreach( $session_tracks as $track ) {
+                                        add_post_meta( $speaker_id, '_wcpt_speaker_tracks', $track->slug, false );
+                                }
+
+                        }
+                }
+        }
+
+        /**
          * Get WordPress.org User
          *
          * Give it login or slug, will return a WP_User object if found.
@@ -1234 +1314 @@
                 delete_post_meta( $post_id, '_wcpt_speaker_id' );
                 foreach ( $speaker_ids as $speaker_id )
                         add_post_meta( $post_id, '_wcpt_speaker_id', $speaker_id );
-                
+
                 // Set the speaker as the author of the session post, so the single
                 // view doesn't confuse users who see "posted by [organizer name]"
                 foreach ( $speaker_ids as $speaker_post ) {
                         $wporg_user_id = get_post_meta( $speaker_post, '_wcpt_user_id', true );
                         $user = get_user_by( 'id', $wporg_user_id );
-                        
+
                         if ( $user ) {
                                 remove_action( 'save_post', array( $this, 'save_post_session' ), 10, 2 );       // avoid infinite recursion
                                 wp_update_post( array(
@@ -1248 +1328 @@
                                         'post_author' => $user->ID
                                 ) );
                                 add_action( 'save_post', array( $this, 'save_post_session' ), 10, 2 );
-                                
+
                                 break;
                         }
                 }
+
+                // Save session tracks as speaker meta for all speakers
+                $this->save_all_speaker_tracks();
         }
 
         /**