* dtable: support for reseting state in dtable plugin.

easysoft · Jul 18, 2024 · f39729e · f39729e
1 parent 11fc739
commit f39729e
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/lib/dtable/src/components/dtable.tsx b/lib/dtable/src/components/dtable.tsx
@@ -223,6 +223,22 @@ export class DTable extends Component<DTableOptions, DTableState> {
     resetState(props?: DTableOptions, init?: boolean) {
         this.#options = undefined;
         this.#layout = undefined;
+
+        props = props || this.props;
+        const newState: Partial<DTableState> = {};
+        this.#plugins.forEach(plugin => {
+            const {resetState, state: pluginState} = plugin;
+            if (resetState) {
+                if (typeof resetState === 'function') {
+                    Object.assign(newState, resetState.call(this, props));
+                } else if (pluginState) {
+                    Object.assign(newState, pluginState.call(this));
+                }
+            }
+        });
+        if (Object.keys(newState).length) {
+            this.setState(newState);
+        }
     }
 
     on(event: string, callback: DTableEventListener, target?: DTableEventTarget) {

diff --git a/lib/dtable/src/types/plugin.ts b/lib/dtable/src/types/plugin.ts
@@ -66,6 +66,7 @@ export type DTablePlugin<T extends DTablePluginTypes = {}, D extends DTablePlugi
     i18n?: Record<string, Record<string, string | object>>;
     data: (this: PluginTable) => {} & T['data'],
     state: (this: PluginTable) => {} & T['state'],
+    resetState: boolean | ((this: PluginTable, options: Options) => {} & T['state']);
     options: (this: PluginTable, options: Options) => Partial<Options>;
     footer: Record<string, CustomRenderResultGenerator<[layout: DTableLayout], PluginTable> | CustomRenderResultItem>;
     onCreate: (this: PluginTable, plugin: DTablePlugin<T, D>) => void;