Redirect to login if not logged in or if not PMPro membership page.

redirect-to-login-if-not-logged-in.php

<?php
/**
 * Redirect to login or homepage if user is logged out or not a member
 * Add this code to your active theme's functions.php file.
 */
function my_template_redirect() {
	global $current_user;

	$okay_pages = array(
		pmpro_getOption( 'billing_page_id' ),
		pmpro_getOption( 'account_page_id' ),
		pmpro_getOption( 'levels_page_id' ),
		pmpro_getOption( 'checkout_page_id' ),
		pmpro_getOption( 'confirmation_page_id' ),
	);

	// if the user doesn't have a membership, send them home
	if ( ! $current_user->ID
		// && ! is_home()
		&& ! is_page( $okay_pages )
		&& ! strpos( $_SERVER['REQUEST_URI'], 'login' ) ) {

		wp_redirect( home_url( 'wp-login.php?redirect_to=' . urlencode( $_SERVER['REQUEST_URI'] ) ) );

	} elseif ( is_page()
		&& ! is_home()
		&& ! is_page( $okay_pages )
		&& ! $current_user->membership_level->ID ) {
		wp_redirect( home_url() );
	}
}
add_action( 'template_redirect', 'my_template_redirect' );