From 3fef2dd61da74043382e956e6a60e5054461de13 Mon Sep 17 00:00:00 2001
From: Przemek Wiech <pwiech@gmail.com>
Date: Fri, 14 Feb 2020 13:31:51 +0100
Subject: [PATCH] WikiTree: Make it work when passing numeric WikiTree ID
 instead of name

---
 src/load_data.ts | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/src/load_data.ts b/src/load_data.ts
index 6a4f1cc..91a248c 100644
--- a/src/load_data.ts
+++ b/src/load_data.ts
@@ -10,10 +10,11 @@ export function getSelection(
   indi?: string,
   generation?: number,
 ): IndiInfo {
-  return {
-    id: indi || data.indis[0].id,
-    generation: generation || 0,
-  };
+  // If ID is not given or it doesn't exist in the data, use the first ID in
+  // the data.
+  const id =
+    indi && data.indis.some((i) => i.id === indi) ? indi : data.indis[0].id;
+  return {id, generation: generation || 0};
 }
 
 function prepareData(