From d595aa42595c4f4fc6b8b794e74df3df8290e70a Mon Sep 17 00:00:00 2001
From: Joonatan Kuosa <joonatan.kuosa@gmail.com>
Date: Tue, 4 Mar 2025 10:14:40 +0200
Subject: [PATCH] fix: edit reservation failing

- Fix: edit mutation called with incorrect end time.
- Fix: edit / new reservation steppers not doing anything.
- Refactor: cleaner edit step components.
---
 apps/ui/components/reservation/EditStep0.tsx  | 25 +----
 apps/ui/components/reservation/EditStep1.tsx  | 68 ++++++++-----
 apps/ui/modules/reservation.ts                |  9 +-
 .../ui/pages/reservation-unit/[...params].tsx | 10 +-
 apps/ui/pages/reservations/[id]/edit.tsx      | 98 +++----------------
 apps/ui/public/locales/en/errors.json         |  1 +
 apps/ui/public/locales/fi/errors.json         |  1 +
 apps/ui/public/locales/sv/errors.json         |  1 +
 8 files changed, 73 insertions(+), 140 deletions(-)

diff --git a/apps/ui/components/reservation/EditStep0.tsx b/apps/ui/components/reservation/EditStep0.tsx
index 4d73dfb2d..fb00ab03a 100644
--- a/apps/ui/components/reservation/EditStep0.tsx
+++ b/apps/ui/components/reservation/EditStep0.tsx
@@ -6,13 +6,7 @@ import type {
   ReservationUnitPageQuery,
 } from "@gql/gql-types";
 import { differenceInMinutes } from "date-fns";
-import {
-  Button,
-  ButtonVariant,
-  IconArrowRight,
-  IconCross,
-  LoadingSpinner,
-} from "hds-react";
+import { Button, ButtonVariant, IconArrowRight, IconCross } from "hds-react";
 import React from "react";
 import { useTranslation } from "next-i18next";
 import styled from "styled-components";
@@ -57,8 +51,6 @@ type Props = {
   activeApplicationRounds: readonly RoundPeriod[];
   blockingReservations: readonly BlockingReservationFieldsFragment[];
   nextStep: () => void;
-  apiBaseUrl: string;
-  isLoading: boolean;
 };
 
 const StyledCalendarWrapper = styled.div`
@@ -120,7 +112,6 @@ export function EditStep0({
   activeApplicationRounds,
   reservationForm,
   nextStep,
-  isLoading,
   blockingReservations: blockingReservationsOrig,
 }: Props): JSX.Element {
   const { t, i18n } = useTranslation();
@@ -266,20 +257,14 @@ export function EditStep0({
             href={getReservationPath(reservation.pk)}
             data-testid="reservation-edit__button--cancel"
           >
-            <IconCross aria-hidden="true" />
+            <IconCross />
             {t("common:stop")}
           </ButtonLikeLink>
           <Button
             type="submit"
-            variant={isLoading ? ButtonVariant.Clear : ButtonVariant.Primary}
-            iconEnd={
-              isLoading ? (
-                <LoadingSpinner small />
-              ) : (
-                <IconArrowRight aria-hidden="true" />
-              )
-            }
-            disabled={!focusSlot.isReservable || !isDirty || isLoading}
+            variant={ButtonVariant.Primary}
+            iconEnd={<IconArrowRight />}
+            disabled={!focusSlot.isReservable || !isDirty}
             data-testid="reservation__button--continue"
           >
             {t("common:next")}
diff --git a/apps/ui/components/reservation/EditStep1.tsx b/apps/ui/components/reservation/EditStep1.tsx
index 9d3b78e57..0edd4b12c 100644
--- a/apps/ui/components/reservation/EditStep1.tsx
+++ b/apps/ui/components/reservation/EditStep1.tsx
@@ -1,6 +1,7 @@
 import {
   type ReservationEditPageQuery,
   type ReservationUnitPageQuery,
+  useAdjustReservationTimeMutation,
 } from "@gql/gql-types";
 import {
   Button,
@@ -27,6 +28,8 @@ import { type UseFormReturn } from "react-hook-form";
 import { convertReservationFormToApi } from "@/modules/reservation";
 import { AcceptTerms } from "./AcceptTerms";
 import { getReservationPath } from "@/modules/urls";
+import { useDisplayError } from "@/hooks/useDisplayError";
+import { useRouter } from "next/router";
 
 type ReservationUnitNodeT = NonNullable<
   ReservationUnitPageQuery["reservationUnit"]
@@ -37,8 +40,6 @@ type Props = {
   reservationUnit: ReservationUnitNodeT;
   options: OptionsRecord;
   onBack: () => void;
-  handleSubmit: () => void;
-  isSubmitting: boolean;
   form: UseFormReturn<PendingReservationFormType>;
 };
 
@@ -97,8 +98,6 @@ export function EditStep1({
   reservationUnit,
   options,
   onBack,
-  handleSubmit,
-  isSubmitting,
   form,
 }: Props): JSX.Element {
   const { t } = useTranslation();
@@ -130,7 +129,13 @@ export function EditStep1({
     frozenReservationUnit?.metadataSet?.supportedFields
   );
 
-  const { watch } = form;
+  const {
+    handleSubmit,
+    watch,
+    formState: { isSubmitting },
+  } = form;
+  const router = useRouter();
+  const displayError = useDisplayError();
 
   const apiValues = convertReservationFormToApi(watch());
   const modifiedReservation = {
@@ -139,6 +144,36 @@ export function EditStep1({
     end: apiValues?.end ?? reservation.end,
   };
 
+  const [mutation, { loading }] = useAdjustReservationTimeMutation();
+
+  const isLoading = loading || isSubmitting;
+
+  // TODO refactor to use form submit instead of getValues
+  const onSubmit = async (formValues: PendingReservationFormType) => {
+    if (!termsAccepted) {
+      errorToast({
+        text: t("reservationCalendar:errors.termsNotAccepted"),
+      });
+      return;
+    }
+    // const formValues = getValues();
+    const times = convertReservationFormToApi(formValues);
+    if (reservation.pk == null || times == null) {
+      return;
+    }
+    try {
+      const input = { pk: reservation.pk, ...times };
+      await mutation({
+        variables: {
+          input,
+        },
+      });
+      router.push(`${getReservationPath(reservation.pk)}?timeUpdated=true`);
+    } catch (e) {
+      displayError(e);
+    }
+  };
+
   const termsAccepted = isTermsAccepted.space && isTermsAccepted.service;
   return (
     <>
@@ -146,18 +181,7 @@ export function EditStep1({
         reservation={modifiedReservation}
         type="confirmed"
       />
-      <StyledForm
-        onSubmit={(e) => {
-          e.preventDefault();
-          if (!termsAccepted) {
-            errorToast({
-              text: t("reservationCalendar:errors.termsNotAccepted"),
-            });
-          } else {
-            handleSubmit();
-          }
-        }}
-      >
+      <StyledForm onSubmit={handleSubmit(onSubmit)}>
         <GeneralFields
           supportedFields={supportedFields}
           reservation={reservation}
@@ -176,7 +200,7 @@ export function EditStep1({
         <Actions>
           <Button
             variant={ButtonVariant.Secondary}
-            iconStart={<IconArrowLeft aria-hidden="true" />}
+            iconStart={<IconArrowLeft />}
             onClick={onBack}
             data-testid="reservation-edit__button--back"
           >
@@ -187,14 +211,14 @@ export function EditStep1({
             size="large"
             data-testid="reservation-edit__button--cancel"
           >
-            <IconCross aria-hidden="true" />
+            <IconCross />
             {t("common:stop")}
           </ButtonLikeLink>
           <Button
             type="submit"
-            variant={isSubmitting ? ButtonVariant.Clear : ButtonVariant.Primary}
-            iconStart={isSubmitting ? <LoadingSpinner small /> : undefined}
-            disabled={isSubmitting || !termsAccepted}
+            variant={isLoading ? ButtonVariant.Clear : ButtonVariant.Primary}
+            iconStart={isLoading ? <LoadingSpinner small /> : undefined}
+            disabled={!termsAccepted || isLoading}
             data-testid="reservation__button--continue"
           >
             {t("reservations:saveNewTime")}
diff --git a/apps/ui/modules/reservation.ts b/apps/ui/modules/reservation.ts
index fc751746e..527b913e9 100644
--- a/apps/ui/modules/reservation.ts
+++ b/apps/ui/modules/reservation.ts
@@ -503,18 +503,19 @@ export function createDateTime(date: string, time: string): Date {
   return new Date();
 }
 
+// NOTE backend throws errors in some cases if we accidentally send seconds or milliseconds that are not 0
 export function convertReservationFormToApi(
   formValues: PendingReservationFormType
 ): { begin: string; end: string } | null {
   const time = formValues.time;
-  const date = fromUIDate(formValues.date ?? "");
+  const date = fromUIDate(formValues.date);
   const duration = formValues.duration;
-  if (date == null || time == null || duration == null) {
+  if (date == null || time === "" || duration === 0) {
     return null;
   }
   const minutes = timeToMinutes(time);
-  const begin: Date = set(date, { minutes });
-  const end: Date = set(begin, { minutes: minutes + duration });
+  const begin: Date = set(date, { minutes, seconds: 0, milliseconds: 0 });
+  const end: Date = addMinutes(begin, duration);
   return { begin: begin.toISOString(), end: end.toISOString() };
 }
 
diff --git a/apps/ui/pages/reservation-unit/[...params].tsx b/apps/ui/pages/reservation-unit/[...params].tsx
index eed0712c1..536f832a1 100644
--- a/apps/ui/pages/reservation-unit/[...params].tsx
+++ b/apps/ui/pages/reservation-unit/[...params].tsx
@@ -380,15 +380,7 @@ function NewReservation(props: PropsNarrowed): JSX.Element | null {
               language={i18n.language}
               selectedStep={step}
               style={{ width: "100%" }}
-              onStepClick={(e) => {
-                const target = e.currentTarget;
-                const s = target
-                  .getAttribute("data-testid")
-                  ?.replace("hds-stepper-step-", "");
-                if (s) {
-                  setStep(parseInt(s, 10));
-                }
-              }}
+              onStepClick={(_, index) => setStep(index)}
               steps={steps}
             />
           )}
diff --git a/apps/ui/pages/reservations/[id]/edit.tsx b/apps/ui/pages/reservations/[id]/edit.tsx
index 9ddd556d4..1ffbf2cd8 100644
--- a/apps/ui/pages/reservations/[id]/edit.tsx
+++ b/apps/ui/pages/reservations/[id]/edit.tsx
@@ -9,9 +9,6 @@ import {
   ReservationUnitPageDocument,
   type ReservationUnitPageQuery,
   type ReservationUnitPageQueryVariables,
-  type Mutation,
-  type MutationAdjustReservationTimeArgs,
-  useAdjustReservationTimeMutation,
   type ReservationEditPageQuery,
   type ReservationEditPageQueryVariables,
   ReservationEditPageDocument,
@@ -25,12 +22,10 @@ import {
 import { toApiDate } from "common/src/common/util";
 import { addYears } from "date-fns";
 import { RELATED_RESERVATION_STATES } from "common/src/const";
-import { gql, type FetchResult } from "@apollo/client";
-import { useRouter } from "next/router";
+import { gql } from "@apollo/client";
 import { StepState, Stepper } from "hds-react";
 import { zodResolver } from "@hookform/resolvers/zod";
 import { useForm } from "react-hook-form";
-import { errorToast } from "common/src/common/toast";
 import { EditStep0 } from "@/components/reservation/EditStep0";
 import { EditStep1 } from "@/components/reservation/EditStep1";
 import {
@@ -40,7 +35,6 @@ import {
 import { ReservationPageWrapper } from "@/components/reservations/styles";
 import { queryOptions } from "@/modules/queryOptions";
 import {
-  convertReservationFormToApi,
   isReservationEditable,
   transformReservation,
 } from "@/modules/reservation";
@@ -61,101 +55,39 @@ const StepperWrapper = styled.div`
 `;
 
 function ReservationEditPage(props: PropsNarrowed): JSX.Element {
-  const {
-    reservation,
-    reservationUnit,
-    apiBaseUrl,
-    options,
-    blockingReservations,
-  } = props;
+  const { reservation, reservationUnit, options, blockingReservations } = props;
   const { t, i18n } = useTranslation();
-  const router = useRouter();
 
   const [step, setStep] = useState<0 | 1>(0);
 
-  const [mutation, { loading: isLoading }] = useAdjustReservationTimeMutation();
-
-  // TODO should rework this so we don't pass a string here (use Dates till we do the mutation)
-  const adjustReservationTime = (
-    input: MutationAdjustReservationTimeArgs["input"]
-  ): Promise<FetchResult<Mutation>> => {
-    if (!input.pk) {
-      throw new Error("No reservation pk provided");
-    }
-    if (!input.begin || !input.end) {
-      throw new Error("No begin or end time provided");
-    }
-    // NOTE backend throws errors in some cases if we accidentally send seconds or milliseconds that are not 0
-    const { begin, end, ...rest } = input;
-    const beginDate = new Date(begin);
-    beginDate.setSeconds(0);
-    beginDate.setMilliseconds(0);
-    const endDate = new Date(end);
-    endDate.setSeconds(0);
-    endDate.setMilliseconds(0);
-    return mutation({
-      variables: {
-        input: {
-          begin: beginDate.toISOString(),
-          end: endDate.toISOString(),
-          ...rest,
-        },
-      },
-    });
-  };
-
-  const reservationForm = useForm<PendingReservationFormType>({
+  const form = useForm<PendingReservationFormType>({
     defaultValues: transformReservation(reservation),
     mode: "onChange",
     resolver: zodResolver(PendingReservationFormSchema),
   });
 
-  const { getValues, reset } = reservationForm;
+  const { reset } = form;
   useEffect(() => {
     reset(transformReservation(reservation));
   }, [reservation, reset]);
 
-  // TODO refactor to use form submit instead of getValues
-  const handleSubmit = async () => {
-    const formValues = getValues();
-    const times = convertReservationFormToApi(formValues);
-    if (reservation.pk == null || times == null) {
-      return;
-    }
-    try {
-      await adjustReservationTime({ pk: reservation.pk, ...times });
-      router.push(`${getReservationPath(reservation.pk)}?timeUpdated=true`);
-    } catch (e) {
-      if (e instanceof Error) {
-        // TODO don't print the error message to the user
-        errorToast({ text: e.message });
-      } else {
-        errorToast({ text: "Unknown error occurred" });
-      }
-    }
-  };
-
   const title =
     step === 0
       ? "reservations:editReservationTime"
       : "reservationCalendar:heading.pendingReservation";
 
-  const handleStepClick = (e: React.MouseEvent<HTMLButtonElement>) => {
-    const target = e.currentTarget;
-    const s = target
-      .getAttribute("data-testid")
-      ?.replace("hds-stepper-step-", "");
-    if (s != null) {
-      const n = parseInt(s, 10);
-      if (n === 0 || n === 1) {
-        setStep(n);
-      }
+  const handleStepClick = (
+    _: React.MouseEvent<HTMLButtonElement>,
+    index: number
+  ) => {
+    if (index === 0 || index === 1) {
+      setStep(index);
     }
   };
 
   const {
     formState: { isValid, dirtyFields },
-  } = reservationForm;
+  } = form;
   // skip control fields
   const isDirty = dirtyFields.date || dirtyFields.time || dirtyFields.duration;
   const steps = [
@@ -193,10 +125,8 @@ function ReservationEditPage(props: PropsNarrowed): JSX.Element {
           reservation={reservation}
           reservationUnit={reservationUnit}
           activeApplicationRounds={activeApplicationRounds}
-          reservationForm={reservationForm}
+          reservationForm={form}
           nextStep={() => setStep(1)}
-          apiBaseUrl={apiBaseUrl}
-          isLoading={false}
           blockingReservations={blockingReservations}
         />
       ) : (
@@ -205,9 +135,7 @@ function ReservationEditPage(props: PropsNarrowed): JSX.Element {
           options={options}
           reservationUnit={reservationUnit}
           onBack={() => setStep(0)}
-          handleSubmit={handleSubmit}
-          form={reservationForm}
-          isSubmitting={isLoading}
+          form={form}
         />
       )}
     </ReservationPageWrapper>
diff --git a/apps/ui/public/locales/en/errors.json b/apps/ui/public/locales/en/errors.json
index a23021fe0..96898293d 100644
--- a/apps/ui/public/locales/en/errors.json
+++ b/apps/ui/public/locales/en/errors.json
@@ -47,6 +47,7 @@
       "APPLICATION_APPLICANT_TYPE_MISSING": "Applicant type is missing.",
       "APPLICATION_STATUS_CANNOT_SEND": "Application cannot be sent.",
       "OVERLAPPING_RESERVATIONS": "There are duplicate bookings.",
+      "RESERVATION_UNITS_MAX_DURATION_EXCEEDED": "The booking time is too long.",
       "invalid": "Invalid value.",
       "generic_validation_error": "An error occurred."
     }
diff --git a/apps/ui/public/locales/fi/errors.json b/apps/ui/public/locales/fi/errors.json
index 2ea786ac6..4b6886967 100644
--- a/apps/ui/public/locales/fi/errors.json
+++ b/apps/ui/public/locales/fi/errors.json
@@ -47,6 +47,7 @@
       "APPLICATION_APPLICANT_TYPE_MISSING": "Hakijan tyyppi puuttuu.",
       "APPLICATION_STATUS_CANNOT_SEND": "Hakemusta ei voi lähettää.",
       "OVERLAPPING_RESERVATIONS": "Päällekkäisiä varauksia.",
+      "RESERVATION_UNITS_MAX_DURATION_EXCEEDED": "Liian pitkä varausaika.",
       "invalid": "Virheellinen arvo.",
       "generic_validation_error": "Tapahtui virhe."
     }
diff --git a/apps/ui/public/locales/sv/errors.json b/apps/ui/public/locales/sv/errors.json
index a664bcc76..429c5a2c8 100644
--- a/apps/ui/public/locales/sv/errors.json
+++ b/apps/ui/public/locales/sv/errors.json
@@ -47,6 +47,7 @@
       "APPLICATION_APPLICANT_TYPE_MISSING": "Sökandens typ saknas.",
       "APPLICATION_STATUS_CANNOT_SEND": "Ansökan kan inte skickas.",
       "OVERLAPPING_RESERVATIONS": "Överlappande bokningar.",
+      "RESERVATION_UNITS_MAX_DURATION_EXCEEDED": "För lång bokningstid.",
       "invalid": "Ogiltigt värde.",
       "generic_validation_error": "Ett fel uppstod."
     }