. */ declare(strict_types=1); namespace Fisharebest\Webtrees\Module; use Fisharebest\Webtrees\Census\CensusInterface; use Fisharebest\Webtrees\Exceptions\HttpNotFoundException; use Fisharebest\Webtrees\I18N; use Fisharebest\Webtrees\Individual; use Fisharebest\Webtrees\Tree; use Psr\Http\Message\ResponseInterface; use Psr\Http\Message\ServerRequestInterface; use function assert; /** * 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 { $census = $request->getParsedBody()['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 = $request->getParsedBody(); $individual = Individual::getInstance($params['xref'], $tree); $head = Individual::getInstance($params['head'], $tree); $census = $params['census']; if ($individual instanceof Individual && $head instanceof Individual) { $html = $this->censusTableRow(new $census(), $individual, $head); return response($html); } throw new HttpNotFoundException(); } /** * @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 = $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 (array_keys($ca_individuals) as $xref) { if ($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 $n => $column) { if ($n === 0) { $text .= "\n"; } else { $text .= ' | '; } $text .= $column->abbreviation(); } foreach ($census->columns() as $n => $column) { if ($n === 0) { $text .= "\n"; } else { $text .= ' | '; } $text .= '-----'; } foreach ($ca_individuals as $xref => $columns) { $text .= "\n" . implode(' | ', $columns); } 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 .= '' . $column->abbreviation() . ''; } return '' . $html . ''; } /** * Generate an HTML row of data for the census * Add prefix cell (store XREF and drag/drop) * Add suffix cell (delete button) * * @param CensusInterface $census * * @return string */ public function censusTableEmptyRow(CensusInterface $census): string { return '' . str_repeat('', count($census->columns())) . '' . view('icons/delete') . ''; } /** * Generate an HTML row of data for the census * Add prefix cell (store XREF and drag/drop) * Add suffix cell (delete button) * * @param CensusInterface $census * @param Individual $individual * @param Individual $head * * @return string */ public function censusTableRow(CensusInterface $census, Individual $individual, Individual $head): string { $html = ''; foreach ($census->columns() as $column) { $html .= ''; } return '' . $individual->xref() . '' . $html . '' . view('icons/delete') . ''; } }