From e6b85c62c1de8540cdd4abc51c26a57348a461c8 Mon Sep 17 00:00:00 2001
From: Arthur Belleville <arthur.belleville@datadoghq.com>
Date: Sun, 19 Oct 2025 11:40:02 +0200
Subject: [PATCH] Fix toasts

---
 ui/src/hooks/availabilities.ts  | 18 ++++++++++++++++++
 ui/src/pages/availabilities.tsx | 27 ++++-----------------------
 2 files changed, 22 insertions(+), 23 deletions(-)

diff --git a/ui/src/hooks/availabilities.ts b/ui/src/hooks/availabilities.ts
index fef1bac..8430660 100644
--- a/ui/src/hooks/availabilities.ts
+++ b/ui/src/hooks/availabilities.ts
@@ -4,6 +4,7 @@ import { supabase } from "@ui/hooks/auth";
 import { queryClient } from "@ui/lib/api";
 import { Database } from "@ui/types/database.types";
 import { useEffect, useState } from "react";
+import { toast } from "src/lib/toast";
 
 export type TimeRange = {
   start: string;
@@ -105,6 +106,18 @@ export function useAvailabilities() {
     },
     onSuccess: () => {
       queryClient.invalidateQueries({ queryKey: ["availabilities"] });
+      toast.add({
+        title: "Succès",
+        description: "Disponibilités mises à jour avec succès",
+        type: "success",
+      });
+    },
+    onError: () => {
+      toast.add({
+        title: "Erreur",
+        description: "Erreur lors de la mise à jour des disponibilités",
+        type: "error",
+      });
     },
   });
 
@@ -135,6 +148,11 @@ export function useAvailabilities() {
     },
     onSuccess: () => {
       queryClient.invalidateQueries({ queryKey: ["availabilities"] });
+      toast.add({
+        title: "Succès",
+        description: "Exception supprimée avec succès",
+        type: "success",
+      });
     },
   });
 
diff --git a/ui/src/pages/availabilities.tsx b/ui/src/pages/availabilities.tsx
index b728a44..93a1a0c 100644
--- a/ui/src/pages/availabilities.tsx
+++ b/ui/src/pages/availabilities.tsx
@@ -126,29 +126,10 @@ export function AvailabilitiesPage() {
                   variant="default"
                   className="[--btn-bg:var(--color-green-800)]"
                   onClick={() => {
-                    updateAvailabilities(
-                      {
-                        updatedAvailabilities: draftAvailabilities,
-                        newException: null,
-                      },
-                      {
-                        onSuccess: () => {
-                          toast.add({
-                            title: "Succès",
-                            description: "Disponibilités enregistrées avec succès",
-                            type: "success",
-                          });
-                        },
-                        onError: (err) => {
-                          console.error(err);
-                          toast.add({
-                            title: "Erreur",
-                            description: "Erreur lors de l'enregistrement des disponibilités",
-                            type: "error",
-                          });
-                        },
-                      }
-                    );
+                    updateAvailabilities({
+                      updatedAvailabilities: draftAvailabilities,
+                      newException: null,
+                    });
                   }}
                 >
                   <SaveIcon /> Enregistrer