From fc4b97adcb55a4fc4a6b92b01bcfab9cd2c606e3 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Tue, 5 Jan 2021 17:28:55 +0100
Subject: [PATCH] table data - reconnect button

---
 packages/web/src/datagrid/DataGridCore.js    | 11 ++++++++++-
 packages/web/src/datagrid/DataGridToolbar.js |  5 ++++-
 packages/web/src/icons.js                    |  1 +
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/packages/web/src/datagrid/DataGridCore.js b/packages/web/src/datagrid/DataGridCore.js
index 570509b8c..c3a975204 100644
--- a/packages/web/src/datagrid/DataGridCore.js
+++ b/packages/web/src/datagrid/DataGridCore.js
@@ -30,6 +30,7 @@ import { FontIcon } from '../icons';
 import useTheme from '../theme/useTheme';
 import { useShowMenu } from '../modals/showMenu';
 import useOpenNewTab from '../utility/useOpenNewTab';
+import axios from '../utility/axios';
 
 const GridContainer = styled.div`
   position: absolute;
@@ -1072,7 +1073,15 @@ export default function DataGridCore(props) {
         props.toolbarPortalRef.current &&
         tabVisible &&
         ReactDOM.createPortal(
-          <DataGridToolbar reload={() => display.reload()} save={handleSave} grider={grider} />,
+          <DataGridToolbar
+            reload={() => display.reload()}
+            save={handleSave}
+            grider={grider}
+            reconnect={async () => {
+              await axios.post('database-connections/refresh', { conid, database });
+              display.reload();
+            }}
+          />,
           props.toolbarPortalRef.current
         )}
       {isLoading && <LoadingInfo wrapper message="Loading data" />}
diff --git a/packages/web/src/datagrid/DataGridToolbar.js b/packages/web/src/datagrid/DataGridToolbar.js
index 9904e36de..180f931a8 100644
--- a/packages/web/src/datagrid/DataGridToolbar.js
+++ b/packages/web/src/datagrid/DataGridToolbar.js
@@ -1,12 +1,15 @@
 import React from 'react';
 import ToolbarButton from '../widgets/ToolbarButton';
 
-export default function DataGridToolbar({ reload, grider, save }) {
+export default function DataGridToolbar({ reload, reconnect, grider, save }) {
   return (
     <>
       <ToolbarButton onClick={reload} icon="icon reload">
         Refresh
       </ToolbarButton>
+      <ToolbarButton onClick={reconnect} icon="icon connection">
+        Reconnect
+      </ToolbarButton>
       <ToolbarButton disabled={!grider.canUndo} onClick={() => grider.undo()} icon="icon undo">
         Undo
       </ToolbarButton>
diff --git a/packages/web/src/icons.js b/packages/web/src/icons.js
index 1bdd25533..143231dcb 100644
--- a/packages/web/src/icons.js
+++ b/packages/web/src/icons.js
@@ -12,6 +12,7 @@ const iconNames = {
   'icon favorite': 'mdi mdi-star',
   'icon share': 'mdi mdi-share-variant',
   'icon add': 'mdi mdi-plus-circle',
+  'icon connection': 'mdi mdi-connection',
 
   'icon database': 'mdi mdi-database',
   'icon server': 'mdi mdi-server',