1 | <?php |
||
2 | |||
3 | /** |
||
4 | * webtrees: online genealogy |
||
5 | * Copyright (C) 2025 webtrees development team |
||
6 | * This program is free software: you can redistribute it and/or modify |
||
7 | * it under the terms of the GNU General Public License as published by |
||
8 | * the Free Software Foundation, either version 3 of the License, or |
||
9 | * (at your option) any later version. |
||
10 | * This program is distributed in the hope that it will be useful, |
||
11 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
||
12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
||
13 | * GNU General Public License for more details. |
||
14 | * You should have received a copy of the GNU General Public License |
||
15 | * along with this program. If not, see <https://www.gnu.org/licenses/>. |
||
16 | */ |
||
17 | |||
18 | declare(strict_types=1); |
||
19 | |||
20 | namespace Fisharebest\Webtrees\Http\RequestHandlers; |
||
21 | |||
22 | use Fisharebest\Webtrees\GedcomRecord; |
||
23 | use Fisharebest\Webtrees\I18N; |
||
24 | use Fisharebest\Webtrees\Module\ModuleDataFixInterface; |
||
25 | use Fisharebest\Webtrees\Services\DataFixService; |
||
26 | use Fisharebest\Webtrees\Services\ModuleService; |
||
27 | use Fisharebest\Webtrees\Tree; |
||
28 | use Fisharebest\Webtrees\Validator; |
||
29 | use Illuminate\Support\Collection; |
||
30 | use Psr\Http\Message\ResponseInterface; |
||
31 | use Psr\Http\Message\ServerRequestInterface; |
||
32 | use Psr\Http\Server\RequestHandlerInterface; |
||
33 | |||
34 | use function assert; |
||
35 | use function json_encode; |
||
36 | use function response; |
||
37 | |||
38 | use const JSON_THROW_ON_ERROR; |
||
39 | |||
40 | /** |
||
41 | * Run a data-fix. |
||
42 | */ |
||
43 | class DataFixUpdateAll implements RequestHandlerInterface |
||
44 | { |
||
45 | // Process this number of records in each HTTP request |
||
46 | private const int CHUNK_SIZE = 250; |
||
0 ignored issues
–
show
Bug
introduced
by
![]() |
|||
47 | |||
48 | private DataFixService $data_fix_service; |
||
49 | |||
50 | private ModuleService $module_service; |
||
51 | |||
52 | /** |
||
53 | * @param DataFixService $data_fix_service |
||
54 | * @param ModuleService $module_service |
||
55 | */ |
||
56 | public function __construct( |
||
57 | DataFixService $data_fix_service, |
||
58 | ModuleService $module_service |
||
59 | ) { |
||
60 | $this->data_fix_service = $data_fix_service; |
||
61 | $this->module_service = $module_service; |
||
62 | } |
||
63 | |||
64 | /** |
||
65 | * @param ServerRequestInterface $request |
||
66 | * |
||
67 | * @return ResponseInterface |
||
68 | */ |
||
69 | public function handle(ServerRequestInterface $request): ResponseInterface |
||
70 | { |
||
71 | $tree = Validator::attributes($request)->tree(); |
||
72 | $data_fix = Validator::attributes($request)->string('data_fix', ''); |
||
73 | $module = $this->module_service->findByName($data_fix); |
||
74 | assert($module instanceof ModuleDataFixInterface); |
||
75 | |||
76 | $params = $request->getQueryParams(); |
||
77 | $rows = $module->recordsToFix($tree, $params); |
||
78 | |||
79 | if ($rows->isEmpty()) { |
||
80 | return response([]); |
||
81 | } |
||
82 | |||
83 | $start = Validator::queryParams($request)->string('start', ''); |
||
84 | $end = Validator::queryParams($request)->string('end', ''); |
||
85 | |||
86 | if ($start === '' || $end === '') { |
||
87 | return $this->createUpdateRanges($tree, $module, $rows, $params); |
||
88 | } |
||
89 | |||
90 | /** @var Collection<int,GedcomRecord> $records */ |
||
91 | $records = $rows |
||
92 | ->map(fn (object $row): GedcomRecord|null => $this->data_fix_service->getRecordByType($row->xref, $tree, $row->type)) |
||
93 | ->filter(static fn (GedcomRecord|null $record): bool => $record instanceof GedcomRecord && !$record->isPendingDeletion() && $module->doesRecordNeedUpdate($record, $params)); |
||
94 | |||
95 | foreach ($records as $record) { |
||
96 | $module->updateRecord($record, $params); |
||
97 | } |
||
98 | |||
99 | return response(); |
||
100 | } |
||
101 | |||
102 | /** |
||
103 | * @param Tree $tree |
||
104 | * @param ModuleDataFixInterface $module |
||
105 | * @param Collection<int,object{xref:string,type:string}> $rows |
||
106 | * @param array<string> $params |
||
107 | * |
||
108 | * @return ResponseInterface |
||
109 | */ |
||
110 | private function createUpdateRanges( |
||
111 | Tree $tree, |
||
112 | ModuleDataFixInterface $module, |
||
113 | Collection $rows, |
||
114 | array $params |
||
115 | ): ResponseInterface { |
||
116 | $total = $rows->count(); |
||
117 | |||
118 | $updates = $rows |
||
119 | ->chunk(self::CHUNK_SIZE) |
||
120 | ->map(static function (Collection $chunk) use ($module, $params, $tree, $total): object { |
||
121 | static $count = 0; |
||
122 | |||
123 | $count += $chunk->count(); |
||
124 | |||
125 | $start = $chunk->first()->xref; |
||
126 | $end = $chunk->last()->xref; |
||
127 | $url = route(self::class, [ |
||
128 | 'tree' => $tree->name(), |
||
129 | 'data_fix' => $module->name(), |
||
130 | 'start' => $start, |
||
131 | 'end' => $end, |
||
132 | ] + $params); |
||
133 | |||
134 | return (object) [ |
||
135 | 'url' => $url, |
||
136 | 'percent' => (100.0 * $count / $total) . '%', |
||
137 | 'progress' => I18N::percentage($count / $total, 1), |
||
138 | ]; |
||
139 | }) |
||
140 | ->all(); |
||
141 | |||
142 | return response(json_encode($updates, JSON_THROW_ON_ERROR)); |
||
143 | } |
||
144 | } |
||
145 |