From 37a452ac08ffc11351d92795b467af898fdb7ee1 Mon Sep 17 00:00:00 2001
From: april <aprl.petersen@gmail.com>
Date: Thu, 11 Jan 2024 14:54:03 -0600
Subject: [PATCH] Implement aircraft updating

---
 web/app/routes/logbook.aircraft/route.tsx | 34 ++++++++++++++++++++++-
 web/app/ui/form/aircraft-form.tsx         | 29 +++++++++++++------
 2 files changed, 54 insertions(+), 9 deletions(-)

diff --git a/web/app/routes/logbook.aircraft/route.tsx b/web/app/routes/logbook.aircraft/route.tsx
index d426a4a..6d85c0e 100644
--- a/web/app/routes/logbook.aircraft/route.tsx
+++ b/web/app/routes/logbook.aircraft/route.tsx
@@ -41,6 +41,19 @@ function AircraftCard({ aircraft }: { aircraft: AircraftSchema }) {
     },
   });
 
+  const [editOpened, { open: openEdit, close: closeEdit }] =
+    useDisclosure(false);
+
+  const updateAircraft = useMutation({
+    mutationFn: async (values: AircraftFormSchema) =>
+      await client
+        .put(`/aircraft/${aircraft.id}`, values)
+        .then((res) => res.data),
+    onSuccess: () => {
+      queryClient.invalidateQueries({ queryKey: ["aircraft-list"] });
+    },
+  });
+
   return (
     <>
       <Modal
@@ -69,12 +82,31 @@ function AircraftCard({ aircraft }: { aircraft: AircraftSchema }) {
           </Group>
         </Stack>
       </Modal>
+      <Modal
+        opened={editOpened}
+        onClose={closeEdit}
+        title="Edit Aircraft"
+        centered
+      >
+        <AircraftForm
+          isError={updateAircraft.isError}
+          error={updateAircraft.error}
+          isPending={updateAircraft.isPending}
+          onSubmit={(values: AircraftFormSchema) =>
+            updateAircraft.mutate(values)
+          }
+          initialValues={aircraft}
+          submitButtonLabel="Update"
+          withCancelButton
+          cancelFunc={closeEdit}
+        />
+      </Modal>
       <Card key={randomId()} withBorder shadow="sm">
         <Stack>
           <Group grow justify="space-between">
             <Title order={4}>{aircraft.tail_no}</Title>
             <Group justify="flex-end">
-              <ActionIcon variant="transparent">
+              <ActionIcon variant="transparent" onClick={openEdit}>
                 <IconPencil />
               </ActionIcon>
               <ActionIcon
diff --git a/web/app/ui/form/aircraft-form.tsx b/web/app/ui/form/aircraft-form.tsx
index 5d439d3..eadf77f 100644
--- a/web/app/ui/form/aircraft-form.tsx
+++ b/web/app/ui/form/aircraft-form.tsx
@@ -72,8 +72,12 @@ export default function AircraftForm({
       await client.get(`/aircraft/categories`).then((res) => res.data),
   });
 
-  const [category, setCategory] = useState("");
-  const [classSelection, setClassSelection] = useState("");
+  const [category, setCategory] = useState(
+    initialValues?.aircraft_category ?? ""
+  );
+  const [classSelection, setClassSelection] = useState<string | null>(
+    initialValues?.aircraft_class ?? ""
+  );
 
   const classes = useQuery({
     queryKey: ["classes", category],
@@ -107,7 +111,6 @@ export default function AircraftForm({
             {...newForm.getInputProps("aircraft_category")}
             label="Category"
             placeholder="Pick a value"
-            name="aircraft_category"
             withAsterisk
             data={
               categories.isFetched && !categories.isError
@@ -115,15 +118,18 @@ export default function AircraftForm({
                 : []
             }
             onChange={(_value, option) => {
+              newForm.setFieldValue("aircraft_category", option.value);
               setCategory(option.value);
-              setClassSelection("");
+              newForm.setFieldValue("aircraft_class", "");
+              setClassSelection(null);
               queryClient.invalidateQueries({
                 queryKey: ["classes", option.value],
               });
-              newForm.setFieldValue("aircraft_category", option.value);
             }}
+            key={classSelection}
           />
           <Select
+            {...newForm.getInputProps("aircraft_class")}
             label="Class"
             placeholder="Pick a value"
             withAsterisk
@@ -133,7 +139,10 @@ export default function AircraftForm({
                 : []
             }
             value={classSelection}
-            {...newForm.getInputProps("aircraft_class")}
+            onChange={(_value, option) => {
+              newForm.setFieldValue("aircraft_class", option.label);
+              setClassSelection(option.label);
+            }}
           />
           <NumberInput
             label="Hobbs"
@@ -147,11 +156,15 @@ export default function AircraftForm({
             {isError ? (
               <Text c="red">
                 {error instanceof AxiosError
-                  ? error?.response?.data?.detail ?? "Error adding aircraft"
+                  ? error?.response?.data?.detail ??
+                    error?.response?.data ??
+                    error?.response ??
+                    error?.message ??
+                    "Failed to submit"
                   : error?.message}
               </Text>
             ) : isPending ? (
-              <Text c="yellow">Adding aircraft...</Text>
+              <Text c="yellow">Loading...</Text>
             ) : null}
             {withCancelButton ? (
               <Button leftSection={<IconX />} color="gray" onClick={cancelFunc}>