app/Module/AncestorsChartModule.php
<?php
/**
* webtrees: online genealogy
* Copyright (C) 2023 webtrees development team
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
declare(strict_types=1);
namespace Fisharebest\Webtrees\Module;
use Fig\Http\Message\RequestMethodInterface;
use Fisharebest\Webtrees\Auth;
use Fisharebest\Webtrees\I18N;
use Fisharebest\Webtrees\Individual;
use Fisharebest\Webtrees\Menu;
use Fisharebest\Webtrees\Registry;
use Fisharebest\Webtrees\Services\ChartService;
use Fisharebest\Webtrees\Validator;
use Psr\Http\Message\ResponseInterface;
use Psr\Http\Message\ServerRequestInterface;
use Psr\Http\Server\RequestHandlerInterface;
use function route;
/**
* Class AncestorsChartModule
*/
class AncestorsChartModule extends AbstractModule implements ModuleChartInterface, RequestHandlerInterface
{
use ModuleChartTrait;
protected const ROUTE_URL = '/tree/{tree}/ancestors-{style}-{generations}/{xref}';
// Chart styles
public const CHART_STYLE_TREE = 'tree';
public const CHART_STYLE_INDIVIDUALS = 'individuals';
public const CHART_STYLE_FAMILIES = 'families';
// Defaults
public const DEFAULT_GENERATIONS = '4';
public const DEFAULT_STYLE = self::CHART_STYLE_TREE;
protected const DEFAULT_PARAMETERS = [
'generations' => self::DEFAULT_GENERATIONS,
'style' => self::DEFAULT_STYLE,
];
// Limits
protected const MINIMUM_GENERATIONS = 2;
protected const MAXIMUM_GENERATIONS = 10;
private ChartService $chart_service;
/**
* @param ChartService $chart_service
*/
public function __construct(ChartService $chart_service)
{
$this->chart_service = $chart_service;
}
/**
* Initialization.
*
* @return void
*/
public function boot(): void
{
Registry::routeFactory()->routeMap()
->get(static::class, static::ROUTE_URL, $this)
->allows(RequestMethodInterface::METHOD_POST)
->tokens([
'generations' => '\d+',
'style' => implode('|', array_keys($this->styles())),
]);
}
/**
* How should this module be identified in the control panel, etc.?
*
* @return string
*/
public function title(): string
{
/* I18N: Name of a module/chart */
return I18N::translate('Ancestors');
}
/**
* A sentence describing what this module does.
*
* @return string
*/
public function description(): string
{
/* I18N: Description of the “AncestorsChart” module */
return I18N::translate('A chart of an individual’s ancestors.');
}
/**
* CSS class for the URL.
*
* @return string
*/
public function chartMenuClass(): string
{
return 'menu-chart-ancestry';
}
/**
* Return a menu item for this chart - for use in individual boxes.
*/
public function chartBoxMenu(Individual $individual): Menu|null
{
return $this->chartMenu($individual);
}
/**
* The title for a specific instance of this chart.
*
* @param Individual $individual
*
* @return string
*/
public function chartTitle(Individual $individual): string
{
/* I18N: %s is an individual’s name */
return I18N::translate('Ancestors of %s', $individual->fullName());
}
/**
* The URL for a page showing chart options.
*
* @param Individual $individual
* @param array<bool|int|string|array<string>|null> $parameters
*
* @return string
*/
public function chartUrl(Individual $individual, array $parameters = []): string
{
return route(static::class, [
'xref' => $individual->xref(),
'tree' => $individual->tree()->name(),
] + $parameters + self::DEFAULT_PARAMETERS);
}
/**
* @param ServerRequestInterface $request
*
* @return ResponseInterface
*/
public function handle(ServerRequestInterface $request): ResponseInterface
{
$tree = Validator::attributes($request)->tree();
$user = Validator::attributes($request)->user();
$style = Validator::attributes($request)->isInArrayKeys($this->styles())->string('style');
$xref = Validator::attributes($request)->isXref()->string('xref');
$generations = Validator::attributes($request)->isBetween(self::MINIMUM_GENERATIONS, self::MAXIMUM_GENERATIONS)->integer('generations');
$ajax = Validator::queryParams($request)->boolean('ajax', false);
// Convert POST requests into GET requests for pretty URLs.
if ($request->getMethod() === RequestMethodInterface::METHOD_POST) {
return redirect(route(static::class, [
'tree' => $tree->name(),
'xref' => Validator::parsedBody($request)->isXref()->string('xref'),
'style' => Validator::parsedBody($request)->isInArrayKeys($this->styles())->string('style'),
'generations' => Validator::parsedBody($request)->isBetween(self::MINIMUM_GENERATIONS, self::MAXIMUM_GENERATIONS)->integer('generations'),
]));
}
Auth::checkComponentAccess($this, ModuleChartInterface::class, $tree, $user);
$individual = Registry::individualFactory()->make($xref, $tree);
$individual = Auth::checkIndividualAccess($individual, false, true);
if ($ajax) {
$this->layout = 'layouts/ajax';
$ancestors = $this->chart_service->sosaStradonitzAncestors($individual, $generations);
switch ($style) {
default:
case self::CHART_STYLE_TREE:
return $this->viewResponse('modules/ancestors-chart/tree', [
'individual' => $individual,
'parents' => $individual->childFamilies()->first(),
'generations' => $generations,
'sosa' => 1,
]);
case self::CHART_STYLE_INDIVIDUALS:
return $this->viewResponse('lists/individuals-table', [
'individuals' => $ancestors,
'sosa' => true,
'tree' => $tree,
]);
case self::CHART_STYLE_FAMILIES:
$families = [];
foreach ($ancestors as $individual) {
foreach ($individual->childFamilies() as $family) {
$families[$family->xref()] = $family;
}
}
return $this->viewResponse('lists/families-table', ['families' => $families, 'tree' => $tree]);
}
}
$ajax_url = $this->chartUrl($individual, [
'ajax' => true,
'generations' => $generations,
'style' => $style,
'xref' => $xref,
]);
return $this->viewResponse('modules/ancestors-chart/page', [
'ajax_url' => $ajax_url,
'generations' => $generations,
'individual' => $individual,
'maximum_generations' => self::MAXIMUM_GENERATIONS,
'minimum_generations' => self::MINIMUM_GENERATIONS,
'module' => $this->name(),
'style' => $style,
'styles' => $this->styles(),
'title' => $this->chartTitle($individual),
'tree' => $tree,
]);
}
/**
* This chart can display its output in a number of styles
*
* @return array<string>
*/
protected function styles(): array
{
return [
self::CHART_STYLE_TREE => I18N::translate('Tree'),
self::CHART_STYLE_INDIVIDUALS => I18N::translate('Individuals'),
self::CHART_STYLE_FAMILIES => I18N::translate('Families'),
];
}
}