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 { getCreditBalanceAction } from '@/actions/get-credit-balance';
|
||||||
import { getCreditStatsAction } from '@/actions/get-credit-stats';
|
import { getCreditStatsAction } from '@/actions/get-credit-stats';
|
||||||
import { getCreditTransactionsAction } from '@/actions/get-credit-transactions';
|
import { getCreditTransactionsAction } from '@/actions/get-credit-transactions';
|
||||||
import { useCreditsStore } from '@/stores/credits-store';
|
|
||||||
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
|
import { useMutation, useQuery, useQueryClient } from '@tanstack/react-query';
|
||||||
import type { SortingState } from '@tanstack/react-table';
|
import type { SortingState } from '@tanstack/react-table';
|
||||||
import { useEffect } from 'react';
|
|
||||||
|
|
||||||
// Query keys
|
// Query keys
|
||||||
export const creditsKeys = {
|
export const creditsKeys = {
|
||||||
@ -23,9 +21,7 @@ export const creditsKeys = {
|
|||||||
|
|
||||||
// Hook to fetch credit balance
|
// Hook to fetch credit balance
|
||||||
export function useCreditBalance() {
|
export function useCreditBalance() {
|
||||||
const updateTrigger = useCreditsStore((state) => state.updateTrigger);
|
return useQuery({
|
||||||
|
|
||||||
const query = useQuery({
|
|
||||||
queryKey: creditsKeys.balance(),
|
queryKey: creditsKeys.balance(),
|
||||||
queryFn: async () => {
|
queryFn: async () => {
|
||||||
console.log('Fetching credit balance...');
|
console.log('Fetching credit balance...');
|
||||||
@ -39,23 +35,11 @@ export function useCreditBalance() {
|
|||||||
return result.data.credits || 0;
|
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
|
// Hook to fetch credit statistics
|
||||||
export function useCreditStats() {
|
export function useCreditStats() {
|
||||||
const updateTrigger = useCreditsStore((state) => state.updateTrigger);
|
return useQuery({
|
||||||
|
|
||||||
const query = useQuery({
|
|
||||||
queryKey: creditsKeys.stats(),
|
queryKey: creditsKeys.stats(),
|
||||||
queryFn: async () => {
|
queryFn: async () => {
|
||||||
console.log('Fetching credit stats...');
|
console.log('Fetching credit stats...');
|
||||||
@ -67,22 +51,11 @@ export function useCreditStats() {
|
|||||||
return result.data.data;
|
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
|
// Hook to consume credits
|
||||||
export function useConsumeCredits() {
|
export function useConsumeCredits() {
|
||||||
const queryClient = useQueryClient();
|
const queryClient = useQueryClient();
|
||||||
const triggerUpdate = useCreditsStore((state) => state.triggerUpdate);
|
|
||||||
|
|
||||||
return useMutation({
|
return useMutation({
|
||||||
mutationFn: async ({
|
mutationFn: async ({
|
||||||
@ -102,9 +75,6 @@ export function useConsumeCredits() {
|
|||||||
return result.data;
|
return result.data;
|
||||||
},
|
},
|
||||||
onSuccess: () => {
|
onSuccess: () => {
|
||||||
// Trigger credits update in store to notify all components
|
|
||||||
triggerUpdate();
|
|
||||||
|
|
||||||
// Invalidate credit balance and stats after consuming credits
|
// Invalidate credit balance and stats after consuming credits
|
||||||
queryClient.invalidateQueries({
|
queryClient.invalidateQueries({
|
||||||
queryKey: creditsKeys.balance(),
|
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