From a1b54011f9e36867e5695efd17345dbcd666f245 Mon Sep 17 00:00:00 2001
From: ultimatemember <umplugin@gmail.com>
Date: Tue, 30 Dec 2014 21:30:16 +0200
Subject: [PATCH] Login checks/validation is now applied to WP login screen

---
 core/um-actions-login.php | 29 ++++++++++++++++++++++++++++-
 core/um-filters-login.php | 35 +++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+), 1 deletion(-)

diff --git a/core/um-actions-login.php b/core/um-actions-login.php
index e0dbb759..024e4005 100644
--- a/core/um-actions-login.php
+++ b/core/um-actions-login.php
@@ -1,5 +1,32 @@
 <?php
 
+	/***
+	***	@Customize WP logo in login screen
+	***/
+	function um_custom_wplogo() {
+		$logo_url = um_url . 'admin/assets/img/logo_small.png';
+		print "<style type='text/css'>                                                                   
+				h1 a { background-image:url($logo_url) !important; border-radius: 999px !important }                            
+		</style>";
+	}
+	add_action('login_head', 'um_custom_wplogo', 999 );
+	
+	/***
+	***	@Customize WP logo url in login screen
+	***/
+	function um_custom_wplogo_url() {
+		return home_url();
+	}
+	add_filter('login_headerurl', 'um_custom_wplogo_url', 999 );
+	
+	/***
+	***	@Customize WP logo title in login screen
+	***/
+	function um_custom_wplogo_title() {
+		return get_option('blogname');
+	}
+	add_filter('login_headertitle', 'um_custom_wplogo_title', 999 );
+	
 	/***
 	***	@Error processing hook : login
 	***/
@@ -78,7 +105,7 @@
 	}
 	
 	/***
-	***	@login checks
+	***	@login checks thru the frontend login
 	***/
 	add_action('um_submit_form_errors_hook_logincheck', 'um_submit_form_errors_hook_logincheck', 9999 );
 	function um_submit_form_errors_hook_logincheck($args){
diff --git a/core/um-filters-login.php b/core/um-filters-login.php
index e69de29b..86ab4c13 100644
--- a/core/um-filters-login.php
+++ b/core/um-filters-login.php
@@ -0,0 +1,35 @@
+<?php
+
+	/***
+	***	@login checks thru the wordpress admin login
+	***/
+	add_filter('authenticate', 'um_wp_form_errors_hook_logincheck', 999, 3);
+	function um_wp_form_errors_hook_logincheck( $user, $username, $password ) {
+		
+		remove_filter( 'authenticate', 'wp_authenticate_username_password', 20, 3 );
+		
+		if ( isset( $user->ID ) ) {
+		
+			um_fetch_user( $user->ID );
+			$status = um_user('account_status');
+
+			switch( $status ) {
+				case 'inactive':
+					return new WP_Error( $status, __('Your account has been disabled.','ultimatemember') );
+					break;
+				case 'awaiting_admin_review':
+					return new WP_Error( $status, __('Your account has not been approved yet.','ultimatemember') );
+					break;
+				case 'awaiting_email_confirmation':
+					return new WP_Error( $status, __('Your account is awaiting e-mail verifications.','ultimatemember') );
+					break;
+				case 'rejected':
+					return new WP_Error( $status, __('Your membership request has been rejected.','ultimatemember') );
+					break;
+			}
+			
+		}
+
+		return wp_authenticate_username_password( null, $username, $password );
+
+	}
\ No newline at end of file