<?php
/*************************************************************
 * 
 * user.class.php
 * 
 * Add Users
 * 
 * 
 * Copyright (c) 2011 Prelovac Media
 * www.prelovac.com
 **************************************************************/

class MMB_User extends MMB_Core
{
    function __construct()
    {
        parent::__construct();
    }
    
    function get_users($args){
			global $wpdb;
			
			//$args: $user_roles;
			if(empty($args))
				return false;
				
			extract($args);
		
		$userlevels = array();
		$level_strings = array();
		foreach($user_roles as $user_role){
		switch(strtolower($user_role)){
			case 'subscriber' : $userlevels[] = 0; $level_strings[] = $user_role; break;
			case 'contributor' : $userlevels[] = 1; $level_strings[] = $user_role; break;
			case 'author' : $userlevels[] = 2; $level_strings[] = $user_role; break;
			case 'editor' : $userlevels[] = 7; $level_strings[] = $user_role; break;
			case 'administrator' : $userlevels[] = 10; $level_strings[] = $user_role; break;
			default: break;
			}
		}
				
		$users = array();
		$userlevel_qry = "('".implode("','",$userlevels)."')";
		$userlevel_fallback_qry = "('%".implode("%','%",$level_strings)."%')";
		$field = $wpdb->prefix."capabilities";
		
		$user_metas = $wpdb->get_results("SELECT * from $wpdb->usermeta WHERE meta_key = '$field' AND meta_value IN $userlevel_fallback_qry");
		if($user_metas == false || empty($user_metas)){
			$user_metas = $wpdb->get_results("SELECT * from $wpdb->usermeta WHERE meta_key = 'wp_user_level' AND meta_value IN $userlevel_qry");	
		}
		
		$include = array();
			if(is_array($user_metas) && !empty($user_metas)){
				foreach($user_metas as $user_meta){
					$include[] = $user_meta->user_id;
				}
			}
			
			$args = array();
			$args['include'] = $include;
			$args['fields'] = 'all_with_meta';
			$temp_users = get_users($args);
			$user = array();
			foreach ((array)$temp_users as $temp){
				$user['user_id'] = $temp->ID;
				$user['user_login'] = $temp->user_login;
				$user['wp_capabilities'] = array_keys($temp->$field);
				$users[] = $user;
			}
			
		return array('users' => $users);
 }
    
    function add_user($args)
    {
    	
    	if(!function_exists('username_exists') || !function_exists('email_exists'))
    	 include_once(ABSPATH . WPINC . '/registration.php');
      
      if(username_exists($args['user_login']))
    	 return array('error' => 'Username already exists');
    	
    	if (email_exists($args['user_email']))
    		return array('error' => 'Email already exists');
    	
			if(!function_exists('wp_insert_user'))
			 include_once (ABSPATH . 'wp-admin/includes/user.php');
			
			$user_id =  wp_insert_user($args);
			
			if($user_id){
			
				if($args['email_notify']){
					//require_once ABSPATH . WPINC . '/pluggable.php';
					wp_new_user_notification($user_id, $args['user_pass']);
				}
				return $user_id;
			}else{
				return array('error' => 'User not added. Please try again.');
			}
			 
    }
    
    function edit_users($args){
    	
    	if(empty($args))
    		return false;
    		if(!function_exists('get_user_to_edit'))
			 		include_once (ABSPATH . 'wp-admin/includes/user.php');
			 	if(!function_exists('wp_update_user'))
			 		include_once (ABSPATH . WPINC.'/user.php');
			 
    	 extract($args);
    	 //$args: $users, $new_role, $new_password, $user_edit_action
    	 
    	 $return = array();
    	 if(count($users)){
    	 foreach($users as $user){
    	 	$result = '';
    	 	$user_obj = $this->mmb_get_user_info( $user );
    	 	if($user_obj != false){
		    	 switch($user_edit_action){
		    		case 'change-password':
		    		 if($new_password){ 
		    		 	$user_data = array();
		    		 	$userdata['user_pass'] = $new_password;
		    	 	 	$userdata['ID'] = $user_obj->ID;
		    	   	$result = wp_update_user($userdata);
		    	  	} else {
		    	  		$result = array('error' => 'No password provided.');
		    	  	}
		    		 break;
		    		case 'change-role':
		    		 if($new_role){
		    		 	if($user != $username){
		    			if(!$this->last_admin($user_obj)){
		    				$user_data = array(); 
		    	 	 		$userdata['ID'] = $user_obj->ID;
		    	 	 		$userdata['role'] = strtolower($new_role);
		    	   		$result = wp_update_user($userdata);
		    	  	} else {
		    	  		$result = array('error' => 'Cannot change role to the only one left admin user.');
		    	  		}
		    	  	} else {
		    	  		$result = array('error' => 'Cannot change role to user assigned for ManageWP.');
		    	  	} 
		    	 	} else {
		    	 		$result = array('error' => 'No role provided.');
		    	 	}
		    			break;
		    		case 'delete-user':
		    			if($user != $username){
			    			if(!$this->last_admin($user_obj)){
				    			if($reassign_user){
				    			$to_user = $this->mmb_get_user_info( $reassign_user );
				    				if($to_user != false){
				    					$result = wp_delete_user($user_obj->ID, $to_user->ID);
				    				} else {
				    					$result = array('error' => 'User not deleted. User to reassign posts doesn\'t exist.');
				    				}
				    			} else {
				    				$result = wp_delete_user($user_obj->ID);
				    			}
				    		} else {
				    			$result = array('error' => 'Cannot delete the only one left admin user.');
				    		}
			    		} else {
			    			$result = array('error' => 'Cannot delete user assigned for ManageWP.');
			    		}
		    		
		    			break;
		    		default:
		    			$result = array('error' => 'Wrong action provided. Please try again.');
		    			break;
		    		}
    			} else {
    				$result = array('error' => 'User not found.');
    			}
    			
    			if(is_wp_error($result)){
    				$result = array('error' => $result->get_error_message()); 
    			}
    			
    			$return[$user] = $result; 
    		}
    	}
    	
    	return $return;
    		
    }
    
    //Check if user is the only one admin on the site
    function last_admin($user_obj){
    	global $wpdb;
    	$field = $wpdb->prefix."capabilities";
    	$capabilities = array_map('strtolower',array_keys($user_obj->$field));
    	$result = count_users();
	    	if(in_array('administrator',$capabilities)){
	    		
	    		if(!function_exists('count_users')){
				 		include_once (ABSPATH . WPINC. '/user.php');
					}
					
	    		$result = count_users();
	    		if($result['avail_roles']['administrator'] == 1){
	    			return true;
	    		}	
	    	}
    		return false;
    }
    
}
?>