fix: correção week days

ctrlmais · Apr 12, 2022 · be8f256 · be8f256
1 parent 0ea07d3
commit be8f256
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 12 deletions.
diff --git a/src/hooks/useSchedule.tsx b/src/hooks/useSchedule.tsx
@@ -1,4 +1,4 @@
-import { useState } from 'react';
+import { useEffect, useState } from 'react';
 
 import { Schedule } from 'types/IContext';
 
@@ -8,17 +8,28 @@ import { useUser } from 'contexts/User';
 
 export function useSchedule() {
   const { barbeiro } = useUser();
-  const weekDay = new Date().getDay();
+  const [weekDay, setWeekDay] = useState<string>(String(new Date().getDay()));
 
-  const schedules = JSON.parse(barbeiro?.schedules || '[]');
+  function getSchedule(weekDay: string) {
+    const schedule = [] as Schedule[];
+    const schedules = JSON.parse(barbeiro?.schedules || '[]');
+    schedules.map((scheduleItem: Schedule) => {
+      if (scheduleItem.week_day === weekDay) {
+        schedule.push(scheduleItem);
+      }
+    });
+    return schedule;
+  }
 
   if (!barbeiro) window.location.assign('/');
 
+  const schedules = JSON.parse(barbeiro?.schedules || '[]');
+
   const [horarioInicialBarbeiroSchedule, setHorarioInicialBarbeiroSchedule] = useState(
-    JSON.parse(barbeiro?.schedules)[weekDay]?.from || '',
+    JSON.parse(barbeiro?.schedules)[weekDay]?.from,
   );
   const [horarioFinalBarbeiroSchedule, setHorarioFinalBarbeiroSchedule] = useState(
-    JSON.parse(barbeiro?.schedules)[weekDay]?.to || '',
+    JSON.parse(barbeiro?.schedules)[weekDay]?.to,
   );
 
   function getHorarioAtual(week_day: string) {
@@ -93,6 +104,11 @@ export function useSchedule() {
     return horariosPosteriores;
   }
 
+  useEffect(() => {
+    setHorarioInicialBarbeiroSchedule(getSchedule(weekDay || '0').map((schedule) => schedule.from));
+    setHorarioFinalBarbeiroSchedule(getSchedule(weekDay || '0').map((schedule) => schedule.to));
+  }, [weekDay]);
+
   return {
     setHorarioInicialBarbeiroSchedule,
     setHorarioFinalBarbeiroSchedule,
@@ -102,5 +118,6 @@ export function useSchedule() {
     desabilitarHorariosPosteriores,
     horarioInicialBarbeiroSchedule,
     horarioFinalBarbeiroSchedule,
+    setWeekDay,
   };
 }
diff --git a/src/pages/Schedule/index.tsx b/src/pages/Schedule/index.tsx
@@ -37,14 +37,13 @@ export function Schedule() {
   } = useUser();
 
   const {
-    setHorarioInicialBarbeiroSchedule,
-    setHorarioFinalBarbeiroSchedule,
     getHorarioAtual,
-    numerosFaltantes,
     desabilitarHorariosAnteriores,
     desabilitarHorariosPosteriores,
     horarioInicialBarbeiroSchedule,
     horarioFinalBarbeiroSchedule,
+    numerosFaltantes,
+    setWeekDay,
   } = useSchedule();
 
   return (
@@ -79,10 +78,10 @@ export function Schedule() {
               fromMonth={new Date()}
               selected={selectDay}
               onDayClick={(day) => {
-                const weekDay = day.getDay();
-                getHorarioAtual(String(weekDay));
-                setHorarioInicialBarbeiroSchedule(JSON.parse(barbeiro?.schedules)[weekDay]?.from);
-                setHorarioFinalBarbeiroSchedule(JSON.parse(barbeiro?.schedules)[weekDay]?.to);
+                const weekDayClick = day.getDay();
+                getHorarioAtual(String(weekDayClick));
+                setWeekDay(String(weekDayClick));
+
                 setSelectDay(day);
               }}
               modifiers={{