Revert "feat: enhance credit hooks to trigger updates on store changes"
This reverts commit 7851a715a3
.
This commit is contained in:
parent
0ae3f27c78
commit
1c0c46fa34
@ -2,10 +2,8 @@ import { consumeCreditsAction } from '@/actions/consume-credits';
|
||||
import { getCreditBalanceAction } from '@/actions/get-credit-balance';
|
||||
import { getCreditStatsAction } from '@/actions/get-credit-stats';
|
||||
import { getCreditTransactionsAction } from '@/actions/get-credit-transactions';
|
||||
import { useCreditsStore } from '@/stores/credits-store';
|
||||
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
|
||||
import type { SortingState } from '@tanstack/react-table';
|
||||
import { useEffect } from 'react';
|
||||
|
||||
// Query keys
|
||||
export const creditsKeys = {
|
||||
@ -23,9 +21,7 @@ export const creditsKeys = {
|
||||
|
||||
// Hook to fetch credit balance
|
||||
export function useCreditBalance() {
|
||||
const updateTrigger = useCreditsStore((state) => state.updateTrigger);
|
||||
|
||||
const query = useQuery({
|
||||
return useQuery({
|
||||
queryKey: creditsKeys.balance(),
|
||||
queryFn: async () => {
|
||||
console.log('Fetching credit balance...');
|
||||
@ -39,23 +35,11 @@ export function useCreditBalance() {
|
||||
return result.data.credits || 0;
|
||||
},
|
||||
});
|
||||
|
||||
// Refetch when updateTrigger changes
|
||||
useEffect(() => {
|
||||
if (updateTrigger > 0) {
|
||||
console.log('Credits update triggered, refetching balance...');
|
||||
query.refetch();
|
||||
}
|
||||
}, [updateTrigger, query]);
|
||||
|
||||
return query;
|
||||
}
|
||||
|
||||
// Hook to fetch credit statistics
|
||||
export function useCreditStats() {
|
||||
const updateTrigger = useCreditsStore((state) => state.updateTrigger);
|
||||
|
||||
const query = useQuery({
|
||||
return useQuery({
|
||||
queryKey: creditsKeys.stats(),
|
||||
queryFn: async () => {
|
||||
console.log('Fetching credit stats...');
|
||||
@ -67,22 +51,11 @@ export function useCreditStats() {
|
||||
return result.data.data;
|
||||
},
|
||||
});
|
||||
|
||||
// Refetch when updateTrigger changes
|
||||
useEffect(() => {
|
||||
if (updateTrigger > 0) {
|
||||
console.log('Credits update triggered, refetching stats...');
|
||||
query.refetch();
|
||||
}
|
||||
}, [updateTrigger, query]);
|
||||
|
||||
return query;
|
||||
}
|
||||
|
||||
// Hook to consume credits
|
||||
export function useConsumeCredits() {
|
||||
const queryClient = useQueryClient();
|
||||
const triggerUpdate = useCreditsStore((state) => state.triggerUpdate);
|
||||
|
||||
return useMutation({
|
||||
mutationFn: async ({
|
||||
@ -102,9 +75,6 @@ export function useConsumeCredits() {
|
||||
return result.data;
|
||||
},
|
||||
onSuccess: () => {
|
||||
// Trigger credits update in store to notify all components
|
||||
triggerUpdate();
|
||||
|
||||
// Invalidate credit balance and stats after consuming credits
|
||||
queryClient.invalidateQueries({
|
||||
queryKey: creditsKeys.balance(),
|
||||
|
@ -1,23 +0,0 @@
|
||||
import { create } from 'zustand';
|
||||
|
||||
interface CreditsState {
|
||||
// trigger for credit updates, incremented each time credits change
|
||||
updateTrigger: number;
|
||||
// method to trigger credit updates
|
||||
triggerUpdate: () => void;
|
||||
}
|
||||
|
||||
/**
|
||||
* Credits store for managing credit balance updates.
|
||||
*
|
||||
* This store provides a simple trigger mechanism to notify components
|
||||
* when credits have been consumed or updated, ensuring UI components can
|
||||
* refetch the latest credit balance.
|
||||
*/
|
||||
export const useCreditsStore = create<CreditsState>((set) => ({
|
||||
updateTrigger: 0,
|
||||
triggerUpdate: () =>
|
||||
set((state) => ({
|
||||
updateTrigger: state.updateTrigger + 1,
|
||||
})),
|
||||
}));
|
Loading…
Reference in New Issue
Block a user