. */ 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 Symfony\Component\HttpFoundation\Request; use Symfony\Component\HttpFoundation\Response; use Symfony\Component\HttpKernel\Exception\NotFoundHttpException; /** * 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 Request $request * * @return Response */ public function getCensusHeaderAction(Request $request): Response { $census = $request->get('census'); $html = $this->censusTableHeader(new $census()); return new Response($html); } /** * @param Request $request * @param Tree $tree * * @return Response */ public function getCensusIndividualAction(Request $request, Tree $tree): Response { $census = $request->get('census', ''); $individual = Individual::getInstance($request->get('xref', ''), $tree); $head = Individual::getInstance($request->get('head', ''), $tree); if ($individual instanceof Individual && $head instanceof Individual) { $html = $this->censusTableRow(new $census(), $individual, $head); return new Response($html); } throw new NotFoundHttpException(); } /** * @param Individual $individual * * @return string */ public function createCensusAssistant(Individual $individual): string { return view('modules/census-assistant', [ 'individual' => $individual, ]); } /** * @param Request $request * @param Individual $individual * @param string $fact_id * @param string $newged * @param bool $keep_chan * * @return string */ public function updateCensusAssistant(Request $request, Individual $individual, string $fact_id, string $newged, bool $keep_chan): string { $ca_title = $request->get('ca_title', ''); $ca_place = $request->get('ca_place', ''); $ca_citation = $request->get('ca_citation', ''); $ca_individuals = (array) $request->get('ca_individuals'); $ca_notes = $request->get('ca_notes', ''); $ca_census = $request->get('ca_census', ''); if ($ca_census !== '' && !empty($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 .= '