Momento/memento-note/lib/query-hooks.ts

'use client'

import { useQueryClient, useQuery } from '@tanstack/react-query'
import { queryKeys } from './query-keys'
import type { Notebook, Label } from '@/lib/types'

// Re-export query keys
export { queryKeys }

// ===== useNotebooks =====
export function useNotebooksQuery() {
  return useQuery({
    queryKey: queryKeys.notebooks(),
    queryFn: async (): Promise<Notebook[]> => {
      const res = await fetch('/api/notebooks', { cache: 'no-store', credentials: 'include' })
      const data = await res.json()
      return data.notebooks || []
    },
  })
}

// ===== useLabels =====
export function useLabelsQuery(notebookId?: string | null) {
  return useQuery({
    queryKey: queryKeys.labels(notebookId),
    queryFn: async (): Promise<Label[]> => {
      const url = new URL('/api/labels', window.location.origin)
      if (notebookId) url.searchParams.set('notebookId', notebookId)
      const res = await fetch(url.toString(), { cache: 'no-store', credentials: 'include' })
      const data = await res.json()
      return data.data || []
    },
  })
}

// ===== invalidateHelpers =====
export function invalidateNotes(queryClient: ReturnType<typeof useQueryClient>, notebookId?: string | null) {
  queryClient.invalidateQueries({ queryKey: queryKeys.notes(notebookId) })
}

export function invalidateNotebooks(queryClient: ReturnType<typeof useQueryClient>) {
  queryClient.invalidateQueries({ queryKey: queryKeys.notebooks() })
}

export function invalidateLabels(queryClient: ReturnType<typeof useQueryClient>, notebookId?: string | null) {
  queryClient.invalidateQueries({ queryKey: queryKeys.labels(notebookId) })
}