From f4fe5b9b53285819cf0c3aaafa513192c01db8f4 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Sun, 7 Mar 2021 11:38:02 +0100
Subject: [PATCH] use editor data

---
 packages/web/src/query/AceEditor.svelte       |   2 +-
 packages/web/src/query/SqlEditor.svelte       |   2 +-
 packages/web/src/query/useEditorData.ts       | 148 ++++++++++++++++++
 packages/web/src/tabs/QueryTab.svelte         |  18 ++-
 packages/web/src/utility/applySqlTemplate.ts  |  32 ++++
 .../web/src/utility/{common.js => common.ts}  |   8 +-
 packages/web/tsconfig.json                    |   2 +
 7 files changed, 204 insertions(+), 8 deletions(-)
 create mode 100644 packages/web/src/query/useEditorData.ts
 create mode 100644 packages/web/src/utility/applySqlTemplate.ts
 rename packages/web/src/utility/{common.js => common.ts} (74%)

diff --git a/packages/web/src/query/AceEditor.svelte b/packages/web/src/query/AceEditor.svelte
index 17ac6a23..21eccf54 100644
--- a/packages/web/src/query/AceEditor.svelte
+++ b/packages/web/src/query/AceEditor.svelte
@@ -6,7 +6,7 @@
 </script>
 
 <div bind:clientWidth bind:clientHeight class="ace-container">
-  <AceEditorCore {...$$props} width={clientWidth} height={clientHeight} />
+  <AceEditorCore {...$$props} width={clientWidth} height={clientHeight} on:input/>
 </div>
 
 <style>
diff --git a/packages/web/src/query/SqlEditor.svelte b/packages/web/src/query/SqlEditor.svelte
index 87c8d8ec..43585b03 100644
--- a/packages/web/src/query/SqlEditor.svelte
+++ b/packages/web/src/query/SqlEditor.svelte
@@ -18,4 +18,4 @@
   }
 </script>
 
-<AceEditor {mode} />
+<AceEditor {mode} {...$$props} on:input />
diff --git a/packages/web/src/query/useEditorData.ts b/packages/web/src/query/useEditorData.ts
new file mode 100644
index 00000000..63a3fa22
--- /dev/null
+++ b/packages/web/src/query/useEditorData.ts
@@ -0,0 +1,148 @@
+import _ from 'lodash';
+import { writable, derived } from 'svelte/store';
+import { onMount, onDestroy } from 'svelte';
+import localforage from 'localforage';
+import { changeTab } from '../utility/common';
+
+function getParsedLocalStorage(key) {
+  const value = localStorage.getItem(key);
+  if (value != null) {
+    try {
+      const res = JSON.parse(value);
+      return res;
+    } catch (e) {
+      // console.log('FAILED LOAD FROM STORAGE', e);
+      // console.log('VALUE', value);
+      localStorage.removeItem(key);
+    }
+  }
+  return null;
+}
+
+export default function useEditorData({ tabid, reloadToken = 0, loadFromArgs = null }) {
+  const localStorageKey = `tabdata_editor_${tabid}`;
+  let changeCounter = 0;
+  let savedCounter = 0;
+
+  const editorState = writable({
+    errorMessage: null,
+    isLoading: true,
+    value: null,
+  });
+
+  const editorValue = derived(editorState, $state => $state.value);
+
+  let initialData = null;
+  let value = null;
+
+  // const valueRef = React.useRef(null);
+
+  const initialLoad = async () => {
+    if (loadFromArgs) {
+      try {
+        const init = await loadFromArgs();
+        changeTab(tabid, tab => ({
+          ...tab,
+          props: _.omit(tab.props, ['initialArgs']),
+        }));
+        editorState.update(x => ({
+          ...x,
+          value: init,
+        }));
+        value = init;
+        initialData = init;
+        // mark as not saved
+        changeCounter += 1;
+      } catch (err) {
+        const message = (err && err.response && err.response.data && err.response.data.error) || 'Loading failed';
+        editorState.update(x => ({
+          ...x,
+          errorMessage: message,
+        }));
+        console.error(err.response);
+      }
+    } else {
+      const initFallback = getParsedLocalStorage(localStorageKey);
+      if (initFallback != null) {
+        editorState.update(x => ({
+          ...x,
+          value: initFallback,
+        }));
+        value = initFallback;
+        // move to local forage
+        await localforage.setItem(localStorageKey, initFallback);
+        localStorage.removeItem(localStorageKey);
+        initialData = initFallback;
+      } else {
+        const init = await localforage.getItem(localStorageKey);
+        if (init) {
+          editorState.update(x => ({
+            ...x,
+            value: init,
+          }));
+          value = init;
+          initialData = init;
+        }
+      }
+    }
+    editorState.update(x => ({
+      ...x,
+      isLoading: false,
+    }));
+  };
+
+  const saveToStorage = async () => {
+    if (value == null) return;
+    try {
+      await localforage.setItem(localStorageKey, value);
+      localStorage.removeItem(localStorageKey);
+      savedCounter = changeCounter;
+    } catch (err) {
+      console.error(err);
+    }
+  };
+
+  const saveToStorageSync = () => {
+    if (value == null) return;
+    if (savedCounter == changeCounter) return; // all saved
+    // on window unload must be synchronous actions, save to local storage instead
+    localStorage.setItem(localStorageKey, JSON.stringify(value));
+  };
+
+  const saveToStorageDebounced = _.debounce(saveToStorage, 5000);
+
+  const setEditorData = newValue => {
+    if (_.isFunction(newValue)) {
+      value = newValue(value);
+    } else {
+      if (newValue != null) value = newValue;
+    }
+    editorState.update(x => ({
+      ...x,
+      value,
+    }));
+
+    changeCounter += 1;
+    saveToStorageDebounced();
+  };
+
+  onMount(() => {
+    window.addEventListener('beforeunload', saveToStorageSync);
+    initialLoad();
+  });
+
+  onDestroy(() => {
+    saveToStorage();
+    window.removeEventListener('beforeunload', saveToStorageSync);
+  });
+
+  return {
+    editorState,
+    editorValue,
+    setEditorData,
+    initialData,
+    saveToStorage,
+    saveToStorageSync,
+    initialLoad,
+  };
+}
diff --git a/packages/web/src/tabs/QueryTab.svelte b/packages/web/src/tabs/QueryTab.svelte
index 7c20f604..31df65e5 100644
--- a/packages/web/src/tabs/QueryTab.svelte
+++ b/packages/web/src/tabs/QueryTab.svelte
@@ -1,17 +1,33 @@
 <script lang="ts">
   import VerticalSplitter from '../elements/VerticalSplitter.svelte';
   import SqlEditor from '../query/SqlEditor.svelte';
+  import useEditorData from '../query/useEditorData';
+  import { extensions } from '../stores';
+  import applySqlTemplate from '../utility/applySqlTemplate';
   import { useConnectionInfo } from '../utility/metadataLoaders';
 
   export let tabid;
   export let conid;
   export let database;
+  export let initialArgs;
 
   $: connection = useConnectionInfo({ conid });
+
+  const { editorState, setEditorData } = useEditorData({
+    tabid,
+    loadFromArgs:
+      initialArgs && initialArgs.sqlTemplate
+        ? () => applySqlTemplate(initialArgs.sqlTemplate, $extensions, $$props)
+        : null,
+  });
 </script>
 
 <VerticalSplitter>
   <svelte:fragment slot="1">
-    <SqlEditor engine={$connection && $connection.engine} />
+    <SqlEditor
+      engine={$connection && $connection.engine}
+      value={$editorState.value || ''}
+      on:input={e => setEditorData(e.detail)}
+    />
   </svelte:fragment>
 </VerticalSplitter>
diff --git a/packages/web/src/utility/applySqlTemplate.ts b/packages/web/src/utility/applySqlTemplate.ts
new file mode 100644
index 00000000..31392f73
--- /dev/null
+++ b/packages/web/src/utility/applySqlTemplate.ts
@@ -0,0 +1,32 @@
+import { getDbCore, getConnectionInfo, getSqlObjectInfo } from './metadataLoaders';
+import sqlFormatter from 'sql-formatter';
+import { driverBase, findEngineDriver } from 'dbgate-tools';
+
+export default async function applySqlTemplate(sqlTemplate, extensions, props) {
+  if (sqlTemplate == 'CREATE TABLE') {
+    const tableInfo = await getDbCore(props, props.objectTypeField || 'tables');
+    const connection = await getConnectionInfo(props);
+    const driver = findEngineDriver(connection, extensions) || driverBase;
+    const dmp = driver.createDumper();
+    if (tableInfo) dmp.createTable(tableInfo);
+    return dmp.s;
+  }
+  if (sqlTemplate == 'CREATE OBJECT') {
+    const objectInfo = await getSqlObjectInfo(props);
+    if (objectInfo) {
+      if (objectInfo.requiresFormat && objectInfo.createSql) return sqlFormatter.format(objectInfo.createSql);
+      else return objectInfo.createSql;
+    }
+  }
+  if (sqlTemplate == 'EXECUTE PROCEDURE') {
+    const procedureInfo = await getSqlObjectInfo(props);
+    const connection = await getConnectionInfo(props);
+
+    const driver = findEngineDriver(connection, extensions) || driverBase;
+    const dmp = driver.createDumper();
+    if (procedureInfo) dmp.put('^execute %f', procedureInfo);
+    return dmp.s;
+  }
+
+  return null;
+}
diff --git a/packages/web/src/utility/common.js b/packages/web/src/utility/common.ts
similarity index 74%
rename from packages/web/src/utility/common.js
rename to packages/web/src/utility/common.ts
index 29928feb..f8df2a33 100644
--- a/packages/web/src/utility/common.js
+++ b/packages/web/src/utility/common.ts
@@ -1,9 +1,7 @@
 import { openedTabs } from '../stores';
 
 export class LoadingToken {
-  constructor() {
-    this.isCanceled = false;
-  }
+  isCanceled = false;
 
   cancel() {
     this.isCanceled = true;
@@ -14,8 +12,8 @@ export function sleep(milliseconds) {
   return new Promise(resolve => window.setTimeout(() => resolve(null), milliseconds));
 }
 
-export function changeTab(tabid, setOpenedTabs, changeFunc) {
-  setOpenedTabs(files => files.map(tab => (tab.tabid == tabid ? changeFunc(tab) : tab)));
+export function changeTab(tabid, changeFunc) {
+  openedTabs.update(files => files.map(tab => (tab.tabid == tabid ? changeFunc(tab) : tab)));
 }
 
 export function setSelectedTabFunc(files, tabid) {
diff --git a/packages/web/tsconfig.json b/packages/web/tsconfig.json
index e7506e66..3b82116b 100644
--- a/packages/web/tsconfig.json
+++ b/packages/web/tsconfig.json
@@ -13,5 +13,7 @@
     "strictNullChecks": false,
     "strict": false,
     "target": "es6",
+    // "allowJs": true,
+    // "checkJs": true,
   }
 }