From 49f22e1a3baad2887c7dfc924fd68a95e8200d34 Mon Sep 17 00:00:00 2001
From: Jan Prochazka <jan.prochazka@gmail.com>
Date: Sat, 6 Mar 2021 18:50:58 +0100
Subject: [PATCH] save commands

---
 packages/datalib/src/ChangeSet.ts             |  1 +
 packages/web/src/datagrid/DataGridCore.svelte | 43 +++++++++++++++++--
 packages/web/src/utility/memberStore.ts       | 19 ++++++++
 3 files changed, 60 insertions(+), 3 deletions(-)
 create mode 100644 packages/web/src/utility/memberStore.ts

diff --git a/packages/datalib/src/ChangeSet.ts b/packages/datalib/src/ChangeSet.ts
index 27384475..37d8c675 100644
--- a/packages/datalib/src/ChangeSet.ts
+++ b/packages/datalib/src/ChangeSet.ts
@@ -347,5 +347,6 @@ export function changeSetInsertNewRow(changeSet: ChangeSet, name?: NamedObjectIn
 }
 
 export function changeSetContainsChanges(changeSet: ChangeSet) {
+  if (!changeSet) return false;
   return changeSet.deletes.length > 0 || changeSet.updates.length > 0 || changeSet.inserts.length > 0;
 }
diff --git a/packages/web/src/datagrid/DataGridCore.svelte b/packages/web/src/datagrid/DataGridCore.svelte
index 6cb4a0b9..0f147a4b 100644
--- a/packages/web/src/datagrid/DataGridCore.svelte
+++ b/packages/web/src/datagrid/DataGridCore.svelte
@@ -1,6 +1,8 @@
 <script lang="ts" context="module">
   const currentDataGrid = writable(null);
 
+  const currentDataGridChangeSet = memberStore(currentDataGrid, grid => grid?.getChangeSetStore() || nullStore);
+
   registerCommand({
     id: 'dataGrid.refresh',
     category: 'Data grid',
@@ -8,7 +10,7 @@
     keyText: 'F5',
     toolbar: true,
     icon: 'icon reload',
-    enabledStore: derived([currentDataGrid], ([grid]) => grid != null),
+    enabledStore: derived(currentDataGrid, grid => grid != null),
     onClick: () => get(currentDataGrid).refresh(),
   });
 
@@ -19,10 +21,27 @@
     keyText: 'Ctrl+S',
     toolbar: true,
     icon: 'icon save',
-    enabledStore: derived([currentDataGrid], ([grid]) => grid?.getGrider()?.allowSave),
+    enabledStore: derived(currentDataGridChangeSet, (changeSet: any) => changeSetContainsChanges(changeSet?.value)),
     onClick: () => get(currentDataGrid).save(),
   });
 
+  registerCommand({
+    id: 'dataGrid.revertRowChanges',
+    category: 'Data grid',
+    name: 'Revert row changes',
+    keyText: 'Ctrl+R',
+    enabledStore: derived(currentDataGridChangeSet, (changeSet: any) => changeSetContainsChanges(changeSet?.value)),
+    onClick: () => get(currentDataGrid).revertRowChanges(),
+  });
+
+  registerCommand({
+    id: 'dataGrid.revertAllChanges',
+    category: 'Data grid',
+    name: 'Revert all changes',
+    enabledStore: derived(currentDataGridChangeSet, (changeSet: any) => changeSetContainsChanges(changeSet?.value)),
+    onClick: () => get(currentDataGrid).revertAllChanges(),
+  });
+
   function getRowCountInfo(selectedCells, grider, realColumnUniqueNames, selectedRowData, allRowCount) {
     if (selectedCells.length > 1 && selectedCells.every(x => _.isNumber(x[0]) && _.isNumber(x[1]))) {
       let sum = _.sumBy(selectedCells, cell => {
@@ -49,7 +68,7 @@
 </script>
 
 <script lang="ts">
-  import { GridDisplay } from 'dbgate-datalib';
+  import { changeSetContainsChanges, GridDisplay } from 'dbgate-datalib';
   import { get_current_component } from 'svelte/internal';
   import _ from 'lodash';
   import { writable, get, derived } from 'svelte/store';
@@ -76,6 +95,7 @@
   import createReducer from '../utility/createReducer';
   import keycodes from '../utility/keycodes';
   import { nullStore } from '../stores';
+  import memberStore from '../utility/memberStore';
 
   export let loadNextData = undefined;
   export let grider = undefined;
@@ -91,6 +111,7 @@
 
   export let isLoadedAll;
   export let loadedTime;
+  export let changeSetStore;
 
   const wheelRowCount = 5;
   const instance = get_current_component();
@@ -124,6 +145,22 @@
     return grider;
   }
 
+  export function getChangeSetStore() {
+    return changeSetStore;
+  }
+
+  export function revertRowChanges() {
+    grider.beginUpdate();
+    for (const index of getSelectedRowIndexes()) {
+      if (_.isNumber(index)) grider.revertRowChanges(index);
+    }
+    grider.endUpdate();
+  }
+
+  export function revertAllChanges() {
+    grider.revertAllChanges();
+  }
+
   $: autofillMarkerCell =
     selectedCells && selectedCells.length > 0 && _.uniq(selectedCells.map(x => x[0])).length == 1
       ? [_.max(selectedCells.map(x => x[0])), _.max(selectedCells.map(x => x[1]))]
diff --git a/packages/web/src/utility/memberStore.ts b/packages/web/src/utility/memberStore.ts
new file mode 100644
index 00000000..98179a38
--- /dev/null
+++ b/packages/web/src/utility/memberStore.ts
@@ -0,0 +1,19 @@
+import { writable } from 'svelte/store';
+
+export default function memberStore(store, extractStore) {
+  let res;
+  let unsubscribeSub = null;
+  store.subscribe(value => {
+    const subStore = extractStore(value);
+    if (unsubscribeSub) unsubscribeSub();
+    unsubscribeSub = subStore.subscribe(subValue => {
+      if (res) {
+        console.log('subValue', subValue);
+        res.set(subValue);
+      } else {
+        res = writable(subValue);
+      }
+    });
+  });
+  return res;
+}