From 491824bd4e403b3259df7df8289f303ff95be96a Mon Sep 17 00:00:00 2001
From: Jake Langlois <langlois.j@northeastern.edu>
Date: Sun, 17 Nov 2024 16:21:05 -0500
Subject: [PATCH] capped years at 5

---
 packages/frontend/components/Plan/AddYearButton.tsx   | 6 +++++-
 packages/frontend/utils/plan/addEmptyDndYearToPlan.ts | 1 +
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/frontend/components/Plan/AddYearButton.tsx b/packages/frontend/components/Plan/AddYearButton.tsx
index b8ec65274..a81d4071a 100644
--- a/packages/frontend/components/Plan/AddYearButton.tsx
+++ b/packages/frontend/components/Plan/AddYearButton.tsx
@@ -17,7 +17,11 @@ export const AddYearButton: React.FC<AddYearButton> = ({
     mutateStudentWithUpdatedPlan(updatedPlan);
   };
   return (
-    <BlueButton onClick={addYear} leftIcon={<AddIcon />}>
+    <BlueButton
+      onClick={addYear}
+      leftIcon={<AddIcon />}
+      isDisabled={plan.schedule.years.length >= 5}
+    >
       Add Year
     </BlueButton>
   );
diff --git a/packages/frontend/utils/plan/addEmptyDndYearToPlan.ts b/packages/frontend/utils/plan/addEmptyDndYearToPlan.ts
index 73b239c3f..05cafc633 100644
--- a/packages/frontend/utils/plan/addEmptyDndYearToPlan.ts
+++ b/packages/frontend/utils/plan/addEmptyDndYearToPlan.ts
@@ -7,6 +7,7 @@ import { getCourseCount, prepareYearForDnd } from "./preparePlanForDnd";
 export const addEmptyDndYearToPlan = (
   plan: PlanModel<string>
 ): PlanModel<string> => {
+  // Max 5 years
   const updatedPlan = produce(plan, (draftPlan) => {
     const schedule = draftPlan.schedule;
     const newYearNum = schedule.years.length + 1;