From 8d5ffc6fed1dab677365ab02738b7439d6f35be2 Mon Sep 17 00:00:00 2001
From: masongga <auckmason@gmail.com>
Date: Sat, 23 May 2026 19:53:18 -0600
Subject: [PATCH] daily quiz entries store in db correctly

---
 components/avatar/avatar-status-card.tsx      |  12 +-
 components/daily-quiz/daily-quiz-form.tsx     |  54 ++++---
 .../daily-quiz/daily-quiz-link-button.tsx     |   8 +-
 components/habits/habit-tracker.tsx           |   2 +-
 lib/avatar-state.ts                           |  39 ++++-
 lib/daily-quiz-storage.ts                     | 151 +++++++++++++-----
 lib/daily-tasks.ts                            |  15 +-
 lib/habit-catalog.ts                          |   2 +-
 lib/journal-storage.ts                        | 110 -------------
 .../20260523120000_journal_entries.sql        |  35 ----
 10 files changed, 197 insertions(+), 231 deletions(-)
 delete mode 100644 lib/journal-storage.ts
 delete mode 100644 supabase/migrations/20260523120000_journal_entries.sql

diff --git a/components/avatar/avatar-status-card.tsx b/components/avatar/avatar-status-card.tsx
index fe6e3a5..17de203 100644
--- a/components/avatar/avatar-status-card.tsx
+++ b/components/avatar/avatar-status-card.tsx
@@ -29,10 +29,14 @@ export function AvatarStatusCard() {
   const [isReady, setIsReady] = useState(false);
 
   useEffect(() => {
-    const checkedIn = hasCompletedDailyQuizToday();
-    setHasCheckedInToday(checkedIn);
-    setCondition(getAvatarConditionForToday() ?? defaultAvatarCondition);
-    setIsReady(true);
+    async function loadStatus() {
+      const checkedIn = await hasCompletedDailyQuizToday();
+      setHasCheckedInToday(checkedIn);
+      setCondition((await getAvatarConditionForToday()) ?? defaultAvatarCondition);
+      setIsReady(true);
+    }
+
+    void loadStatus();
   }, []);
 
   if (!isReady) {
diff --git a/components/daily-quiz/daily-quiz-form.tsx b/components/daily-quiz/daily-quiz-form.tsx
index eedb2b2..00fee4b 100644
--- a/components/daily-quiz/daily-quiz-form.tsx
+++ b/components/daily-quiz/daily-quiz-form.tsx
@@ -16,19 +16,18 @@ import {
 import { Label } from "@/components/ui/label";
 import {
   defaultDailyQuizAnswers,
+  formatSleepHours,
+  SLEEP_HOURS_MAX,
+  SLEEP_HOURS_MIN,
+  SLEEP_HOURS_STEP,
   WELLNESS_SCALE_MAX,
   type DailyQuizAnswers,
   type DailyQuizSubmission,
 } from "@/lib/avatar-state";
 import {
-  getDailyQuizSubmission,
-  getTodayDateKey,
-  saveDailyQuizSubmission,
+  getDailyEntryForToday,
+  saveDailyEntry,
 } from "@/lib/daily-quiz-storage";
-import {
-  getJournalEntryForToday,
-  saveJournalEntry,
-} from "@/lib/journal-storage";
 
 export function DailyQuizForm() {
   const [answers, setAnswers] = useState<DailyQuizAnswers>(
@@ -44,19 +43,23 @@ export function DailyQuizForm() {
 
   useEffect(() => {
     async function loadDailyCheckIn() {
-      const existingSubmission = getDailyQuizSubmission();
-      const existingJournal = await getJournalEntryForToday();
+      try {
+        const existingEntry = await getDailyEntryForToday();
 
-      if (existingSubmission?.date === getTodayDateKey()) {
-        setSubmission(existingSubmission);
-        setAnswers(existingSubmission.answers);
+        if (existingEntry) {
+          setSubmission(existingEntry);
+          setAnswers(existingEntry.answers);
+          setJournal(existingEntry.journal);
+        }
+      } catch (loadError) {
+        setError(
+          loadError instanceof Error
+            ? loadError.message
+            : "Could not load today's check-in.",
+        );
+      } finally {
+        setIsReady(true);
       }
-
-      if (existingJournal) {
-        setJournal(existingJournal.content);
-      }
-
-      setIsReady(true);
     }
 
     void loadDailyCheckIn();
@@ -84,9 +87,8 @@ export function DailyQuizForm() {
     setError(null);
 
     try {
-      const savedSubmission = saveDailyQuizSubmission(answers);
-      await saveJournalEntry(journal);
-      setSubmission(savedSubmission);
+      const savedEntry = await saveDailyEntry(answers, journal);
+      setSubmission(savedEntry);
     } catch (submitError) {
       setError(
         submitError instanceof Error
@@ -117,7 +119,7 @@ export function DailyQuizForm() {
         <CardHeader>
           <CardTitle className="text-base">Wellness</CardTitle>
           <CardDescription>
-            Sliders from 1 to 5 for how you&apos;re doing today.
+            Rate your wellness from 1 to 5. Sleep length uses hours.
           </CardDescription>
         </CardHeader>
         <CardContent className="space-y-5">
@@ -146,6 +148,10 @@ export function DailyQuizForm() {
             id="sleep-length"
             label="How long did you sleep?"
             value={answers.sleepLength}
+            min={SLEEP_HOURS_MIN}
+            max={SLEEP_HOURS_MAX}
+            step={SLEEP_HOURS_STEP}
+            formatValue={formatSleepHours}
             disabled={isCompleted}
             onChange={(value) => updateAnswer("sleepLength", value)}
           />
@@ -162,9 +168,7 @@ export function DailyQuizForm() {
       <Card>
         <CardHeader>
           <CardTitle className="text-base">Daily journal</CardTitle>
-          <CardDescription>
-            Write about your day. Saved separately from your quiz answers.
-          </CardDescription>
+          <CardDescription>Write about your day.</CardDescription>
         </CardHeader>
         <CardContent className="space-y-3">
           <Label htmlFor="journal" className="sr-only">
diff --git a/components/daily-quiz/daily-quiz-link-button.tsx b/components/daily-quiz/daily-quiz-link-button.tsx
index 0b63957..2dd6862 100644
--- a/components/daily-quiz/daily-quiz-link-button.tsx
+++ b/components/daily-quiz/daily-quiz-link-button.tsx
@@ -12,8 +12,12 @@ export function DailyQuizLinkButton() {
   const [isReady, setIsReady] = useState(false);
 
   useEffect(() => {
-    setIsCompleted(hasCompletedDailyQuizToday());
-    setIsReady(true);
+    async function loadStatus() {
+      setIsCompleted(await hasCompletedDailyQuizToday());
+      setIsReady(true);
+    }
+
+    void loadStatus();
   }, []);
 
   if (!isReady || isCompleted) {
diff --git a/components/habits/habit-tracker.tsx b/components/habits/habit-tracker.tsx
index 093c701..ad97611 100644
--- a/components/habits/habit-tracker.tsx
+++ b/components/habits/habit-tracker.tsx
@@ -168,7 +168,7 @@ export function HabitTracker({ mode = "daily" }: HabitTrackerProps) {
       setDailyTasks(tasks);
       setCustomHabits(customs);
       setFocusTopics(getProfilePreferences().focusTopics);
-      setQuizCompletedToday(hasCompletedDailyQuizToday());
+      setQuizCompletedToday(await hasCompletedDailyQuizToday());
       setIsReady(true);
     } catch (refreshError) {
       setError(
diff --git a/lib/avatar-state.ts b/lib/avatar-state.ts
index 46f617f..b65d46c 100644
--- a/lib/avatar-state.ts
+++ b/lib/avatar-state.ts
@@ -17,26 +17,49 @@ export type AvatarCondition = {
 export type DailyQuizSubmission = {
   date: string;
   answers: DailyQuizAnswers;
+  journal: string;
   condition: AvatarCondition;
 };
 
 export const WELLNESS_SCALE_MAX = 5;
 
+export const SLEEP_HOURS_MIN = 0.5;
+export const SLEEP_HOURS_MAX = 12;
+export const SLEEP_HOURS_STEP = 0.5;
+export const DEFAULT_SLEEP_HOURS = 7.5;
+
 export const defaultDailyQuizAnswers: DailyQuizAnswers = {
   feeling: 3,
   stress: 3,
   energy: 3,
-  sleepLength: 3,
+  sleepLength: DEFAULT_SLEEP_HOURS,
   sleepQuality: 3,
 };
 
+export function clampSleepHours(hours: number) {
+  const stepped = Math.round(hours / SLEEP_HOURS_STEP) * SLEEP_HOURS_STEP;
+  return Math.min(SLEEP_HOURS_MAX, Math.max(SLEEP_HOURS_MIN, stepped));
+}
+
+export function formatSleepHours(hours: number) {
+  return hours === 1 ? "1 hour" : `${hours} hours`;
+}
+
+function sleepHoursToWellnessScore(hours: number) {
+  const score =
+    1 + ((clampSleepHours(hours) - SLEEP_HOURS_MIN) / (SLEEP_HOURS_MAX - SLEEP_HOURS_MIN)) * 4;
+
+  return clampStat(Math.round(score));
+}
+
 export function computeAvatarCondition(
   answers: DailyQuizAnswers,
 ): AvatarCondition {
   const { feeling, stress, energy, sleepLength, sleepQuality } = answers;
 
   const wellnessAverage = (feeling + energy + (WELLNESS_SCALE_MAX + 1 - stress)) / 3;
-  const sleepAverage = (sleepLength + sleepQuality) / 2;
+  const sleepScore = sleepHoursToWellnessScore(sleepLength);
+  const sleepAverage = (sleepScore + sleepQuality) / 2;
   const health = Math.round((wellnessAverage + sleepAverage) / 2);
   const energyLevel = Math.round((energy + sleepQuality) / 2);
 
@@ -44,7 +67,7 @@ export function computeAvatarCondition(
 
   if (energyLevel >= 4 && feeling >= 4 && health >= 4) {
     avatarMood = "happy";
-  } else if (energy <= 2 || sleepLength <= 2 || sleepQuality <= 2) {
+  } else if (energy <= 2 || sleepLength < 6 || sleepQuality <= 2) {
     avatarMood = "tired";
   }
 
@@ -82,11 +105,19 @@ export function normalizeDailyQuizAnswers(
       : value;
   };
 
+  const normalizeSleepLength = (value: number | undefined) => {
+    if (value === undefined) {
+      return defaultDailyQuizAnswers.sleepLength;
+    }
+
+    return clampSleepHours(value);
+  };
+
   return {
     feeling: scaleValue(answers.feeling ?? answers.mood, defaultDailyQuizAnswers.feeling),
     stress: scaleValue(answers.stress, defaultDailyQuizAnswers.stress),
     energy: scaleValue(answers.energy, defaultDailyQuizAnswers.energy),
-    sleepLength: scaleValue(answers.sleepLength, defaultDailyQuizAnswers.sleepLength),
+    sleepLength: normalizeSleepLength(answers.sleepLength),
     sleepQuality: scaleValue(
       answers.sleepQuality,
       defaultDailyQuizAnswers.sleepQuality,
diff --git a/lib/daily-quiz-storage.ts b/lib/daily-quiz-storage.ts
index 73c1101..fa30185 100644
--- a/lib/daily-quiz-storage.ts
+++ b/lib/daily-quiz-storage.ts
@@ -6,69 +6,142 @@ import {
   type DailyQuizSubmission,
 } from "@/lib/avatar-state";
 import { notifyHabitPetDataUpdated } from "@/lib/app-events";
-const STORAGE_KEY = "habit-pet-daily-quiz";
+import { createClient } from "@/lib/supabase/client";
+
+type DailyEntryRow = {
+  entry_date: string;
+  mood: number;
+  stress: number;
+  energy: number;
+  sleep_hours: number;
+  sleep_quality: number;
+  journal: string | null;
+};
 
 export function getTodayDateKey(date = new Date()) {
   return date.toISOString().slice(0, 10);
 }
 
-function normalizeSubmission(
-  submission: DailyQuizSubmission,
-): DailyQuizSubmission {
-  const answers = normalizeDailyQuizAnswers(submission.answers);
+function mapRowToSubmission(row: DailyEntryRow): DailyQuizSubmission {
+  const answers = normalizeDailyQuizAnswers({
+    feeling: row.mood,
+    stress: row.stress,
+    energy: row.energy,
+    sleepLength: Number(row.sleep_hours),
+    sleepQuality: row.sleep_quality,
+  });
 
   return {
-    ...submission,
+    date: row.entry_date,
     answers,
+    journal: row.journal ?? "",
     condition: computeAvatarCondition(answers),
   };
 }
 
-export function getDailyQuizSubmission(): DailyQuizSubmission | null {
-  if (typeof window === "undefined") {
+async function getAuthenticatedUserId() {
+  const supabase = createClient();
+  const {
+    data: { user },
+    error,
+  } = await supabase.auth.getUser();
+
+  if (error || !user) {
     return null;
   }
 
-  const raw = window.localStorage.getItem(STORAGE_KEY);
-
-  if (!raw) {
-    return null;
-  }
-
-  try {
-    return normalizeSubmission(JSON.parse(raw) as DailyQuizSubmission);
-  } catch {
-    return null;
-  }
+  return user.id;
 }
 
-export function hasCompletedDailyQuizToday() {
-  const submission = getDailyQuizSubmission();
-  return submission?.date === getTodayDateKey();
+export async function getDailyEntryForToday(): Promise<DailyQuizSubmission | null> {
+  const userId = await getAuthenticatedUserId();
+
+  if (!userId) {
+    return null;
+  }
+
+  const supabase = createClient();
+  const entryDate = getTodayDateKey();
+  const { data, error } = await supabase
+    .from("daily_entries")
+    .select(
+      "entry_date, mood, stress, energy, sleep_hours, sleep_quality, journal",
+    )
+    .eq("user_id", userId)
+    .eq("entry_date", entryDate)
+    .maybeSingle();
+
+  if (error) {
+    throw new Error(error.message);
+  }
+
+  if (!data) {
+    return null;
+  }
+
+  return mapRowToSubmission(data as DailyEntryRow);
 }
 
-export function saveDailyQuizSubmission(
+export async function getDailyQuizSubmission() {
+  return getDailyEntryForToday();
+}
+
+export async function hasCompletedDailyQuizToday() {
+  const entry = await getDailyEntryForToday();
+  return entry !== null;
+}
+
+export async function saveDailyEntry(
   answers: DailyQuizAnswers,
-): DailyQuizSubmission {
-  const submission: DailyQuizSubmission = {
-    date: getTodayDateKey(),
-    answers,
-    condition: computeAvatarCondition(answers),
-  };
+  journal: string,
+): Promise<DailyQuizSubmission> {
+  const userId = await getAuthenticatedUserId();
 
-  window.localStorage.setItem(STORAGE_KEY, JSON.stringify(submission));
+  if (!userId) {
+    throw new Error("You must be signed in to save your daily check-in.");
+  }
 
+  const supabase = createClient();
+  const entryDate = getTodayDateKey();
+  const normalizedAnswers = normalizeDailyQuizAnswers(answers);
+
+  const { data, error } = await supabase
+    .from("daily_entries")
+    .upsert(
+      {
+        user_id: userId,
+        entry_date: entryDate,
+        mood: normalizedAnswers.feeling,
+        stress: normalizedAnswers.stress,
+        energy: normalizedAnswers.energy,
+        sleep_hours: normalizedAnswers.sleepLength,
+        sleep_quality: normalizedAnswers.sleepQuality,
+        journal,
+      },
+      { onConflict: "user_id,entry_date" },
+    )
+    .select(
+      "entry_date, mood, stress, energy, sleep_hours, sleep_quality, journal",
+    )
+    .single();
+
+  if (error) {
+    throw new Error(error.message);
+  }
+
+  const submission = mapRowToSubmission(data as DailyEntryRow);
   notifyHabitPetDataUpdated();
-
   return submission;
 }
 
-export function getAvatarConditionForToday(): AvatarCondition | null {
-  const submission = getDailyQuizSubmission();
-
-  if (!submission || submission.date !== getTodayDateKey()) {
-    return null;
-  }
-
-  return submission.condition;
+export async function saveDailyQuizSubmission(
+  answers: DailyQuizAnswers,
+  journal = "",
+) {
+  return saveDailyEntry(answers, journal);
+}
+
+export async function getAvatarConditionForToday(): Promise<AvatarCondition | null> {
+  const entry = await getDailyEntryForToday();
+  return entry?.condition ?? null;
 }
diff --git a/lib/daily-tasks.ts b/lib/daily-tasks.ts
index 9000aed..ef5ed6d 100644
--- a/lib/daily-tasks.ts
+++ b/lib/daily-tasks.ts
@@ -1,5 +1,5 @@
 import type { DailyQuizAnswers } from "@/lib/avatar-state";
-import { getDailyQuizSubmission, getTodayDateKey } from "@/lib/daily-quiz-storage";
+import { getDailyEntryForToday } from "@/lib/daily-quiz-storage";
 import {
   catalogHabitIds,
   getCatalogEntry,
@@ -23,14 +23,9 @@ export type DailyTask = Habit & {
   reason: DailyTaskReason;
 };
 
-function getQuizAnswersForToday(): DailyQuizAnswers | null {
-  const submission = getDailyQuizSubmission();
-
-  if (!submission || submission.date !== getTodayDateKey()) {
-    return null;
-  }
-
-  return submission.answers;
+async function getQuizAnswersForToday(): Promise<DailyQuizAnswers | null> {
+  const entry = await getDailyEntryForToday();
+  return entry?.answers ?? null;
 }
 
 function getCatalogTaskReason(
@@ -54,7 +49,7 @@ function getCatalogTaskReason(
 
 export async function getDailyTasks(): Promise<DailyTask[]> {
   const preferences = getProfilePreferences();
-  const quizAnswers = getQuizAnswersForToday();
+  const quizAnswers = await getQuizAnswersForToday();
 
   const selectedCatalogIds = habitCatalog
     .map((entry) => {
diff --git a/lib/habit-catalog.ts b/lib/habit-catalog.ts
index 021235e..ee4db1e 100644
--- a/lib/habit-catalog.ts
+++ b/lib/habit-catalog.ts
@@ -50,7 +50,7 @@ export const habitCatalog: HabitCatalogEntry[] = [
     label: "Wind down before bed",
     focusTopics: ["Sleep"],
     suggestedByQuiz: (answers) =>
-      answers.sleepLength <= 2 || answers.sleepQuality <= 2,
+      answers.sleepLength < 7 || answers.sleepQuality <= 2,
   },
 ];
 
diff --git a/lib/journal-storage.ts b/lib/journal-storage.ts
deleted file mode 100644
index 2e1f757..0000000
--- a/lib/journal-storage.ts
+++ /dev/null
@@ -1,110 +0,0 @@
-import { createClient } from "@/lib/supabase/client";
-
-export type JournalEntry = {
-  entryDate: string;
-  content: string;
-};
-
-const LOCAL_STORAGE_KEY = "habit-pet-journal";
-
-export function getTodayDateKey(date = new Date()) {
-  return date.toISOString().slice(0, 10);
-}
-
-function getLocalJournalEntry(entryDate = getTodayDateKey()): JournalEntry | null {
-  if (typeof window === "undefined") {
-    return null;
-  }
-
-  const raw = window.localStorage.getItem(LOCAL_STORAGE_KEY);
-
-  if (!raw) {
-    return null;
-  }
-
-  try {
-    const entry = JSON.parse(raw) as JournalEntry;
-    return entry.entryDate === entryDate ? entry : null;
-  } catch {
-    return null;
-  }
-}
-
-function saveLocalJournalEntry(content: string, entryDate = getTodayDateKey()) {
-  const entry: JournalEntry = { entryDate, content };
-  window.localStorage.setItem(LOCAL_STORAGE_KEY, JSON.stringify(entry));
-  return entry;
-}
-
-export async function getJournalEntryForToday(): Promise<JournalEntry | null> {
-  const entryDate = getTodayDateKey();
-  const supabase = createClient();
-  const {
-    data: { user },
-  } = await supabase.auth.getUser();
-
-  if (!user) {
-    return getLocalJournalEntry(entryDate);
-  }
-
-  const { data, error } = await supabase
-    .from("journal_entries")
-    .select("entry_date, content")
-    .eq("user_id", user.id)
-    .eq("entry_date", entryDate)
-    .maybeSingle();
-
-  if (error) {
-    console.error("Failed to load journal entry:", error.message);
-    return getLocalJournalEntry(entryDate);
-  }
-
-  if (!data) {
-    return null;
-  }
-
-  return {
-    entryDate: data.entry_date,
-    content: data.content,
-  };
-}
-
-export async function saveJournalEntry(content: string): Promise<JournalEntry> {
-  const entryDate = getTodayDateKey();
-  const supabase = createClient();
-  const {
-    data: { user },
-  } = await supabase.auth.getUser();
-
-  if (!user) {
-    return saveLocalJournalEntry(content, entryDate);
-  }
-
-  const { data, error } = await supabase
-    .from("journal_entries")
-    .upsert(
-      {
-        user_id: user.id,
-        entry_date: entryDate,
-        content,
-        updated_at: new Date().toISOString(),
-      },
-      { onConflict: "user_id,entry_date" },
-    )
-    .select("entry_date, content")
-    .single();
-
-  if (error) {
-    console.error("Failed to save journal entry:", error.message);
-    return saveLocalJournalEntry(content, entryDate);
-  }
-
-  return {
-    entryDate: data.entry_date,
-    content: data.content,
-  };
-}
-
-export function hasJournalEntryToday(entry: JournalEntry | null) {
-  return entry?.entryDate === getTodayDateKey() && entry.content.trim().length > 0;
-}
diff --git a/supabase/migrations/20260523120000_journal_entries.sql b/supabase/migrations/20260523120000_journal_entries.sql
deleted file mode 100644
index fbf645a..0000000
--- a/supabase/migrations/20260523120000_journal_entries.sql
+++ /dev/null
@@ -1,35 +0,0 @@
-create table if not exists public.journal_entries (
-  id uuid primary key default gen_random_uuid(),
-  user_id uuid not null references auth.users (id) on delete cascade,
-  entry_date date not null,
-  content text not null default '',
-  created_at timestamptz not null default now(),
-  updated_at timestamptz not null default now(),
-  unique (user_id, entry_date)
-);
-
-create index if not exists journal_entries_user_id_entry_date_idx
-  on public.journal_entries (user_id, entry_date desc);
-
-alter table public.journal_entries enable row level security;
-
-create policy "Users can read own journal entries"
-  on public.journal_entries
-  for select
-  using (auth.uid() = user_id);
-
-create policy "Users can insert own journal entries"
-  on public.journal_entries
-  for insert
-  with check (auth.uid() = user_id);
-
-create policy "Users can update own journal entries"
-  on public.journal_entries
-  for update
-  using (auth.uid() = user_id)
-  with check (auth.uid() = user_id);
-
-create policy "Users can delete own journal entries"
-  on public.journal_entries
-  for delete
-  using (auth.uid() = user_id);