1<?php 2 3/** 4 * webtrees: online genealogy 5 * Copyright (C) 2022 webtrees development team 6 * This program is free software: you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License as published by 8 * the Free Software Foundation, either version 3 of the License, or 9 * (at your option) any later version. 10 * This program is distributed in the hope that it will be useful, 11 * but WITHOUT ANY WARRANTY; without even the implied warranty of 12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 * GNU General Public License for more details. 14 * You should have received a copy of the GNU General Public License 15 * along with this program. If not, see <https://www.gnu.org/licenses/>. 16 */ 17 18declare(strict_types=1); 19 20namespace Fisharebest\Webtrees\Http\RequestHandlers; 21 22use Fisharebest\Webtrees\Exceptions\FileUploadException; 23use Fisharebest\Webtrees\FlashMessages; 24use Fisharebest\Webtrees\I18N; 25use Fisharebest\Webtrees\Registry; 26use Fisharebest\Webtrees\Services\TreeService; 27use Fisharebest\Webtrees\Validator; 28use League\Flysystem\FilesystemException; 29use League\Flysystem\UnableToReadFile; 30use Psr\Http\Message\ResponseInterface; 31use Psr\Http\Message\ServerRequestInterface; 32use Psr\Http\Message\StreamFactoryInterface; 33use Psr\Http\Server\RequestHandlerInterface; 34 35use function basename; 36use function redirect; 37use function route; 38 39use const UPLOAD_ERR_NO_FILE; 40use const UPLOAD_ERR_OK; 41 42/** 43 * Import a GEDCOM file into a tree. 44 */ 45class ImportGedcomAction implements RequestHandlerInterface 46{ 47 private StreamFactoryInterface $stream_factory; 48 49 private TreeService $tree_service; 50 51 /** 52 * @param StreamFactoryInterface $stream_factory 53 * @param TreeService $tree_service 54 */ 55 public function __construct(StreamFactoryInterface $stream_factory, TreeService $tree_service) 56 { 57 $this->tree_service = $tree_service; 58 $this->stream_factory = $stream_factory; 59 } 60 61 /** 62 * @param ServerRequestInterface $request 63 * 64 * @return ResponseInterface 65 * @throws FilesystemException 66 * @throws UnableToReadFile 67 */ 68 public function handle(ServerRequestInterface $request): ResponseInterface 69 { 70 $tree = Validator::attributes($request)->tree(); 71 72 $data_filesystem = Registry::filesystem()->data(); 73 74 $params = (array) $request->getParsedBody(); 75 $source = $params['source']; 76 $keep_media = (bool) ($params['keep_media'] ?? false); 77 $WORD_WRAPPED_NOTES = (bool) ($params['WORD_WRAPPED_NOTES'] ?? false); 78 $GEDCOM_MEDIA_PATH = $params['GEDCOM_MEDIA_PATH']; 79 $encoding = $params['encoding'] ?? ''; 80 81 // Save these choices as defaults 82 $tree->setPreference('keep_media', $keep_media ? '1' : '0'); 83 $tree->setPreference('WORD_WRAPPED_NOTES', $WORD_WRAPPED_NOTES ? '1' : '0'); 84 $tree->setPreference('GEDCOM_MEDIA_PATH', $GEDCOM_MEDIA_PATH); 85 86 if ($source === 'client') { 87 $upload = $request->getUploadedFiles()['tree_name'] ?? null; 88 89 if ($upload === null || $upload->getError() === UPLOAD_ERR_NO_FILE) { 90 FlashMessages::addMessage(I18N::translate('No GEDCOM file was received.'), 'danger'); 91 } elseif ($upload->getError() !== UPLOAD_ERR_OK) { 92 throw new FileUploadException($upload); 93 } else { 94 $this->tree_service->importGedcomFile($tree, $upload->getStream(), basename($upload->getClientFilename()), $encoding); 95 } 96 } 97 98 if ($source === 'server') { 99 $basename = basename($params['tree_name'] ?? ''); 100 101 if ($basename === '') { 102 FlashMessages::addMessage(I18N::translate('No GEDCOM file was received.'), 'danger'); 103 } else { 104 $resource = $data_filesystem->readStream($basename); 105 $stream = $this->stream_factory->createStreamFromResource($resource); 106 $this->tree_service->importGedcomFile($tree, $stream, $basename, $encoding); 107 } 108 } 109 110 $url = route(ManageTrees::class, ['tree' => $tree->name()]); 111 112 return redirect($url); 113 } 114} 115