GPT

Custom Hooks for Firestore Operations

import { useDocumentQuery, useCollectionQuery } from '@tanstack-query-firebase/react/firestore'; import { useSubscription } from '@tanstack-query-firebase/react/firestore'; import { useMutation, useQueryClient } from '@t

Prompt text

Original English text. Replace placeholders with your own details.

import { useDocumentQuery, useCollectionQuery } from '@tanstack-query-firebase/react/firestore';
import { useSubscription } from '@tanstack-query-firebase/react/firestore';
import { useMutation, useQueryClient } from '@tanstack/react-query';
import { doc, collection, query, where, orderBy, setDoc, deleteDoc, DocumentData } from 'firebase/firestore';
import { db } from '../lib/firebase';

// Document hooks
export function useDocument<T = DocumentData>(path: string, id: string) {
 const docRef = doc(db, path, id);
 return useDocumentQuery<T>(docRef, {
 queryKey: [path, id],
 });
}

export function useDocumentRealtime<T = DocumentData>(path: string, id: string) {
 const docRef = doc(db, path, id);
 return useSubscription<T>(docRef, {
 queryKey: [path, id],
 });
}

// Collection hooks
export function useCollection<T = DocumentData>(
 path: string,
 constraints?: {
 where?: [string, firebase.firestore.WhereFilterOp, any][];
 orderBy?: [string, 'asc' | 'desc'][];
 limit?: number;
 }
) {
 let collectionRef = collection(db, path);
 
 if (constraints) {
 let q = query(collectionRef);
 
 if (constraints.where) {
 constraints.where.forEach(([field, op, value]) => {
 q = query(q, where(field, op, value));
 });
 }
 
 if (constraints.orderBy) {
 constraints.orderBy.forEach(([field, direction]) => {
 q = query(q, orderBy(field, direction));
 });
 }
 
 collectionRef = q;
 }

 return useCollectionQuery<T>(collectionRef, {
 queryKey: [path, constraints],
 });
}

export function useCollectionRealtime<T = DocumentData>(
 path: string,
 constraints?: {
 where?: [string, firebase.firestore.WhereFilterOp, any][];
 orderBy?: [string, 'asc' | 'desc'][];
 limit?: number;
 }
) {
 let collectionRef = collection(db, path);
 
 if (constraints) {
 let q = query(collectionRef);
 
 if (constraints.where) {
 constraints.where.forEach(([field, op, value]) => {
 q = query(q, where(field, op, value));
 });
 }
 
 if (constraints.orderBy) {
 constraints.orderBy.forEach(([field, direction]) => {
 q = query(q, orderBy(field, direction));
 });
 }
 
 collectionRef = q;
 }

 return useSubscription<T>(collectionRef, {
 queryKey: [path, constraints],
 });
}

// Mutation hooks
export function useDocumentMutation<T = DocumentData>(path: string) {
 const queryClient = useQueryClient();

 return useMutation({
 mutationFn: async ({ id, data }: { id: string; data: T }) => {
 const docRef = doc(db, path, id);
 await setDoc(docRef, data, { merge: true });
 return { id, data };
 },
 onSuccess: (_, { id }) => {
 queryClient.invalidateQueries([path, id]);
 queryClient.invalidateQueries([path]);
 }
 });
}

export function useDocumentDelete(path: string) {
 const queryClient = useQueryClient();

 return useMutation({
 mutationFn: async (id: string) => {
 const docRef = doc(db, path, id);
 await deleteDoc(docRef);
 return id;
 },
 onSuccess: (id) => {
 queryClient.invalidateQueries([path, id]);
 queryClient.invalidateQueries([path]);
 }
 });
}

Source & attribution

Instructa AI Prompts · Instructa AI Prompts

Source-listed license: CC-BY-4.0. Prompt text is preserved; titles and previews may be shortened for display. Source examples are not NexGateHub test results.

View original source ↗ · Source credits & licenses

Explore more prompts →