first sync

core/um-access.php

@@ -0,0 +1,95 @@
+<?php
+
+class UM_Access {
+
+	function __construct() {
+	
+		// hook into template redirects [home]
+		add_action('template_redirect',  array(&$this, 'custom_homepage_per_role'), 1000);
+
+		// hook into template redirects
+		add_action('template_redirect',  array(&$this, 'post_page_access_control'), 999);
+		
+	}
+	
+	/***
+	***	@get meta
+	***/
+	function get_meta() {
+		global $post;
+		$post_id = $post->ID;
+		$meta = get_post_custom( $post_id );
+		foreach ($meta as $k => $v){
+			if ( strstr($k, '_um_') ) {
+				$k = str_replace('_um_', '', $k);
+				$array[$k] = $v[0];
+			}
+		}
+		if ( isset( $array ) )
+			return (array)$array;
+		else
+			return array('');
+	}
+	
+	/***
+	***	@custom homepage per role
+	***/
+	function custom_homepage_per_role(){
+	
+		if ( !is_user_logged_in() ) return;
+		if ( is_admin() ) return;
+		if ( um_user('default_homepage') ) return;
+		if ( !um_user('redirect_homepage') ) return;
+		
+		if( is_home() || is_front_page() )
+			exit( wp_redirect( um_user('redirect_homepage') ) );
+		
+	}
+	
+	/***
+	***	@the main restrict function
+	***/
+	function post_page_access_control(){
+		global $post;
+		
+		if ( !get_post_type() || !isset($post->ID) ) return;
+		
+		$args = $this->get_meta();
+		extract($args);
+
+		$redirect = null;
+
+		if ( !isset( $accessible ) ) return;
+		
+		switch( $accessible ) {
+			
+			case 0:
+				break;
+			
+			case 1:
+			
+				if ( is_user_logged_in() )
+					$redirect = $access_redirect2;
+				break;
+				
+			case 2:
+				
+				if ( !is_user_logged_in() )
+					$redirect = $access_redirect;
+					
+				if ( isset( $access_roles ) && !empty( $access_roles ) )
+					if ( !in_array( um_user('role'), unserialize( $access_roles ) ) )
+						$redirect = $access_redirect;
+						
+				break;
+				
+		}
+		
+		if ( $redirect ) {
+			wp_redirect( $redirect );
+			exit;
+		}
+		
+	}
+
+}