Files
DESKTOP-GBA0BK8\Admin 7c8c8b1c76 first commit
2023-04-08 12:19:53 -04:00

159 lines
4.5 KiB
PHP

<?php
/*
Copyright 2013-2019 Yellow Tree, Siegen, Germany
Author: Benjamin Pick (wp-geoip-detect| |posteo.de)
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*/
namespace YellowTree\GeoipDetect\DataSources\Header;
use YellowTree\GeoipDetect\DataSources\AbstractDataSource;
class HeaderReader extends \YellowTree\GeoipDetect\DataSources\AbstractReader {
protected $providers = array(
'aws' => 'Amazon AWS CloudFront',
'cloudflare' => 'Cloudflare',
);
public function country($ip) {
$r = array();
$isoCode = '';
switch ($this->options['provider']) {
case 'aws':
if (isset($_SERVER['CloudFront-Viewer-Country'])) {
$isoCode = $_SERVER['CloudFront-Viewer-Country'];
}
break;
case 'cloudflare';
if (isset($_SERVER["HTTP_CF_IPCOUNTRY"])) {
$isoCode = $_SERVER["HTTP_CF_IPCOUNTRY"];
if ($isoCode == 'xx' /* not a country / unknown */)
$isoCode = '';
}
break;
}
$country = '';
if (!$isoCode) {
return null;
}
$r['country']['iso_code'] = strtoupper($isoCode);
$r['traits']['ip_address'] = $ip;
$record = new \GeoIp2\Model\City($r, array('en'));
return $record;
}
}
class HeaderDataSource extends AbstractDataSource {
public function getId() { return 'header'; }
public function getLabel() { return __('Special Hosting Providers (Cloudflare, Amazon AWS CloudFront)', 'geoip-detect'); }
public function getDescriptionHTML() { return __('These servers already do geodetection, but only of the visitor\'s country.', 'geoip-detect'); }
public function getStatusInformationHTML() {
$provider = get_option('geoip-detect-header-provider');
$html = '';
$link = '';
if ($provider == 'cloudflare') {
$link = 'https://support.cloudflare.com/hc/en-us/articles/200168236-What-does-CloudFlare-IP-Geolocation-do-';
} elseif ($provider == 'aws') {
$link = 'https://aws.amazon.com/blogs/aws/enhanced-cloudfront-customization/';
}
if ($link)
$html = sprintf(__('This needs to be enabled in the admin panel: see <a href="%s">Help</a>.', 'geoip-detect'), $link);
return $html;
}
public function getParameterHTML() {
$provider = get_option('geoip-detect-header-provider');
$checked_cloudflare = $provider == 'cloudflare' ? 'checked' : '';
$checked_aws = $provider == 'aws' ? 'checked' : '';
$label = __('Which Hosting Provider:', 'geoip-detect');
$html = <<<HTML
<p>$label<br>
<label><input type="radio" name="options_header[provider]" value="cloudflare" $checked_cloudflare /> Cloudflare</label>
<label><input type="radio" name="options_header[provider]" value="aws" $checked_aws /> Amazon AWS CloudFront</label>
</p>
<br />
HTML;
return $html;
}
public function saveParameters($post) {
$message = '';
$value = isset($post['options_header']['provider']) ? $post['options_header']['provider'] : '';
if (!empty($value)) {
update_option('geoip-detect-header-provider', $value);
}
return $message;
}
public function getShortLabel() {
$provider = get_option('geoip-detect-header-provider');
$labels = array(
'' => __('None', 'geoip-detect'),
'aws' => 'Amazon AWS CloudFront',
'cloudflare' => 'Cloudflare',
);
if (!isset($labels[$provider]))
$provider = '';
$html = __('Hosting Provider:', 'geoip-detect') . ' ' . $labels[$provider];
return $html;
}
public function getReader($locales = array('en'), $options = array()) {
$reader = null;
$provider = get_option('geoip-detect-header-provider');
if ($provider) {
try {
$reader = new HeaderReader( array(
'provider' => $provider,
) );
} catch ( \Exception $e ) {
if (WP_DEBUG)
echo printf(__('Error while creating reader for "%s": %s', 'geoip-detect'), $filename, $e->getMessage ());
}
}
return $reader;
}
public function isWorking() {
$working = (bool) get_option('geoip-detect-header-provider');
return $working;
}
}
geoip_detect2_register_source(new HeaderDataSource());