first commit
This commit is contained in:
DESKTOP-GBA0BK8\Admin
@@ -0,0 +1,259 @@
|
||||
<?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\Auto;
|
||||
|
||||
use YellowTree\GeoipDetect\DataSources\Manual\ManualDataSource;
|
||||
|
||||
define('GEOIP_DETECT_DATA_UPDATE_FILENAME', 'GeoLite2-City.mmdb');
|
||||
|
||||
class AutoDataSource extends ManualDataSource
|
||||
{
|
||||
public function getId() { return 'auto'; }
|
||||
public function getLabel() { return __('Automatic download & update of Maxmind GeoIP Lite City', 'geoip-detect'); }
|
||||
public function getShortLabel() { return sprintf(__('%s (updated monthly)', 'geoip-detect'), parent::getShortLabel()); }
|
||||
public function getDescriptionHTML() { return __('(License: Creative Commons Attribution-ShareAlike 3.0 Unported. See <a href="https://github.com/yellowtree/wp-geoip-detect/wiki/FAQ#the-maxmind-lite-databases-are-licensed-creative-commons-sharealike-attribution-when-do-i-need-to-give-attribution" target="_blank">Licensing FAQ</a> for more details.)', 'geoip-detect'); }
|
||||
|
||||
public function getStatusInformationHTML() {
|
||||
$html = parent::getStatusInformationHTML();
|
||||
$date_format = get_option('date_format') . ' ' . get_option('time_format');
|
||||
|
||||
$rescheduled = '';
|
||||
$next_cron_update = wp_next_scheduled( 'geoipdetectupdate' );
|
||||
if ($next_cron_update === false) {
|
||||
$rescheduled = ' ' . __('(Was rescheduled just now)', 'geoip-detect');
|
||||
$this->set_cron_schedule();
|
||||
$next_cron_update = wp_next_scheduled( 'geoipdetectupdate' );
|
||||
}
|
||||
$html .= '<br />' . sprintf(__('Next update: %s', 'geoip-detect'), $next_cron_update !== false ? date_i18n($date_format, $next_cron_update) : __('Never', 'geoip-detect'));
|
||||
$html .= $rescheduled;
|
||||
|
||||
return $html;
|
||||
}
|
||||
|
||||
public function getParameterHTML() {
|
||||
$text_update = __('Update now', 'geoip-detect');
|
||||
$nonce_field = wp_nonce_field( 'geoip_detect_update' );
|
||||
$html = <<<HTML
|
||||
<form method="post" action="#">
|
||||
$nonce_field
|
||||
<input type="hidden" name="action" value="update" />
|
||||
<input type="submit" class="button button-secondary" value="$text_update" />
|
||||
</form>
|
||||
HTML;
|
||||
return $html;
|
||||
}
|
||||
|
||||
public function saveParameters($post) {}
|
||||
|
||||
public function __construct() {
|
||||
parent::__construct();
|
||||
add_action('geoipdetectupdate', array($this, 'hook_cron'), 10, 1);
|
||||
add_action('plugins_loaded', array($this, 'on_plugins_loaded'));
|
||||
}
|
||||
|
||||
public function on_plugins_loaded() {
|
||||
if (!defined('GEOIP_DETECT_AUTO_UPDATE_DEACTIVATED'))
|
||||
define('GEOIP_DETECT_AUTO_UPDATE_DEACTIVATED', false);
|
||||
}
|
||||
|
||||
public function maxmindGetFilename() {
|
||||
$data_filename = $this->maxmindGetUploadFilename();
|
||||
if (!is_readable($data_filename))
|
||||
$data_filename = '';
|
||||
|
||||
$data_filename = apply_filters('geoip_detect_get_abs_db_filename', $data_filename);
|
||||
return $data_filename;
|
||||
}
|
||||
|
||||
protected function maxmindGetUploadFilename() {
|
||||
$upload_dir = wp_upload_dir();
|
||||
$dir = $upload_dir['basedir'];
|
||||
|
||||
$filename = $dir . '/' . GEOIP_DETECT_DATA_UPDATE_FILENAME;
|
||||
return $filename;
|
||||
}
|
||||
|
||||
protected function download_url($url, $modified = 0) {
|
||||
// Similar to wordpress download_url, but with custom UA
|
||||
$url_filename = basename( parse_url( $url, PHP_URL_PATH ) );
|
||||
|
||||
$tmpfname = wp_tempnam( $url_filename );
|
||||
if ( ! $tmpfname )
|
||||
return new \WP_Error('http_no_file', __('Could not create temporary file.', 'geoip-detect'));
|
||||
|
||||
$headers = array();
|
||||
$headers['User-Agent'] = GEOIP_DETECT_USER_AGENT;
|
||||
if ($modified) {
|
||||
$headers['If-Modified-Since'] = date('r', $modified);
|
||||
}
|
||||
|
||||
$response = wp_safe_remote_get( $url, array('timeout' => 300, 'stream' => true, 'filename' => $tmpfname, 'headers' => $headers ) );
|
||||
$http_response_code = wp_remote_retrieve_response_code( $response );
|
||||
if (304 === $http_response_code) {
|
||||
return new \WP_Error( 'http_304', __('It has not changed since the last update.', 'geoip-detect') );
|
||||
}
|
||||
if (is_wp_error( $response ) || 200 != $http_response_code) {
|
||||
unlink($tmpfname);
|
||||
$body = wp_remote_retrieve_body($response);
|
||||
return new \WP_Error( 'http_404', $http_response_code . ': ' . trim( wp_remote_retrieve_response_message( $response ) ) . ' ' . $body );
|
||||
}
|
||||
|
||||
return $tmpfname;
|
||||
}
|
||||
|
||||
public function maxmindUpdate()
|
||||
{
|
||||
require_once(ABSPATH.'/wp-admin/includes/file.php');
|
||||
|
||||
$download_url = 'http://geolite.maxmind.com/download/geoip/database/GeoLite2-City.tar.gz';
|
||||
//$download_url = 'http://geolite.maxmind.com/download/geoip/database/GeoLite2-Country.tar.gz';
|
||||
$download_url = apply_filters('geoip_detect2_download_url', $download_url);
|
||||
|
||||
$outFile = $this->maxmindGetUploadFilename();
|
||||
$modified = 0;
|
||||
if (\is_readable($outFile)) {
|
||||
$modified = filemtime($outFile);
|
||||
}
|
||||
|
||||
// Check if existing download should be resumed
|
||||
$tmpFile = get_option('geoip-detect-auto_downloaded_file');
|
||||
if (!$tmpFile || !file_exists($tmpFile)) {
|
||||
// Download file
|
||||
$tmpFile = $this->download_url($download_url, $modified);
|
||||
}
|
||||
|
||||
if (is_wp_error($tmpFile)) {
|
||||
return $tmpFile->get_error_message();
|
||||
}
|
||||
update_option('geoip-detect-auto_downloaded_file', $tmpFile);
|
||||
|
||||
// Unpack tar.gz
|
||||
$ret = $this->unpackArchive($tmpFile, $outFile);
|
||||
if (is_string($ret)) {
|
||||
return $ret;
|
||||
}
|
||||
|
||||
if (!is_readable($outFile)) {
|
||||
return 'Something went wrong: the unpacked file cannot be found.';
|
||||
}
|
||||
|
||||
update_option('geoip-detect-auto_downloaded_file', '');
|
||||
unlink($tmpFile);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
// Ungzip File
|
||||
protected function unpackArchive($downloadedFilename, $outFile) {
|
||||
if (!is_readable($downloadedFilename))
|
||||
return __('Downloaded file could not be opened for reading.', 'geoip-detect');
|
||||
if (!\is_writable(dirname($outFile)))
|
||||
return sprintf(__('Database could not be written (%s).', 'geoip-detect'), $outFile);
|
||||
|
||||
$phar = new \PharData( $downloadedFilename );
|
||||
|
||||
$outDir = get_temp_dir() . 'geoip-detect/';
|
||||
|
||||
global $wp_filesystem;
|
||||
if (!$wp_filesystem) {
|
||||
\WP_Filesystem(false, get_temp_dir());
|
||||
}
|
||||
if (\is_dir($outDir)) {
|
||||
$wp_filesystem->rmdir($outDir, true);
|
||||
}
|
||||
|
||||
mkdir($outDir);
|
||||
$phar->extractTo($outDir, null, true);
|
||||
|
||||
$files = new \RecursiveIteratorIterator(new \RecursiveDirectoryIterator($outDir));
|
||||
|
||||
$inFile = '';
|
||||
foreach($files as $file) {
|
||||
if (!$file->isDir() && mb_substr($file->getFilename(), -5) == '.mmdb') {
|
||||
$inFile = $file->getPathname();
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (!\is_readable($inFile))
|
||||
return __('Downloaded file could not be opened for reading.', 'geoip-detect');
|
||||
|
||||
$ret = copy($inFile, $outFile);
|
||||
if (!$ret)
|
||||
return sprintf(__('Downloaded file could not write or overwrite %s.', 'geoip-detect'), $outFile);
|
||||
|
||||
$wp_filesystem->rmdir($outDir, true);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
public function hook_cron() {
|
||||
/**
|
||||
* Filter:
|
||||
* Cron has fired.
|
||||
* Find out if file should be updated now.
|
||||
*
|
||||
* @param $do_it False if deactivated by define
|
||||
* @param $immediately_after_activation True if this is fired because the plugin was recently activated (deprecated, will now always be false)
|
||||
*/
|
||||
$do_it = apply_filters('geoip_detect_cron_do_update', !GEOIP_DETECT_AUTO_UPDATE_DEACTIVATED, false);
|
||||
|
||||
$this->schedule_next_cron_run();
|
||||
|
||||
if ($do_it)
|
||||
$this->maxmindUpdate();
|
||||
}
|
||||
|
||||
public function set_cron_schedule()
|
||||
{
|
||||
$next = wp_next_scheduled( 'geoipdetectupdate' );
|
||||
if ( $next === false ) {
|
||||
$this->schedule_next_cron_run();
|
||||
}
|
||||
}
|
||||
|
||||
public function schedule_next_cron_run() {
|
||||
// Try to update every 1-2 weeks
|
||||
$next = time() + WEEK_IN_SECONDS;
|
||||
$next += mt_rand(1, WEEK_IN_SECONDS);
|
||||
wp_schedule_single_event($next, 'geoipdetectupdate');
|
||||
}
|
||||
|
||||
public function activate() {
|
||||
$this->set_cron_schedule();
|
||||
}
|
||||
|
||||
public function deactivate()
|
||||
{
|
||||
wp_clear_scheduled_hook('geoipdetectupdate');
|
||||
}
|
||||
|
||||
public function uninstall() {
|
||||
// Delete the automatically downloaded file, if it exists
|
||||
$filename = $this->maxmindGetFilename();
|
||||
if ($filename) {
|
||||
unlink($filename);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
geoip_detect2_register_source(new AutoDataSource());
|
||||
|
||||
Reference in New Issue
Block a user