fisharebest /
webtrees
| Total Complexity | 137 |
| Total Lines | 965 |
| Duplicated Lines | 0 % |
| Changes | 3 | ||
| Bugs | 0 | Features | 0 |
Complex classes like IndividualListModule often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.
Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.
While breaking up the class, it is a good idea to analyze how other classes use IndividualListModule, and based on these observations, apply Extract Interface, too.
| 1 | <?php |
||
| 56 | class IndividualListModule extends AbstractModule implements ModuleListInterface, RequestHandlerInterface |
||
| 57 | { |
||
| 58 | use ModuleListTrait; |
||
| 59 | |||
| 60 | protected const ROUTE_URL = '/tree/{tree}/individual-list'; |
||
| 61 | |||
| 62 | /** |
||
| 63 | * Initialization. |
||
| 64 | * |
||
| 65 | * @return void |
||
| 66 | */ |
||
| 67 | public function boot(): void |
||
| 68 | { |
||
| 69 | Registry::routeFactory()->routeMap() |
||
| 70 | ->get(static::class, static::ROUTE_URL, $this); |
||
| 71 | } |
||
| 72 | |||
| 73 | /** |
||
| 74 | * How should this module be identified in the control panel, etc.? |
||
| 75 | * |
||
| 76 | * @return string |
||
| 77 | */ |
||
| 78 | public function title(): string |
||
| 79 | { |
||
| 80 | /* I18N: Name of a module/list */ |
||
| 81 | return I18N::translate('Individuals'); |
||
| 82 | } |
||
| 83 | |||
| 84 | /** |
||
| 85 | * A sentence describing what this module does. |
||
| 86 | * |
||
| 87 | * @return string |
||
| 88 | */ |
||
| 89 | public function description(): string |
||
| 90 | { |
||
| 91 | /* I18N: Description of the “Individuals” module */ |
||
| 92 | return I18N::translate('A list of individuals.'); |
||
| 93 | } |
||
| 94 | |||
| 95 | /** |
||
| 96 | * CSS class for the URL. |
||
| 97 | * |
||
| 98 | * @return string |
||
| 99 | */ |
||
| 100 | public function listMenuClass(): string |
||
| 101 | { |
||
| 102 | return 'menu-list-indi'; |
||
| 103 | } |
||
| 104 | |||
| 105 | /** |
||
| 106 | * @param Tree $tree |
||
| 107 | * @param array<bool|int|string|array<string>|null> $parameters |
||
| 108 | * |
||
| 109 | * @return string |
||
| 110 | */ |
||
| 111 | public function listUrl(Tree $tree, array $parameters = []): string |
||
| 112 | { |
||
| 113 | $request = app(ServerRequestInterface::class); |
||
| 114 | assert($request instanceof ServerRequestInterface); |
||
| 115 | |||
| 116 | $xref = Validator::attributes($request)->isXref()->string('xref', ''); |
||
| 117 | |||
| 118 | if ($xref !== '') { |
||
| 119 | $individual = Registry::individualFactory()->make($xref, $tree); |
||
| 120 | |||
| 121 | if ($individual instanceof Individual && $individual->canShow()) { |
||
| 122 | $primary_name = $individual->getPrimaryName(); |
||
| 123 | |||
| 124 | $parameters['surname'] = $parameters['surname'] ?? $individual->getAllNames()[$primary_name]['surn'] ?? null; |
||
| 125 | } |
||
| 126 | } |
||
| 127 | |||
| 128 | $parameters['tree'] = $tree->name(); |
||
| 129 | |||
| 130 | return route(static::class, $parameters); |
||
| 131 | } |
||
| 132 | |||
| 133 | /** |
||
| 134 | * @return array<string> |
||
| 135 | */ |
||
| 136 | public function listUrlAttributes(): array |
||
| 137 | { |
||
| 138 | return []; |
||
| 139 | } |
||
| 140 | |||
| 141 | /** |
||
| 142 | * Handle URLs generated by older versions of webtrees |
||
| 143 | * |
||
| 144 | * @param ServerRequestInterface $request |
||
| 145 | * |
||
| 146 | * @return ResponseInterface |
||
| 147 | */ |
||
| 148 | public function getListAction(ServerRequestInterface $request): ResponseInterface |
||
| 149 | { |
||
| 150 | $tree = Validator::attributes($request)->tree(); |
||
| 151 | |||
| 152 | return redirect($this->listUrl($tree, $request->getQueryParams())); |
||
| 153 | } |
||
| 154 | |||
| 155 | /** |
||
| 156 | * @param ServerRequestInterface $request |
||
| 157 | * |
||
| 158 | * @return ResponseInterface |
||
| 159 | */ |
||
| 160 | public function handle(ServerRequestInterface $request): ResponseInterface |
||
| 161 | { |
||
| 162 | $tree = Validator::attributes($request)->tree(); |
||
| 163 | $user = Validator::attributes($request)->user(); |
||
| 164 | |||
| 165 | Auth::checkComponentAccess($this, ModuleListInterface::class, $tree, $user); |
||
| 166 | |||
| 167 | return $this->createResponse($tree, $user, $request->getQueryParams(), false); |
||
| 168 | } |
||
| 169 | |||
| 170 | /** |
||
| 171 | * @param Tree $tree |
||
| 172 | * @param UserInterface $user |
||
| 173 | * @param array<string> $params |
||
| 174 | * @param bool $families |
||
| 175 | * |
||
| 176 | * @return ResponseInterface |
||
| 177 | */ |
||
| 178 | protected function createResponse(Tree $tree, UserInterface $user, array $params, bool $families): ResponseInterface |
||
| 179 | { |
||
| 180 | ob_start(); |
||
| 181 | |||
| 182 | // We show three different lists: initials, surnames and individuals |
||
| 183 | |||
| 184 | // All surnames beginning with this letter where "@"=unknown and ","=none |
||
| 185 | $alpha = $params['alpha'] ?? ''; |
||
| 186 | |||
| 187 | // All individuals with this surname |
||
| 188 | $surname = $params['surname'] ?? ''; |
||
| 189 | |||
| 190 | // All individuals |
||
| 191 | $show_all = $params['show_all'] ?? 'no'; |
||
| 192 | |||
| 193 | // Long lists can be broken down by given name |
||
| 194 | $show_all_firstnames = $params['show_all_firstnames'] ?? 'no'; |
||
| 195 | if ($show_all_firstnames === 'yes') { |
||
| 196 | $falpha = ''; |
||
| 197 | } else { |
||
| 198 | // All first names beginning with this letter |
||
| 199 | $falpha = $params['falpha'] ?? ''; |
||
| 200 | } |
||
| 201 | |||
| 202 | $show_marnm = $params['show_marnm'] ?? ''; |
||
| 203 | switch ($show_marnm) { |
||
| 204 | case 'no': |
||
| 205 | case 'yes': |
||
| 206 | $user->setPreference($families ? 'family-list-marnm' : 'individual-list-marnm', $show_marnm); |
||
| 207 | break; |
||
| 208 | default: |
||
| 209 | $show_marnm = $user->getPreference($families ? 'family-list-marnm' : 'individual-list-marnm'); |
||
| 210 | } |
||
| 211 | |||
| 212 | // Make sure selections are consistent. |
||
| 213 | // i.e. can’t specify show_all and surname at the same time. |
||
| 214 | if ($show_all === 'yes') { |
||
| 215 | $alpha = ''; |
||
| 216 | $surname = ''; |
||
| 217 | |||
| 218 | if ($show_all_firstnames === 'yes') { |
||
| 219 | $legend = I18N::translate('All'); |
||
| 220 | $params = [ |
||
| 221 | 'tree' => $tree->name(), |
||
| 222 | 'show_all' => 'yes', |
||
| 223 | ]; |
||
| 224 | $show = 'indi'; |
||
| 225 | } elseif ($falpha !== '') { |
||
| 226 | $legend = I18N::translate('All') . ', ' . e($falpha) . '…'; |
||
| 227 | $params = [ |
||
| 228 | 'tree' => $tree->name(), |
||
| 229 | 'show_all' => 'yes', |
||
| 230 | ]; |
||
| 231 | $show = 'indi'; |
||
| 232 | } else { |
||
| 233 | $legend = I18N::translate('All'); |
||
| 234 | $show = $params['show'] ?? 'surn'; |
||
| 235 | $params = [ |
||
| 236 | 'tree' => $tree->name(), |
||
| 237 | 'show_all' => 'yes', |
||
| 238 | ]; |
||
| 239 | } |
||
| 240 | } elseif ($surname !== '') { |
||
| 241 | $alpha = I18N::language()->initialLetter($surname); // so we can highlight the initial letter |
||
| 242 | $show_all = 'no'; |
||
| 243 | if ($surname === Individual::NOMEN_NESCIO) { |
||
| 244 | $legend = I18N::translateContext('Unknown surname', '…'); |
||
| 245 | } else { |
||
| 246 | // The surname parameter is a root/canonical form. |
||
| 247 | // Display it as the actual surname |
||
| 248 | $legend = implode('/', array_keys($this->surnames($tree, $surname, $alpha, $show_marnm === 'yes', $families, I18N::locale()))); |
||
| 249 | } |
||
| 250 | $params = [ |
||
| 251 | 'tree' => $tree->name(), |
||
| 252 | 'surname' => $surname, |
||
| 253 | 'falpha' => $falpha, |
||
| 254 | ]; |
||
| 255 | switch ($falpha) { |
||
| 256 | case '': |
||
| 257 | break; |
||
| 258 | case '@': |
||
| 259 | $legend .= ', ' . I18N::translateContext('Unknown given name', '…'); |
||
| 260 | break; |
||
| 261 | default: |
||
| 262 | $legend .= ', ' . e($falpha) . '…'; |
||
| 263 | break; |
||
| 264 | } |
||
| 265 | $show = 'indi'; // SURN list makes no sense here |
||
| 266 | } elseif ($alpha === '@') { |
||
| 267 | $show_all = 'no'; |
||
| 268 | $legend = I18N::translateContext('Unknown surname', '…'); |
||
| 269 | $params = [ |
||
| 270 | 'alpha' => $alpha, |
||
| 271 | 'tree' => $tree->name(), |
||
| 272 | ]; |
||
| 273 | $show = 'indi'; // SURN list makes no sense here |
||
| 274 | } elseif ($alpha === ',') { |
||
| 275 | $show_all = 'no'; |
||
| 276 | $legend = I18N::translate('No surname'); |
||
| 277 | $params = [ |
||
| 278 | 'alpha' => $alpha, |
||
| 279 | 'tree' => $tree->name(), |
||
| 280 | ]; |
||
| 281 | $show = 'indi'; // SURN list makes no sense here |
||
| 282 | } elseif ($alpha !== '') { |
||
| 283 | $show_all = 'no'; |
||
| 284 | $legend = e($alpha) . '…'; |
||
| 285 | $show = $params['show'] ?? 'surn'; |
||
| 286 | $params = [ |
||
| 287 | 'alpha' => $alpha, |
||
| 288 | 'tree' => $tree->name(), |
||
| 289 | ]; |
||
| 290 | } else { |
||
| 291 | $show_all = 'no'; |
||
| 292 | $legend = '…'; |
||
| 293 | $params = [ |
||
| 294 | 'tree' => $tree->name(), |
||
| 295 | ]; |
||
| 296 | $show = 'none'; // Don't show lists until something is chosen |
||
| 297 | } |
||
| 298 | $legend = '<bdi>' . $legend . '</bdi>'; |
||
| 299 | |||
| 300 | if ($families) { |
||
| 301 | $title = I18N::translate('Families') . ' — ' . $legend; |
||
| 302 | } else { |
||
| 303 | $title = I18N::translate('Individuals') . ' — ' . $legend; |
||
| 304 | } ?> |
||
| 305 | <div class="d-flex flex-column wt-page-options wt-page-options-individual-list d-print-none"> |
||
| 306 | <ul class="d-flex flex-wrap list-unstyled justify-content-center wt-initials-list wt-initials-list-surname"> |
||
| 307 | |||
| 308 | <?php foreach ($this->surnameAlpha($tree, $show_marnm === 'yes', $families, I18N::locale()) as $letter => $count) : ?> |
||
| 309 | <li class="wt-initials-list-item d-flex"> |
||
| 310 | <?php if ($count > 0) : ?> |
||
| 311 | <a href="<?= e($this->listUrl($tree, ['alpha' => $letter, 'tree' => $tree->name()])) ?>" class="wt-initial px-1<?= $letter === $alpha ? ' active' : '' ?> '" title="<?= I18N::number($count) ?>"><?= $this->surnameInitial((string) $letter) ?></a> |
||
| 312 | <?php else : ?> |
||
| 313 | <span class="wt-initial px-1 text-muted"><?= $this->surnameInitial((string) $letter) ?></span> |
||
| 314 | |||
| 315 | <?php endif ?> |
||
| 316 | </li> |
||
| 317 | <?php endforeach ?> |
||
| 318 | |||
| 319 | <?php if (Session::has('initiated')) : ?> |
||
| 320 | <!-- Search spiders don't get the "show all" option as the other links give them everything. --> |
||
| 321 | <li class="wt-initials-list-item d-flex"> |
||
| 322 | <a class="wt-initial px-1<?= $show_all === 'yes' ? ' active' : '' ?>" href="<?= e($this->listUrl($tree, ['show_all' => 'yes'] + $params)) ?>"><?= I18N::translate('All') ?></a> |
||
| 323 | </li> |
||
| 324 | <?php endif ?> |
||
| 325 | </ul> |
||
| 326 | |||
| 327 | <!-- Search spiders don't get an option to show/hide the surname sublists, nor does it make sense on the all/unknown/surname views --> |
||
| 328 | <?php if ($show !== 'none' && Session::has('initiated')) : ?> |
||
| 329 | <?php if ($show_marnm === 'yes') : ?> |
||
| 330 | <p> |
||
| 331 | <a href="<?= e($this->listUrl($tree, ['show' => $show, 'show_marnm' => 'no'] + $params)) ?>"> |
||
| 332 | <?= I18N::translate('Exclude individuals with “%s” as a married name', $legend) ?> |
||
| 333 | </a> |
||
| 334 | </p> |
||
| 335 | <?php else : ?> |
||
| 336 | <p> |
||
| 337 | <a href="<?= e($this->listUrl($tree, ['show' => $show, 'show_marnm' => 'yes'] + $params)) ?>"> |
||
| 338 | <?= I18N::translate('Include individuals with “%s” as a married name', $legend) ?> |
||
| 339 | </a> |
||
| 340 | </p> |
||
| 341 | <?php endif ?> |
||
| 342 | |||
| 343 | <?php if ($alpha !== '@' && $alpha !== ',' && $surname === '') : ?> |
||
| 344 | <?php if ($show === 'surn') : ?> |
||
| 345 | <p> |
||
| 346 | <a href="<?= e($this->listUrl($tree, ['show' => 'indi', 'show_marnm' => 'no'] + $params)) ?>"> |
||
| 347 | <?= I18N::translate('Show the list of individuals') ?> |
||
| 348 | </a> |
||
| 349 | </p> |
||
| 350 | <?php else : ?> |
||
| 351 | <p> |
||
| 352 | <a href="<?= e($this->listUrl($tree, ['show' => 'surn', 'show_marnm' => 'no'] + $params)) ?>"> |
||
| 353 | <?= I18N::translate('Show the list of surnames') ?> |
||
| 354 | </a> |
||
| 355 | </p> |
||
| 356 | <?php endif ?> |
||
| 357 | <?php endif ?> |
||
| 358 | <?php endif ?> |
||
| 359 | </div> |
||
| 360 | |||
| 361 | <div class="wt-page-content"> |
||
| 362 | <?php |
||
| 363 | |||
| 364 | if ($show === 'indi' || $show === 'surn') { |
||
| 365 | $surns = $this->surnames($tree, $surname, $alpha, $show_marnm === 'yes', $families, I18N::locale()); |
||
| 366 | if ($show === 'surn') { |
||
| 367 | // Show the surname list |
||
| 368 | switch ($tree->getPreference('SURNAME_LIST_STYLE')) { |
||
| 369 | case 'style1': |
||
| 370 | echo view('lists/surnames-column-list', [ |
||
| 371 | 'module' => $this, |
||
| 372 | 'surnames' => $surns, |
||
| 373 | 'totals' => true, |
||
| 374 | 'tree' => $tree, |
||
| 375 | ]); |
||
| 376 | break; |
||
| 377 | case 'style3': |
||
| 378 | echo view('lists/surnames-tag-cloud', [ |
||
| 379 | 'module' => $this, |
||
| 380 | 'surnames' => $surns, |
||
| 381 | 'totals' => true, |
||
| 382 | 'tree' => $tree, |
||
| 383 | ]); |
||
| 384 | break; |
||
| 385 | case 'style2': |
||
| 386 | default: |
||
| 387 | echo view('lists/surnames-table', [ |
||
| 388 | 'families' => $families, |
||
| 389 | 'module' => $this, |
||
| 390 | 'order' => [[0, 'asc']], |
||
| 391 | 'surnames' => $surns, |
||
| 392 | 'tree' => $tree, |
||
| 393 | ]); |
||
| 394 | break; |
||
| 395 | } |
||
| 396 | } else { |
||
| 397 | // Show the list |
||
| 398 | $count = 0; |
||
| 399 | foreach ($surns as $surnames) { |
||
| 400 | foreach ($surnames as $total) { |
||
| 401 | $count += $total; |
||
| 402 | } |
||
| 403 | } |
||
| 404 | // Don't sublist short lists. |
||
| 405 | if ($count < $tree->getPreference('SUBLIST_TRIGGER_I')) { |
||
| 406 | $falpha = ''; |
||
| 407 | } else { |
||
| 408 | $givn_initials = $this->givenAlpha($tree, $surname, $alpha, $show_marnm === 'yes', $families, I18N::locale()); |
||
| 409 | // Break long lists by initial letter of given name |
||
| 410 | if ($surname !== '' || $show_all === 'yes') { |
||
| 411 | if ($show_all === 'no') { |
||
| 412 | echo '<h2 class="wt-page-title">', I18N::translate('Individuals with surname %s', $legend), '</h2>'; |
||
| 413 | } |
||
| 414 | // Don't show the list until we have some filter criteria |
||
| 415 | $show = $falpha !== '' || $show_all_firstnames === 'yes' ? 'indi' : 'none'; |
||
| 416 | $list = []; |
||
| 417 | echo '<ul class="d-flex flex-wrap list-unstyled justify-content-center wt-initials-list wt-initials-list-given-names">'; |
||
| 418 | foreach ($givn_initials as $givn_initial => $given_count) { |
||
| 419 | echo '<li class="wt-initials-list-item d-flex">'; |
||
| 420 | if ($given_count > 0) { |
||
| 421 | if ($show === 'indi' && $givn_initial === $falpha && $show_all_firstnames === 'no') { |
||
| 422 | echo '<a class="wt-initial px-1 active" href="' . e($this->listUrl($tree, ['falpha' => $givn_initial] + $params)) . '" title="' . I18N::number($given_count) . '">' . $this->givenNameInitial((string) $givn_initial) . '</a>'; |
||
| 423 | } else { |
||
| 424 | echo '<a class="wt-initial px-1" href="' . e($this->listUrl($tree, ['falpha' => $givn_initial] + $params)) . '" title="' . I18N::number($given_count) . '">' . $this->givenNameInitial((string) $givn_initial) . '</a>'; |
||
| 425 | } |
||
| 426 | } else { |
||
| 427 | echo '<span class="wt-initial px-1 text-muted">' . $this->givenNameInitial((string) $givn_initial) . '</span>'; |
||
| 428 | } |
||
| 429 | echo '</li>'; |
||
| 430 | } |
||
| 431 | // Search spiders don't get the "show all" option as the other links give them everything. |
||
| 432 | if (Session::has('initiated')) { |
||
| 433 | echo '<li class="wt-initials-list-item d-flex">'; |
||
| 434 | if ($show_all_firstnames === 'yes') { |
||
| 435 | echo '<span class="wt-initial px-1 active">' . I18N::translate('All') . '</span>'; |
||
| 436 | } else { |
||
| 437 | echo '<a class="wt-initial px-1" href="' . e($this->listUrl($tree, ['show_all_firstnames' => 'yes'] + $params)) . '" title="' . I18N::number($count) . '">' . I18N::translate('All') . '</a>'; |
||
| 438 | } |
||
| 439 | echo '</li>'; |
||
| 440 | } |
||
| 441 | echo '</ul>'; |
||
| 442 | echo '<p class="text-center alpha_index">', implode(' | ', $list), '</p>'; |
||
| 443 | } |
||
| 444 | } |
||
| 445 | if ($show === 'indi') { |
||
| 446 | if (!$families) { |
||
| 447 | echo view('lists/individuals-table', [ |
||
| 448 | 'individuals' => $this->individuals($tree, $surname, $alpha, $falpha, $show_marnm === 'yes', false, I18N::locale()), |
||
| 449 | 'sosa' => false, |
||
| 450 | 'tree' => $tree, |
||
| 451 | ]); |
||
| 452 | } else { |
||
| 453 | echo view('lists/families-table', [ |
||
| 454 | 'families' => $this->families($tree, $surname, $alpha, $falpha, $show_marnm === 'yes', I18N::locale()), |
||
| 455 | 'tree' => $tree, |
||
| 456 | ]); |
||
| 457 | } |
||
| 458 | } |
||
| 459 | } |
||
| 460 | } ?> |
||
| 461 | </div> |
||
| 462 | <?php |
||
| 463 | |||
| 464 | $html = ob_get_clean(); |
||
| 465 | |||
| 466 | return $this->viewResponse('modules/individual-list/page', [ |
||
| 467 | 'content' => $html, |
||
| 468 | 'title' => $title, |
||
| 469 | 'tree' => $tree, |
||
| 470 | ]); |
||
| 471 | } |
||
| 472 | |||
| 473 | /** |
||
| 474 | * Some initial letters have a special meaning |
||
| 475 | * |
||
| 476 | * @param string $initial |
||
| 477 | * |
||
| 478 | * @return string |
||
| 479 | */ |
||
| 480 | protected function givenNameInitial(string $initial): string |
||
| 487 | } |
||
| 488 | |||
| 489 | /** |
||
| 490 | * Some initial letters have a special meaning |
||
| 491 | * |
||
| 492 | * @param string $initial |
||
| 493 | * |
||
| 494 | * @return string |
||
| 495 | */ |
||
| 496 | protected function surnameInitial(string $initial): string |
||
| 507 | } |
||
| 508 | |||
| 509 | /** |
||
| 510 | * Restrict a query to individuals that are a spouse in a family record. |
||
| 511 | * |
||
| 512 | * @param bool $fams |
||
| 513 | * @param Builder $query |
||
| 514 | */ |
||
| 515 | protected function whereFamily(bool $fams, Builder $query): void |
||
| 523 | }); |
||
| 524 | } |
||
| 525 | } |
||
| 526 | |||
| 527 | /** |
||
| 528 | * Restrict a query to include/exclude married names. |
||
| 529 | * |
||
| 530 | * @param bool $marnm |
||
| 531 | * @param Builder $query |
||
| 532 | */ |
||
| 533 | protected function whereMarriedName(bool $marnm, Builder $query): void |
||
| 537 | } |
||
| 538 | } |
||
| 539 | |||
| 540 | /** |
||
| 541 | * Get a list of initial surname letters. |
||
| 542 | * |
||
| 543 | * @param Tree $tree |
||
| 544 | * @param bool $marnm if set, include married names |
||
| 545 | * @param bool $fams if set, only consider individuals with FAMS records |
||
| 546 | * @param LocaleInterface $locale |
||
| 547 | * |
||
| 548 | * @return array<int> |
||
| 549 | */ |
||
| 550 | public function surnameAlpha(Tree $tree, bool $marnm, bool $fams, LocaleInterface $locale): array |
||
| 551 | { |
||
| 552 | $n_surn = $this->fieldWithCollation('n_surn'); |
||
| 553 | $alphas = []; |
||
| 554 | |||
| 555 | $query = DB::table('name')->where('n_file', '=', $tree->id()); |
||
| 556 | |||
| 557 | $this->whereFamily($fams, $query); |
||
| 558 | $this->whereMarriedName($marnm, $query); |
||
| 559 | |||
| 560 | // Fetch all the letters in our alphabet, whether or not there |
||
| 561 | // are any names beginning with that letter. It looks better to |
||
| 562 | // show the full alphabet, rather than omitting rare letters such as X. |
||
| 563 | foreach (I18N::language()->alphabet() as $letter) { |
||
| 564 | $query2 = clone $query; |
||
| 565 | |||
| 566 | $this->whereInitial($query2, 'n_surn', $letter, $locale); |
||
| 567 | |||
| 568 | $alphas[$letter] = $query2->count(); |
||
| 569 | } |
||
| 570 | |||
| 571 | // Now fetch initial letters that are not in our alphabet, |
||
| 572 | // including "@" (for "@N.N.") and "" for no surname. |
||
| 573 | foreach (I18N::language()->alphabet() as $letter) { |
||
| 574 | $query->where($n_surn, 'NOT LIKE', $letter . '%'); |
||
| 575 | } |
||
| 576 | |||
| 577 | $rows = $query |
||
| 578 | ->groupBy(['initial']) |
||
| 579 | ->orderBy('initial') |
||
| 580 | ->pluck(new Expression('COUNT(*) AS aggregate'), new Expression('SUBSTR(n_surn, 1, 1) AS initial')); |
||
| 581 | |||
| 582 | $specials = ['@', '']; |
||
| 583 | |||
| 584 | foreach ($rows as $alpha => $count) { |
||
| 585 | if (!in_array($alpha, $specials, true)) { |
||
| 586 | $alphas[$alpha] = (int) $count; |
||
| 587 | } |
||
| 588 | } |
||
| 589 | |||
| 590 | // Empty surnames have a special code ',' - as we search for SURN,GIVN |
||
| 591 | foreach ($specials as $special) { |
||
| 592 | if ($rows->has($special)) { |
||
| 593 | $alphas[$special ?: ','] = (int) $rows[$special]; |
||
| 594 | } |
||
| 595 | } |
||
| 596 | |||
| 597 | return $alphas; |
||
| 598 | } |
||
| 599 | |||
| 600 | /** |
||
| 601 | * Get a list of initial given name letters for indilist.php and famlist.php |
||
| 602 | * |
||
| 603 | * @param Tree $tree |
||
| 604 | * @param string $surn if set, only consider people with this surname |
||
| 605 | * @param string $salpha if set, only consider surnames starting with this letter |
||
| 606 | * @param bool $marnm if set, include married names |
||
| 607 | * @param bool $fams if set, only consider individuals with FAMS records |
||
| 608 | * @param LocaleInterface $locale |
||
| 609 | * |
||
| 610 | * @return array<int> |
||
| 611 | */ |
||
| 612 | public function givenAlpha(Tree $tree, string $surn, string $salpha, bool $marnm, bool $fams, LocaleInterface $locale): array |
||
| 663 | } |
||
| 664 | |||
| 665 | /** |
||
| 666 | * Get a count of actual surnames and variants, based on a "root" surname. |
||
| 667 | * |
||
| 668 | * @param Tree $tree |
||
| 669 | * @param string $surn if set, only count people with this surname |
||
| 670 | * @param string $salpha if set, only consider surnames starting with this letter |
||
| 671 | * @param bool $marnm if set, include married names |
||
| 672 | * @param bool $fams if set, only consider individuals with FAMS records |
||
| 673 | * @param LocaleInterface $locale |
||
| 674 | * |
||
| 675 | * @return array<array<int>> |
||
| 676 | */ |
||
| 677 | protected function surnames( |
||
| 678 | Tree $tree, |
||
| 679 | string $surn, |
||
| 680 | string $salpha, |
||
| 681 | bool $marnm, |
||
| 682 | bool $fams, |
||
| 683 | LocaleInterface $locale |
||
| 684 | ): array { |
||
| 685 | $query = DB::table('name') |
||
| 686 | ->where('n_file', '=', $tree->id()) |
||
| 687 | ->select([ |
||
| 688 | new Expression('n_surn /*! COLLATE utf8_bin */ AS n_surn'), |
||
| 689 | new Expression('n_surname /*! COLLATE utf8_bin */ AS n_surname'), |
||
| 690 | new Expression('COUNT(*) AS total'), |
||
| 691 | ]); |
||
| 692 | |||
| 693 | $this->whereFamily($fams, $query); |
||
| 694 | $this->whereMarriedName($marnm, $query); |
||
| 695 | |||
| 696 | if ($surn !== '') { |
||
| 697 | $query->where('n_surn', '=', $surn); |
||
| 698 | } elseif ($salpha === ',') { |
||
| 699 | $query->where('n_surn', '=', ''); |
||
| 700 | } elseif ($salpha === '@') { |
||
| 701 | $query->where('n_surn', '=', Individual::NOMEN_NESCIO); |
||
| 702 | } elseif ($salpha !== '') { |
||
| 703 | $this->whereInitial($query, 'n_surn', $salpha, $locale); |
||
| 704 | } else { |
||
| 705 | // All surnames |
||
| 706 | $query->whereNotIn('n_surn', ['', Individual::NOMEN_NESCIO]); |
||
| 707 | } |
||
| 708 | $query->groupBy([ |
||
| 709 | new Expression('n_surn /*! COLLATE utf8_bin */'), |
||
| 710 | new Expression('n_surname /*! COLLATE utf8_bin */'), |
||
| 711 | ]); |
||
| 712 | |||
| 713 | $list = []; |
||
| 714 | |||
| 715 | foreach ($query->get() as $row) { |
||
| 716 | $row->n_surn = strtr(I18N::strtoupper($row->n_surn), I18N::language()->equivalentLetters()); |
||
| 717 | $row->total += $list[$row->n_surn][$row->n_surname] ?? 0; |
||
| 718 | |||
| 719 | $list[$row->n_surn][$row->n_surname] = (int) $row->total; |
||
| 720 | } |
||
| 721 | |||
| 722 | uksort($list, I18N::comparator()); |
||
| 723 | |||
| 724 | return $list; |
||
| 725 | } |
||
| 726 | |||
| 727 | /** |
||
| 728 | * Fetch a list of individuals with specified names |
||
| 729 | * To search for unknown names, use $surn="@N.N.", $salpha="@" or $galpha="@" |
||
| 730 | * To search for names with no surnames, use $salpha="," |
||
| 731 | * |
||
| 732 | * @param Tree $tree |
||
| 733 | * @param string $surn if set, only fetch people with this surname |
||
| 734 | * @param string $salpha if set, only fetch surnames starting with this letter |
||
| 735 | * @param string $galpha if set, only fetch given names starting with this letter |
||
| 736 | * @param bool $marnm if set, include married names |
||
| 737 | * @param bool $fams if set, only fetch individuals with FAMS records |
||
| 738 | * @param LocaleInterface $locale |
||
| 739 | * |
||
| 740 | * @return Collection<Individual> |
||
| 741 | */ |
||
| 742 | protected function individuals( |
||
| 743 | Tree $tree, |
||
| 744 | string $surn, |
||
| 745 | string $salpha, |
||
| 746 | string $galpha, |
||
| 747 | bool $marnm, |
||
| 748 | bool $fams, |
||
| 749 | LocaleInterface $locale |
||
| 750 | ): Collection { |
||
| 751 | // Use specific collation for name fields. |
||
| 752 | $n_givn = $this->fieldWithCollation('n_givn'); |
||
| 753 | $n_surn = $this->fieldWithCollation('n_surn'); |
||
| 754 | |||
| 755 | $query = DB::table('individuals') |
||
| 756 | ->join('name', static function (JoinClause $join): void { |
||
| 757 | $join |
||
| 758 | ->on('n_id', '=', 'i_id') |
||
| 759 | ->on('n_file', '=', 'i_file'); |
||
| 760 | }) |
||
| 761 | ->where('i_file', '=', $tree->id()) |
||
| 762 | ->select(['i_id AS xref', 'i_gedcom AS gedcom', 'n_givn', 'n_surn']); |
||
| 763 | |||
| 764 | $this->whereFamily($fams, $query); |
||
| 765 | $this->whereMarriedName($marnm, $query); |
||
| 766 | |||
| 767 | if ($surn) { |
||
| 768 | $query->where($n_surn, '=', $surn); |
||
| 769 | } elseif ($salpha === ',') { |
||
| 770 | $query->where($n_surn, '=', ''); |
||
| 771 | } elseif ($salpha === '@') { |
||
| 772 | $query->where($n_surn, '=', Individual::NOMEN_NESCIO); |
||
| 773 | } elseif ($salpha) { |
||
| 774 | $this->whereInitial($query, 'n_surn', $salpha, $locale); |
||
| 775 | } else { |
||
| 776 | // All surnames |
||
| 777 | $query->whereNotIn($n_surn, ['', Individual::NOMEN_NESCIO]); |
||
| 778 | } |
||
| 779 | if ($galpha) { |
||
| 780 | $this->whereInitial($query, 'n_givn', $galpha, $locale); |
||
| 781 | } |
||
| 782 | |||
| 783 | $query |
||
| 784 | ->orderBy(new Expression("CASE n_surn WHEN '" . Individual::NOMEN_NESCIO . "' THEN 1 ELSE 0 END")) |
||
| 785 | ->orderBy($n_surn) |
||
| 786 | ->orderBy(new Expression("CASE n_givn WHEN '" . Individual::NOMEN_NESCIO . "' THEN 1 ELSE 0 END")) |
||
| 787 | ->orderBy($n_givn); |
||
| 788 | |||
| 789 | $individuals = new Collection(); |
||
| 790 | $rows = $query->get(); |
||
| 791 | |||
| 792 | foreach ($rows as $row) { |
||
| 793 | $individual = Registry::individualFactory()->make($row->xref, $tree, $row->gedcom); |
||
| 794 | assert($individual instanceof Individual); |
||
| 795 | |||
| 796 | // The name from the database may be private - check the filtered list... |
||
| 797 | foreach ($individual->getAllNames() as $n => $name) { |
||
| 798 | if ($name['givn'] === $row->n_givn && $name['surn'] === $row->n_surn) { |
||
| 799 | $individual->setPrimaryName($n); |
||
| 800 | // We need to clone $individual, as we may have multiple references to the |
||
| 801 | // same individual in this list, and the "primary name" would otherwise |
||
| 802 | // be shared amongst all of them. |
||
| 803 | $individuals->push(clone $individual); |
||
| 804 | break; |
||
| 805 | } |
||
| 806 | } |
||
| 807 | } |
||
| 808 | |||
| 809 | return $individuals; |
||
| 810 | } |
||
| 811 | |||
| 812 | /** |
||
| 813 | * Fetch a list of families with specified names |
||
| 814 | * To search for unknown names, use $surn="@N.N.", $salpha="@" or $galpha="@" |
||
| 815 | * To search for names with no surnames, use $salpha="," |
||
| 816 | * |
||
| 817 | * @param Tree $tree |
||
| 818 | * @param string $surn if set, only fetch people with this surname |
||
| 819 | * @param string $salpha if set, only fetch surnames starting with this letter |
||
| 820 | * @param string $galpha if set, only fetch given names starting with this letter |
||
| 821 | * @param bool $marnm if set, include married names |
||
| 822 | * @param LocaleInterface $locale |
||
| 823 | * |
||
| 824 | * @return Collection<Family> |
||
| 825 | */ |
||
| 826 | protected function families(Tree $tree, string $surn, string $salpha, string $galpha, bool $marnm, LocaleInterface $locale): Collection |
||
| 837 | } |
||
| 838 | |||
| 839 | /** |
||
| 840 | * Use MySQL-specific comments so we can run these queries on other RDBMS. |
||
| 841 | * |
||
| 842 | * @param string $field |
||
| 843 | * |
||
| 844 | * @return Expression |
||
| 845 | */ |
||
| 846 | protected function fieldWithCollation(string $field): Expression |
||
| 847 | { |
||
| 848 | return new Expression($field . ' /*! COLLATE ' . I18N::collation() . ' */'); |
||
| 849 | } |
||
| 850 | |||
| 851 | /** |
||
| 852 | * Modify a query to restrict a field to a given initial letter. |
||
| 853 | * Take account of digraphs, equialent letters, etc. |
||
| 854 | * |
||
| 855 | * @param Builder $query |
||
| 856 | * @param string $field |
||
| 857 | * @param string $letter |
||
| 858 | * @param LocaleInterface $locale |
||
| 859 | * |
||
| 860 | * @return void |
||
| 861 | */ |
||
| 862 | protected function whereInitial( |
||
| 863 | Builder $query, |
||
| 864 | string $field, |
||
| 865 | string $letter, |
||
| 866 | LocaleInterface $locale |
||
| 867 | ): void { |
||
| 868 | // Use MySQL-specific comments so we can run these queries on other RDBMS. |
||
| 869 | $field_with_collation = $this->fieldWithCollation($field); |
||
| 870 | |||
| 871 | switch ($locale->languageTag()) { |
||
| 872 | case 'cs': |
||
| 873 | $this->whereInitialCzech($query, $field_with_collation, $letter); |
||
| 874 | break; |
||
| 875 | |||
| 876 | case 'da': |
||
| 877 | case 'nb': |
||
| 878 | case 'nn': |
||
| 879 | $this->whereInitialNorwegian($query, $field_with_collation, $letter); |
||
| 880 | break; |
||
| 881 | |||
| 882 | case 'sv': |
||
| 883 | case 'fi': |
||
| 884 | $this->whereInitialSwedish($query, $field_with_collation, $letter); |
||
| 885 | break; |
||
| 886 | |||
| 887 | case 'hu': |
||
| 888 | $this->whereInitialHungarian($query, $field_with_collation, $letter); |
||
| 889 | break; |
||
| 890 | |||
| 891 | case 'nl': |
||
| 892 | $this->whereInitialDutch($query, $field_with_collation, $letter); |
||
| 893 | break; |
||
| 894 | |||
| 895 | default: |
||
| 896 | $query->where($field_with_collation, 'LIKE', '\\' . $letter . '%'); |
||
| 897 | } |
||
| 898 | } |
||
| 899 | |||
| 900 | /** |
||
| 901 | * @param Builder $query |
||
| 902 | * @param Expression $field |
||
| 903 | * @param string $letter |
||
| 904 | */ |
||
| 905 | protected function whereInitialCzech(Builder $query, Expression $field, string $letter): void |
||
| 906 | { |
||
| 907 | if ($letter === 'C') { |
||
| 908 | $query->where($field, 'LIKE', 'C%')->where($field, 'NOT LIKE', 'CH%'); |
||
| 909 | } else { |
||
| 910 | $query->where($field, 'LIKE', '\\' . $letter . '%'); |
||
| 911 | } |
||
| 912 | } |
||
| 913 | |||
| 914 | /** |
||
| 915 | * @param Builder $query |
||
| 916 | * @param Expression $field |
||
| 917 | * @param string $letter |
||
| 918 | */ |
||
| 919 | protected function whereInitialDutch(Builder $query, Expression $field, string $letter): void |
||
| 925 | } |
||
| 926 | } |
||
| 927 | |||
| 928 | /** |
||
| 929 | * Hungarian has many digraphs and trigraphs, so exclude these from prefixes. |
||
| 930 | * |
||
| 931 | * @param Builder $query |
||
| 932 | * @param Expression $field |
||
| 933 | * @param string $letter |
||
| 934 | */ |
||
| 935 | protected function whereInitialHungarian(Builder $query, Expression $field, string $letter): void |
||
| 977 | } |
||
| 978 | } |
||
| 979 | |||
| 980 | /** |
||
| 981 | * In Norwegian and Danish, AA gets listed under Å, NOT A |
||
| 982 | * |
||
| 983 | * @param Builder $query |
||
| 984 | * @param Expression $field |
||
| 985 | * @param string $letter |
||
| 986 | */ |
||
| 987 | protected function whereInitialNorwegian(Builder $query, Expression $field, string $letter): void |
||
| 1005 | } |
||
| 1006 | } |
||
| 1007 | |||
| 1008 | /** |
||
| 1009 | * In Swedish and Finnish, AA gets listed under A, NOT Å (even though Swedish collation says they should). |
||
| 1010 | * |
||
| 1011 | * @param Builder $query |
||
| 1012 | * @param Expression $field |
||
| 1013 | * @param string $letter |
||
| 1014 | */ |
||
| 1015 | protected function whereInitialSwedish(Builder $query, Expression $field, string $letter): void |
||
| 1021 | } |
||
| 1022 | } |
||
| 1023 | } |
||
| 1024 |
