1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
<?php
/**
* Matomo - free/libre analytics platform
*
* @link https://matomo.org
* @license http://www.gnu.org/licenses/gpl-3.0.html GPL v3 or later
*
*/
namespace Piwik\Plugins\GeoIp2\Commands;
use Piwik\Common;
use Piwik\Db;
use Piwik\DbHelper;
use Piwik\Option;
use Piwik\Plugin\ConsoleCommand;
use Piwik\Plugins\UserCountry\LocationProvider;
use Piwik\Plugins\GeoIp2\LocationProvider\GeoIp2;
use Symfony\Component\Console\Input\InputInterface;
use Symfony\Component\Console\Output\OutputInterface;
class ConvertRegionCodesToIso extends ConsoleCommand
{
const OPTION_NAME = 'regioncodes_converted';
const MAPPING_TABLE_NAME = 'fips2iso';
protected function configure()
{
$this->setName('usercountry:convert-region-codes');
$this->setDescription("Convert FIPS region codes saved by GeoIP legacy provider to ISO.");
}
public function isEnabled()
{
return (LocationProvider::getCurrentProvider() instanceof GeoIp2);
}
/**
* @param InputInterface $input
* @param OutputInterface $output
* @return void|int
*/
protected function execute(InputInterface $input, OutputInterface $output)
{
// chick if option is set to disable second run
if (Option::get(self::OPTION_NAME)) {
$output->writeln('Converting region codes already done.');
return;
}
$output->setDecorated(true);
$output->write('Creating mapping table in database');
Db::query('DROP table if exists ' . self::MAPPING_TABLE_NAME);
DbHelper::createTable(self::MAPPING_TABLE_NAME,
"`country_code` VARCHAR(2) NOT NULL,
`fips_code` VARCHAR(2) NOT NULL,
`iso_code` VARCHAR(4) NULL DEFAULT NULL,
PRIMARY KEY (`country_code`, `fips_code`)");
$output->writeln(' <fg=green>✓</>');
$mappings = include __DIR__ . '/../data/regionMapping.php';
$output->write('Inserting mapping data ');
$counter = 0;
foreach ($mappings as $country => $regionMapping) {
foreach ($regionMapping as $fips => $iso) {
if ($fips == $iso) {
continue; // nothing needs to be changed, so ignore the mapping
}
Db::query('INSERT INTO `'.Common::prefixTable(self::MAPPING_TABLE_NAME).'` VALUES (?, ?, ?)', [$country, $fips, $iso]);
$counter++;
if ($counter%50 == 0) {
$output->write('.');
}
}
}
$output->writeln(' <fg=green>✓</>');
$output->writeln('Updating Matomo log tables:');
$activationTime = Option::get(GeoIp2::SWITCH_TO_ISO_REGIONS_OPTION_NAME);
$activationDateTime = date('Y-m-d H:i:s', $activationTime);
// fix country and region of tibet so it will be updated correctly afterwards
$tibetFixQuery = 'UPDATE %s SET location_country = "cn", location_region = "14" WHERE location_country = "ti"';
// replace invalid country codes used by GeoIP Legacy
$fixInvalidCountriesQuery = 'UPDATE %s SET location_country = "" WHERE location_country IN("AP", "EU", "A1", "A2")';
$query = "UPDATE %s INNER JOIN %s ON location_country = country_code AND location_region = fips_code SET location_region = iso_code
WHERE `%s` < ?";
$logTables = ['log_visit' => 'visit_first_action_time', 'log_conversion' => 'server_time'];
foreach ($logTables as $logTable => $dateField) {
$output->write('- Updating ' . $logTable);
Db::query(sprintf($tibetFixQuery, Common::prefixTable($logTable)));
Db::query(sprintf($fixInvalidCountriesQuery, Common::prefixTable($logTable)));
$sql = sprintf($query, Common::prefixTable($logTable), Common::prefixTable(self::MAPPING_TABLE_NAME), $dateField);
Db::query($sql, $activationDateTime);
$output->writeln(' <fg=green>✓</>');
}
$output->write('Removing mapping table from database ');
Db::dropTables(Common::prefixTable(self::MAPPING_TABLE_NAME));
$output->writeln(' <fg=green>✓</>');
// save option to prevent a second run
Option::set(self::OPTION_NAME, true);
$output->writeln('All region codes converted.');
}
}
|