Redirect away from PMPro subsite

pmpro-redirect-away-from-subsite.php

<?php
/**
 * Plugin Name: PMPro Customizations
 */

add_action( 'template_redirect', 'redirect_away_from_site_two' );
function redirect_away_from_site_two() {
	global $current_user;
	// Administrators are unaffected by the redirection.
	if ( is_user_logged_in() && current_user_can( 'manage_options' ) ) {
		return;
	}
	$blog_id = get_current_blog_id();
	// Set the location of the redirection.
	$location = network_site_url();

	// Redirect visitors away.
	if ( ! is_user_logged_in() && 2 === $blog_id ) {
		wp_safe_redirect( $location );
		exit;
	}

	// Redirect members away from site 2 if not in level 2.
	if ( is_user_logged_in() ) {
		$level = pmpro_getMembershipLevelForUser( $current_user->ID );
		if ( 2 === $blog_id && ! in_array( $level->id, array( 2 ) ) ) {
			wp_safe_redirect( $location );
			exit;
		}
	}
}