From d4bd801980ddffa3e8dcc16fdc4286b4c6b2dac1 Mon Sep 17 00:00:00 2001
From: Przemek Wiech <pwiech@gmail.com>
Date: Mon, 20 Jan 2020 19:34:20 +0100
Subject: [PATCH] When showing data from WikiTree, don't show "all relatives"
 chart option

---
 src/app.tsx     |  5 +++++
 src/top_bar.tsx | 20 +++++++++++++-------
 2 files changed, 18 insertions(+), 7 deletions(-)

diff --git a/src/app.tsx b/src/app.tsx
index 8ff04ee..9d1d46f 100644
--- a/src/app.tsx
+++ b/src/app.tsx
@@ -97,6 +97,8 @@ interface State {
   chartType: ChartType;
   /** Whether to show the error popup. */
   showErrorPopup: boolean;
+  /** True if data is loaded from WikiTree. */
+  wikiTreeSource: boolean;
 }
 
 export class App extends React.Component<RouteComponentProps, {}> {
@@ -106,6 +108,7 @@ export class App extends React.Component<RouteComponentProps, {}> {
     standalone: true,
     chartType: ChartType.Hourglass,
     showErrorPopup: false,
+    wikiTreeSource: false,
   };
   chartRef: Chart | null = null;
 
@@ -291,6 +294,7 @@ export class App extends React.Component<RouteComponentProps, {}> {
             showSidePanel,
             standalone,
             chartType,
+            wikiTreeSource: source === 'wikitree',
           }),
         );
       } catch (error) {
@@ -427,6 +431,7 @@ export class App extends React.Component<RouteComponentProps, {}> {
             <TopBar
               {...props}
               gedcom={this.state.data && this.state.data.gedcom}
+              allowAllRelativesChart={!this.state.wikiTreeSource}
               showingChart={
                 !!(
                   this.props.history.location.pathname === '/view' &&
diff --git a/src/top_bar.tsx b/src/top_bar.tsx
index 4ba49ef..651bdab 100644
--- a/src/top_bar.tsx
+++ b/src/top_bar.tsx
@@ -40,9 +40,13 @@ interface EventHandlers {
 }
 
 interface Props {
+  /** True if the application is currently showing a chart. */
   showingChart: boolean;
+  /** Data used for the search index. */
   gedcom?: GedcomData;
   standalone: boolean;
+  /** Whether to show the "All relatives" chart type in the menu. */
+  allowAllRelativesChart: boolean;
   eventHandlers: EventHandlers;
 }
 
@@ -320,13 +324,15 @@ export class TopBar extends React.Component<
                 defaultMessage="Hourglass chart"
               />
             </Dropdown.Item>
-            <Dropdown.Item onClick={() => this.changeView('relatives')}>
-              <Icon name="users" />
-              <FormattedMessage
-                id="menu.relatives"
-                defaultMessage="All relatives"
-              />
-            </Dropdown.Item>
+            {this.props.allowAllRelativesChart ? (
+              <Dropdown.Item onClick={() => this.changeView('relatives')}>
+                <Icon name="users" />
+                <FormattedMessage
+                  id="menu.relatives"
+                  defaultMessage="All relatives"
+                />
+              </Dropdown.Item>
+            ) : null}
             <Dropdown.Item onClick={() => this.changeView('fancy')}>
               <Icon name="users" />
               <FormattedMessage