From 9ef979ebd29b16db96b90a04a1ffe7d196ef4fa0 Mon Sep 17 00:00:00 2001
From: CREDO23 <thierrybakera12@gmail.com>
Date: Wed, 2 Oct 2024 20:25:06 +0200
Subject: [PATCH] fix dropdowns in 'See plan' modal

---
 apps/web/lib/components/modal.tsx             |  7 +++-
 .../add-task-estimation-hours-modal.tsx       | 41 +++++++++----------
 .../features/daily-plan/all-plans-modal.tsx   |  7 +++-
 3 files changed, 31 insertions(+), 24 deletions(-)

diff --git a/apps/web/lib/components/modal.tsx b/apps/web/lib/components/modal.tsx
index 33ea9eed9..84985fc01 100644
--- a/apps/web/lib/components/modal.tsx
+++ b/apps/web/lib/components/modal.tsx
@@ -11,6 +11,7 @@ type Props = {
 	description?: string;
 	isOpen: boolean;
 	closeModal: () => void;
+	customCloseModal?: () => void;
 	className?: string;
 	alignCloseIcon?: boolean;
 	showCloseIcon?: boolean;
@@ -19,6 +20,7 @@ type Props = {
 export function Modal({
 	isOpen,
 	closeModal,
+	customCloseModal,
 	children,
 	title,
 	titleClass,
@@ -54,7 +56,10 @@ export function Modal({
 						{description && <Dialog.Description>{description}</Dialog.Description>}
 						{showCloseIcon && (
 							<div
-								onClick={closeModal}
+								onClick={() => {
+									closeModal();
+									customCloseModal?.();
+								}}
 								className={`absolute ${
 									alignCloseIcon ? 'right-2 top-3' : 'right-3 top-3'
 								}  md:right-2 md:top-3 cursor-pointer z-50`}
diff --git a/apps/web/lib/features/daily-plan/add-task-estimation-hours-modal.tsx b/apps/web/lib/features/daily-plan/add-task-estimation-hours-modal.tsx
index 8914ae272..69a8b0184 100644
--- a/apps/web/lib/features/daily-plan/add-task-estimation-hours-modal.tsx
+++ b/apps/web/lib/features/daily-plan/add-task-estimation-hours-modal.tsx
@@ -89,13 +89,6 @@ export function AddTasksEstimationHoursModal(props: IAddTasksEstimationHoursModa
 		draft: false,
 		saved: false
 	});
-	const isTomorrowPlan = useMemo(
-		() =>
-			plan &&
-			new Date(moment().add(1, 'days').toDate()).toLocaleDateString('en') ==
-				new Date(plan.date).toLocaleDateString('en'),
-		[plan]
-	);
 
 	const canStartWorking = useMemo(() => {
 		const isTodayPlan =
@@ -260,8 +253,8 @@ export function AddTasksEstimationHoursModal(props: IAddTasksEstimationHoursModa
 	const StartWorkingButton = (
 		<Button
 			disabled={
-				(!isTomorrowPlan ? warning : false) || loading || timerStatus?.running
-					? planEditState.draft
+				(canStartWorking && warning) || loading || (canStartWorking && timerStatus?.running)
+					? planEditState.draft && !warning
 						? false
 						: true
 					: false
@@ -270,7 +263,7 @@ export function AddTasksEstimationHoursModal(props: IAddTasksEstimationHoursModa
 			type="submit"
 			className={clsxm(
 				'py-3 px-5 w-full  rounded-md font-light flex items-center justify-center text-md dark:text-white',
-				(!isTomorrowPlan ? warning : false) && 'bg-gray-400'
+				canStartWorking && warning && 'bg-gray-400'
 			)}
 			onClick={handleSubmit}
 		>
@@ -704,21 +697,13 @@ function TaskCard(props: ITaskCardProps) {
 			if (plan && plan.id) {
 				await addTaskToPlan({ taskId: task.id }, plan.id);
 			} else {
-				if (plan) {
-					await createDailyPlan({
-						workTimePlanned: 0,
-						taskId: task.id,
-						date: new Date(moment(plan.date).format('YYYY-MM-DD')),
-						status: DailyPlanStatusEnum.OPEN,
-						tenantId: user?.tenantId ?? '',
-						employeeId: user?.employee.id,
-						organizationId: user?.employee.organizationId
-					});
-				} else if (selectedDate) {
+				const planDate = plan ? plan.date : selectedDate;
+
+				if (planDate) {
 					await createDailyPlan({
 						workTimePlanned: 0,
 						taskId: task.id,
-						date: new Date(moment(selectedDate).format('YYYY-MM-DD')),
+						date: new Date(moment(planDate).format('YYYY-MM-DD')),
 						status: DailyPlanStatusEnum.OPEN,
 						tenantId: user?.tenantId ?? '',
 						employeeId: user?.employee.id,
@@ -810,6 +795,12 @@ function TaskCard(props: ITaskCardProps) {
 	);
 }
 
+/**
+ * ----------------------------------------------------------------
+ * 		----------------- TASK CARD ACTIONS -------------------
+ * ----------------------------------------------------------------
+ */
+
 interface ITaskCardActionsProps {
 	task: ITeamTask;
 	selectedPlan: IDailyPlan;
@@ -964,6 +955,12 @@ function TaskCardActions(props: ITaskCardActionsProps) {
 	);
 }
 
+/**
+ * ----------------------------------------------------------------
+ * 		---------------- UNPLAN TASK ACTIONS ----------------
+ * ----------------------------------------------------------------
+ */
+
 interface IUnplanTaskProps {
 	taskId: string;
 	selectedPlanId: string;
diff --git a/apps/web/lib/features/daily-plan/all-plans-modal.tsx b/apps/web/lib/features/daily-plan/all-plans-modal.tsx
index 1550263c3..33526dd2a 100644
--- a/apps/web/lib/features/daily-plan/all-plans-modal.tsx
+++ b/apps/web/lib/features/daily-plan/all-plans-modal.tsx
@@ -135,7 +135,12 @@ export const AllPlansModal = memo(function AllPlansModal(props: IAllPlansModal)
 	}, [createDailyPlan, customDate, user?.employee.id, user?.employee.organizationId, user?.tenantId]);
 
 	return (
-		<Modal isOpen={isOpen} closeModal={handleCloseModal} className={clsxm('w-[36rem]')}>
+		<Modal
+			isOpen={isOpen}
+			customCloseModal={handleCloseModal}
+			closeModal={() => null}
+			className={clsxm('w-[36rem]')}
+		>
 			<Card className="w-full  h-full overflow-hidden" shadow="custom">
 				<div className="w-full flex flex-col gap-3 ">
 					<div className="relative w-full h-12  flex items-center justify-center">