Jacek Kowalski
2019-06-17 e3e91e07770ff533fbe1493095a32a9e19e3161b
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
125
126
<?php
require_once(__DIR__.'/lib/database.php');
require_once(__DIR__.'/lib/fetch.php');
require_once(__DIR__.'/lib/mapper.php');
 
$logger = new Monolog\Logger('Parse changes');
 
$sources = [
    'bus' => [
        'gtfsrt' => 'ftp://ztp.krakow.pl/VehiclePositions_A.pb',
        'gtfsrt_file' => 'VehiclePositions_A.pb',
        'ttss' => 'http://ttss.mpk.krakow.pl/internetservice/geoserviceDispatcher/services/vehicleinfo/vehicles',
        'ttss_file' => 'vehicles_A.json',
        'database' => 'mapping_A.sqlite3',
        'result' => 'mapping_A.json',
        'mapper' => 'numToTypeB',
    ],
    'tram' => [
        'gtfsrt' => 'ftp://ztp.krakow.pl/VehiclePositions.pb',
        'gtfsrt_file' => 'VehiclePositions_T.pb',
        'ttss' => 'http://www.ttss.krakow.pl/internetservice/geoserviceDispatcher/services/vehicleinfo/vehicles',
        'ttss_file' => 'vehicles_T.json',
        'database' => 'mapping_T.sqlite3',
        'result' => 'mapping_T.json',
        'mapper' => 'numToTypeT',
    ],
];
 
foreach($sources as $name => $source) {
    $logger = new Monolog\Logger('fetch_'.$name);
    try {
        foreach(['gtfsrt_file', 'ttss_file', 'database', 'result'] as $field) {
            $source[$field] = __DIR__.'/data/'.$source[$field];
        }
        $source['result_temp'] = $source['result'].'.tmp';
        
        $logger->info('Fetching '.$name.' position data from FTP...');
        $updated = ftp_fetch_if_newer($source['gtfsrt'], $source['gtfsrt_file']);
        if(!$updated) {
            $logger->info('Nothing to do, remote file not newer than local one');
            continue;
        }
        
        $logger->info('Fetching '.$name.' position data from TTSS...');
        fetch($source['ttss'], $source['ttss_file']);
        
        $logger->info('Loading data...');
        $mapper = new Mapper();
        
        $mapper->loadTTSS($source['ttss_file']);
        $timeDifference = time() - $mapper->getTTSSDate();
        if(abs($timeDifference) > 120) {
            throw new Exception('TTSS timestamp difference ('.$timeDifference.'s) is too high, aborting!');
        }
        
        $mapper->loadGTFSRT($source['gtfsrt_file']);
        $timeDifference = time() - $mapper->getGTFSRTDate();
        if(abs($timeDifference) > 120) {
            throw new Exception('GTFSRT timestamp difference ('.$timeDifference.'s) is too high, aborting!');
        }
        
        $db = new Database($source['database']);
        
        $logger->info('Finding correct offset...');
        $offset = $mapper->findOffset();
        if(!$offset) {
            throw new Exception('Offset not found');
        }
        
        $logger->info('Got offset '.$offset.', creating mapping...');
        $mapping = $mapper->mapUsingOffset($offset, $source['mapper']);
        
        $logger->info('Checking the data for correctness...');
        $weight = count($mapping);
        
        $correct = 0;
        $incorrect = 0;
        $old = 0;
        $maxWeight = 0;
        foreach($mapping as $id => $vehicle) {
            $dbVehicle = $db->getById($id);
            if($dbVehicle) {
                $maxWeight = max($maxWeight, $dbVehicle['weight']);
                if((int)substr($vehicle['num'], 2) == (int)$dbVehicle['num']) {
                    $correct += 1;
                } else {
                    $incorrect += 1;
                }
                continue;
            }
            
            $dbVehicle = $db->getByNum($vehicle['num']);
            if($dbVehicle && $dbVehicle['id'] != $id) {
                $old += 1;
            }
        }
        $logger->info('Weight: '.$weight.', correct: '.$correct.', incorrect: '.$incorrect.', old: '.$old);
        
        $previousMapping = NULL;
        if($incorrect > $correct && $maxWeight > $weight) {
            throw new Exception('Ignoring result due to better data already present');
        } elseif($old > $correct) {
            $logger->warn('Replacing DB data with the new mapping');
            $db->clear();
        } else {
            $previousMapping = @json_decode(@file_get_contents($source['result']), TRUE);
        }
        
        $db->addMapping($mapping);
        
        if(is_array($previousMapping)) {
            $logger->info('Merging previous data with current mapping');
            $mapping = $mapping + $previousMapping;
            ksort($mapping);
        }
        
        $json = json_encode($mapping);
        if(!file_put_contents($source['result_temp'], $json)) {
            throw new Exception('Result save failed');
        }
        rename($source['result_temp'], $source['result']);
        $logger->info('Finished');
    } catch(Throwable $e) {
        $logger->error($e->getMessage(), ['exception' => $e, 'exception_string' => (string)$e]);
    }
}