From a5cc7624adf393a5a358e02aabd1f8b5ae5219b9 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Sun, 17 May 2020 20:51:16 +0200
Subject: [PATCH] running as portal

---
 packages/api/.env                          |  1 +
 packages/api/src/controllers/config.js     |  2 +
 packages/web/src/tabs/InfoPageTab.js       |  8 +++-
 packages/web/src/widgets/DatabaseWidget.js | 10 ++++-
 packages/web/src/widgets/Toolbar.js        | 49 +++++++++++++++-------
 5 files changed, 53 insertions(+), 17 deletions(-)

diff --git a/packages/api/.env b/packages/api/.env
index 0f4a7a58..8d6ae537 100644
--- a/packages/api/.env
+++ b/packages/api/.env
@@ -10,5 +10,6 @@ TOOLBAR=home
 ICON_home=fas fa-home
 TITLE_home=Home
 PAGE_home=home.html
+STARTUP_PAGES=home
 
 PAGES_DIRECTORY=/home/jena/jenasoft/dbgate-web/pages
diff --git a/packages/api/src/controllers/config.js b/packages/api/src/controllers/config.js
index a4dcb23c..c4ddb3a1 100644
--- a/packages/api/src/controllers/config.js
+++ b/packages/api/src/controllers/config.js
@@ -10,9 +10,11 @@ module.exports = {
           page: process.env[`PAGE_${name}`],
         }))
       : null;
+    const startupPages = process.env.STARTUP_PAGES ? process.env.STARTUP_PAGES.split(',') : [];
     return {
       runAsPortal: !!process.env.CONNECTIONS,
       toolbar,
+      startupPages,
     };
   },
 };
diff --git a/packages/web/src/tabs/InfoPageTab.js b/packages/web/src/tabs/InfoPageTab.js
index 740d04c1..daa581a4 100644
--- a/packages/web/src/tabs/InfoPageTab.js
+++ b/packages/web/src/tabs/InfoPageTab.js
@@ -1,6 +1,12 @@
 import React from 'react';
+import styled from 'styled-components';
 import resolveApi from '../utility/resolveApi';
 
+const Frame = styled.iframe`
+  flex: 1;
+  border: 0px solid gray;
+`;
+
 export default function InfoPageTab({ page }) {
-  return <iframe src={`${resolveApi()}/pages/${page}`} />;
+  return <Frame src={`${resolveApi()}/pages/${page}`} />;
 }
diff --git a/packages/web/src/widgets/DatabaseWidget.js b/packages/web/src/widgets/DatabaseWidget.js
index 8215b95f..7a4b3d9c 100644
--- a/packages/web/src/widgets/DatabaseWidget.js
+++ b/packages/web/src/widgets/DatabaseWidget.js
@@ -25,6 +25,7 @@ import {
 import axios from '../utility/axios';
 import LoadingInfo from './LoadingInfo';
 import SearchInput from './SearchInput';
+import ErrorInfo from './ErrorInfo';
 
 function SubDatabaseList({ data }) {
   const setDb = useSetCurrentDatabase();
@@ -126,7 +127,14 @@ function SqlObjectList({ conid, database }) {
 function SqlObjectListWrapper() {
   const db = useCurrentDatabase();
 
-  if (!db) return <div>(Choose database)</div>;
+  if (!db) {
+    return (
+      <>
+        <WidgetTitle>Tables, views, functions</WidgetTitle>
+        <ErrorInfo message="Database not selected" />
+      </>
+    );
+  }
   const { name, connection } = db;
 
   return <SqlObjectList conid={connection._id} database={name} />;
diff --git a/packages/web/src/widgets/Toolbar.js b/packages/web/src/widgets/Toolbar.js
index a51eef5e..23979c80 100644
--- a/packages/web/src/widgets/Toolbar.js
+++ b/packages/web/src/widgets/Toolbar.js
@@ -5,7 +5,7 @@ import styled from 'styled-components';
 import ToolbarButton from './ToolbarButton';
 import useNewQuery from '../query/useNewQuery';
 import { useConfig } from '../utility/metadataLoaders';
-import { useSetOpenedTabs } from '../utility/globalState';
+import { useSetOpenedTabs, useOpenedTabs } from '../utility/globalState';
 import { openNewTab } from '../utility/common';
 
 const ToolbarContainer = styled.div`
@@ -19,25 +19,44 @@ export default function ToolBar({ toolbarPortalRef }) {
   const config = useConfig();
   const toolbar = config.toolbar || [];
   const setOpenedTabs = useSetOpenedTabs();
+  const openedTabs = useOpenedTabs();
+
+  function openTabFromButton(button) {
+    if (openedTabs.find((x) => x.tabComponent == 'InfoPageTab' && x.props && x.props.page == button.page)) {
+      setOpenedTabs((tabs) =>
+        tabs.map((tab) => ({
+          ...tab,
+          selected: tab.tabComponent == 'InfoPageTab' && tab.props && tab.props.page == button.page,
+        }))
+      );
+    } else {
+      openNewTab(setOpenedTabs, {
+        title: button.title,
+        tabComponent: 'InfoPageTab',
+        icon: button.icon,
+        props: {
+          page: button.page,
+        },
+      });
+    }
+  }
+
+  React.useEffect(() => {
+    if (config.startupPages) {
+      for (const page of config.startupPages) {
+        const button = toolbar.find((x) => x.name == page);
+        if (button) {
+          openTabFromButton(button);
+        }
+      }
+    }
+  }, config && config.startupPages);
 
   return (
     <ToolbarContainer>
       <ConnectionModal modalState={modalState} />
       {toolbar.map((button) => (
-        <ToolbarButton
-          key={button.name}
-          onClick={() => {
-            openNewTab(setOpenedTabs, {
-              title: button.title,
-              tabComponent: 'InfoPageTab',
-              icon: button.icon,
-              props: {
-                page: button.page,
-              },
-            });
-          }}
-          icon={button.icon}
-        >
+        <ToolbarButton key={button.name} onClick={() => openTabFromButton(button)} icon={button.icon}>
           {button.title}
         </ToolbarButton>
       ))}