xref: /webtrees/app/Tree.php (revision 518bbdc107415f0c334ca15f70a0105a5786df0e)
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 */
26*518bbdc1SGreg Roach	private $tree_id;
27*518bbdc1SGreg Roach
28a25f0a04SGreg Roach	/** @var string The tree's name */
29a25f0a04SGreg Roach	private $name;
30*518bbdc1SGreg Roach
31a25f0a04SGreg Roach	/** @var string The tree's title */
32a25f0a04SGreg Roach	private $title;
33a25f0a04SGreg Roach
34*518bbdc1SGreg Roach	/** @var integer[] Default access rules for facts in this tree */
35*518bbdc1SGreg Roach	private $fact_privacy;
36*518bbdc1SGreg Roach
37*518bbdc1SGreg Roach	/** @var integer[] Default access rules for individuals in this tree */
38*518bbdc1SGreg Roach	private $individual_privacy;
39*518bbdc1SGreg Roach
40*518bbdc1SGreg Roach	/** @var integer[][] Default access rules for individual facts in this tree */
41*518bbdc1SGreg Roach	private $individual_fact_privacy;
42*518bbdc1SGreg 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	 *
56*518bbdc1SGreg Roach	 * @param integer $tree_id
57a25f0a04SGreg Roach	 * @param string  $tree_name
58a25f0a04SGreg Roach	 * @param string  $tree_title
59a25f0a04SGreg Roach	 */
60*518bbdc1SGreg Roach	private function __construct($tree_id, $tree_name, $tree_title) {
61*518bbdc1SGreg Roach		$this->tree_id                 = $tree_id;
62a25f0a04SGreg Roach		$this->name                    = $tree_name;
63a25f0a04SGreg Roach		$this->title                   = $tree_title;
64*518bbdc1SGreg Roach		$this->fact_privacy            = array();
65*518bbdc1SGreg Roach		$this->individual_privacy      = array();
66*518bbdc1SGreg Roach		$this->individual_fact_privacy = array();
67*518bbdc1SGreg Roach
68*518bbdc1SGreg Roach		// Load the privacy settings for this tree
69*518bbdc1SGreg Roach		$rows = Database::prepare(
70*518bbdc1SGreg 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" .
71*518bbdc1SGreg Roach			" FROM `##default_resn` WHERE gedcom_id = :tree_id"
72*518bbdc1SGreg Roach		)->execute(array(
73*518bbdc1SGreg Roach			'priv_public' => WT_PRIV_PUBLIC,
74*518bbdc1SGreg Roach			'priv_user'   => WT_PRIV_USER,
75*518bbdc1SGreg Roach			'priv_none'   => WT_PRIV_NONE,
76*518bbdc1SGreg Roach			'priv_hide'   => WT_PRIV_HIDE,
77*518bbdc1SGreg Roach			'tree_id'     => $this->tree_id
78*518bbdc1SGreg Roach		))->fetchAll();
79*518bbdc1SGreg Roach
80*518bbdc1SGreg Roach		foreach ($rows as $row) {
81*518bbdc1SGreg Roach			if ($row->xref !== null) {
82*518bbdc1SGreg Roach				if ($row->tag_type !== null) {
83*518bbdc1SGreg Roach					$this->individual_fact_privacy[$row->xref][$row->tag_type] = (int) $row->resn;
84*518bbdc1SGreg Roach				} else {
85*518bbdc1SGreg Roach					$this->individual_privacy[$row->xref] = (int) $row->resn;
86*518bbdc1SGreg Roach				}
87*518bbdc1SGreg Roach			} else {
88*518bbdc1SGreg Roach				$this->fact_privacy[$row->tag_type] = (int) $row->resn;
89*518bbdc1SGreg Roach			}
90*518bbdc1SGreg Roach		}
91*518bbdc1SGreg Roach
92*518bbdc1SGreg Roach
93a25f0a04SGreg Roach	}
94a25f0a04SGreg Roach
95a25f0a04SGreg Roach	/**
96a25f0a04SGreg Roach	 * The ID of this tree
97a25f0a04SGreg Roach	 *
98a25f0a04SGreg Roach	 * @return integer
99a25f0a04SGreg Roach	 */
100*518bbdc1SGreg Roach	public function getTreeId() {
101*518bbdc1SGreg 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	/**
150*518bbdc1SGreg Roach	 * The fact-level privacy for this tree.
151*518bbdc1SGreg Roach	 *
152*518bbdc1SGreg Roach	 * @return string
153*518bbdc1SGreg Roach	 */
154*518bbdc1SGreg Roach	public function getFactPrivacy() {
155*518bbdc1SGreg Roach		return $this->fact_privacy;
156*518bbdc1SGreg Roach	}
157*518bbdc1SGreg Roach
158*518bbdc1SGreg Roach	/**
159*518bbdc1SGreg Roach	 * The individual-level privacy for this tree.
160*518bbdc1SGreg Roach	 *
161*518bbdc1SGreg Roach	 * @return string
162*518bbdc1SGreg Roach	 */
163*518bbdc1SGreg Roach	public function getIndividualPrivacy() {
164*518bbdc1SGreg Roach		return $this->individual_privacy;
165*518bbdc1SGreg Roach	}
166*518bbdc1SGreg Roach
167*518bbdc1SGreg Roach	/**
168*518bbdc1SGreg Roach	 * The individual-fact-level privacy for this tree.
169*518bbdc1SGreg Roach	 *
170*518bbdc1SGreg Roach	 * @return string
171*518bbdc1SGreg Roach	 */
172*518bbdc1SGreg Roach	public function getIndividualFactPrivacy() {
173*518bbdc1SGreg Roach		return $this->individual_fact_privacy;
174*518bbdc1SGreg Roach	}
175*518bbdc1SGreg Roach
176*518bbdc1SGreg 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 = ?"
188*518bbdc1SGreg 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(
213*518bbdc1SGreg 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(
221*518bbdc1SGreg 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 = ?"
250*518bbdc1SGreg 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(
276*518bbdc1SGreg 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(),
285*518bbdc1SGreg 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	/**
345a25f0a04SGreg Roach	 * Get the tree with a specific ID.
346a25f0a04SGreg Roach	 *
347a25f0a04SGreg Roach	 * @param integer $tree_id
348a25f0a04SGreg Roach	 *
349a25f0a04SGreg Roach	 * @return Tree
350a25f0a04SGreg Roach	 */
351a25f0a04SGreg Roach	public static function get($tree_id) {
352a25f0a04SGreg Roach		$trees = self::getAll();
353a25f0a04SGreg Roach
354a25f0a04SGreg Roach		return $trees[$tree_id];
355a25f0a04SGreg Roach	}
356a25f0a04SGreg Roach
357a25f0a04SGreg Roach	/**
358a25f0a04SGreg Roach	 * Create arguments to select_edit_control()
359a25f0a04SGreg Roach	 * Note - these will be escaped later
360a25f0a04SGreg Roach	 *
361a25f0a04SGreg Roach	 * @return string[]
362a25f0a04SGreg Roach	 */
363a25f0a04SGreg Roach	public static function getIdList() {
364a25f0a04SGreg Roach		$list = array();
365a25f0a04SGreg Roach		foreach (self::getAll() as $tree) {
366*518bbdc1SGreg Roach			$list[$tree->tree_id] = $tree->title;
367a25f0a04SGreg Roach		}
368a25f0a04SGreg Roach
369a25f0a04SGreg Roach		return $list;
370a25f0a04SGreg Roach	}
371a25f0a04SGreg Roach
372a25f0a04SGreg Roach	/**
373a25f0a04SGreg Roach	 * Create arguments to select_edit_control()
374a25f0a04SGreg Roach	 * Note - these will be escaped later
375a25f0a04SGreg Roach	 *
376a25f0a04SGreg Roach	 * @return string[]
377a25f0a04SGreg Roach	 */
378a25f0a04SGreg Roach	public static function getNameList() {
379a25f0a04SGreg Roach		$list = array();
380a25f0a04SGreg Roach		foreach (self::getAll() as $tree) {
381a25f0a04SGreg Roach			$list[$tree->name] = $tree->title;
382a25f0a04SGreg Roach		}
383a25f0a04SGreg Roach
384a25f0a04SGreg Roach		return $list;
385a25f0a04SGreg Roach	}
386a25f0a04SGreg Roach
387a25f0a04SGreg Roach	/**
388a25f0a04SGreg Roach	 * Find the ID number for a tree name
389a25f0a04SGreg Roach	 *
390a25f0a04SGreg Roach	 * @param integer $tree_name
391a25f0a04SGreg Roach	 *
392a25f0a04SGreg Roach	 * @return integer|null
393a25f0a04SGreg Roach	 */
394a25f0a04SGreg Roach	public static function getIdFromName($tree_name) {
395a25f0a04SGreg Roach		foreach (self::getAll() as $tree_id => $tree) {
396a25f0a04SGreg Roach			if ($tree->name == $tree_name) {
397a25f0a04SGreg Roach				return $tree_id;
398a25f0a04SGreg Roach			}
399a25f0a04SGreg Roach		}
400a25f0a04SGreg Roach
401a25f0a04SGreg Roach		return null;
402a25f0a04SGreg Roach	}
403a25f0a04SGreg Roach
404a25f0a04SGreg Roach	/**
405a25f0a04SGreg Roach	 * Find the tree name from a numeric ID.
406a25f0a04SGreg Roach	 * @param integer $tree_id
407a25f0a04SGreg Roach	 *
408a25f0a04SGreg Roach	 * @return string
409a25f0a04SGreg Roach	 */
410a25f0a04SGreg Roach	public static function getNameFromId($tree_id) {
411a25f0a04SGreg Roach		return self::get($tree_id)->name;
412a25f0a04SGreg Roach	}
413a25f0a04SGreg Roach
414a25f0a04SGreg Roach	/**
415a25f0a04SGreg Roach	 * Create a new tree
416a25f0a04SGreg Roach	 *
417a25f0a04SGreg Roach	 * @param string $tree_name
418a25f0a04SGreg Roach	 * @param string $tree_title
419a25f0a04SGreg Roach	 *
420a25f0a04SGreg Roach	 * @return Tree
421a25f0a04SGreg Roach	 */
422a25f0a04SGreg Roach	public static function create($tree_name, $tree_title) {
423a25f0a04SGreg Roach		try {
424a25f0a04SGreg Roach			// Create a new tree
425a25f0a04SGreg Roach			Database::prepare(
426a25f0a04SGreg Roach				"INSERT INTO `##gedcom` (gedcom_name) VALUES (?)"
427a25f0a04SGreg Roach			)->execute(array($tree_name));
428a25f0a04SGreg Roach			$tree_id = Database::prepare("SELECT LAST_INSERT_ID()")->fetchOne();
429a25f0a04SGreg Roach		} catch (PDOException $ex) {
430a25f0a04SGreg Roach			// A tree with that name already exists?
431a25f0a04SGreg Roach			return self::get(self::getIdFromName($tree_name));
432a25f0a04SGreg Roach		}
433a25f0a04SGreg Roach
434a25f0a04SGreg Roach		// Update the list of trees - to include this new one
435a25f0a04SGreg Roach		self::$trees = null;
436a25f0a04SGreg Roach		$tree        = self::get($tree_id);
437a25f0a04SGreg Roach
438a25f0a04SGreg Roach		$tree->setPreference('imported', '0');
439a25f0a04SGreg Roach		$tree->setPreference('title', $tree_title);
440a25f0a04SGreg Roach
441a25f0a04SGreg Roach		// Module privacy
442a25f0a04SGreg Roach		Module::setDefaultAccess($tree_id);
443a25f0a04SGreg Roach
444a25f0a04SGreg Roach		// Gedcom and privacy settings
445a25f0a04SGreg Roach		$tree->setPreference('ADVANCED_NAME_FACTS', 'NICK,_AKA');
446a25f0a04SGreg Roach		$tree->setPreference('ADVANCED_PLAC_FACTS', '');
447a25f0a04SGreg Roach		$tree->setPreference('ALLOW_THEME_DROPDOWN', '1');
448a25f0a04SGreg Roach		$tree->setPreference('CALENDAR_FORMAT', 'gregorian');
449a25f0a04SGreg Roach		$tree->setPreference('CHART_BOX_TAGS', '');
450a25f0a04SGreg Roach		$tree->setPreference('COMMON_NAMES_ADD', '');
451a25f0a04SGreg Roach		$tree->setPreference('COMMON_NAMES_REMOVE', '');
452a25f0a04SGreg Roach		$tree->setPreference('COMMON_NAMES_THRESHOLD', '40');
453a25f0a04SGreg Roach		$tree->setPreference('CONTACT_USER_ID', Auth::id());
454a25f0a04SGreg Roach		$tree->setPreference('DEFAULT_PEDIGREE_GENERATIONS', '4');
455a25f0a04SGreg Roach		$tree->setPreference('EXPAND_RELATIVES_EVENTS', '0');
456a25f0a04SGreg Roach		$tree->setPreference('EXPAND_SOURCES', '0');
457a25f0a04SGreg Roach		$tree->setPreference('FAM_FACTS_ADD', 'CENS,MARR,RESI,SLGS,MARR_CIVIL,MARR_RELIGIOUS,MARR_PARTNERS,RESN');
458a25f0a04SGreg Roach		$tree->setPreference('FAM_FACTS_QUICK', 'MARR,DIV,_NMR');
459a25f0a04SGreg Roach		$tree->setPreference('FAM_FACTS_UNIQUE', 'NCHI,MARL,DIV,ANUL,DIVF,ENGA,MARB,MARC,MARS');
460a25f0a04SGreg Roach		$tree->setPreference('FAM_ID_PREFIX', 'F');
461a25f0a04SGreg Roach		$tree->setPreference('FORMAT_TEXT', 'markdown');
462a25f0a04SGreg Roach		$tree->setPreference('FULL_SOURCES', '0');
463a25f0a04SGreg Roach		$tree->setPreference('GEDCOM_ID_PREFIX', 'I');
464a25f0a04SGreg Roach		$tree->setPreference('GEDCOM_MEDIA_PATH', '');
465a25f0a04SGreg Roach		$tree->setPreference('GENERATE_UIDS', '0');
466a25f0a04SGreg Roach		$tree->setPreference('HIDE_GEDCOM_ERRORS', '1');
467a25f0a04SGreg Roach		$tree->setPreference('HIDE_LIVE_PEOPLE', '1');
468a25f0a04SGreg 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');
469a25f0a04SGreg Roach		$tree->setPreference('INDI_FACTS_QUICK', 'BIRT,BURI,BAPM,CENS,DEAT,OCCU,RESI');
470a25f0a04SGreg Roach		$tree->setPreference('INDI_FACTS_UNIQUE', '');
471a25f0a04SGreg Roach		$tree->setPreference('KEEP_ALIVE_YEARS_BIRTH', '');
472a25f0a04SGreg Roach		$tree->setPreference('KEEP_ALIVE_YEARS_DEATH', '');
473a25f0a04SGreg Roach		$tree->setPreference('LANGUAGE', WT_LOCALE); // Default to the current admin’s language
474a25f0a04SGreg Roach		$tree->setPreference('MAX_ALIVE_AGE', 120);
475a25f0a04SGreg Roach		$tree->setPreference('MAX_DESCENDANCY_GENERATIONS', '15');
476a25f0a04SGreg Roach		$tree->setPreference('MAX_PEDIGREE_GENERATIONS', '10');
477a25f0a04SGreg Roach		$tree->setPreference('MEDIA_DIRECTORY', 'media/');
478a25f0a04SGreg Roach		$tree->setPreference('MEDIA_ID_PREFIX', 'M');
479a25f0a04SGreg Roach		$tree->setPreference('MEDIA_UPLOAD', WT_PRIV_USER);
480a25f0a04SGreg Roach		$tree->setPreference('META_DESCRIPTION', '');
481a25f0a04SGreg Roach		$tree->setPreference('META_TITLE', WT_WEBTREES);
482a25f0a04SGreg Roach		$tree->setPreference('NOTE_FACTS_ADD', 'SOUR,RESN');
483a25f0a04SGreg Roach		$tree->setPreference('NOTE_FACTS_QUICK', '');
484a25f0a04SGreg Roach		$tree->setPreference('NOTE_FACTS_UNIQUE', '');
485a25f0a04SGreg Roach		$tree->setPreference('NOTE_ID_PREFIX', 'N');
486a25f0a04SGreg Roach		$tree->setPreference('NO_UPDATE_CHAN', '0');
487a25f0a04SGreg Roach		$tree->setPreference('PEDIGREE_FULL_DETAILS', '1');
488a25f0a04SGreg Roach		$tree->setPreference('PEDIGREE_LAYOUT', '1');
489a25f0a04SGreg Roach		$tree->setPreference('PEDIGREE_ROOT_ID', '');
490a25f0a04SGreg Roach		$tree->setPreference('PEDIGREE_SHOW_GENDER', '0');
491a25f0a04SGreg Roach		$tree->setPreference('PREFER_LEVEL2_SOURCES', '1');
492a25f0a04SGreg Roach		$tree->setPreference('QUICK_REQUIRED_FACTS', 'BIRT,DEAT');
493a25f0a04SGreg Roach		$tree->setPreference('QUICK_REQUIRED_FAMFACTS', 'MARR');
494a25f0a04SGreg Roach		$tree->setPreference('REPO_FACTS_ADD', 'PHON,EMAIL,FAX,WWW,RESN');
495a25f0a04SGreg Roach		$tree->setPreference('REPO_FACTS_QUICK', '');
496a25f0a04SGreg Roach		$tree->setPreference('REPO_FACTS_UNIQUE', 'NAME,ADDR');
497a25f0a04SGreg Roach		$tree->setPreference('REPO_ID_PREFIX', 'R');
498a25f0a04SGreg Roach		$tree->setPreference('REQUIRE_AUTHENTICATION', '0');
499a25f0a04SGreg Roach		$tree->setPreference('SAVE_WATERMARK_IMAGE', '0');
500a25f0a04SGreg Roach		$tree->setPreference('SAVE_WATERMARK_THUMB', '0');
501a25f0a04SGreg Roach		$tree->setPreference('SHOW_AGE_DIFF', '0');
502a25f0a04SGreg Roach		$tree->setPreference('SHOW_COUNTER', '1');
503a25f0a04SGreg Roach		$tree->setPreference('SHOW_DEAD_PEOPLE', WT_PRIV_PUBLIC);
504a25f0a04SGreg Roach		$tree->setPreference('SHOW_EST_LIST_DATES', '0');
505a25f0a04SGreg Roach		$tree->setPreference('SHOW_FACT_ICONS', '1');
506a25f0a04SGreg Roach		$tree->setPreference('SHOW_GEDCOM_RECORD', '0');
507a25f0a04SGreg Roach		$tree->setPreference('SHOW_HIGHLIGHT_IMAGES', '1');
508a25f0a04SGreg Roach		$tree->setPreference('SHOW_LDS_AT_GLANCE', '0');
509a25f0a04SGreg Roach		$tree->setPreference('SHOW_LEVEL2_NOTES', '1');
510a25f0a04SGreg Roach		$tree->setPreference('SHOW_LIVING_NAMES', WT_PRIV_USER);
511a25f0a04SGreg Roach		$tree->setPreference('SHOW_MEDIA_DOWNLOAD', '0');
512a25f0a04SGreg Roach		$tree->setPreference('SHOW_NO_WATERMARK', WT_PRIV_USER);
513a25f0a04SGreg Roach		$tree->setPreference('SHOW_PARENTS_AGE', '1');
514a25f0a04SGreg Roach		$tree->setPreference('SHOW_PEDIGREE_PLACES', '9');
515a25f0a04SGreg Roach		$tree->setPreference('SHOW_PEDIGREE_PLACES_SUFFIX', '0');
516a25f0a04SGreg Roach		$tree->setPreference('SHOW_PRIVATE_RELATIONSHIPS', '1');
517a25f0a04SGreg Roach		$tree->setPreference('SHOW_RELATIVES_EVENTS', '_BIRT_CHIL,_BIRT_SIBL,_MARR_CHIL,_MARR_PARE,_DEAT_CHIL,_DEAT_PARE,_DEAT_GPAR,_DEAT_SIBL,_DEAT_SPOU');
518a25f0a04SGreg Roach		$tree->setPreference('SHOW_STATS', '0');
519a25f0a04SGreg Roach		$tree->setPreference('SOURCE_ID_PREFIX', 'S');
520a25f0a04SGreg Roach		$tree->setPreference('SOUR_FACTS_ADD', 'NOTE,REPO,SHARED_NOTE,RESN');
521a25f0a04SGreg Roach		$tree->setPreference('SOUR_FACTS_QUICK', 'TEXT,NOTE,REPO');
522a25f0a04SGreg Roach		$tree->setPreference('SOUR_FACTS_UNIQUE', 'AUTH,ABBR,TITL,PUBL,TEXT');
523a25f0a04SGreg Roach		$tree->setPreference('SUBLIST_TRIGGER_I', '200');
524a25f0a04SGreg Roach		$tree->setPreference('SURNAME_LIST_STYLE', 'style2');
525a25f0a04SGreg Roach		switch (WT_LOCALE) {
526a25f0a04SGreg Roach		case 'es':
527a25f0a04SGreg Roach			$tree->setPreference('SURNAME_TRADITION', 'spanish');
528a25f0a04SGreg Roach			break;
529a25f0a04SGreg Roach		case 'is':
530a25f0a04SGreg Roach			$tree->setPreference('SURNAME_TRADITION', 'icelandic');
531a25f0a04SGreg Roach			break;
532a25f0a04SGreg Roach		case 'lt':
533a25f0a04SGreg Roach			$tree->setPreference('SURNAME_TRADITION', 'lithuanian');
534a25f0a04SGreg Roach			break;
535a25f0a04SGreg Roach		case 'pl':
536a25f0a04SGreg Roach			$tree->setPreference('SURNAME_TRADITION', 'polish');
537a25f0a04SGreg Roach			break;
538a25f0a04SGreg Roach		case 'pt':
539a25f0a04SGreg Roach		case 'pt-BR':
540a25f0a04SGreg Roach			$tree->setPreference('SURNAME_TRADITION', 'portuguese');
541a25f0a04SGreg Roach			break;
542a25f0a04SGreg Roach		default:
543a25f0a04SGreg Roach			$tree->setPreference('SURNAME_TRADITION', 'paternal');
544a25f0a04SGreg Roach			break;
545a25f0a04SGreg Roach		}
546a25f0a04SGreg Roach		$tree->setPreference('THEME_DIR', 'webtrees');
547a25f0a04SGreg Roach		$tree->setPreference('THUMBNAIL_WIDTH', '100');
548a25f0a04SGreg Roach		$tree->setPreference('USE_RIN', '0');
549a25f0a04SGreg Roach		$tree->setPreference('USE_SILHOUETTE', '1');
550a25f0a04SGreg Roach		$tree->setPreference('WATERMARK_THUMB', '0');
551a25f0a04SGreg Roach		$tree->setPreference('WEBMASTER_USER_ID', Auth::id());
552a25f0a04SGreg Roach		$tree->setPreference('WEBTREES_EMAIL', '');
553a25f0a04SGreg Roach		$tree->setPreference('WORD_WRAPPED_NOTES', '0');
554a25f0a04SGreg Roach
555a25f0a04SGreg Roach		// Default restriction settings
556a25f0a04SGreg Roach		$statement = Database::prepare(
557a25f0a04SGreg Roach			"INSERT INTO `##default_resn` (gedcom_id, xref, tag_type, resn) VALUES (?, NULL, ?, ?)"
558a25f0a04SGreg Roach		);
559a25f0a04SGreg Roach		$statement->execute(array($tree_id, 'SSN', 'confidential'));
560a25f0a04SGreg Roach		$statement->execute(array($tree_id, 'SOUR', 'privacy'));
561a25f0a04SGreg Roach		$statement->execute(array($tree_id, 'REPO', 'privacy'));
562a25f0a04SGreg Roach		$statement->execute(array($tree_id, 'SUBM', 'confidential'));
563a25f0a04SGreg Roach		$statement->execute(array($tree_id, 'SUBN', 'confidential'));
564a25f0a04SGreg Roach
565a25f0a04SGreg Roach		// Genealogy data
566a25f0a04SGreg Roach		// It is simpler to create a temporary/unimported GEDCOM than to populate all the tables...
567a25f0a04SGreg Roach		$john_doe = /* I18N: This should be a common/default/placeholder name of an individual.  Put slashes around the surname. */
568a25f0a04SGreg Roach			I18N::translate('John /DOE/');
569a25f0a04SGreg Roach		$note     = I18N::translate('Edit this individual and replace their details with your own');
570a25f0a04SGreg Roach		Database::prepare("INSERT INTO `##gedcom_chunk` (gedcom_id, chunk_data) VALUES (?, ?)")->execute(array(
571a25f0a04SGreg Roach			$tree_id,
572a25f0a04SGreg 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"
573a25f0a04SGreg Roach		));
574a25f0a04SGreg Roach
575a25f0a04SGreg Roach		// Set the initial blocks
576a25f0a04SGreg Roach		Database::prepare(
577a25f0a04SGreg Roach			"INSERT INTO `##block` (gedcom_id, location, block_order, module_name)" .
578a25f0a04SGreg Roach			" SELECT ?, location, block_order, module_name" .
579a25f0a04SGreg Roach			" FROM `##block`" .
580a25f0a04SGreg Roach			" WHERE gedcom_id = -1"
581a25f0a04SGreg Roach		)->execute(array($tree_id));
582a25f0a04SGreg Roach
583a25f0a04SGreg Roach		// Update our cache
584*518bbdc1SGreg Roach		self::$trees[$tree->tree_id] = $tree;
585a25f0a04SGreg Roach
586a25f0a04SGreg Roach		return $tree;
587a25f0a04SGreg Roach	}
588a25f0a04SGreg Roach
589a25f0a04SGreg Roach	/**
590a25f0a04SGreg Roach	 * Delete all the genealogy data from a tree - in preparation for importing
591a25f0a04SGreg Roach	 * new data.  Optionally retain the media data, for when the user has been
592a25f0a04SGreg Roach	 * editing their data offline using an application which deletes (or does not
593a25f0a04SGreg Roach	 * support) media data.
594a25f0a04SGreg Roach	 *
595a25f0a04SGreg Roach	 * @param bool $keep_media
596a25f0a04SGreg Roach	 */
597a25f0a04SGreg Roach	public function deleteGenealogyData($keep_media) {
598*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##gedcom_chunk` WHERE gedcom_id = ?")->execute(array($this->tree_id));
599*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##individuals`  WHERE i_file    = ?")->execute(array($this->tree_id));
600*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##families`     WHERE f_file    = ?")->execute(array($this->tree_id));
601*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##sources`      WHERE s_file    = ?")->execute(array($this->tree_id));
602*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##other`        WHERE o_file    = ?")->execute(array($this->tree_id));
603*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##places`       WHERE p_file    = ?")->execute(array($this->tree_id));
604*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##placelinks`   WHERE pl_file   = ?")->execute(array($this->tree_id));
605*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##name`         WHERE n_file    = ?")->execute(array($this->tree_id));
606*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##dates`        WHERE d_file    = ?")->execute(array($this->tree_id));
607*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##change`       WHERE gedcom_id = ?")->execute(array($this->tree_id));
608a25f0a04SGreg Roach
609a25f0a04SGreg Roach		if ($keep_media) {
610*518bbdc1SGreg Roach			Database::prepare("DELETE FROM `##link` WHERE l_file =? AND l_type<>'OBJE'")->execute(array($this->tree_id));
611a25f0a04SGreg Roach		} else {
612*518bbdc1SGreg Roach			Database::prepare("DELETE FROM `##link`  WHERE l_file =?")->execute(array($this->tree_id));
613*518bbdc1SGreg Roach			Database::prepare("DELETE FROM `##media` WHERE m_file =?")->execute(array($this->tree_id));
614a25f0a04SGreg Roach		}
615a25f0a04SGreg Roach	}
616a25f0a04SGreg Roach
617a25f0a04SGreg Roach	/**
618a25f0a04SGreg Roach	 * Delete everything relating to a tree
619a25f0a04SGreg Roach	 */
620a25f0a04SGreg Roach	public function delete() {
621a25f0a04SGreg Roach		// If this is the default tree, then unset it
622*518bbdc1SGreg Roach		if (Site::getPreference('DEFAULT_GEDCOM') === self::getNameFromId($this->tree_id)) {
623a25f0a04SGreg Roach			Site::setPreference('DEFAULT_GEDCOM', '');
624a25f0a04SGreg Roach		}
625a25f0a04SGreg Roach
626a25f0a04SGreg Roach		$this->deleteGenealogyData(false);
627a25f0a04SGreg Roach
628*518bbdc1SGreg Roach		Database::prepare("DELETE `##block_setting` FROM `##block_setting` JOIN `##block` USING (block_id) WHERE gedcom_id=?")->execute(array($this->tree_id));
629*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##block`               WHERE gedcom_id = ?")->execute(array($this->tree_id));
630*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##user_gedcom_setting` WHERE gedcom_id = ?")->execute(array($this->tree_id));
631*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##gedcom_setting`      WHERE gedcom_id = ?")->execute(array($this->tree_id));
632*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##module_privacy`      WHERE gedcom_id = ?")->execute(array($this->tree_id));
633*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##next_id`             WHERE gedcom_id = ?")->execute(array($this->tree_id));
634*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##hit_counter`         WHERE gedcom_id = ?")->execute(array($this->tree_id));
635*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##default_resn`        WHERE gedcom_id = ?")->execute(array($this->tree_id));
636*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##gedcom_chunk`        WHERE gedcom_id = ?")->execute(array($this->tree_id));
637*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##log`                 WHERE gedcom_id = ?")->execute(array($this->tree_id));
638*518bbdc1SGreg Roach		Database::prepare("DELETE FROM `##gedcom`              WHERE gedcom_id = ?")->execute(array($this->tree_id));
639a25f0a04SGreg Roach
640a25f0a04SGreg Roach		// After updating the database, we need to fetch a new (sorted) copy
641a25f0a04SGreg Roach		self::$trees = null;
642a25f0a04SGreg Roach	}
643a25f0a04SGreg Roach
644a25f0a04SGreg Roach	/**
645a25f0a04SGreg Roach	 * Export the tree to a GEDCOM file
646a25f0a04SGreg Roach	 *
647a25f0a04SGreg Roach	 * @param $gedcom_file
648a25f0a04SGreg Roach	 *
649a25f0a04SGreg Roach	 * @return bool
650a25f0a04SGreg Roach	 */
651a25f0a04SGreg Roach	public function exportGedcom($gedcom_file) {
652a25f0a04SGreg Roach		// To avoid partial trees on timeout/diskspace/etc, write to a temporary file first
653a25f0a04SGreg Roach		$tmp_file = $gedcom_file . '.tmp';
654a25f0a04SGreg Roach
655a25f0a04SGreg Roach		$file_pointer = @fopen($tmp_file, 'w');
656a25f0a04SGreg Roach		if ($file_pointer === false) {
657a25f0a04SGreg Roach			return false;
658a25f0a04SGreg Roach		}
659a25f0a04SGreg Roach
660a25f0a04SGreg Roach		$buffer = reformat_record_export(gedcom_header($this->name));
661a25f0a04SGreg Roach
662a25f0a04SGreg Roach		$stmt = Database::prepare(
663a25f0a04SGreg Roach			"SELECT i_gedcom AS gedcom FROM `##individuals` WHERE i_file = ?" .
664a25f0a04SGreg Roach			" UNION ALL " .
665a25f0a04SGreg Roach			"SELECT f_gedcom AS gedcom FROM `##families`    WHERE f_file = ?" .
666a25f0a04SGreg Roach			" UNION ALL " .
667a25f0a04SGreg Roach			"SELECT s_gedcom AS gedcom FROM `##sources`     WHERE s_file = ?" .
668a25f0a04SGreg Roach			" UNION ALL " .
669a25f0a04SGreg Roach			"SELECT o_gedcom AS gedcom FROM `##other`       WHERE o_file = ? AND o_type NOT IN ('HEAD', 'TRLR')" .
670a25f0a04SGreg Roach			" UNION ALL " .
671a25f0a04SGreg Roach			"SELECT m_gedcom AS gedcom FROM `##media`       WHERE m_file = ?"
672*518bbdc1SGreg Roach		)->execute(array($this->tree_id, $this->tree_id, $this->tree_id, $this->tree_id, $this->tree_id));
673a25f0a04SGreg Roach
674a25f0a04SGreg Roach		while ($row = $stmt->fetch()) {
675a25f0a04SGreg Roach			$buffer .= reformat_record_export($row->gedcom);
676a25f0a04SGreg Roach			if (strlen($buffer) > 65535) {
677a25f0a04SGreg Roach				fwrite($file_pointer, $buffer);
678a25f0a04SGreg Roach				$buffer = '';
679a25f0a04SGreg Roach			}
680a25f0a04SGreg Roach		}
681a25f0a04SGreg Roach
682a25f0a04SGreg Roach		fwrite($file_pointer, $buffer . '0 TRLR' . WT_EOL);
683a25f0a04SGreg Roach		fclose($file_pointer);
684a25f0a04SGreg Roach
685a25f0a04SGreg Roach		return @rename($tmp_file, $gedcom_file);
686a25f0a04SGreg Roach	}
687a25f0a04SGreg Roach
688a25f0a04SGreg Roach	/**
689a25f0a04SGreg Roach	 * Import data from a gedcom file into this tree.
690a25f0a04SGreg Roach	 *
691a25f0a04SGreg Roach	 * @param string  $path       The full path to the (possibly temporary) file.
692a25f0a04SGreg Roach	 * @param string  $filename   The preferred filename, for export/download.
693a25f0a04SGreg Roach	 * @param boolean $keep_media Whether to retain any existing media records
694a25f0a04SGreg Roach	 *
695a25f0a04SGreg Roach	 * @throws \Exception
696a25f0a04SGreg Roach	 */
697a25f0a04SGreg Roach	public function importGedcomFile($path, $filename, $keep_media) {
698a25f0a04SGreg Roach		// Read the file in blocks of roughly 64K.  Ensure that each block
699a25f0a04SGreg Roach		// contains complete gedcom records.  This will ensure we don’t split
700a25f0a04SGreg Roach		// multi-byte characters, as well as simplifying the code to import
701a25f0a04SGreg Roach		// each block.
702a25f0a04SGreg Roach
703a25f0a04SGreg Roach		$file_data = '';
704a25f0a04SGreg Roach		$fp = fopen($path, 'rb');
705a25f0a04SGreg Roach
706a25f0a04SGreg Roach		// Don’t allow the user to cancel the request.  We do not want to be left with an incomplete transaction.
707a25f0a04SGreg Roach		ignore_user_abort(true);
708a25f0a04SGreg Roach
709a25f0a04SGreg Roach		Database::beginTransaction();
710a25f0a04SGreg Roach		$this->deleteGenealogyData($keep_media);
711a25f0a04SGreg Roach		$this->setPreference('gedcom_filename', $filename);
712a25f0a04SGreg Roach		$this->setPreference('imported', '0');
713a25f0a04SGreg Roach
714a25f0a04SGreg Roach		while (!feof($fp)) {
715a25f0a04SGreg Roach			$file_data .= fread($fp, 65536);
716a25f0a04SGreg Roach			// There is no strrpos() function that searches for substrings :-(
717a25f0a04SGreg Roach			for ($pos = strlen($file_data) - 1; $pos > 0; --$pos) {
718a25f0a04SGreg Roach				if ($file_data[$pos] === '0' && ($file_data[$pos - 1] === "\n" || $file_data[$pos - 1] === "\r")) {
719a25f0a04SGreg Roach					// We’ve found the last record boundary in this chunk of data
720a25f0a04SGreg Roach					break;
721a25f0a04SGreg Roach				}
722a25f0a04SGreg Roach			}
723a25f0a04SGreg Roach			if ($pos) {
724a25f0a04SGreg Roach				Database::prepare(
725a25f0a04SGreg Roach					"INSERT INTO `##gedcom_chunk` (gedcom_id, chunk_data) VALUES (?, ?)"
726*518bbdc1SGreg Roach				)->execute(array($this->tree_id, substr($file_data, 0, $pos)));
727a25f0a04SGreg Roach				$file_data = substr($file_data, $pos);
728a25f0a04SGreg Roach			}
729a25f0a04SGreg Roach		}
730a25f0a04SGreg Roach		Database::prepare(
731a25f0a04SGreg Roach			"INSERT INTO `##gedcom_chunk` (gedcom_id, chunk_data) VALUES (?, ?)"
732*518bbdc1SGreg Roach		)->execute(array($this->tree_id, $file_data));
733a25f0a04SGreg Roach
734a25f0a04SGreg Roach		Database::commit();
735a25f0a04SGreg Roach		fclose($fp);
736a25f0a04SGreg Roach	}
737a25f0a04SGreg Roach}
738