. */ declare(strict_types=1); namespace Fisharebest\Webtrees\Module; use Fisharebest\Webtrees\Census\CensusInterface; use Fisharebest\Webtrees\I18N; use Fisharebest\Webtrees\Individual; use Fisharebest\Webtrees\Tree; use Psr\Http\Message\ResponseInterface; use Psr\Http\Message\ServerRequestInterface; use function array_keys; use function assert; use function count; use function e; use function response; use function str_repeat; use function str_replace; use function view; /** * Class CensusAssistantModule */ class CensusAssistantModule extends AbstractModule { /** * How should this module be identified in the control panel, etc.? * * @return string */ public function title(): string { /* I18N: Name of a module */ return I18N::translate('Census assistant'); } /** * A sentence describing what this module does. * * @return string */ public function description(): string { /* I18N: Description of the “Census assistant” module */ return I18N::translate('An alternative way to enter census transcripts and link them to individuals.'); } /** * @param ServerRequestInterface $request * * @return ResponseInterface */ public function postCensusHeaderAction(ServerRequestInterface $request): ResponseInterface { $params = (array) $request->getParsedBody(); $census = $params['census']; $html = $this->censusTableHeader(new $census()); return response($html); } /** * @param ServerRequestInterface $request * * @return ResponseInterface */ public function postCensusIndividualAction(ServerRequestInterface $request): ResponseInterface { $tree = $request->getAttribute('tree'); assert($tree instanceof Tree); $params = (array) $request->getParsedBody(); $individual = Individual::getInstance($params['xref'], $tree); $head = Individual::getInstance($params['head'], $tree); $census_class = $params['census']; $census = new $census_class(); if ($individual instanceof Individual && $head instanceof Individual) { $html = $this->censusTableRow($census, $individual, $head); } else { $html = $this->censusTableEmptyRow($census); } return response($html); } /** * @param Individual $individual * * @return string */ public function createCensusAssistant(Individual $individual): string { return view('modules/census-assistant', [ 'individual' => $individual, ]); } /** * @param ServerRequestInterface $request * @param Individual $individual * @param string $fact_id * @param string $newged * @param bool $keep_chan * * @return string */ public function updateCensusAssistant(ServerRequestInterface $request, Individual $individual, string $fact_id, string $newged, bool $keep_chan): string { $params = (array) $request->getParsedBody(); $ca_title = $params['ca_title'] ?? ''; $ca_place = $params['ca_place'] ?? ''; $ca_citation = $params['ca_citation'] ?? ''; $ca_individuals = $params['ca_individuals'] ?? []; $ca_notes = $params['ca_notes'] ?? ''; $ca_census = $params['ca_census'] ?? ''; if ($ca_census !== '' && $ca_individuals !== []) { $census = new $ca_census(); $note_text = $this->createNoteText($census, $ca_title, $ca_place, $ca_citation, $ca_individuals, $ca_notes); $note_gedcom = '0 @@ NOTE ' . str_replace("\n", "\n1 CONT ", $note_text); $note = $individual->tree()->createRecord($note_gedcom); $newged .= "\n2 NOTE @" . $note->xref() . '@'; // Add the census fact to the rest of the household foreach ($ca_individuals['xref'] ?? [] as $xref) { if ($xref !== '' && $xref !== $individual->xref()) { Individual::getInstance($xref, $individual->tree()) ->updateFact($fact_id, $newged, !$keep_chan); } } } return $newged; } /** * @param CensusInterface $census * @param string $ca_title * @param string $ca_place * @param string $ca_citation * @param string[][] $ca_individuals * @param string $ca_notes * * @return string */ private function createNoteText(CensusInterface $census, $ca_title, $ca_place, $ca_citation, $ca_individuals, $ca_notes): string { $text = $ca_title . "\n" . $ca_citation . "\n" . $ca_place . "\n\n|"; foreach ($census->columns() as $column) { $text .= ' ' . $column->abbreviation() . ' |'; } $text .= "\n|" . str_repeat(' ----- |', count($census->columns())); foreach (array_keys($ca_individuals['xref'] ?? []) as $key) { $text .= "\n|"; foreach ($census->columns() as $n => $column) { $text .= ' ' . $ca_individuals[$n][$key] . ' |'; } } return $text . "\n\n" . $ca_notes; } /** * Generate an HTML row of data for the census header * Add prefix cell (store XREF and drag/drop) * Add suffix cell (delete button) * * @param CensusInterface $census * * @return string */ protected function censusTableHeader(CensusInterface $census): string { $html = ''; foreach ($census->columns() as $column) { $html .= '