1a25f0a04SGreg Roach<?php 2dd04c183SGreg Roachnamespace Fisharebest\Webtrees; 3a25f0a04SGreg Roach 4a25f0a04SGreg Roach/** 5a25f0a04SGreg Roach * webtrees: online genealogy 6a25f0a04SGreg Roach * Copyright (C) 2015 webtrees development team 7a25f0a04SGreg Roach * This program is free software: you can redistribute it and/or modify 8a25f0a04SGreg Roach * it under the terms of the GNU General Public License as published by 9a25f0a04SGreg Roach * the Free Software Foundation, either version 3 of the License, or 10a25f0a04SGreg Roach * (at your option) any later version. 11a25f0a04SGreg Roach * This program is distributed in the hope that it will be useful, 12a25f0a04SGreg Roach * but WITHOUT ANY WARRANTY; without even the implied warranty of 13a25f0a04SGreg Roach * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14a25f0a04SGreg Roach * GNU General Public License for more details. 15a25f0a04SGreg Roach * You should have received a copy of the GNU General Public License 16a25f0a04SGreg Roach * along with this program. If not, see <http://www.gnu.org/licenses/>. 17a25f0a04SGreg Roach */ 18a25f0a04SGreg Roach 19a25f0a04SGreg Roachuse PDOException; 20a25f0a04SGreg Roach 21a25f0a04SGreg Roach/** 22a25f0a04SGreg Roach * Class Tree - Provide an interface to the wt_gedcom table 23a25f0a04SGreg Roach */ 24a25f0a04SGreg Roachclass Tree { 25a25f0a04SGreg Roach /** @var integer The tree's ID number */ 26518bbdc1SGreg Roach private $tree_id; 27518bbdc1SGreg Roach 28a25f0a04SGreg Roach /** @var string The tree's name */ 29a25f0a04SGreg Roach private $name; 30518bbdc1SGreg Roach 31a25f0a04SGreg Roach /** @var string The tree's title */ 32a25f0a04SGreg Roach private $title; 33a25f0a04SGreg Roach 34518bbdc1SGreg Roach /** @var integer[] Default access rules for facts in this tree */ 35518bbdc1SGreg Roach private $fact_privacy; 36518bbdc1SGreg Roach 37518bbdc1SGreg Roach /** @var integer[] Default access rules for individuals in this tree */ 38518bbdc1SGreg Roach private $individual_privacy; 39518bbdc1SGreg Roach 40518bbdc1SGreg Roach /** @var integer[][] Default access rules for individual facts in this tree */ 41518bbdc1SGreg Roach private $individual_fact_privacy; 42518bbdc1SGreg Roach 43a25f0a04SGreg Roach /** @var Tree[] All trees that we have permission to see. */ 44a25f0a04SGreg Roach private static $trees; 45a25f0a04SGreg Roach 46a25f0a04SGreg Roach /** @var string[] Cached copy of the wt_gedcom_setting table. */ 47a25f0a04SGreg Roach private $preferences; 48a25f0a04SGreg Roach 49a25f0a04SGreg Roach /** @var string[][] Cached copy of the wt_user_gedcom_setting table. */ 50a25f0a04SGreg Roach private $user_preferences = array(); 51a25f0a04SGreg Roach 52a25f0a04SGreg Roach /** 53a25f0a04SGreg Roach * Create a tree object. This is a private constructor - it can only 54a25f0a04SGreg Roach * be called from Tree::getAll() to ensure proper initialisation. 55a25f0a04SGreg Roach * 56518bbdc1SGreg Roach * @param integer $tree_id 57a25f0a04SGreg Roach * @param string $tree_name 58a25f0a04SGreg Roach * @param string $tree_title 59a25f0a04SGreg Roach */ 60518bbdc1SGreg Roach private function __construct($tree_id, $tree_name, $tree_title) { 61518bbdc1SGreg Roach $this->tree_id = $tree_id; 62a25f0a04SGreg Roach $this->name = $tree_name; 63a25f0a04SGreg Roach $this->title = $tree_title; 64518bbdc1SGreg Roach $this->fact_privacy = array(); 65518bbdc1SGreg Roach $this->individual_privacy = array(); 66518bbdc1SGreg Roach $this->individual_fact_privacy = array(); 67518bbdc1SGreg Roach 68518bbdc1SGreg Roach // Load the privacy settings for this tree 69518bbdc1SGreg Roach $rows = Database::prepare( 70518bbdc1SGreg Roach "SELECT SQL_CACHE xref, tag_type, CASE resn WHEN 'none' THEN :priv_public WHEN 'privacy' THEN :priv_user WHEN 'confidential' THEN :priv_none WHEN 'hidden' THEN :priv_hide END AS resn" . 71518bbdc1SGreg Roach " FROM `##default_resn` WHERE gedcom_id = :tree_id" 72518bbdc1SGreg Roach )->execute(array( 73518bbdc1SGreg Roach 'priv_public' => WT_PRIV_PUBLIC, 74518bbdc1SGreg Roach 'priv_user' => WT_PRIV_USER, 75518bbdc1SGreg Roach 'priv_none' => WT_PRIV_NONE, 76518bbdc1SGreg Roach 'priv_hide' => WT_PRIV_HIDE, 77518bbdc1SGreg Roach 'tree_id' => $this->tree_id 78518bbdc1SGreg Roach ))->fetchAll(); 79518bbdc1SGreg Roach 80518bbdc1SGreg Roach foreach ($rows as $row) { 81518bbdc1SGreg Roach if ($row->xref !== null) { 82518bbdc1SGreg Roach if ($row->tag_type !== null) { 83518bbdc1SGreg Roach $this->individual_fact_privacy[$row->xref][$row->tag_type] = (int) $row->resn; 84518bbdc1SGreg Roach } else { 85518bbdc1SGreg Roach $this->individual_privacy[$row->xref] = (int) $row->resn; 86518bbdc1SGreg Roach } 87518bbdc1SGreg Roach } else { 88518bbdc1SGreg Roach $this->fact_privacy[$row->tag_type] = (int) $row->resn; 89518bbdc1SGreg Roach } 90518bbdc1SGreg Roach } 91518bbdc1SGreg Roach 92518bbdc1SGreg Roach 93a25f0a04SGreg Roach } 94a25f0a04SGreg Roach 95a25f0a04SGreg Roach /** 96a25f0a04SGreg Roach * The ID of this tree 97a25f0a04SGreg Roach * 98a25f0a04SGreg Roach * @return integer 99a25f0a04SGreg Roach */ 100518bbdc1SGreg Roach public function getTreeId() { 101518bbdc1SGreg Roach return $this->tree_id; 102a25f0a04SGreg Roach } 103a25f0a04SGreg Roach 104a25f0a04SGreg Roach /** 105a25f0a04SGreg Roach * The name of this tree 106a25f0a04SGreg Roach * 107a25f0a04SGreg Roach * @return string 108a25f0a04SGreg Roach */ 109000959d9SGreg Roach public function getName() { 110a25f0a04SGreg Roach return $this->name; 111a25f0a04SGreg Roach } 112a25f0a04SGreg Roach 113a25f0a04SGreg Roach /** 114a25f0a04SGreg Roach * The name of this tree 115a25f0a04SGreg Roach * 116a25f0a04SGreg Roach * @return string 117a25f0a04SGreg Roach */ 118000959d9SGreg Roach public function getNameHtml() { 119a25f0a04SGreg Roach return Filter::escapeHtml($this->name); 120a25f0a04SGreg Roach } 121a25f0a04SGreg Roach 122a25f0a04SGreg Roach /** 123a25f0a04SGreg Roach * The name of this tree 124a25f0a04SGreg Roach * 125a25f0a04SGreg Roach * @return string 126a25f0a04SGreg Roach */ 127000959d9SGreg Roach public function getNameUrl() { 128a25f0a04SGreg Roach return Filter::escapeUrl($this->name); 129a25f0a04SGreg Roach } 130a25f0a04SGreg Roach 131a25f0a04SGreg Roach /** 132a25f0a04SGreg Roach * The title of this tree 133a25f0a04SGreg Roach * 134a25f0a04SGreg Roach * @return string 135a25f0a04SGreg Roach */ 136000959d9SGreg Roach public function getTitle() { 137a25f0a04SGreg Roach return $this->title; 138a25f0a04SGreg Roach } 139a25f0a04SGreg Roach 140a25f0a04SGreg Roach /** 141a25f0a04SGreg Roach * The title of this tree, with HTML markup 142a25f0a04SGreg Roach * 143a25f0a04SGreg Roach * @return string 144a25f0a04SGreg Roach */ 145000959d9SGreg Roach public function getTitleHtml() { 146a25f0a04SGreg Roach return Filter::escapeHtml($this->title); 147a25f0a04SGreg Roach } 148a25f0a04SGreg Roach 149a25f0a04SGreg Roach /** 150518bbdc1SGreg Roach * The fact-level privacy for this tree. 151518bbdc1SGreg Roach * 1524f381cf1SGreg Roach * @return integer[] 153518bbdc1SGreg Roach */ 154518bbdc1SGreg Roach public function getFactPrivacy() { 155518bbdc1SGreg Roach return $this->fact_privacy; 156518bbdc1SGreg Roach } 157518bbdc1SGreg Roach 158518bbdc1SGreg Roach /** 159518bbdc1SGreg Roach * The individual-level privacy for this tree. 160518bbdc1SGreg Roach * 1614f381cf1SGreg Roach * @return integer[] 162518bbdc1SGreg Roach */ 163518bbdc1SGreg Roach public function getIndividualPrivacy() { 164518bbdc1SGreg Roach return $this->individual_privacy; 165518bbdc1SGreg Roach } 166518bbdc1SGreg Roach 167518bbdc1SGreg Roach /** 168518bbdc1SGreg Roach * The individual-fact-level privacy for this tree. 169518bbdc1SGreg Roach * 1704f381cf1SGreg Roach * @return integer[][] 171518bbdc1SGreg Roach */ 172518bbdc1SGreg Roach public function getIndividualFactPrivacy() { 173518bbdc1SGreg Roach return $this->individual_fact_privacy; 174518bbdc1SGreg Roach } 175518bbdc1SGreg Roach 176518bbdc1SGreg Roach /** 177a25f0a04SGreg Roach * Get the tree’s configuration settings. 178a25f0a04SGreg Roach * 179a25f0a04SGreg Roach * @param string $setting_name 180a25f0a04SGreg Roach * @param string|null $default 181a25f0a04SGreg Roach * 182a25f0a04SGreg Roach * @return string|null 183a25f0a04SGreg Roach */ 184a25f0a04SGreg Roach public function getPreference($setting_name, $default = null) { 185a25f0a04SGreg Roach if ($this->preferences === null) { 186a25f0a04SGreg Roach $this->preferences = Database::prepare( 187a25f0a04SGreg Roach "SELECT SQL_CACHE setting_name, setting_value FROM `##gedcom_setting` WHERE gedcom_id = ?" 188518bbdc1SGreg Roach )->execute(array($this->tree_id))->fetchAssoc(); 189a25f0a04SGreg Roach } 190a25f0a04SGreg Roach 191a25f0a04SGreg Roach if (array_key_exists($setting_name, $this->preferences)) { 192a25f0a04SGreg Roach return $this->preferences[$setting_name]; 193a25f0a04SGreg Roach } else { 194a25f0a04SGreg Roach return $default; 195a25f0a04SGreg Roach } 196a25f0a04SGreg Roach } 197a25f0a04SGreg Roach 198a25f0a04SGreg Roach /** 199a25f0a04SGreg Roach * Set the tree’s configuration settings. 200a25f0a04SGreg Roach * 201a25f0a04SGreg Roach * @param string $setting_name 202a25f0a04SGreg Roach * @param string $setting_value 203a25f0a04SGreg Roach * 204a25f0a04SGreg Roach * @return $this 205a25f0a04SGreg Roach */ 206a25f0a04SGreg Roach public function setPreference($setting_name, $setting_value) { 207a25f0a04SGreg Roach if ($setting_value !== $this->getPreference($setting_name)) { 208a25f0a04SGreg Roach // Update the database 209a25f0a04SGreg Roach if ($setting_value === null) { 210a25f0a04SGreg Roach Database::prepare( 211a25f0a04SGreg Roach "DELETE FROM `##gedcom_setting` WHERE gedcom_id = :tree_id AND setting_name = :setting_name" 212a25f0a04SGreg Roach )->execute(array( 213518bbdc1SGreg Roach 'tree_id' => $this->tree_id, 214a25f0a04SGreg Roach 'setting_name' => $setting_name, 215a25f0a04SGreg Roach )); 216a25f0a04SGreg Roach } else { 217a25f0a04SGreg Roach Database::prepare( 218a25f0a04SGreg Roach "REPLACE INTO `##gedcom_setting` (gedcom_id, setting_name, setting_value)" . 219a25f0a04SGreg Roach " VALUES (:tree_id, :setting_name, LEFT(:setting_value, 255))" 220a25f0a04SGreg Roach )->execute(array( 221518bbdc1SGreg Roach 'tree_id' => $this->tree_id, 222a25f0a04SGreg Roach 'setting_name' => $setting_name, 223a25f0a04SGreg Roach 'setting_value' => $setting_value, 224a25f0a04SGreg Roach )); 225a25f0a04SGreg Roach } 226a25f0a04SGreg Roach // Update our cache 227a25f0a04SGreg Roach $this->preferences[$setting_name] = $setting_value; 228a25f0a04SGreg Roach // Audit log of changes 229a25f0a04SGreg Roach Log::addConfigurationLog('Tree setting "' . $setting_name . '" set to "' . $setting_value . '"', $this); 230a25f0a04SGreg Roach } 231a25f0a04SGreg Roach 232a25f0a04SGreg Roach return $this; 233a25f0a04SGreg Roach } 234a25f0a04SGreg Roach 235a25f0a04SGreg Roach /** 236a25f0a04SGreg Roach * Get the tree’s user-configuration settings. 237a25f0a04SGreg Roach * 238a25f0a04SGreg Roach * @param User $user 239a25f0a04SGreg Roach * @param string $setting_name 240a25f0a04SGreg Roach * @param string|null $default 241a25f0a04SGreg Roach * 242a25f0a04SGreg Roach * @return string 243a25f0a04SGreg Roach */ 244a25f0a04SGreg Roach public function getUserPreference(User $user, $setting_name, $default = null) { 245a25f0a04SGreg Roach // There are lots of settings, and we need to fetch lots of them on every page 246a25f0a04SGreg Roach // so it is quicker to fetch them all in one go. 247a25f0a04SGreg Roach if (!array_key_exists($user->getUserId(), $this->user_preferences)) { 248a25f0a04SGreg Roach $this->user_preferences[$user->getUserId()] = Database::prepare( 249a25f0a04SGreg Roach "SELECT SQL_CACHE setting_name, setting_value FROM `##user_gedcom_setting` WHERE user_id = ? AND gedcom_id = ?" 250518bbdc1SGreg Roach )->execute(array($user->getUserId(), $this->tree_id))->fetchAssoc(); 251a25f0a04SGreg Roach } 252a25f0a04SGreg Roach 253a25f0a04SGreg Roach if (array_key_exists($setting_name, $this->user_preferences[$user->getUserId()])) { 254a25f0a04SGreg Roach return $this->user_preferences[$user->getUserId()][$setting_name]; 255a25f0a04SGreg Roach } else { 256a25f0a04SGreg Roach return $default; 257a25f0a04SGreg Roach } 258a25f0a04SGreg Roach } 259a25f0a04SGreg Roach 260a25f0a04SGreg Roach /** 261a25f0a04SGreg Roach * Set the tree’s user-configuration settings. 262a25f0a04SGreg Roach * 263a25f0a04SGreg Roach * @param User $user 264a25f0a04SGreg Roach * @param string $setting_name 265a25f0a04SGreg Roach * @param string $setting_value 266a25f0a04SGreg Roach * 267a25f0a04SGreg Roach * @return $this 268a25f0a04SGreg Roach */ 269a25f0a04SGreg Roach public function setUserPreference(User $user, $setting_name, $setting_value) { 270a25f0a04SGreg Roach if ($this->getUserPreference($user, $setting_name) !== $setting_value) { 271a25f0a04SGreg Roach // Update the database 272a25f0a04SGreg Roach if ($setting_value === null) { 273a25f0a04SGreg Roach Database::prepare( 274a25f0a04SGreg Roach "DELETE FROM `##user_gedcom_setting` WHERE gedcom_id = :tree_id AND user_id = :user_id AND setting_name = :setting_name" 275a25f0a04SGreg Roach )->execute(array( 276518bbdc1SGreg Roach 'tree_id' => $this->tree_id, 277a25f0a04SGreg Roach 'user_id' => $user->getUserId(), 278a25f0a04SGreg Roach 'setting_name' => $setting_name, 279a25f0a04SGreg Roach )); 280a25f0a04SGreg Roach } else { 281a25f0a04SGreg Roach Database::prepare( 282a25f0a04SGreg Roach "REPLACE INTO `##user_gedcom_setting` (user_id, gedcom_id, setting_name, setting_value) VALUES (:user_id, :tree_id, :setting_name, LEFT(:setting_value, 255))" 283a25f0a04SGreg Roach )->execute(array( 284a25f0a04SGreg Roach 'user_id' => $user->getUserId(), 285518bbdc1SGreg Roach 'tree_id' => $this->tree_id, 286a25f0a04SGreg Roach 'setting_name' => $setting_name, 287a25f0a04SGreg Roach 'setting_value' => $setting_value 288a25f0a04SGreg Roach )); 289a25f0a04SGreg Roach } 290a25f0a04SGreg Roach // Update our cache 291a25f0a04SGreg Roach $this->user_preferences[$user->getUserId()][$setting_name] = $setting_value; 292a25f0a04SGreg Roach // Audit log of changes 293a25f0a04SGreg Roach Log::addConfigurationLog('Tree setting "' . $setting_name . '" set to "' . $setting_value . '" for user "' . $user->getUserName() . '"', $this); 294a25f0a04SGreg Roach } 295a25f0a04SGreg Roach 296a25f0a04SGreg Roach return $this; 297a25f0a04SGreg Roach } 298a25f0a04SGreg Roach 299a25f0a04SGreg Roach /** 300a25f0a04SGreg Roach * Can a user accept changes for this tree? 301a25f0a04SGreg Roach * 302a25f0a04SGreg Roach * @param User $user 303a25f0a04SGreg Roach * 304a25f0a04SGreg Roach * @return boolean 305a25f0a04SGreg Roach */ 306a25f0a04SGreg Roach public function canAcceptChanges(User $user) { 307a25f0a04SGreg Roach return Auth::isModerator($this, $user); 308a25f0a04SGreg Roach } 309a25f0a04SGreg Roach 310a25f0a04SGreg Roach /** 311a25f0a04SGreg Roach * Fetch all the trees that we have permission to access. 312a25f0a04SGreg Roach * 313a25f0a04SGreg Roach * @return Tree[] 314a25f0a04SGreg Roach */ 315a25f0a04SGreg Roach public static function getAll() { 316a25f0a04SGreg Roach if (self::$trees === null) { 317a25f0a04SGreg Roach self::$trees = array(); 318a25f0a04SGreg Roach $rows = Database::prepare( 319a25f0a04SGreg Roach "SELECT SQL_CACHE g.gedcom_id AS tree_id, g.gedcom_name AS tree_name, gs1.setting_value AS tree_title" . 320a25f0a04SGreg Roach " FROM `##gedcom` g" . 321a25f0a04SGreg Roach " LEFT JOIN `##gedcom_setting` gs1 ON (g.gedcom_id=gs1.gedcom_id AND gs1.setting_name='title')" . 322a25f0a04SGreg Roach " LEFT JOIN `##gedcom_setting` gs2 ON (g.gedcom_id=gs2.gedcom_id AND gs2.setting_name='imported')" . 323a25f0a04SGreg Roach " LEFT JOIN `##gedcom_setting` gs3 ON (g.gedcom_id=gs3.gedcom_id AND gs3.setting_name='REQUIRE_AUTHENTICATION')" . 324a25f0a04SGreg Roach " LEFT JOIN `##user_gedcom_setting` ugs ON (g.gedcom_id=ugs.gedcom_id AND ugs.setting_name='canedit' AND ugs.user_id=?)" . 325a25f0a04SGreg Roach " WHERE " . 326a25f0a04SGreg Roach " g.gedcom_id>0 AND (" . // exclude the "template" tree 327a25f0a04SGreg Roach " EXISTS (SELECT 1 FROM `##user_setting` WHERE user_id=? AND setting_name='canadmin' AND setting_value=1)" . // Admin sees all 328a25f0a04SGreg Roach " ) OR (" . 329a25f0a04SGreg Roach " gs2.setting_value = 1 AND (" . // Allow imported trees, with either: 330a25f0a04SGreg Roach " gs3.setting_value <> 1 OR" . // visitor access 331a25f0a04SGreg Roach " IFNULL(ugs.setting_value, 'none')<>'none'" . // explicit access 332a25f0a04SGreg Roach " )" . 333a25f0a04SGreg Roach " )" . 334a25f0a04SGreg Roach " ORDER BY g.sort_order, 3" 335a25f0a04SGreg Roach )->execute(array(Auth::id(), Auth::id()))->fetchAll(); 336a25f0a04SGreg Roach foreach ($rows as $row) { 337a25f0a04SGreg Roach self::$trees[$row->tree_id] = new self($row->tree_id, $row->tree_name, $row->tree_title); 338a25f0a04SGreg Roach } 339a25f0a04SGreg Roach } 340a25f0a04SGreg Roach 341a25f0a04SGreg Roach return self::$trees; 342a25f0a04SGreg Roach } 343a25f0a04SGreg Roach 344a25f0a04SGreg Roach /** 345d2cdeb3fSGreg Roach * Find the tree with a specific ID. 346a25f0a04SGreg Roach * 347a25f0a04SGreg Roach * @param integer $tree_id 348a25f0a04SGreg Roach * 349a25f0a04SGreg Roach * @return Tree 350a25f0a04SGreg Roach */ 351d2cdeb3fSGreg Roach public static function findById($tree_id) { 352a25f0a04SGreg Roach $trees = self::getAll(); 353a25f0a04SGreg Roach 354a25f0a04SGreg Roach return $trees[$tree_id]; 355a25f0a04SGreg Roach } 356a25f0a04SGreg Roach 357a25f0a04SGreg Roach /** 358d2cdeb3fSGreg Roach * Find the tree with a specific name. 359cf4bcc09SGreg Roach * 360cf4bcc09SGreg Roach * @param string $tree_name 361cf4bcc09SGreg Roach * 362cf4bcc09SGreg Roach * @return Tree|null 363cf4bcc09SGreg Roach */ 364cf4bcc09SGreg Roach public static function findByName($tree_name) { 365cf4bcc09SGreg Roach foreach (self::getAll() as $tree) { 366d2cdeb3fSGreg Roach if ($tree->getName() === $tree_name) { 367cf4bcc09SGreg Roach return $tree; 368cf4bcc09SGreg Roach } 369cf4bcc09SGreg Roach } 370cf4bcc09SGreg Roach 371cf4bcc09SGreg Roach return null; 372cf4bcc09SGreg Roach } 373cf4bcc09SGreg Roach 374cf4bcc09SGreg Roach /** 375a25f0a04SGreg Roach * Create arguments to select_edit_control() 376a25f0a04SGreg Roach * Note - these will be escaped later 377a25f0a04SGreg Roach * 378a25f0a04SGreg Roach * @return string[] 379a25f0a04SGreg Roach */ 380a25f0a04SGreg Roach public static function getIdList() { 381a25f0a04SGreg Roach $list = array(); 382a25f0a04SGreg Roach foreach (self::getAll() as $tree) { 383518bbdc1SGreg Roach $list[$tree->tree_id] = $tree->title; 384a25f0a04SGreg Roach } 385a25f0a04SGreg Roach 386a25f0a04SGreg Roach return $list; 387a25f0a04SGreg Roach } 388a25f0a04SGreg Roach 389a25f0a04SGreg Roach /** 390a25f0a04SGreg Roach * Create arguments to select_edit_control() 391a25f0a04SGreg Roach * Note - these will be escaped later 392a25f0a04SGreg Roach * 393a25f0a04SGreg Roach * @return string[] 394a25f0a04SGreg Roach */ 395a25f0a04SGreg Roach public static function getNameList() { 396a25f0a04SGreg Roach $list = array(); 397a25f0a04SGreg Roach foreach (self::getAll() as $tree) { 398a25f0a04SGreg Roach $list[$tree->name] = $tree->title; 399a25f0a04SGreg Roach } 400a25f0a04SGreg Roach 401a25f0a04SGreg Roach return $list; 402a25f0a04SGreg Roach } 403a25f0a04SGreg Roach 404a25f0a04SGreg Roach /** 405a25f0a04SGreg Roach * Find the ID number for a tree name 406a25f0a04SGreg Roach * 407a25f0a04SGreg Roach * @param integer $tree_name 408a25f0a04SGreg Roach * 409a25f0a04SGreg Roach * @return integer|null 410a25f0a04SGreg Roach */ 411a25f0a04SGreg Roach public static function getIdFromName($tree_name) { 412a25f0a04SGreg Roach foreach (self::getAll() as $tree_id => $tree) { 413a25f0a04SGreg Roach if ($tree->name == $tree_name) { 414a25f0a04SGreg Roach return $tree_id; 415a25f0a04SGreg Roach } 416a25f0a04SGreg Roach } 417a25f0a04SGreg Roach 418a25f0a04SGreg Roach return null; 419a25f0a04SGreg Roach } 420a25f0a04SGreg Roach 421a25f0a04SGreg Roach /** 422a25f0a04SGreg Roach * Find the tree name from a numeric ID. 423a25f0a04SGreg Roach * @param integer $tree_id 424a25f0a04SGreg Roach * 425a25f0a04SGreg Roach * @return string 426a25f0a04SGreg Roach */ 427a25f0a04SGreg Roach public static function getNameFromId($tree_id) { 428d2cdeb3fSGreg Roach return self::findById($tree_id)->name; 429a25f0a04SGreg Roach } 430a25f0a04SGreg Roach 431a25f0a04SGreg Roach /** 432a25f0a04SGreg Roach * Create a new tree 433a25f0a04SGreg Roach * 434a25f0a04SGreg Roach * @param string $tree_name 435a25f0a04SGreg Roach * @param string $tree_title 436a25f0a04SGreg Roach * 437a25f0a04SGreg Roach * @return Tree 438a25f0a04SGreg Roach */ 439a25f0a04SGreg Roach public static function create($tree_name, $tree_title) { 440a25f0a04SGreg Roach try { 441a25f0a04SGreg Roach // Create a new tree 442a25f0a04SGreg Roach Database::prepare( 443a25f0a04SGreg Roach "INSERT INTO `##gedcom` (gedcom_name) VALUES (?)" 444a25f0a04SGreg Roach )->execute(array($tree_name)); 445a25f0a04SGreg Roach $tree_id = Database::prepare("SELECT LAST_INSERT_ID()")->fetchOne(); 446a25f0a04SGreg Roach } catch (PDOException $ex) { 447a25f0a04SGreg Roach // A tree with that name already exists? 448d2cdeb3fSGreg Roach return self::findById(self::getIdFromName($tree_name)); 449a25f0a04SGreg Roach } 450a25f0a04SGreg Roach 451a25f0a04SGreg Roach // Update the list of trees - to include this new one 452a25f0a04SGreg Roach self::$trees = null; 453d2cdeb3fSGreg Roach $tree = self::findById($tree_id); 454a25f0a04SGreg Roach 455a25f0a04SGreg Roach $tree->setPreference('imported', '0'); 456a25f0a04SGreg Roach $tree->setPreference('title', $tree_title); 457a25f0a04SGreg Roach 458a25f0a04SGreg Roach // Module privacy 459a25f0a04SGreg Roach Module::setDefaultAccess($tree_id); 460a25f0a04SGreg Roach 461a25f0a04SGreg Roach // Gedcom and privacy settings 462a25f0a04SGreg Roach $tree->setPreference('ADVANCED_NAME_FACTS', 'NICK,_AKA'); 463a25f0a04SGreg Roach $tree->setPreference('ADVANCED_PLAC_FACTS', ''); 464a25f0a04SGreg Roach $tree->setPreference('ALLOW_THEME_DROPDOWN', '1'); 465a25f0a04SGreg Roach $tree->setPreference('CALENDAR_FORMAT', 'gregorian'); 466a25f0a04SGreg Roach $tree->setPreference('CHART_BOX_TAGS', ''); 467a25f0a04SGreg Roach $tree->setPreference('COMMON_NAMES_ADD', ''); 468a25f0a04SGreg Roach $tree->setPreference('COMMON_NAMES_REMOVE', ''); 469a25f0a04SGreg Roach $tree->setPreference('COMMON_NAMES_THRESHOLD', '40'); 470a25f0a04SGreg Roach $tree->setPreference('CONTACT_USER_ID', Auth::id()); 471a25f0a04SGreg Roach $tree->setPreference('DEFAULT_PEDIGREE_GENERATIONS', '4'); 472a25f0a04SGreg Roach $tree->setPreference('EXPAND_RELATIVES_EVENTS', '0'); 473a25f0a04SGreg Roach $tree->setPreference('EXPAND_SOURCES', '0'); 474a25f0a04SGreg Roach $tree->setPreference('FAM_FACTS_ADD', 'CENS,MARR,RESI,SLGS,MARR_CIVIL,MARR_RELIGIOUS,MARR_PARTNERS,RESN'); 475a25f0a04SGreg Roach $tree->setPreference('FAM_FACTS_QUICK', 'MARR,DIV,_NMR'); 476a25f0a04SGreg Roach $tree->setPreference('FAM_FACTS_UNIQUE', 'NCHI,MARL,DIV,ANUL,DIVF,ENGA,MARB,MARC,MARS'); 477a25f0a04SGreg Roach $tree->setPreference('FAM_ID_PREFIX', 'F'); 478a25f0a04SGreg Roach $tree->setPreference('FORMAT_TEXT', 'markdown'); 479a25f0a04SGreg Roach $tree->setPreference('FULL_SOURCES', '0'); 480a25f0a04SGreg Roach $tree->setPreference('GEDCOM_ID_PREFIX', 'I'); 481a25f0a04SGreg Roach $tree->setPreference('GEDCOM_MEDIA_PATH', ''); 482a25f0a04SGreg Roach $tree->setPreference('GENERATE_UIDS', '0'); 483a25f0a04SGreg Roach $tree->setPreference('HIDE_GEDCOM_ERRORS', '1'); 484a25f0a04SGreg Roach $tree->setPreference('HIDE_LIVE_PEOPLE', '1'); 485a25f0a04SGreg Roach $tree->setPreference('INDI_FACTS_ADD', 'AFN,BIRT,DEAT,BURI,CREM,ADOP,BAPM,BARM,BASM,BLES,CHRA,CONF,FCOM,ORDN,NATU,EMIG,IMMI,CENS,PROB,WILL,GRAD,RETI,DSCR,EDUC,IDNO,NATI,NCHI,NMR,OCCU,PROP,RELI,RESI,SSN,TITL,BAPL,CONL,ENDL,SLGC,_MILI,ASSO,RESN'); 486a25f0a04SGreg Roach $tree->setPreference('INDI_FACTS_QUICK', 'BIRT,BURI,BAPM,CENS,DEAT,OCCU,RESI'); 487a25f0a04SGreg Roach $tree->setPreference('INDI_FACTS_UNIQUE', ''); 488a25f0a04SGreg Roach $tree->setPreference('KEEP_ALIVE_YEARS_BIRTH', ''); 489a25f0a04SGreg Roach $tree->setPreference('KEEP_ALIVE_YEARS_DEATH', ''); 490a25f0a04SGreg Roach $tree->setPreference('LANGUAGE', WT_LOCALE); // Default to the current admin’s language 491a25f0a04SGreg Roach $tree->setPreference('MAX_ALIVE_AGE', 120); 492a25f0a04SGreg Roach $tree->setPreference('MAX_DESCENDANCY_GENERATIONS', '15'); 493a25f0a04SGreg Roach $tree->setPreference('MAX_PEDIGREE_GENERATIONS', '10'); 494a25f0a04SGreg Roach $tree->setPreference('MEDIA_DIRECTORY', 'media/'); 495a25f0a04SGreg Roach $tree->setPreference('MEDIA_ID_PREFIX', 'M'); 496a25f0a04SGreg Roach $tree->setPreference('MEDIA_UPLOAD', WT_PRIV_USER); 497a25f0a04SGreg Roach $tree->setPreference('META_DESCRIPTION', ''); 498a25f0a04SGreg Roach $tree->setPreference('META_TITLE', WT_WEBTREES); 499a25f0a04SGreg Roach $tree->setPreference('NOTE_FACTS_ADD', 'SOUR,RESN'); 500a25f0a04SGreg Roach $tree->setPreference('NOTE_FACTS_QUICK', ''); 501a25f0a04SGreg Roach $tree->setPreference('NOTE_FACTS_UNIQUE', ''); 502a25f0a04SGreg Roach $tree->setPreference('NOTE_ID_PREFIX', 'N'); 503a25f0a04SGreg Roach $tree->setPreference('NO_UPDATE_CHAN', '0'); 504a25f0a04SGreg Roach $tree->setPreference('PEDIGREE_FULL_DETAILS', '1'); 505a25f0a04SGreg Roach $tree->setPreference('PEDIGREE_LAYOUT', '1'); 506a25f0a04SGreg Roach $tree->setPreference('PEDIGREE_ROOT_ID', ''); 507a25f0a04SGreg Roach $tree->setPreference('PEDIGREE_SHOW_GENDER', '0'); 508a25f0a04SGreg Roach $tree->setPreference('PREFER_LEVEL2_SOURCES', '1'); 509a25f0a04SGreg Roach $tree->setPreference('QUICK_REQUIRED_FACTS', 'BIRT,DEAT'); 510a25f0a04SGreg Roach $tree->setPreference('QUICK_REQUIRED_FAMFACTS', 'MARR'); 511a25f0a04SGreg Roach $tree->setPreference('REPO_FACTS_ADD', 'PHON,EMAIL,FAX,WWW,RESN'); 512a25f0a04SGreg Roach $tree->setPreference('REPO_FACTS_QUICK', ''); 513a25f0a04SGreg Roach $tree->setPreference('REPO_FACTS_UNIQUE', 'NAME,ADDR'); 514a25f0a04SGreg Roach $tree->setPreference('REPO_ID_PREFIX', 'R'); 515a25f0a04SGreg Roach $tree->setPreference('REQUIRE_AUTHENTICATION', '0'); 516a25f0a04SGreg Roach $tree->setPreference('SAVE_WATERMARK_IMAGE', '0'); 517a25f0a04SGreg Roach $tree->setPreference('SAVE_WATERMARK_THUMB', '0'); 518a25f0a04SGreg Roach $tree->setPreference('SHOW_AGE_DIFF', '0'); 519a25f0a04SGreg Roach $tree->setPreference('SHOW_COUNTER', '1'); 520a25f0a04SGreg Roach $tree->setPreference('SHOW_DEAD_PEOPLE', WT_PRIV_PUBLIC); 521a25f0a04SGreg Roach $tree->setPreference('SHOW_EST_LIST_DATES', '0'); 522a25f0a04SGreg Roach $tree->setPreference('SHOW_FACT_ICONS', '1'); 523a25f0a04SGreg Roach $tree->setPreference('SHOW_GEDCOM_RECORD', '0'); 524a25f0a04SGreg Roach $tree->setPreference('SHOW_HIGHLIGHT_IMAGES', '1'); 525a25f0a04SGreg Roach $tree->setPreference('SHOW_LDS_AT_GLANCE', '0'); 526a25f0a04SGreg Roach $tree->setPreference('SHOW_LEVEL2_NOTES', '1'); 527a25f0a04SGreg Roach $tree->setPreference('SHOW_LIVING_NAMES', WT_PRIV_USER); 528a25f0a04SGreg Roach $tree->setPreference('SHOW_MEDIA_DOWNLOAD', '0'); 529a25f0a04SGreg Roach $tree->setPreference('SHOW_NO_WATERMARK', WT_PRIV_USER); 530a25f0a04SGreg Roach $tree->setPreference('SHOW_PARENTS_AGE', '1'); 531a25f0a04SGreg Roach $tree->setPreference('SHOW_PEDIGREE_PLACES', '9'); 532a25f0a04SGreg Roach $tree->setPreference('SHOW_PEDIGREE_PLACES_SUFFIX', '0'); 533a25f0a04SGreg Roach $tree->setPreference('SHOW_PRIVATE_RELATIONSHIPS', '1'); 534a25f0a04SGreg Roach $tree->setPreference('SHOW_RELATIVES_EVENTS', '_BIRT_CHIL,_BIRT_SIBL,_MARR_CHIL,_MARR_PARE,_DEAT_CHIL,_DEAT_PARE,_DEAT_GPAR,_DEAT_SIBL,_DEAT_SPOU'); 535a25f0a04SGreg Roach $tree->setPreference('SHOW_STATS', '0'); 536a25f0a04SGreg Roach $tree->setPreference('SOURCE_ID_PREFIX', 'S'); 537a25f0a04SGreg Roach $tree->setPreference('SOUR_FACTS_ADD', 'NOTE,REPO,SHARED_NOTE,RESN'); 538a25f0a04SGreg Roach $tree->setPreference('SOUR_FACTS_QUICK', 'TEXT,NOTE,REPO'); 539a25f0a04SGreg Roach $tree->setPreference('SOUR_FACTS_UNIQUE', 'AUTH,ABBR,TITL,PUBL,TEXT'); 540a25f0a04SGreg Roach $tree->setPreference('SUBLIST_TRIGGER_I', '200'); 541a25f0a04SGreg Roach $tree->setPreference('SURNAME_LIST_STYLE', 'style2'); 542a25f0a04SGreg Roach switch (WT_LOCALE) { 543a25f0a04SGreg Roach case 'es': 544a25f0a04SGreg Roach $tree->setPreference('SURNAME_TRADITION', 'spanish'); 545a25f0a04SGreg Roach break; 546a25f0a04SGreg Roach case 'is': 547a25f0a04SGreg Roach $tree->setPreference('SURNAME_TRADITION', 'icelandic'); 548a25f0a04SGreg Roach break; 549a25f0a04SGreg Roach case 'lt': 550a25f0a04SGreg Roach $tree->setPreference('SURNAME_TRADITION', 'lithuanian'); 551a25f0a04SGreg Roach break; 552a25f0a04SGreg Roach case 'pl': 553a25f0a04SGreg Roach $tree->setPreference('SURNAME_TRADITION', 'polish'); 554a25f0a04SGreg Roach break; 555a25f0a04SGreg Roach case 'pt': 556a25f0a04SGreg Roach case 'pt-BR': 557a25f0a04SGreg Roach $tree->setPreference('SURNAME_TRADITION', 'portuguese'); 558a25f0a04SGreg Roach break; 559a25f0a04SGreg Roach default: 560a25f0a04SGreg Roach $tree->setPreference('SURNAME_TRADITION', 'paternal'); 561a25f0a04SGreg Roach break; 562a25f0a04SGreg Roach } 563a25f0a04SGreg Roach $tree->setPreference('THEME_DIR', 'webtrees'); 564a25f0a04SGreg Roach $tree->setPreference('THUMBNAIL_WIDTH', '100'); 565a25f0a04SGreg Roach $tree->setPreference('USE_RIN', '0'); 566a25f0a04SGreg Roach $tree->setPreference('USE_SILHOUETTE', '1'); 567a25f0a04SGreg Roach $tree->setPreference('WATERMARK_THUMB', '0'); 568a25f0a04SGreg Roach $tree->setPreference('WEBMASTER_USER_ID', Auth::id()); 569a25f0a04SGreg Roach $tree->setPreference('WEBTREES_EMAIL', ''); 570a25f0a04SGreg Roach $tree->setPreference('WORD_WRAPPED_NOTES', '0'); 571a25f0a04SGreg Roach 572a25f0a04SGreg Roach // Default restriction settings 573a25f0a04SGreg Roach $statement = Database::prepare( 574a25f0a04SGreg Roach "INSERT INTO `##default_resn` (gedcom_id, xref, tag_type, resn) VALUES (?, NULL, ?, ?)" 575a25f0a04SGreg Roach ); 576a25f0a04SGreg Roach $statement->execute(array($tree_id, 'SSN', 'confidential')); 577a25f0a04SGreg Roach $statement->execute(array($tree_id, 'SOUR', 'privacy')); 578a25f0a04SGreg Roach $statement->execute(array($tree_id, 'REPO', 'privacy')); 579a25f0a04SGreg Roach $statement->execute(array($tree_id, 'SUBM', 'confidential')); 580a25f0a04SGreg Roach $statement->execute(array($tree_id, 'SUBN', 'confidential')); 581a25f0a04SGreg Roach 582a25f0a04SGreg Roach // Genealogy data 583a25f0a04SGreg Roach // It is simpler to create a temporary/unimported GEDCOM than to populate all the tables... 584a25f0a04SGreg Roach $john_doe = /* I18N: This should be a common/default/placeholder name of an individual. Put slashes around the surname. */ 585a25f0a04SGreg Roach I18N::translate('John /DOE/'); 586a25f0a04SGreg Roach $note = I18N::translate('Edit this individual and replace their details with your own'); 587a25f0a04SGreg Roach Database::prepare("INSERT INTO `##gedcom_chunk` (gedcom_id, chunk_data) VALUES (?, ?)")->execute(array( 588a25f0a04SGreg Roach $tree_id, 589a25f0a04SGreg Roach "0 HEAD\n1 CHAR UTF-8\n0 @I1@ INDI\n1 NAME {$john_doe}\n1 SEX M\n1 BIRT\n2 DATE 01 JAN 1850\n2 NOTE {$note}\n0 TRLR\n" 590a25f0a04SGreg Roach )); 591a25f0a04SGreg Roach 592a25f0a04SGreg Roach // Set the initial blocks 593a25f0a04SGreg Roach Database::prepare( 594a25f0a04SGreg Roach "INSERT INTO `##block` (gedcom_id, location, block_order, module_name)" . 595a25f0a04SGreg Roach " SELECT ?, location, block_order, module_name" . 596a25f0a04SGreg Roach " FROM `##block`" . 597a25f0a04SGreg Roach " WHERE gedcom_id = -1" 598a25f0a04SGreg Roach )->execute(array($tree_id)); 599a25f0a04SGreg Roach 600a25f0a04SGreg Roach // Update our cache 601518bbdc1SGreg Roach self::$trees[$tree->tree_id] = $tree; 602a25f0a04SGreg Roach 603a25f0a04SGreg Roach return $tree; 604a25f0a04SGreg Roach } 605a25f0a04SGreg Roach 606a25f0a04SGreg Roach /** 607a25f0a04SGreg Roach * Delete all the genealogy data from a tree - in preparation for importing 608a25f0a04SGreg Roach * new data. Optionally retain the media data, for when the user has been 609a25f0a04SGreg Roach * editing their data offline using an application which deletes (or does not 610a25f0a04SGreg Roach * support) media data. 611a25f0a04SGreg Roach * 612a25f0a04SGreg Roach * @param bool $keep_media 613a25f0a04SGreg Roach */ 614a25f0a04SGreg Roach public function deleteGenealogyData($keep_media) { 615518bbdc1SGreg Roach Database::prepare("DELETE FROM `##gedcom_chunk` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 616518bbdc1SGreg Roach Database::prepare("DELETE FROM `##individuals` WHERE i_file = ?")->execute(array($this->tree_id)); 617518bbdc1SGreg Roach Database::prepare("DELETE FROM `##families` WHERE f_file = ?")->execute(array($this->tree_id)); 618518bbdc1SGreg Roach Database::prepare("DELETE FROM `##sources` WHERE s_file = ?")->execute(array($this->tree_id)); 619518bbdc1SGreg Roach Database::prepare("DELETE FROM `##other` WHERE o_file = ?")->execute(array($this->tree_id)); 620518bbdc1SGreg Roach Database::prepare("DELETE FROM `##places` WHERE p_file = ?")->execute(array($this->tree_id)); 621518bbdc1SGreg Roach Database::prepare("DELETE FROM `##placelinks` WHERE pl_file = ?")->execute(array($this->tree_id)); 622518bbdc1SGreg Roach Database::prepare("DELETE FROM `##name` WHERE n_file = ?")->execute(array($this->tree_id)); 623518bbdc1SGreg Roach Database::prepare("DELETE FROM `##dates` WHERE d_file = ?")->execute(array($this->tree_id)); 624518bbdc1SGreg Roach Database::prepare("DELETE FROM `##change` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 625a25f0a04SGreg Roach 626a25f0a04SGreg Roach if ($keep_media) { 627518bbdc1SGreg Roach Database::prepare("DELETE FROM `##link` WHERE l_file =? AND l_type<>'OBJE'")->execute(array($this->tree_id)); 628a25f0a04SGreg Roach } else { 629518bbdc1SGreg Roach Database::prepare("DELETE FROM `##link` WHERE l_file =?")->execute(array($this->tree_id)); 630518bbdc1SGreg Roach Database::prepare("DELETE FROM `##media` WHERE m_file =?")->execute(array($this->tree_id)); 631a25f0a04SGreg Roach } 632a25f0a04SGreg Roach } 633a25f0a04SGreg Roach 634a25f0a04SGreg Roach /** 635a25f0a04SGreg Roach * Delete everything relating to a tree 636a25f0a04SGreg Roach */ 637a25f0a04SGreg Roach public function delete() { 638a25f0a04SGreg Roach // If this is the default tree, then unset it 639518bbdc1SGreg Roach if (Site::getPreference('DEFAULT_GEDCOM') === self::getNameFromId($this->tree_id)) { 640a25f0a04SGreg Roach Site::setPreference('DEFAULT_GEDCOM', ''); 641a25f0a04SGreg Roach } 642a25f0a04SGreg Roach 643a25f0a04SGreg Roach $this->deleteGenealogyData(false); 644a25f0a04SGreg Roach 645518bbdc1SGreg Roach Database::prepare("DELETE `##block_setting` FROM `##block_setting` JOIN `##block` USING (block_id) WHERE gedcom_id=?")->execute(array($this->tree_id)); 646518bbdc1SGreg Roach Database::prepare("DELETE FROM `##block` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 647518bbdc1SGreg Roach Database::prepare("DELETE FROM `##user_gedcom_setting` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 648518bbdc1SGreg Roach Database::prepare("DELETE FROM `##gedcom_setting` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 649518bbdc1SGreg Roach Database::prepare("DELETE FROM `##module_privacy` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 650518bbdc1SGreg Roach Database::prepare("DELETE FROM `##next_id` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 651518bbdc1SGreg Roach Database::prepare("DELETE FROM `##hit_counter` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 652518bbdc1SGreg Roach Database::prepare("DELETE FROM `##default_resn` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 653518bbdc1SGreg Roach Database::prepare("DELETE FROM `##gedcom_chunk` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 654518bbdc1SGreg Roach Database::prepare("DELETE FROM `##log` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 655518bbdc1SGreg Roach Database::prepare("DELETE FROM `##gedcom` WHERE gedcom_id = ?")->execute(array($this->tree_id)); 656a25f0a04SGreg Roach 657a25f0a04SGreg Roach // After updating the database, we need to fetch a new (sorted) copy 658a25f0a04SGreg Roach self::$trees = null; 659a25f0a04SGreg Roach } 660a25f0a04SGreg Roach 661a25f0a04SGreg Roach /** 662a25f0a04SGreg Roach * Export the tree to a GEDCOM file 663a25f0a04SGreg Roach * 664a25f0a04SGreg Roach * @param $gedcom_file 665a25f0a04SGreg Roach * 666a25f0a04SGreg Roach * @return bool 667a25f0a04SGreg Roach */ 668a25f0a04SGreg Roach public function exportGedcom($gedcom_file) { 669a25f0a04SGreg Roach // To avoid partial trees on timeout/diskspace/etc, write to a temporary file first 670a25f0a04SGreg Roach $tmp_file = $gedcom_file . '.tmp'; 671a25f0a04SGreg Roach 672a25f0a04SGreg Roach $file_pointer = @fopen($tmp_file, 'w'); 673a25f0a04SGreg Roach if ($file_pointer === false) { 674a25f0a04SGreg Roach return false; 675a25f0a04SGreg Roach } 676a25f0a04SGreg Roach 677a25f0a04SGreg Roach $buffer = reformat_record_export(gedcom_header($this->name)); 678a25f0a04SGreg Roach 679a25f0a04SGreg Roach $stmt = Database::prepare( 680a25f0a04SGreg Roach "SELECT i_gedcom AS gedcom FROM `##individuals` WHERE i_file = ?" . 681a25f0a04SGreg Roach " UNION ALL " . 682a25f0a04SGreg Roach "SELECT f_gedcom AS gedcom FROM `##families` WHERE f_file = ?" . 683a25f0a04SGreg Roach " UNION ALL " . 684a25f0a04SGreg Roach "SELECT s_gedcom AS gedcom FROM `##sources` WHERE s_file = ?" . 685a25f0a04SGreg Roach " UNION ALL " . 686a25f0a04SGreg Roach "SELECT o_gedcom AS gedcom FROM `##other` WHERE o_file = ? AND o_type NOT IN ('HEAD', 'TRLR')" . 687a25f0a04SGreg Roach " UNION ALL " . 688a25f0a04SGreg Roach "SELECT m_gedcom AS gedcom FROM `##media` WHERE m_file = ?" 689518bbdc1SGreg Roach )->execute(array($this->tree_id, $this->tree_id, $this->tree_id, $this->tree_id, $this->tree_id)); 690a25f0a04SGreg Roach 691a25f0a04SGreg Roach while ($row = $stmt->fetch()) { 692a25f0a04SGreg Roach $buffer .= reformat_record_export($row->gedcom); 693a25f0a04SGreg Roach if (strlen($buffer) > 65535) { 694a25f0a04SGreg Roach fwrite($file_pointer, $buffer); 695a25f0a04SGreg Roach $buffer = ''; 696a25f0a04SGreg Roach } 697a25f0a04SGreg Roach } 698a25f0a04SGreg Roach 699a25f0a04SGreg Roach fwrite($file_pointer, $buffer . '0 TRLR' . WT_EOL); 700a25f0a04SGreg Roach fclose($file_pointer); 701a25f0a04SGreg Roach 702a25f0a04SGreg Roach return @rename($tmp_file, $gedcom_file); 703a25f0a04SGreg Roach } 704a25f0a04SGreg Roach 705a25f0a04SGreg Roach /** 706a25f0a04SGreg Roach * Import data from a gedcom file into this tree. 707a25f0a04SGreg Roach * 708a25f0a04SGreg Roach * @param string $path The full path to the (possibly temporary) file. 709a25f0a04SGreg Roach * @param string $filename The preferred filename, for export/download. 710a25f0a04SGreg Roach * @param boolean $keep_media Whether to retain any existing media records 711a25f0a04SGreg Roach * 712a25f0a04SGreg Roach * @throws \Exception 713a25f0a04SGreg Roach */ 714a25f0a04SGreg Roach public function importGedcomFile($path, $filename, $keep_media) { 715a25f0a04SGreg Roach // Read the file in blocks of roughly 64K. Ensure that each block 716a25f0a04SGreg Roach // contains complete gedcom records. This will ensure we don’t split 717a25f0a04SGreg Roach // multi-byte characters, as well as simplifying the code to import 718a25f0a04SGreg Roach // each block. 719a25f0a04SGreg Roach 720a25f0a04SGreg Roach $file_data = ''; 721a25f0a04SGreg Roach $fp = fopen($path, 'rb'); 722a25f0a04SGreg Roach 723a25f0a04SGreg Roach // Don’t allow the user to cancel the request. We do not want to be left with an incomplete transaction. 724a25f0a04SGreg Roach ignore_user_abort(true); 725a25f0a04SGreg Roach 726a25f0a04SGreg Roach Database::beginTransaction(); 727a25f0a04SGreg Roach $this->deleteGenealogyData($keep_media); 728a25f0a04SGreg Roach $this->setPreference('gedcom_filename', $filename); 729a25f0a04SGreg Roach $this->setPreference('imported', '0'); 730a25f0a04SGreg Roach 731a25f0a04SGreg Roach while (!feof($fp)) { 732a25f0a04SGreg Roach $file_data .= fread($fp, 65536); 733a25f0a04SGreg Roach // There is no strrpos() function that searches for substrings :-( 734a25f0a04SGreg Roach for ($pos = strlen($file_data) - 1; $pos > 0; --$pos) { 735a25f0a04SGreg Roach if ($file_data[$pos] === '0' && ($file_data[$pos - 1] === "\n" || $file_data[$pos - 1] === "\r")) { 736a25f0a04SGreg Roach // We’ve found the last record boundary in this chunk of data 737a25f0a04SGreg Roach break; 738a25f0a04SGreg Roach } 739a25f0a04SGreg Roach } 740a25f0a04SGreg Roach if ($pos) { 741a25f0a04SGreg Roach Database::prepare( 742a25f0a04SGreg Roach "INSERT INTO `##gedcom_chunk` (gedcom_id, chunk_data) VALUES (?, ?)" 743518bbdc1SGreg Roach )->execute(array($this->tree_id, substr($file_data, 0, $pos))); 744a25f0a04SGreg Roach $file_data = substr($file_data, $pos); 745a25f0a04SGreg Roach } 746a25f0a04SGreg Roach } 747a25f0a04SGreg Roach Database::prepare( 748a25f0a04SGreg Roach "INSERT INTO `##gedcom_chunk` (gedcom_id, chunk_data) VALUES (?, ?)" 749518bbdc1SGreg Roach )->execute(array($this->tree_id, $file_data)); 750a25f0a04SGreg Roach 751a25f0a04SGreg Roach Database::commit(); 752a25f0a04SGreg Roach fclose($fp); 753a25f0a04SGreg Roach } 754304f20d5SGreg Roach 755304f20d5SGreg Roach /** 756*b90d8accSGreg Roach * Generate a new XREF, unique across all family trees 757*b90d8accSGreg Roach * 758*b90d8accSGreg Roach * @param string $type 759*b90d8accSGreg Roach * 760*b90d8accSGreg Roach * @return string 761*b90d8accSGreg Roach */ 762*b90d8accSGreg Roach function getNewXref($type = 'INDI') { 763*b90d8accSGreg Roach /** @var string[] Which tree preference is used for which record type */ 764*b90d8accSGreg Roach static $type_to_preference = array( 765*b90d8accSGreg Roach 'INDI' => 'GEDCOM_ID_PREFIX', 766*b90d8accSGreg Roach 'FAM' => 'FAM_ID_PREFIX', 767*b90d8accSGreg Roach 'OBJE' => 'MEDIA_ID_PREFIX', 768*b90d8accSGreg Roach 'NOTE' => 'NOTE_ID_PREFIX', 769*b90d8accSGreg Roach 'SOUR' => 'SOURCE_ID_PREFIX', 770*b90d8accSGreg Roach 'REPO' => 'REPO_ID_PREFIX', 771*b90d8accSGreg Roach ); 772*b90d8accSGreg Roach 773*b90d8accSGreg Roach if (array_key_exists($type, $type_to_preference)) { 774*b90d8accSGreg Roach $prefix = $this->getPreference($type_to_preference[$type]); 775*b90d8accSGreg Roach } else { 776*b90d8accSGreg Roach // Use the first non-underscore character 777*b90d8accSGreg Roach $prefix = substr(trim($type, '_'), 0, 1); 778*b90d8accSGreg Roach } 779*b90d8accSGreg Roach 780*b90d8accSGreg Roach do { 781*b90d8accSGreg Roach // Use LAST_INSERT_ID(expr) to provide a transaction-safe sequence. See 782*b90d8accSGreg Roach // http://dev.mysql.com/doc/refman/5.6/en/information-functions.html#function_last-insert-id 783*b90d8accSGreg Roach $statement = Database::prepare( 784*b90d8accSGreg Roach "UPDATE `##next_id` SET next_id = LAST_INSERT_ID(next_id + 1) WHERE record_type = :record_type AND gedcom_id = :tree_id" 785*b90d8accSGreg Roach ); 786*b90d8accSGreg Roach $statement->execute(array( 787*b90d8accSGreg Roach 'record_type' => $type, 788*b90d8accSGreg Roach 'tree_id' => $this->getTreeId(), 789*b90d8accSGreg Roach )); 790*b90d8accSGreg Roach 791*b90d8accSGreg Roach if ($statement->rowCount() === 0) { 792*b90d8accSGreg Roach // First time we've used this record type. 793*b90d8accSGreg Roach Database::prepare( 794*b90d8accSGreg Roach "INSERT INTO `##next_id` (gedcom_id, record_type, next_id) VALUES(:tree_id, :record_type, 1)" 795*b90d8accSGreg Roach )->execute(array( 796*b90d8accSGreg Roach 'record_type' => $type, 797*b90d8accSGreg Roach 'tree_id' => $this->getTreeId(), 798*b90d8accSGreg Roach )); 799*b90d8accSGreg Roach $num = 1; 800*b90d8accSGreg Roach } else { 801*b90d8accSGreg Roach $num = Database::prepare("SELECT LAST_INSERT_ID()")->fetchOne(); 802*b90d8accSGreg Roach } 803*b90d8accSGreg Roach 804*b90d8accSGreg Roach // Records may already exist with this sequence number. 805*b90d8accSGreg Roach $already_used = Database::prepare( 806*b90d8accSGreg Roach "SELECT i_id FROM `##individuals` WHERE i_id = :i_id" . 807*b90d8accSGreg Roach " UNION ALL " . 808*b90d8accSGreg Roach "SELECT f_id FROM `##families` WHERE f_id = :f_id" . 809*b90d8accSGreg Roach " UNION ALL " . 810*b90d8accSGreg Roach "SELECT s_id FROM `##sources` WHERE s_id = :s_id" . 811*b90d8accSGreg Roach " UNION ALL " . 812*b90d8accSGreg Roach "SELECT m_id FROM `##media` WHERE m_id = :m_id" . 813*b90d8accSGreg Roach " UNION ALL " . 814*b90d8accSGreg Roach "SELECT o_id FROM `##other` WHERE o_id = :o_id" . 815*b90d8accSGreg Roach " UNION ALL " . 816*b90d8accSGreg Roach "SELECT xref FROM `##change` WHERE xref = :xref" 817*b90d8accSGreg Roach )->execute(array( 818*b90d8accSGreg Roach 'i_id' => $prefix . $num, 819*b90d8accSGreg Roach 'f_id' => $prefix . $num, 820*b90d8accSGreg Roach 's_id' => $prefix . $num, 821*b90d8accSGreg Roach 'm_id' => $prefix . $num, 822*b90d8accSGreg Roach 'o_id' => $prefix . $num, 823*b90d8accSGreg Roach 'xref' => $prefix . $num, 824*b90d8accSGreg Roach ))->fetchOne(); 825*b90d8accSGreg Roach } while ($already_used); 826*b90d8accSGreg Roach 827*b90d8accSGreg Roach return $prefix . $num; 828*b90d8accSGreg Roach } 829*b90d8accSGreg Roach 830*b90d8accSGreg Roach /** 831304f20d5SGreg Roach * Create a new record from GEDCOM data. 832304f20d5SGreg Roach * 833304f20d5SGreg Roach * @param string $gedcom 834304f20d5SGreg Roach * 835304f20d5SGreg Roach * @return GedcomRecord 836304f20d5SGreg Roach * @throws \Exception 837304f20d5SGreg Roach */ 838304f20d5SGreg Roach public function createRecord($gedcom) { 839304f20d5SGreg Roach if (preg_match('/^0 @(' . WT_REGEX_XREF . ')@ (' . WT_REGEX_TAG . ')/', $gedcom, $match)) { 840304f20d5SGreg Roach $xref = $match[1]; 841304f20d5SGreg Roach $type = $match[2]; 842304f20d5SGreg Roach } else { 843304f20d5SGreg Roach throw new \Exception('Invalid argument to GedcomRecord::createRecord(' . $gedcom . ')'); 844304f20d5SGreg Roach } 845304f20d5SGreg Roach if (strpos("\r", $gedcom) !== false) { 846304f20d5SGreg Roach // MSDOS line endings will break things in horrible ways 847304f20d5SGreg Roach throw new \Exception('Evil line endings found in GedcomRecord::createRecord(' . $gedcom . ')'); 848304f20d5SGreg Roach } 849304f20d5SGreg Roach 850304f20d5SGreg Roach // webtrees creates XREFs containing digits. Anything else (e.g. “new”) is just a placeholder. 851304f20d5SGreg Roach if (!preg_match('/\d/', $xref)) { 852*b90d8accSGreg Roach $xref = $this->getNewXref($type); 853304f20d5SGreg Roach $gedcom = preg_replace('/^0 @(' . WT_REGEX_XREF . ')@/', '0 @' . $xref . '@', $gedcom); 854304f20d5SGreg Roach } 855304f20d5SGreg Roach 856304f20d5SGreg Roach // Create a change record, if not already present 857304f20d5SGreg Roach if (!preg_match('/\n1 CHAN/', $gedcom)) { 858304f20d5SGreg Roach $gedcom .= "\n1 CHAN\n2 DATE " . date('d M Y') . "\n3 TIME " . date('H:i:s') . "\n2 _WT_USER " . Auth::user()->getUserName(); 859304f20d5SGreg Roach } 860304f20d5SGreg Roach 861304f20d5SGreg Roach // Create a pending change 862304f20d5SGreg Roach Database::prepare( 863304f20d5SGreg Roach "INSERT INTO `##change` (gedcom_id, xref, old_gedcom, new_gedcom, user_id) VALUES (?, ?, '', ?, ?)" 864304f20d5SGreg Roach )->execute(array( 865304f20d5SGreg Roach $this->getTreeId(), 866304f20d5SGreg Roach $xref, 867304f20d5SGreg Roach $gedcom, 868304f20d5SGreg Roach Auth::id() 869304f20d5SGreg Roach )); 870304f20d5SGreg Roach 871304f20d5SGreg Roach Log::addEditLog('Create: ' . $type . ' ' . $xref); 872304f20d5SGreg Roach 873304f20d5SGreg Roach // Accept this pending change 874304f20d5SGreg Roach if (Auth::user()->getPreference('auto_accept')) { 875304f20d5SGreg Roach accept_all_changes($xref, $this->getTreeId()); 876304f20d5SGreg Roach // Return the newly created record 877304f20d5SGreg Roach return new GedcomRecord($xref, $gedcom, null, $this->getTreeId()); 878304f20d5SGreg Roach } else { 879304f20d5SGreg Roach // Return the newly created record 880304f20d5SGreg Roach return new GedcomRecord($xref, null, $gedcom, $this->getTreeId()); 881304f20d5SGreg Roach } 882304f20d5SGreg Roach 883304f20d5SGreg Roach } 884a25f0a04SGreg Roach} 885