From c2dc4d76ba29c581aee615700041787b3b0267b9 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Thu, 4 Mar 2021 17:55:45 +0100
Subject: [PATCH] connection modal converted

---
 packages/web/src/TabContent.svelte            |  16 ++-
 .../web/src/appobj/ConnectionAppObject.svelte |   9 ++
 packages/web/src/forms/CheckboxField.svelte   |   2 +-
 .../web/src/forms/FormCheckboxField.svelte    |  16 ++-
 .../src/forms/FormElectronFileSelector.svelte |  14 +++
 .../forms/FormElectronFileSelectorRaw.svelte  |  29 +++++
 .../web/src/forms/FromCheckboxFieldRaw.svelte |   7 +-
 packages/web/src/forms/SelectField.svelte     |   6 +-
 .../web/src/modals/ConnectionModal.svelte     |  79 +++++++++++--
 .../ConnectionModalSshTunnelFields.svelte     | 107 ++++++++++++++++++
 .../modals/ConnectionModalSslFields.svelte    |  19 ++++
 packages/web/src/utility/metadataLoaders.ts   |   4 +-
 packages/web/src/widgets/TabsPanel.svelte     |   1 +
 13 files changed, 284 insertions(+), 25 deletions(-)
 create mode 100644 packages/web/src/forms/FormElectronFileSelector.svelte
 create mode 100644 packages/web/src/forms/FormElectronFileSelectorRaw.svelte
 create mode 100644 packages/web/src/modals/ConnectionModalSshTunnelFields.svelte
 create mode 100644 packages/web/src/modals/ConnectionModalSslFields.svelte

diff --git a/packages/web/src/TabContent.svelte b/packages/web/src/TabContent.svelte
index 70cf0fbff..dc6162dad 100644
--- a/packages/web/src/TabContent.svelte
+++ b/packages/web/src/TabContent.svelte
@@ -4,7 +4,7 @@
     if (tabComponent) {
       return {
         tabComponent,
-        props: selectedTab.props,
+        props: selectedTab && selectedTab.props,
       };
     }
     return null;
@@ -38,11 +38,15 @@
   $: {
     if (selectedTab) {
       const { tabid } = selectedTab;
-      if (tabid && !mountedTabs[tabid])
-        mountedTabs = {
-          ...mountedTabs,
-          [tabid]: createTabComponent(selectedTab),
-        };
+      if (tabid && !mountedTabs[tabid]) {
+        const newTab = createTabComponent(selectedTab);
+        if (newTab) {
+          mountedTabs = {
+            ...mountedTabs,
+            [tabid]: newTab,
+          };
+        }
+      }
     }
   }
 </script>
diff --git a/packages/web/src/appobj/ConnectionAppObject.svelte b/packages/web/src/appobj/ConnectionAppObject.svelte
index 84c19d608..05823a149 100644
--- a/packages/web/src/appobj/ConnectionAppObject.svelte
+++ b/packages/web/src/appobj/ConnectionAppObject.svelte
@@ -9,8 +9,15 @@
     const handleConnect = () => {
       openedConnections.update(list => _.uniq([...list, data._id]));
     };
+    const handleEdit = () => {
+      showModal(ConnectionModal, { connection: data });
+    };
 
     return [
+      {
+        text: 'Edit',
+        onClick: handleEdit,
+      },
       !$openedConnections.includes(data._id) && {
         text: 'Connect',
         onClick: handleConnect,
@@ -37,6 +44,8 @@
   import { currentDatabase, extensions, openedConnections } from '../stores';
   import axios from '../utility/axios';
   import { filterName } from 'dbgate-datalib';
+  import { showModal } from '../modals/modalTools';
+  import ConnectionModal from '../modals/ConnectionModal.svelte';
 
   export let data;
 
diff --git a/packages/web/src/forms/CheckboxField.svelte b/packages/web/src/forms/CheckboxField.svelte
index 8331051ba..301fff3c4 100644
--- a/packages/web/src/forms/CheckboxField.svelte
+++ b/packages/web/src/forms/CheckboxField.svelte
@@ -1 +1 @@
-<input type="checkbox" {...$$restProps} />
+<input type="checkbox" {...$$restProps} on:change />
diff --git a/packages/web/src/forms/FormCheckboxField.svelte b/packages/web/src/forms/FormCheckboxField.svelte
index 63b0d1d3c..ff5b13948 100644
--- a/packages/web/src/forms/FormCheckboxField.svelte
+++ b/packages/web/src/forms/FormCheckboxField.svelte
@@ -1,14 +1,22 @@
 <script lang="ts">
   import { getFormContext } from './FormProviderCore.svelte';
-  import FromCheckboxFieldRaw from './FromTextFieldRaw.svelte';
+  import FromCheckboxFieldRaw from './FromCheckboxFieldRaw.svelte';
 
   export let label;
   export let name;
+  export let disabled;
   export let templateProps = {};
 
-  const { template } = getFormContext();
+  const { template, setFieldValue, values } = getFormContext();
 </script>
 
-<svelte:component this={template} type="checkbox" {label} {...templateProps}>
-  <FromCheckboxFieldRaw {name} {...$$restProps} />
+<svelte:component
+  this={template}
+  type="checkbox"
+  {label}
+  {disabled}
+  {...templateProps}
+  labelProps={disabled ? { disabled: true } : { onClick: () => setFieldValue(name, !$values[name]) }}
+>
+  <FromCheckboxFieldRaw {name} {...$$restProps} {disabled} />
 </svelte:component>
diff --git a/packages/web/src/forms/FormElectronFileSelector.svelte b/packages/web/src/forms/FormElectronFileSelector.svelte
new file mode 100644
index 000000000..2a0639d6d
--- /dev/null
+++ b/packages/web/src/forms/FormElectronFileSelector.svelte
@@ -0,0 +1,14 @@
+<script lang="ts">
+  import FormElectronFileSelectorRaw from './FormElectronFileSelectorRaw.svelte';
+  import { getFormContext } from './FormProviderCore.svelte';
+
+  export let label;
+  export let name;
+  export let templateProps = {};
+
+  const { template } = getFormContext();
+</script>
+
+<svelte:component this={template} type="select" {label} {...templateProps}>
+  <FormElectronFileSelectorRaw {name} {...$$restProps} />
+</svelte:component>
diff --git a/packages/web/src/forms/FormElectronFileSelectorRaw.svelte b/packages/web/src/forms/FormElectronFileSelectorRaw.svelte
new file mode 100644
index 000000000..fb2b0eb83
--- /dev/null
+++ b/packages/web/src/forms/FormElectronFileSelectorRaw.svelte
@@ -0,0 +1,29 @@
+<script lang="ts">
+  import getElectron from '../utility/getElectron';
+
+  import InlineButton from '../widgets/InlineButton.svelte';
+
+  import { getFormContext } from './FormProviderCore.svelte';
+  import TextField from './TextField.svelte';
+
+  export let name;
+  export let disabled = false;
+
+  const { values, setFieldValue } = getFormContext();
+
+  function handleBrowse() {
+    const electron = getElectron();
+    if (!electron) return;
+    const filePaths = electron.remote.dialog.showOpenDialogSync(electron.remote.getCurrentWindow(), {
+      defaultPath: values[name],
+      properties: ['showHiddenFiles'],
+    });
+    const filePath = filePaths && filePaths[0];
+    if (filePath) setFieldValue(name, filePath);
+  }
+</script>
+
+<div class="flex">
+  <TextField {...$$restProps} value={$values[name]} onClick={handleBrowse} readOnly {disabled} />
+  <InlineButton on:click={handleBrowse} {disabled}>Browse</InlineButton>
+</div>
diff --git a/packages/web/src/forms/FromCheckboxFieldRaw.svelte b/packages/web/src/forms/FromCheckboxFieldRaw.svelte
index eb5328772..b381e071e 100644
--- a/packages/web/src/forms/FromCheckboxFieldRaw.svelte
+++ b/packages/web/src/forms/FromCheckboxFieldRaw.svelte
@@ -1,10 +1,13 @@
 <script lang="ts">
   import { getFormContext } from './FormProviderCore.svelte';
-  import TextField from './TextField.svelte';
+  import CheckboxField from './CheckboxField.svelte';
 
   export let name;
 
   const { values, setFieldValue } = getFormContext();
+  function handleChange(e) {
+    setFieldValue(name, e.target['checked']);
+  }
 </script>
 
-<TextField {...$$restProps} value={$values[name]} on:change={e => setFieldValue(name, e.target['checked'])} />
+<CheckboxField {...$$restProps} checked={$values[name]} on:change={handleChange} />
diff --git a/packages/web/src/forms/SelectField.svelte b/packages/web/src/forms/SelectField.svelte
index f3b92f7a3..810bc5761 100644
--- a/packages/web/src/forms/SelectField.svelte
+++ b/packages/web/src/forms/SelectField.svelte
@@ -1,12 +1,12 @@
 <script lang="ts">
+  import _ from 'lodash';
+
   export let options = [];
   export let value;
-
-  $: console.log('FIELD', $$props.value);
 </script>
 
 <select {...$$restProps} on:change>
-  {#each options as x (x.value)}
+  {#each _.compact(options) as x (x.value)}
     <option value={x.value} selected={value == x.value}>
       {x.label}
     </option>
diff --git a/packages/web/src/modals/ConnectionModal.svelte b/packages/web/src/modals/ConnectionModal.svelte
index 8be51e852..3fcc19f0c 100644
--- a/packages/web/src/modals/ConnectionModal.svelte
+++ b/packages/web/src/modals/ConnectionModal.svelte
@@ -3,20 +3,52 @@
   import FormButton from '../forms/FormButton.svelte';
   import FormProvider from '../forms/FormProvider.svelte';
   import FormSubmit from '../forms/FormSubmit.svelte';
+  import FontIcon from '../icons/FontIcon.svelte';
+  import axios from '../utility/axios';
   import TabControl from '../widgets/TabControl.svelte';
   import ConnectionModalDriverFields from './ConnectionModalDriverFields.svelte';
+  import ConnectionModalSshTunnelFields from './ConnectionModalSshTunnelFields.svelte';
+  import ConnectionModalSslFields from './ConnectionModalSslFields.svelte';
   import FormFieldTemplateLarge from './FormFieldTemplateLarge.svelte';
 
   import ModalBase from './ModalBase.svelte';
+  import { closeModal } from './modalTools';
 
   export let connection;
+  export let modalId;
+
+  let isTesting;
+  let sqlConnectResult;
+
+  const testIdRef = { current: 0 };
+
+  async function handleTest(e) {
+    isTesting = true;
+    testIdRef.current += 1;
+    const testid = testIdRef.current;
+    const resp = await axios.post('connections/test', e.detail);
+    if (testIdRef.current != testid) return;
+
+    isTesting = false;
+    sqlConnectResult = resp.data;
+  }
+
+  function handleCancelTest() {
+    testIdRef.current += 1; // invalidate current test
+    isTesting = false;
+  }
+
+  async function handleSubmit(e) {
+    axios.post('connections/save', e.detail);
+    closeModal(modalId);
+  }
 </script>
 
 <FormProvider
   template={FormFieldTemplateLarge}
   initialValues={connection || { server: 'localhost', engine: 'mssql@dbgate-plugin-mssql' }}
 >
-  <ModalBase {...$$restProps} noPadding>
+  <ModalBase {...$$restProps} {modalId} noPadding>
     <div slot="header">Add connection</div>
 
     <TabControl
@@ -28,17 +60,42 @@
         },
         {
           label: 'SSH Tunnel',
-          slot: 1,
+          component: ConnectionModalSshTunnelFields,
+        },
+        {
+          label: 'SSL',
+          component: ConnectionModalSslFields,
         },
       ]}
-    >
-      <div slot="1">SSH</div>
-    </TabControl>
+    />
 
     <div slot="footer" class="flex">
       <div class="buttons">
-        <FormButton value="Test" />
-        <FormSubmit value="Save" on:click={v => console.log('SAVE', v.detail)} />
+        {#if isTesting}
+          <FormButton value="Cancel test" on:click={handleCancelTest} />
+        {:else}
+          <FormButton value="Test" on:click={handleTest} />
+        {/if}
+        <FormSubmit value="Save" on:click={handleSubmit} />
+      </div>
+      <div class="test-result">
+        {#if !isTesting && sqlConnectResult && sqlConnectResult.msgtype == 'connected'}
+          <div>
+            Connected: <FontIcon icon="img ok" />
+            {sqlConnectResult.version}
+          </div>
+        {/if}
+        {#if !isTesting && sqlConnectResult && sqlConnectResult.msgtype == 'error'}
+          <div>
+            Connect failed: <FontIcon icon="img error" />
+            {sqlConnectResult.error}
+          </div>
+        {/if}
+        {#if isTesting}
+          <div>
+            <FontIcon icon="icon loading" /> Testing connection
+          </div>
+        {/if}
       </div>
     </div>
   </ModalBase>
@@ -48,4 +105,12 @@
   .buttons {
     flex-shrink: 0;
   }
+
+  .test-result {
+    margin-left: 10px;
+    align-self: center;
+    white-space: nowrap;
+    overflow: hidden;
+    text-overflow: ellipsis;
+  }
 </style>
diff --git a/packages/web/src/modals/ConnectionModalSshTunnelFields.svelte b/packages/web/src/modals/ConnectionModalSshTunnelFields.svelte
new file mode 100644
index 000000000..235e8f99f
--- /dev/null
+++ b/packages/web/src/modals/ConnectionModalSshTunnelFields.svelte
@@ -0,0 +1,107 @@
+<script lang="ts">
+  import FormElectronFileSelector from '../forms/FormElectronFileSelector.svelte';
+
+  import FormPasswordField from '../forms/FormPasswordField.svelte';
+
+  import { getFormContext } from '../forms/FormProviderCore.svelte';
+  import FormSelectField from '../forms/FormSelectField.svelte';
+
+  import FormTextField from '../forms/FormTextField.svelte';
+  import FormCheckboxField from '../forms/FormCheckboxField.svelte';
+  import getElectron from '../utility/getElectron';
+  import { usePlatformInfo } from '../utility/metadataLoaders';
+  import FontIcon from '../icons/FontIcon.svelte';
+
+  const { values, setFieldValue } = getFormContext();
+  const electron = getElectron();
+
+  $: useSshTunnel = $values.useSshTunnel;
+  $: platformInfo = usePlatformInfo();
+
+  $: {
+    if (!$values.sshMode) setFieldValue('sshMode', 'userPassword');
+    if (!$values.sshPort) setFieldValue('sshPort', '22');
+    if (!$values.sshKeyfile && $platformInfo) setFieldValue('sshKeyfile', $platformInfo.defaultKeyFile);
+  }
+</script>
+
+<FormCheckboxField label="Use SSH tunnel" name="useSshTunnel" />
+
+<div class="row">
+  <div class="col-9 mr-1">
+    <FormTextField label="Host" name="sshHost" disabled={!useSshTunnel} templateProps={{ noMargin: true }} />
+  </div>
+  <div class="col-3">
+    <FormTextField label="Port" name="sshPort" disabled={!useSshTunnel} templateProps={{ noMargin: true }} />
+  </div>
+</div>
+<FormTextField label="Bastion host (Jump host)" name="sshBastionHost" disabled={!useSshTunnel} />
+
+<FormSelectField
+  label="SSH Authentication"
+  name="sshMode"
+  disabled={!useSshTunnel}
+  options={[
+    { value: 'userPassword', label: 'Username & password' },
+    { value: 'agent', label: 'SSH agent' },
+    electron && { value: 'keyFile', label: 'Key file' },
+  ]}
+/>
+
+{#if $values.sshMode != 'userPassword'}
+  <FormTextField label="Login" name="sshLogin" disabled={!useSshTunnel} />
+{/if}
+
+{#if $values.sshMode == 'userPassword'}
+  <div class="row">
+    <div class="col-6 mr-1">
+      <FormTextField label="Login" name="sshLogin" disabled={!useSshTunnel} templateProps={{ noMargin: true }} />
+    </div>
+    <div class="col-6">
+      <FormPasswordField
+        label="Password"
+        name="sshPassword"
+        disabled={!useSshTunnel}
+        templateProps={{ noMargin: true }}
+      />
+    </div>
+  </div>
+{/if}
+
+{#if $values.sshMode == 'keyFile'}
+  <div class="row">
+    <div class="col-6 mr-1">
+      <FormElectronFileSelector
+        label="Private key file"
+        name="sshKeyfile"
+        disabled={!useSshTunnel}
+        templateProps={{ noMargin: true }}
+      />
+    </div>
+    <div class="col-6">
+      <FormPasswordField
+        label="Key file passphrase"
+        name="sshKeyfilePassword"
+        disabled={!useSshTunnel}
+        templateProps={{ noMargin: true }}
+      />
+    </div>
+  </div>
+{/if}
+
+{#if useSshTunnel && $values.sshMode == 'agent'}
+  <div class="ml-3 mb-3">
+    {#if $platformInfo && $platformInfo.sshAuthSock}
+      <FontIcon icon="img ok" /> SSH Agent found
+    {:else}
+      <FontIcon icon="img error" /> SSH Agent not found
+    {/if}
+  </div>
+{/if}
+
+<style>
+  .row {
+    margin: var(--dim-large-form-margin);
+    display: flex;
+  }
+</style>
diff --git a/packages/web/src/modals/ConnectionModalSslFields.svelte b/packages/web/src/modals/ConnectionModalSslFields.svelte
new file mode 100644
index 000000000..64e3c250c
--- /dev/null
+++ b/packages/web/src/modals/ConnectionModalSslFields.svelte
@@ -0,0 +1,19 @@
+<script lang="ts">
+  import FormElectronFileSelector from '../forms/FormElectronFileSelector.svelte';
+
+  import { getFormContext } from '../forms/FormProviderCore.svelte';
+
+  import FormCheckboxField from '../forms/FormCheckboxField.svelte';
+  import getElectron from '../utility/getElectron';
+
+  const { values, setFieldValue } = getFormContext();
+  const electron = getElectron();
+
+  $: useSsl = $values.useSsl;
+</script>
+
+<FormCheckboxField label="Use SSL" name="useSsl" />
+<FormElectronFileSelector label="CA Cert (optional)" name="sslCaFile" disabled={!useSsl || !electron} />
+<FormElectronFileSelector label="Certificate (optional)" name="sslCertFile" disabled={!useSsl || !electron} />
+<FormElectronFileSelector label="Key file (optional)" name="sslKeyFile" disabled={!useSsl || !electron} />
+<FormCheckboxField label="Reject unauthorized" name="sslRejectUnauthorized" disabled={!useSsl} />
diff --git a/packages/web/src/utility/metadataLoaders.ts b/packages/web/src/utility/metadataLoaders.ts
index 4ffc6f059..32f6a8d0c 100644
--- a/packages/web/src/utility/metadataLoaders.ts
+++ b/packages/web/src/utility/metadataLoaders.ts
@@ -335,10 +335,10 @@ export function useConfig() {
 }
 
 export function getPlatformInfo() {
-  return getCore(platformInfoLoader, {}) || {};
+  return getCore(platformInfoLoader, {});
 }
 export function usePlatformInfo() {
-  return useCore(platformInfoLoader, {}) || {};
+  return useCore(platformInfoLoader, {});
 }
 
 export function getArchiveFiles(args) {
diff --git a/packages/web/src/widgets/TabsPanel.svelte b/packages/web/src/widgets/TabsPanel.svelte
index 5778859d5..705b02bd2 100644
--- a/packages/web/src/widgets/TabsPanel.svelte
+++ b/packages/web/src/widgets/TabsPanel.svelte
@@ -210,6 +210,7 @@
   }
   .file-name {
     margin-left: 5px;
+    white-space: nowrap;
   }
   .close-button {
     margin-left: 5px;