Hustle - geolocation integration with Defender plugin

hustle-defender-geoip-integration.php

<?php 

/**
* Plugin Name: Hustle - geolocation integration with Defender plugin 
* Plugin URI:  https://gist.github.com/adczk
* Description: Makes Hustle use Defender  plugin country detection for country-based visibility rules
* Author:      adczk
*
* Author URI:  https://gist.github.com/adczk
* License:     GPLv2 or later
*
* Use as MU plugin; 
*
* Tested with Hustle 7.8.4 and Defender Pro 4.7.1
* REQUIRES DEFENDER TO BE ACTIVE AND HAVE MAXMIND DB CONFIGURED AND DOWNLOADED
*
*/

add_filter( 'hustle_get_user_country', 'hustle_get_geoip_from_defender', 11, 2);
function hustle_get_geoip_from_defender( $country, $ip ) {
	
    // defaults to null in case detection fails
	$country = null;
	
	// if Defender active
	if ( class_exists( 'WP_Defender\Bootstrap' ) ) {
		
		// if MaxMind DB is downloaded
		$hgip_service = wd_di()->get( \WP_Defender\Component\Blacklist_Lockout::class );
		if ( $hgip_service->is_geodb_downloaded() ) {
						
			$hgip_model = wd_di()->get( \WP_Defender\Model\Setting\Blacklist_Lockout::class );
			// Additional check if MaxMind dir is deleted.
			if ( is_null( $hgip_model->geodb_path ) || ! is_file( $hgip_model->geodb_path ) ) {
				// if so, skip detection
				return $country;
			}
			
			// all set so let's geolocate IP
			$defgeo = new WP_Defender\Extra\GeoIp( $hgip_model->geodb_path );
			$geo = $defgeo->ip_to_country( $ip );
			if ( is_array($geo) ) {
				$country = $geo['iso'];
			}			
		}
	} 
	
	return $country;
	
}