.
*/
declare(strict_types=1);
namespace Fisharebest\Webtrees\Module;
use Fisharebest\Webtrees\Census\CensusInterface;
use Fisharebest\Webtrees\Registry;
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 = Registry::individualFactory()->make($params['xref'], $tree);
$head = Registry::individualFactory()->make($params['head'], $tree);
$census_class = $params['census'];
$census = new $census_class();
// No head of household? Create a fake one.
$head = $head ?? Registry::individualFactory()->new('X', '0 @X@ INDI', null, $tree);
// Generate columns (e.g. relationship name) using the correct language.
I18N::init($census->censusLanguage());
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()) {
Registry::individualFactory()->make($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, string $ca_title, string $ca_place, string $ca_citation, array $ca_individuals, string $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" . strtr($ca_notes, ["\r" => '']);
}
/**
* 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
{
$html = '