From 95dc5fb8492dee14ae3eccf33b4b424a9060d169 Mon Sep 17 00:00:00 2001
From: Ali BARIN <ali.barin53@gmail.com>
Date: Mon, 2 Sep 2024 15:29:38 +0000
Subject: [PATCH] refactor(RoleMappings): rewrite mutations with REST API
 endpoints

---
 ...AdminUpdateSamlAuthProviderRoleMappings.js | 31 +++++++++++++++++
 .../src/pages/Authentication/RoleMappings.jsx | 34 +++++++------------
 2 files changed, 44 insertions(+), 21 deletions(-)
 create mode 100644 packages/web/src/hooks/useAdminUpdateSamlAuthProviderRoleMappings.js

diff --git a/packages/web/src/hooks/useAdminUpdateSamlAuthProviderRoleMappings.js b/packages/web/src/hooks/useAdminUpdateSamlAuthProviderRoleMappings.js
new file mode 100644
index 00000000..4909e540
--- /dev/null
+++ b/packages/web/src/hooks/useAdminUpdateSamlAuthProviderRoleMappings.js
@@ -0,0 +1,31 @@
+import { useMutation, useQueryClient } from '@tanstack/react-query';
+import api from 'helpers/api';
+
+export default function useAdminUpdateSamlAuthProviderRoleMappings(
+  samlAuthProviderId,
+) {
+  const queryClient = useQueryClient();
+
+  const query = useMutation({
+    mutationFn: async (payload) => {
+      const { data } = await api.patch(
+        `/v1/admin/saml-auth-providers/${samlAuthProviderId}/role-mappings`,
+        payload,
+      );
+
+      return data;
+    },
+    onSuccess: () => {
+      queryClient.invalidateQueries({
+        queryKey: [
+          'admin',
+          'samlAuthProviders',
+          samlAuthProviderId,
+          'roleMappings',
+        ],
+      });
+    },
+  });
+
+  return query;
+}
diff --git a/packages/web/src/pages/Authentication/RoleMappings.jsx b/packages/web/src/pages/Authentication/RoleMappings.jsx
index a2605eef..7a46c35a 100644
--- a/packages/web/src/pages/Authentication/RoleMappings.jsx
+++ b/packages/web/src/pages/Authentication/RoleMappings.jsx
@@ -1,5 +1,4 @@
 import PropTypes from 'prop-types';
-import { useMutation } from '@apollo/client';
 import LoadingButton from '@mui/lab/LoadingButton';
 import Divider from '@mui/material/Divider';
 import Stack from '@mui/material/Stack';
@@ -8,9 +7,9 @@ import useEnqueueSnackbar from 'hooks/useEnqueueSnackbar';
 import { useMemo } from 'react';
 
 import Form from 'components/Form';
-import { UPSERT_SAML_AUTH_PROVIDERS_ROLE_MAPPINGS } from 'graphql/mutations/upsert-saml-auth-providers-role-mappings';
 import useFormatMessage from 'hooks/useFormatMessage';
 import useAdminSamlAuthProviderRoleMappings from 'hooks/useAdminSamlAuthProviderRoleMappings';
+import useAdminUpdateSamlAuthProviderRoleMappings from 'hooks/useAdminUpdateSamlAuthProviderRoleMappings';
 import RoleMappingsFieldArray from './RoleMappingsFieldsArray';
 
 function generateFormRoleMappings(roleMappings) {
@@ -28,33 +27,26 @@ function RoleMappings({ provider, providerLoading }) {
   const formatMessage = useFormatMessage();
   const enqueueSnackbar = useEnqueueSnackbar();
 
+  const {
+    mutateAsync: updateSamlAuthProvidersRoleMappings,
+    isPending: isUpdateSamlAuthProvidersRoleMappingsPending,
+  } = useAdminUpdateSamlAuthProviderRoleMappings(provider?.id);
+
   const { data, isLoading: isAdminSamlAuthProviderRoleMappingsLoading } =
     useAdminSamlAuthProviderRoleMappings({
       adminSamlAuthProviderId: provider?.id,
     });
   const roleMappings = data?.data;
 
-  const [
-    upsertSamlAuthProvidersRoleMappings,
-    { loading: upsertRoleMappingsLoading },
-  ] = useMutation(UPSERT_SAML_AUTH_PROVIDERS_ROLE_MAPPINGS);
-
   const handleRoleMappingsUpdate = async (values) => {
     try {
       if (provider?.id) {
-        await upsertSamlAuthProvidersRoleMappings({
-          variables: {
-            input: {
-              samlAuthProviderId: provider.id,
-              samlAuthProvidersRoleMappings: values.roleMappings.map(
-                ({ roleId, remoteRoleName }) => ({
-                  roleId,
-                  remoteRoleName,
-                }),
-              ),
-            },
-          },
-        });
+        await updateSamlAuthProvidersRoleMappings(
+          values.roleMappings.map(({ roleId, remoteRoleName }) => ({
+            roleId,
+            remoteRoleName,
+          })),
+        );
 
         enqueueSnackbar(formatMessage('roleMappingsForm.successfullySaved'), {
           variant: 'success',
@@ -97,7 +89,7 @@ function RoleMappings({ provider, providerLoading }) {
             variant="contained"
             color="primary"
             sx={{ boxShadow: 2 }}
-            loading={upsertRoleMappingsLoading}
+            loading={isUpdateSamlAuthProvidersRoleMappingsPending}
           >
             {formatMessage('roleMappingsForm.save')}
           </LoadingButton>