From 628a8e2d4307886850ecef9b473bb82a281c000c Mon Sep 17 00:00:00 2001
From: Vincent Shuali <vincent.shuali@gmail.com>
Date: Thu, 9 Jan 2025 21:11:24 -0800
Subject: [PATCH] Restore getBenchmarkAndTrialData to also grab task, at least
 temporarily

---
 src/backend/routers/iep.ts                    | 16 +++++++++++++---
 src/pages/benchmarks/[benchmark_id]/index.tsx |  2 +-
 2 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/src/backend/routers/iep.ts b/src/backend/routers/iep.ts
index 06d94c12..44fb5cca 100644
--- a/src/backend/routers/iep.ts
+++ b/src/backend/routers/iep.ts
@@ -393,18 +393,21 @@ export const iep = router({
   getBenchmarkAndTrialData: hasPara
     .input(
       z.object({
-        benchmark_id: z.string(),
+        task_id: z.string(),
       })
     )
     .query(async (req) => {
-      const { benchmark_id } = req.input;
+      const { task_id } = req.input;
 
       const result = await req.ctx.db
         .selectFrom("benchmark")
+        .innerJoin("task", "benchmark.benchmark_id", "task.benchmark_id")
         .innerJoin("goal", "benchmark.goal_id", "goal.goal_id")
         .innerJoin("iep", "goal.iep_id", "iep.iep_id")
         .innerJoin("student", "iep.student_id", "student.student_id")
+        .where("task.task_id", "=", task_id)
         .select((eb) => [
+          "task.task_id",
           "benchmark.benchmark_id",
           "student.first_name",
           "student.last_name",
@@ -414,6 +417,9 @@ export const iep = router({
           "benchmark.frequency",
           "benchmark.number_of_trials",
           "benchmark.benchmark_id",
+          "task.due_date",
+          "task.seen",
+          "task.trial_count",
           jsonArrayFrom(
             eb
               .selectFrom("trial_data")
@@ -440,10 +446,14 @@ export const iep = router({
                 "=",
                 "benchmark.benchmark_id"
               )
+              .whereRef(
+                "trial_data.created_by_user_id",
+                "=",
+                "task.assignee_id"
+              )
               .orderBy("trial_data.created_at")
           ).as("trials"),
         ])
-        .where("benchmark.benchmark_id", "=", benchmark_id)
         .executeTakeFirstOrThrow();
 
       return result;
diff --git a/src/pages/benchmarks/[benchmark_id]/index.tsx b/src/pages/benchmarks/[benchmark_id]/index.tsx
index e3706ca6..e3b08a0f 100644
--- a/src/pages/benchmarks/[benchmark_id]/index.tsx
+++ b/src/pages/benchmarks/[benchmark_id]/index.tsx
@@ -35,7 +35,7 @@ const BenchmarkPage = () => {
     isError,
   } = trpc.iep.getBenchmarkAndTrialData.useQuery(
     {
-      task_id: benchmark_id as string,
+      task_id: benchmark_id as string, // how does this line make sense?
     },
     {
       enabled: Boolean(benchmark_id),