Initial commit

frontend/src/hooks/useSleepSession.ts

@@ -0,0 +1,73 @@
+import { useQuery, useMutation, useQueryClient } from '@tanstack/react-query';
+import { fetchActiveSession, createSession, endSession, fetchSessions, deleteSession, updateSession } from '../api/client';
+import type { SleepSessionRead } from '../types';
+
+export function useSleepSession() {
+  const queryClient = useQueryClient();
+
+  const activeSessionQuery = useQuery<SleepSessionRead | null>({
+    queryKey: ['activeSession'],
+    queryFn: fetchActiveSession,
+    refetchInterval: 30000, // Poll every 30 seconds
+  });
+
+  const createMutation = useMutation({
+    mutationFn: createSession,
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ['activeSession'] });
+      queryClient.invalidateQueries({ queryKey: ['sessions'] });
+    },
+  });
+
+  const endMutation = useMutation({
+    mutationFn: ({ sessionId, endTime }: { sessionId: number; endTime: string }) =>
+      endSession(sessionId, { end_time: endTime }),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ['activeSession'] });
+      queryClient.invalidateQueries({ queryKey: ['sessions'] });
+    },
+  });
+
+  return {
+    activeSession: activeSessionQuery.data,
+    isLoading: activeSessionQuery.isLoading,
+    isFetching: activeSessionQuery.isFetching,
+    createSession: createMutation.mutateAsync,
+    endSession: endMutation.mutateAsync,
+    isCreating: createMutation.isPending,
+    isEnding: endMutation.isPending,
+  };
+}
+
+export function useSessions() {
+  const queryClient = useQueryClient();
+
+  const sessionsQuery = useQuery({
+    queryKey: ['sessions'],
+    queryFn: () => fetchSessions(),
+  });
+
+  const deleteMutation = useMutation({
+    mutationFn: deleteSession,
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ['sessions'] });
+    },
+  });
+
+  const updateMutation = useMutation({
+    mutationFn: ({ sessionId, payload }: { sessionId: number; payload: any }) => 
+      updateSession(sessionId, payload),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ['sessions'] });
+    },
+  });
+
+  return {
+    sessions: sessionsQuery.data?.sessions || [],
+    total: sessionsQuery.data?.total || 0,
+    isLoading: sessionsQuery.isLoading,
+    isDeleting: deleteMutation.isPending,
+    deleteSession: deleteMutation.mutateAsync,
+    updateSession: updateMutation.mutateAsync,
+  };
+}