@ventlio/tanstack-query 0.2.63-beta.7 → 0.2.64

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ventlio/tanstack-query",
-  "version": "0.2.63-beta.7",
+  "version": "0.2.64",
   "license": "MIT",
   "main": "dist/index.js",
   "contributors": [

package/src/model/model.interface.ts

@@ -1,7 +1,10 @@
 export interface QueryModelBuilder<T> {
+  add: (data: T, position?: QueryModelAddPosition, path?: string) => T | undefined;
   findAll: (path?: string) => T[] | undefined;
   findMany: (selector: (record: T) => boolean, path?: string) => T[];
   find: (id: number | string, path?: string) => T | undefined;
   update: (id: number | string, data: Partial<T>, path?: string) => T | undefined;
   remove: (id: number, path?: string) => boolean;
 }
+
+export type QueryModelAddPosition = 'start' | 'end';

package/src/model/useQueryModel.ts

@@ -2,7 +2,7 @@ import { useQueryClient } from '@tanstack/react-query';
 import result from 'lodash.result';
 import set from 'lodash.set';
 import type { TanstackQueryConfig } from '../types';
-import type { QueryModelBuilder } from './model.interface';
+import type { QueryModelAddPosition, QueryModelBuilder } from './model.interface';
 import { useKeyTrackerModel } from './useKeyTrackerModel';
 
 export const useQueryModel = <T>(keyTracker: string, exact: boolean = true): QueryModelBuilder<T> => {
@@ -10,6 +10,26 @@ export const useQueryModel = <T>(keyTracker: string, exact: boolean = true): Que
   const { getQueryKey } = useKeyTrackerModel(keyTracker);
   const queryKey = getQueryKey() as any[];
 
+  const add = (data: T, position?: QueryModelAddPosition, path?: string): T | undefined => {
+    let records = findAll(path) ?? [];
+
+    if (!position || position === 'end') {
+      records = [...records, data];
+      // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition
+    } else if (position === 'start') {
+      records = [data, ...records];
+    }
+
+    if (!path) {
+      queryClient.setQueryData(queryKey, records);
+    } else {
+      const queryData = queryClient.getQueryData(queryKey, { exact }) ?? {};
+      queryClient.setQueryData(queryKey, set(queryData, path, records));
+    }
+
+    return data;
+  };
+
   const findAll = (path?: string): T[] | undefined => {
     const data = queryClient.getQueryData(queryKey, { exact });
 
@@ -104,5 +124,5 @@ export const useQueryModel = <T>(keyTracker: string, exact: boolean = true): Que
     return updated;
   };
 
-  return { find, findAll, findMany, remove, update };
+  return { find, findAll, findMany, remove, update, add };
 };