edr-platform/apps/edr-freight-web/backoffice/src/hooks/useWagons.ts

import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
import { wagonService } from '@/services/wagon.service';

export type WagonListFilters = import('@/services/wagon.service').WagonListFilters;

export const wagonKeys = {
  all: ['wagons'] as const,
  list: (filters?: WagonListFilters) => [...wagonKeys.all, 'list', filters ?? {}] as const,
  byTrain: (trainId: string) => [...wagonKeys.all, 'train', trainId] as const,
  details: () => [...wagonKeys.all, 'detail'] as const,
  detail: (id: string) => [...wagonKeys.details(), id] as const,
};

export function useWagons(filters?: WagonListFilters) {
  return useQuery({
    queryKey: wagonKeys.list(filters),
    queryFn: () => wagonService.getAll(filters ?? {}).then((res) => res.data),
  });
}

export const useGetWagons = useWagons;

export function useWagonsByTrain(trainId: string) {
  return useQuery({ queryKey: wagonKeys.byTrain(trainId), queryFn: () => wagonService.getByTrain(trainId).then(res => res.data), enabled: !!trainId });
}

export function useWagon(id: string) {
  return useQuery({ queryKey: wagonKeys.detail(id), queryFn: () => wagonService.getById(id).then(res => res.data), enabled: !!id });
}

export const useGetWagon = useWagon;

export function useAssignWagonToTrain() {
  const qc = useQueryClient();
  return useMutation({ mutationFn: ({ wagonId, trainId, sequenceNumber }: any) => wagonService.assignToTrain(wagonId, trainId, sequenceNumber), onSuccess: (_, { trainId }) => qc.invalidateQueries({ queryKey: wagonKeys.byTrain(trainId) }) });
}

export function useUnassignWagon() {
  const qc = useQueryClient();
  return useMutation({ mutationFn: wagonService.unassign, onSuccess: () => qc.invalidateQueries({ queryKey: wagonKeys.all }) });
}

export function useReorderWagons() {
  const qc = useQueryClient();
  return useMutation({ mutationFn: ({ trainId, wagonIds }: any) => wagonService.reorder(trainId, wagonIds), onSuccess: (_, { trainId }) => qc.invalidateQueries({ queryKey: wagonKeys.byTrain(trainId) }) });
}

export function useCreateWagon() {
  const qc = useQueryClient();
  return useMutation({ mutationFn: wagonService.create, onSuccess: () => qc.invalidateQueries({ queryKey: wagonKeys.all }) });
}

export function useUpdateWagon() {
  const qc = useQueryClient();
  return useMutation({ mutationFn: ({ id, data }: any) => wagonService.update(id, data), onSuccess: (_, { id }) => {
    qc.invalidateQueries({ queryKey: wagonKeys.all });
    qc.invalidateQueries({ queryKey: wagonKeys.detail(id) });
  } });
}

export function useDeleteWagon() {
  const qc = useQueryClient();
  return useMutation({ mutationFn: wagonService.delete, onSuccess: () => qc.invalidateQueries({ queryKey: wagonKeys.all }) });
}