Function useIntegration

packages/shared/src/hooks/integrations/useIntegration.ts:39–138

()

};

export const useIntegration = (): UseIntegration => {
  const { user } = useAuthContext();
  const queryClient = useQueryClient();
  const { showPrompt } = usePrompt();
  const { logEvent } = useLogContext();

  const { mutateAsync: removeIntegration } = useMutation({
    mutationFn: async ({
      integrationId,
      integrationType,
    }: {
      integrationId: string;
      integrationType: UserIntegrationType;
    }) => {
      const deleteConfirmed = await showPrompt(deleteIntegrationPromptOptions);

      if (!deleteConfirmed) {
        throw new Error('User cancelled integration deletion');
      }

      logEvent({
        event_name: LogEvent.RevokeIntegrationAccess,
        target_id: integrationType,
      });

      await gqlClient.request(REMOVE_INTEGRATION_MUTATION, {
        integrationId,
      });
    },

    onSuccess: async (data, { integrationId }) => {
      queryClient.setQueryData<UserIntegration[]>(
        generateQueryKey(RequestKey.UserIntegrations, user),
        (currentIntegrations) => {
          return currentIntegrations?.filter(
            (integration) => integration.id !== integrationId,
          );
        },
      );
    },
  });

  const { mutateAsync: removeSourceIntegration } = useMutation({
    mutationFn: async ({
      sourceId,
      integrationId,
      integrationType,
    }: {
      sourceId: string;
      integrationId: string;
      integrationType: UserIntegrationType;
    }) => {
      const deleteConfirmed = await showPrompt(
        deleteSourceIntegrationPromptOptions,
      );

      if (!deleteConfirmed) {
        throw new Error('User cancelled source integration deletion');