use-active-table-update.ts

import { useMutation } from "@tanstack/react-query";

import type {
  AdapterUpdateDetails,
  AdapterUpdateOptions,
} from "../../data/adapter";
import { useActiveTableRowsCollection } from "./use-active-table-rows-collection";
import { useFiltering } from "./use-filtering";
import { usePagination } from "./use-pagination";
import { useSorting } from "./use-sorting";

export interface UseActiveTableUpdateParams {
  details: AdapterUpdateDetails;
  options: AdapterUpdateOptions;
}

export function useActiveTableUpdate() {
  const { paginationState } = usePagination();
  const { sortingState } = useSorting();
  const { appliedFilter } = useFiltering();
  const { activeTable, collection } = useActiveTableRowsCollection({
    pageIndex: paginationState.pageIndex,
    pageSize: paginationState.pageSize,
    sortOrder: sortingState,
    filter: appliedFilter,
  });
  const queryKeyPrefix = [
    "schema",
    activeTable?.schema ?? null,
    "table",
    activeTable?.name ?? null,
  ] as const;

  return useMutation({
    mutationFn: async (params: UseActiveTableUpdateParams) => {
      const rowId = String(params.details.row.__ps_rowid ?? "");

      if (!collection || !activeTable || !rowId) {
        throw new Error("Active table collection is not available");
      }

      const transaction = collection.update(rowId, (draft) => {
        Object.assign(draft, params.details.changes);
      });

      await transaction.isPersisted.promise;

      return {
        row: collection.get(rowId),
      };
    },
    mutationKey: [...queryKeyPrefix, "update"],
    retry: false,
  });
}