From a60d2faee5c8ce08a86e4c6d115954289b5e829d Mon Sep 17 00:00:00 2001
From: "U-THOMASB\\thoma" <t.ferb1@gmail.com>
Date: Fri, 16 Aug 2024 15:58:04 +0200
Subject: [PATCH] fix: diploma creation for existing programs

---
 .../[location]/diplomas/_actions/fetch.ts     | 29 ++++++++++++-
 .../_components/create-dialog-client.tsx      | 42 ++++++++++++++++++-
 apps/web/src/lib/nwd.ts                       | 24 ++++++++---
 .../core/src/models/student/curriculum.ts     | 27 ++++++++++++
 4 files changed, 114 insertions(+), 8 deletions(-)

diff --git a/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_actions/fetch.ts b/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_actions/fetch.ts
index bb1daa25..110d86ea 100644
--- a/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_actions/fetch.ts
+++ b/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_actions/fetch.ts
@@ -1,6 +1,11 @@
 "use server";
 
-import { listCurriculaByProgram, listGearTypesByCurriculum } from "~/lib/nwd";
+import {
+  listCompletedCompetenciesByStudentCurriculumId,
+  listCurriculaByProgram,
+  listGearTypesByCurriculum,
+  retrieveStudentCurriculumByPersonIdAndCurriculumId,
+} from "~/lib/nwd";
 
 export async function getCurriculaByProgram(programId: string) {
   return await listCurriculaByProgram(programId, true);
@@ -9,3 +14,25 @@ export async function getCurriculaByProgram(programId: string) {
 export async function getGearTypesByCurriculum(curriculumId: string) {
   return await listGearTypesByCurriculum(curriculumId);
 }
+
+export async function getCompletedCompetencies(
+  personId: string,
+  curriculumId: string,
+  gearTypeId: string,
+) {
+  const studentCurriculum =
+    await retrieveStudentCurriculumByPersonIdAndCurriculumId(
+      personId,
+      curriculumId,
+      gearTypeId,
+    );
+
+  if (!studentCurriculum) {
+    return null;
+  }
+
+  const completedCompetencies =
+    await listCompletedCompetenciesByStudentCurriculumId(studentCurriculum.id);
+
+  return completedCompetencies.map((cc) => cc.competencyId);
+}
diff --git a/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_components/create-dialog-client.tsx b/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_components/create-dialog-client.tsx
index fa64b5c8..0d04e0b3 100644
--- a/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_components/create-dialog-client.tsx
+++ b/apps/web/src/app/(dashboard)/(management)/locatie/[location]/diplomas/_components/create-dialog-client.tsx
@@ -46,6 +46,7 @@ import type {
 } from "~/lib/nwd";
 import { createCertificate } from "../_actions/create";
 import {
+  getCompletedCompetencies,
   getCurriculaByProgram,
   getGearTypesByCurriculum,
 } from "../_actions/fetch";
@@ -93,16 +94,26 @@ function CreateDialogClient({
       setIsOpen(false);
     }
 
+    if (result.message === "Error" && Object.keys(result.errors).length < 1) {
+      toast.error("Er is iets misgegaan");
+    }
+
     return result;
   };
 
   const [state, formAction] = useActionState(submit, undefined);
 
+  const [selectedPerson, setSelectedPerson] = useState<
+    (typeof persons)[number] | null
+  >(null);
   const [selectedProgram, setSelectedProgram] = useState<string | null>(null);
   const [selectedGearType, setSelectedGearType] = useState<string | null>(null);
   const [selectedCurriculum, setSelectedCurriculum] = useState<
     Awaited<ReturnType<typeof listCurriculaByProgram>>[number] | null
   >(null);
+  const [completedCompetencies, setCompletedCompetencies] = useState<Awaited<
+    ReturnType<typeof getCompletedCompetencies>
+  > | null>(null);
 
   const [programQuery, setProgramQuery] = useState("");
   const [personQuery, setPersonQuery] = useState("");
@@ -131,6 +142,7 @@ function CreateDialogClient({
       const [curriculum] = await getCurriculaByProgram(selectedProgram);
 
       if (!curriculum) {
+        setCompletedCompetencies(null);
         setSelectedCurriculum(null);
         setGearTypes([]);
         return;
@@ -144,6 +156,24 @@ function CreateDialogClient({
     void fetchCurricula();
   }, [selectedProgram]);
 
+  useEffect(() => {
+    async function fetchCompletedCompetencies() {
+      if (!selectedPerson || !selectedCurriculum || !selectedGearType) {
+        return;
+      }
+
+      const completedCompetencies = await getCompletedCompetencies(
+        selectedPerson.id,
+        selectedCurriculum.id,
+        selectedGearType,
+      );
+
+      setCompletedCompetencies(completedCompetencies ?? []);
+    }
+
+    void fetchCompletedCompetencies();
+  }, [selectedPerson, selectedCurriculum, selectedGearType]);
+
   return (
     <>
       <Button
@@ -171,6 +201,8 @@ function CreateDialogClient({
                     onClose={() => setPersonQuery("")}
                     virtual={{ options: filteredPersons }}
                     multiple={false}
+                    value={selectedPerson}
+                    onChange={(value) => setSelectedPerson(value)}
                   >
                     <div
                       className={clsx([
@@ -369,6 +401,7 @@ function CreateDialogClient({
                         setSelectedGearType(null);
                         setGearTypes([]);
                         setSelectedCurriculum(null);
+                        setCompletedCompetencies(null);
                       }}
                       invalid={!!state?.errors.curriculumId}
                     >
@@ -414,7 +447,7 @@ function CreateDialogClient({
 
                 <Fieldset>
                   <Legend>Modules</Legend>
-                  {selectedCurriculum ? (
+                  {selectedCurriculum && completedCompetencies !== null ? (
                     <div className="grid grid-cols-1 lg:grid-cols-2 mt-2 gap-x-4">
                       <CheckboxGroup>
                         <Legend>Kernmodules</Legend>
@@ -422,7 +455,12 @@ function CreateDialogClient({
                           .sort((a, b) => a.weight - b.weight)
                           .filter((module) => module.isRequired)
                           .map((module) => (
-                            <CheckboxField key={module.id}>
+                            <CheckboxField
+                              key={module.id}
+                              disabled={module.competencies.every((c) =>
+                                completedCompetencies?.includes(c.id),
+                              )}
+                            >
                               <Checkbox
                                 name="competencies[]"
                                 value={module.competencies
diff --git a/apps/web/src/lib/nwd.ts b/apps/web/src/lib/nwd.ts
index 1a1e54d5..5cfe1db7 100644
--- a/apps/web/src/lib/nwd.ts
+++ b/apps/web/src/lib/nwd.ts
@@ -746,11 +746,13 @@ export const createCompletedCertificate = async (
       });
 
       // Start student curriculum
-      const { id: studentCurriculumId } = await Student.Curriculum.start({
-        curriculumId,
-        personId,
-        gearTypeId,
-      });
+      const { id: studentCurriculumId } = await Student.Curriculum.findOrEnroll(
+        {
+          curriculumId,
+          personId,
+          gearTypeId,
+        },
+      );
 
       // Start certificate
       const { id: certificateId } = await Student.Certificate.startCertificate({
@@ -1341,6 +1343,18 @@ export const retrieveStudentAllocationWithCurriculumForPerson = cache(
   },
 );
 
+export const retrieveStudentCurriculumByPersonIdAndCurriculumId = cache(
+  async (personId: string, curriculumId: string, gearTypeId: string) => {
+    return makeRequest(async () => {
+      return await Student.Curriculum.find({
+        personId,
+        curriculumId,
+        gearTypeId,
+      });
+    });
+  },
+);
+
 export const listCompletedCompetenciesByStudentCurriculumId = cache(
   async (studentCurriculumId: string) => {
     return makeRequest(async () => {
diff --git a/packages/core/src/models/student/curriculum.ts b/packages/core/src/models/student/curriculum.ts
index f1bd481d..2e7318ac 100644
--- a/packages/core/src/models/student/curriculum.ts
+++ b/packages/core/src/models/student/curriculum.ts
@@ -41,6 +41,33 @@ export const start = withZod(
   },
 )
 
+export const find = withZod(
+  insertSchema.pick({
+    personId: true,
+    curriculumId: true,
+    gearTypeId: true,
+  }),
+  successfulCreateResponse.nullable(),
+  async (input) => {
+    const query = useQuery()
+
+    const studentCurriculum = await query
+      .select({ id: s.studentCurriculum.id })
+      .from(s.studentCurriculum)
+      .where(
+        and(
+          eq(s.studentCurriculum.personId, input.personId),
+          eq(s.studentCurriculum.curriculumId, input.curriculumId),
+          eq(s.studentCurriculum.gearTypeId, input.gearTypeId),
+          isNull(s.studentCurriculum.deletedAt),
+        ),
+      )
+      .then(possibleSingleRow)
+
+    return studentCurriculum ?? null
+  },
+)
+
 export const findOrEnroll = withZod(
   insertSchema.pick({
     personId: true,