AJAX
3 years ago
Migrations
3 years ago
Models
3 years ago
Queries
3 years ago
Tables
3 years ago
Utils
3 years ago
Campaign_Builder.php
3 years ago
Capability_Manager.php
3 years ago
Chart.php
3 years ago
Chart_Geo.php
3 years ago
Chart_SVG.php
3 years ago
Current_Resource.php
3 years ago
Dashboard_Options.php
3 years ago
Dashboard_Widget.php
3 years ago
Email_Reports.php
3 years ago
Filters.php
3 years ago
Geo_Database.php
3 years ago
Geo_Database_Download_Job.php
3 years ago
Health_Check.php
3 years ago
Independent_Analytics.php
3 years ago
Known_Referrers.php
3 years ago
PDF.php
3 years ago
Query.php
3 years ago
Quick_Stats.php
3 years ago
REST_API.php
3 years ago
Real_Time.php
3 years ago
Settings.php
3 years ago
Track_Resource_Changes.php
3 years ago
View_Counter.php
3 years ago
WP_Option_Cache_Bust.php
3 years ago
Geo_Database.php
142 lines
| 1 | <?php |
| 2 | |
| 3 | namespace IAWP; |
| 4 | |
| 5 | class Geo_Database |
| 6 | { |
| 7 | private $latest_geo_database_version = '2'; |
| 8 | private $latest_geo_database_checksum = '320bfae90bc6257f0cb52191cc410995'; |
| 9 | private $database_file_name = 'iawp-geo-db.mmdb'; |
| 10 | |
| 11 | /** |
| 12 | * TODO - Move this out |
| 13 | * @param $ip |
| 14 | * @return array|null |
| 15 | */ |
| 16 | public function ip_to_geo($ip): ?array |
| 17 | { |
| 18 | try { |
| 19 | $reader = new MaxMind\Db\Reader($this->database_path()); |
| 20 | $geo = $reader->get($ip); |
| 21 | $reader->close(); |
| 22 | |
| 23 | return $geo; |
| 24 | } catch (\Exception $e) { |
| 25 | return null; |
| 26 | } |
| 27 | } |
| 28 | |
| 29 | /** |
| 30 | * Determine if an update to the geo database should occur |
| 31 | * |
| 32 | * @return bool |
| 33 | */ |
| 34 | private function skip_update(): bool |
| 35 | { |
| 36 | $is_downloading = get_option('iawp_is_database_downloading', '0') === '1'; |
| 37 | |
| 38 | if ($is_downloading) { |
| 39 | return true; |
| 40 | } |
| 41 | |
| 42 | $file_missing = file_exists($this->database_path()) === false; |
| 43 | |
| 44 | if ($file_missing) { |
| 45 | return false; |
| 46 | } |
| 47 | |
| 48 | $valid_checksum = verify_file_md5($this->database_path(), $this->latest_geo_database_checksum) === true; |
| 49 | |
| 50 | if ($valid_checksum) { |
| 51 | return true; |
| 52 | } |
| 53 | |
| 54 | return false; |
| 55 | } |
| 56 | |
| 57 | /** |
| 58 | * @return void |
| 59 | */ |
| 60 | public function maybe_dispatch_download_job() |
| 61 | { |
| 62 | if ($this->skip_update()) { |
| 63 | return; |
| 64 | } |
| 65 | |
| 66 | $download_job = new Geo_Database_Download_Job(); |
| 67 | $download_job->dispatch(); |
| 68 | } |
| 69 | |
| 70 | /** |
| 71 | * @return void |
| 72 | */ |
| 73 | public function download() |
| 74 | { |
| 75 | if ($this->skip_update()) { |
| 76 | return; |
| 77 | } |
| 78 | |
| 79 | update_option('iawp_is_database_downloading', '1'); |
| 80 | |
| 81 | try { |
| 82 | wp_remote_get($this->database_download_url(), [ |
| 83 | 'stream' => true, |
| 84 | 'filename' => $this->zip_path(), |
| 85 | 'timeout' => 60, |
| 86 | ]); |
| 87 | |
| 88 | $zip = new \ZipArchive; |
| 89 | if ($zip->open($this->zip_path()) === true) { |
| 90 | $zip->extractTo($this->wordpress_upload_folder_path()); |
| 91 | $zip->close(); |
| 92 | } |
| 93 | |
| 94 | wp_delete_file($this->zip_path()); |
| 95 | } catch (\Exception $e) { |
| 96 | // Do nothing |
| 97 | } |
| 98 | |
| 99 | delete_option('iawp_is_database_downloading'); |
| 100 | } |
| 101 | |
| 102 | /** |
| 103 | * @return string Name of the zip file for the latest geo database version |
| 104 | */ |
| 105 | private function zip_file_name(): string |
| 106 | { |
| 107 | return 'iawp-geo-db-' . $this->latest_geo_database_version . '.zip'; |
| 108 | } |
| 109 | |
| 110 | /** |
| 111 | * @return string Path to WordPress upload directory |
| 112 | */ |
| 113 | private function wordpress_upload_folder_path(): string |
| 114 | { |
| 115 | return trailingslashit(wp_upload_dir()['basedir']); |
| 116 | } |
| 117 | |
| 118 | /** |
| 119 | * @return string Path to downloaded database |
| 120 | */ |
| 121 | private function zip_path(): string |
| 122 | { |
| 123 | return $this->wordpress_upload_folder_path() . $this->zip_file_name(); |
| 124 | } |
| 125 | |
| 126 | /** |
| 127 | * @return string Path to downloaded database |
| 128 | */ |
| 129 | private function database_path(): string |
| 130 | { |
| 131 | return $this->wordpress_upload_folder_path() . $this->database_file_name; |
| 132 | } |
| 133 | |
| 134 | /** |
| 135 | * @return string AWS Cloudfront URL |
| 136 | */ |
| 137 | private function database_download_url(): string |
| 138 | { |
| 139 | return 'https://d12l9frnj72cgq.cloudfront.net/' . $this->zip_file_name(); |
| 140 | } |
| 141 | } |
| 142 |