Improve challenge navigation, query optimization, and fix delete goal

app/(tabs)/challenges.tsx

@@ -12,7 +12,7 @@ import {
   isBefore,
   isAfter,
 } from "date-fns";
-import { Link } from "expo-router";
+import { Link, router } from "expo-router";
 
 const getChallengeStatus = (startDate: Date, endDate: Date) => {
   const currentDate = new Date();
@@ -52,11 +52,13 @@ export default function ChallengesScreen() {
         </Text>
 
         {/* TODO: Allow only admin to create challenges */}
-        <Link href="/challenges/create" asChild>
-          <Button size="icon" className="h-14 w-14 rounded-full">
-            <Plus color="#fff" size={30} />
-          </Button>
-        </Link>
+        <Button 
+          size="icon" 
+          className="h-14 w-14 rounded-full"
+          onPress={() => router.push("/challenges/create")}
+        >
+          <Plus color="#fff" size={30} />
+        </Button>
 
         <FlatList
           contentContainerStyle={{

app/challenges/[id]/index.tsx

@@ -37,10 +37,7 @@ export default function ChallengeScreen() {
   const createGoal = useMutation(api.goals.createGoal);
   const deleteGoalAndGoalLogs = useMutation(api.goals.deleteGoalAndGoalLogs);
   const userGoals = useQuery(api.goals.listGoals);
-  const challengeGoals = useQuery(api.challengeGoals.listChallengeGoals);
-  const filteredChallengeGoals = challengeGoals?.filter(
-    (goal) => goal.challengeId === id
-  );
+  const filteredChallengeGoals = useQuery(api.challengeGoals.listChallengeGoalsById, { goalId: id });
   const createGoalLogsFromGoal = useMutation(
     api.goalLogs.createGoalLogsFromGoal
   );

app/challenges/create/index.tsx

@@ -93,23 +93,25 @@ function CreateChallengeForm() {
 
       <Button
         onPress={async () => {
-          const newChallenge = {
-            name,
-            description,
-            repeat: ["daily"],
-            unitType: "General",
-            unitValue: 1,
-            unit: "times",
-            recurrence: "per day",
-            startDate: startDate.getTime(),
-            endDate: endDate.getTime(),
-            points: points ? parseInt(points) : 0,
-          };
-          const challengeId = await createChallenge(newChallenge);
-          if (!challengeId) {
+          try {
+            const newChallenge = {
+              name,
+              description,
+              repeat: ["daily"],
+              unitType: "General",
+              unitValue: 1,
+              unit: "times",
+              recurrence: "per day",
+              startDate: startDate.getTime(),
+              endDate: endDate.getTime(),
+              points: points ? parseInt(points) : 0,
+            };
+
+            await createChallenge(newChallenge);
+            router.dismiss();
+          } catch (error) {
             throw new Error("Failed to create challenge");
           }
-          router.navigate("/challenges");
         }}
       >
         <Text>Add a sample challenge</Text>

app/goals/[goalId]/[goalLogId]/index.tsx

@@ -132,8 +132,8 @@ export default function GoalScreen() {
         {
           text: "Yes",
           onPress: async () => {
+            router.dismiss(); // Dismiss first, page will break if goal does not exist before exit. Read nonexistent data
             await deleteGoalAndGoalLogs({ goalId });
-            router.dismiss();
           },
           style: "destructive",
         },

convex/challengeGoals.ts

@@ -47,3 +47,19 @@ export const listChallengeGoals = query({
     return goals;
   },
 });
+
+export const listChallengeGoalsById = query({
+  args: { goalId: v.id("challenges") },
+  handler: async (ctx, { goalId }) => {
+    const userId = await getAuthUserId(ctx);
+    if (userId === null) {
+      return null;
+    }
+    const goals = await ctx.db
+    .query("challengeGoals")
+    .filter((q) => q.eq(q.field("challengeId"), goalId))
+    .collect();
+
+    return goals;
+  },
+});