cross-financial-management/client/src/hooks/useSegments.ts

import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
import {
  createSegmentDefinition,
  createSegmentValue,
  fetchSegmentDefinitions,
} from '../api/segments.api';

export function useSegmentDefinitions() {
  return useQuery({ queryKey: ['segmentDefinitions'], queryFn: fetchSegmentDefinitions });
}

export function useCreateSegmentDefinition() {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: createSegmentDefinition,
    onSuccess: () => queryClient.invalidateQueries({ queryKey: ['segmentDefinitions'] }),
  });
}

export function useCreateSegmentValue() {
  const queryClient = useQueryClient();
  return useMutation({
    mutationFn: createSegmentValue,
    onSuccess: () => {
      queryClient.invalidateQueries({ queryKey: ['segmentDefinitions'] });
      queryClient.invalidateQueries({ queryKey: ['branches'] });
    },
  });
}