prodobit/packages/react-sdk/src/hooks/useBoms.ts

Open-core business application development platform

import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
import { useProdobitClient } from '../providers/ProdobitProvider';
import { queryKeys } from '../utils/query-keys';
import type { QueryOptions, MutationOptions, BomQuery } from '../types';
import type { Bom, CreateBomRequest, UpdateBomRequest, CreateBomComponentRequest, Response, PaginatedResponse } from '@prodobit/types';

export const useBoms = (
  filters?: BomQuery,
  options?: QueryOptions
) => {
  const client = useProdobitClient();

  return useQuery<Response<Bom[]>, Error>({
    queryKey: queryKeys.boms.list(filters),
    queryFn: () => client.getBoms(filters),
    ...options,
  });
};

export const useBom = (id: string, options?: QueryOptions) => {
  const client = useProdobitClient();

  return useQuery<Response<Bom>, Error>({
    queryKey: queryKeys.boms.detail(id),
    queryFn: () => client.getBomById(id),
    enabled: !!id && options?.enabled !== false,
    ...options,
  });
};

export const useCreateBom = (options?: MutationOptions) => {
  const client = useProdobitClient();
  const queryClient = useQueryClient();

  return useMutation<Response<Bom>, Error, CreateBomRequest>({
    mutationFn: (data: CreateBomRequest) => client.createBom(data),
    onSuccess: (data) => {
      queryClient.invalidateQueries({ queryKey: queryKeys.boms.all() });
      options?.onSuccess?.(data);
    },
    onError: options?.onError,
  });
};

export const useUpdateBom = (options?: MutationOptions) => {
  const client = useProdobitClient();
  const queryClient = useQueryClient();

  return useMutation<Response<Bom>, Error, { id: string; data: UpdateBomRequest }>({
    mutationFn: ({ id, data }: { id: string; data: UpdateBomRequest }) => 
      client.updateBom(id, data),
    onSuccess: (data, variables) => {
      queryClient.invalidateQueries({ queryKey: queryKeys.boms.all() });
      queryClient.invalidateQueries({ queryKey: queryKeys.boms.detail(variables.id) });
      options?.onSuccess?.(data);
    },
    onError: options?.onError,
  });
};

export const useDeleteBom = (options?: MutationOptions) => {
  const client = useProdobitClient();
  const queryClient = useQueryClient();

  return useMutation<Response<void>, Error, string>({
    mutationFn: (id: string) => client.deleteBom(id),
    onSuccess: (data, id) => {
      queryClient.invalidateQueries({ queryKey: queryKeys.boms.all() });
      queryClient.removeQueries({ queryKey: queryKeys.boms.detail(id) });
      options?.onSuccess?.(data);
    },
    onError: options?.onError,
  });
};

// BOM Components hooks
export const useBomComponents = (
  filters?: { bomId?: string; itemId?: string; page?: number; limit?: number },
  options?: QueryOptions
) => {
  const client = useProdobitClient();

  return useQuery<Response<any[]>, Error>({
    queryKey: ['bomComponents', filters],
    queryFn: () => client.getBomComponents(filters),
    ...options,
  });
};

export const useCreateBomComponent = (options?: MutationOptions) => {
  const client = useProdobitClient();
  const queryClient = useQueryClient();

  return useMutation<Response<any>, Error, CreateBomComponentRequest>({
    mutationFn: (data: CreateBomComponentRequest) => client.createBomComponent(data),
    onSuccess: (data) => {
      queryClient.invalidateQueries({ queryKey: ['bomComponents'] });
      options?.onSuccess?.(data);
    },
    onError: options?.onError,
  });
};

export const useUpdateBomComponent = (options?: MutationOptions) => {
  const client = useProdobitClient();
  const queryClient = useQueryClient();

  return useMutation<Response<any>, Error, { id: string; data: any }>({
    mutationFn: ({ id, data }) => client.updateBomComponent(id, data),
    onSuccess: (data) => {
      queryClient.invalidateQueries({ queryKey: ['bomComponents'] });
      options?.onSuccess?.(data);
    },
    onError: options?.onError,
  });
};

export const useDeleteBomComponent = (options?: MutationOptions) => {
  const client = useProdobitClient();
  const queryClient = useQueryClient();

  return useMutation<Response<void>, Error, string>({
    mutationFn: (id: string) => client.deleteBomComponent(id),
    onSuccess: (data) => {
      queryClient.invalidateQueries({ queryKey: ['bomComponents'] });
      options?.onSuccess?.(data);
    },
    onError: options?.onError,
  });
};