From 0d9bdd1360573d6675414a74d5618d53db4acd0e Mon Sep 17 00:00:00 2001
From: Mihaela Balutoiu <mbalutoiu@cloudbasesolutions.com>
Date: Thu, 7 May 2026 21:26:41 +0300
Subject: [PATCH] Add loading state to Export VM Inventory button

Signed-off-by: Mihaela Balutoiu <mbalutoiu@cloudbasesolutions.com>
---
 .../EndpointDetailsContent.tsx                | 15 +++++++----
 .../EndpointDetailsPage.tsx                   |  3 +++
 src/stores/EndpointStore.ts                   | 25 +++++++++++++------
 3 files changed, 30 insertions(+), 13 deletions(-)

diff --git a/src/components/modules/EndpointModule/EndpointDetailsContent/EndpointDetailsContent.tsx b/src/components/modules/EndpointModule/EndpointDetailsContent/EndpointDetailsContent.tsx
index fbb7d025..4d582817 100644
--- a/src/components/modules/EndpointModule/EndpointDetailsContent/EndpointDetailsContent.tsx
+++ b/src/components/modules/EndpointModule/EndpointDetailsContent/EndpointDetailsContent.tsx
@@ -23,6 +23,7 @@ import { Region } from "@src/@types/Region";
 import EndpointLogos from "@src/components/modules/EndpointModule/EndpointLogos";
 import { ThemePalette, ThemeProps } from "@src/components/Theme";
 import Button from "@src/components/ui/Button";
+import LoadingButton from "@src/components/ui/LoadingButton";
 import CopyMultilineValue from "@src/components/ui/CopyMultilineValue";
 import CopyValue from "@src/components/ui/CopyValue";
 import PasswordValue from "@src/components/ui/PasswordValue";
@@ -100,6 +101,7 @@ type Props = {
   transfers: TransferItem[];
   connectionInfoSchema: FieldType[];
   supportsInventoryExport?: boolean;
+  exportingInventoryCsv?: boolean;
   onDeleteClick: () => void;
   onValidateClick: () => void;
   onExportInventoryCsvClick?: () => void;
@@ -199,11 +201,14 @@ class EndpointDetailsContent extends React.Component<Props> {
             Validate Endpoint
           </Button>
         </MainButtons>
-        {this.props.supportsInventoryExport && (
-          <Button onClick={this.props.onExportInventoryCsvClick}>
-            Export VM Inventory
-          </Button>
-        )}
+        {this.props.supportsInventoryExport &&
+          (this.props.exportingInventoryCsv ? (
+            <LoadingButton>Export VM Inventory</LoadingButton>
+          ) : (
+            <Button onClick={this.props.onExportInventoryCsvClick}>
+              Export VM Inventory
+            </Button>
+          ))}
         <DeleteButton>
           <Button hollow alert onClick={this.props.onDeleteClick}>
             Delete Endpoint
diff --git a/src/components/smart/EndpointDetailsPage/EndpointDetailsPage.tsx b/src/components/smart/EndpointDetailsPage/EndpointDetailsPage.tsx
index 7523a951..7098ba52 100644
--- a/src/components/smart/EndpointDetailsPage/EndpointDetailsPage.tsx
+++ b/src/components/smart/EndpointDetailsPage/EndpointDetailsPage.tsx
@@ -302,6 +302,8 @@ class EndpointDetailsPage extends React.Component<Props, State> {
               action: () => {
                 this.handleExportInventoryCsvClick();
               },
+              loading: endpointStore.exportingInventoryCsv,
+              disabled: endpointStore.exportingInventoryCsv,
             },
           ]
         : []),
@@ -347,6 +349,7 @@ class EndpointDetailsPage extends React.Component<Props, State> {
               connectionInfo={endpointStore.connectionInfo}
               connectionInfoSchema={providerStore.connectionInfoSchema}
               supportsInventoryExport={this.supportsInventoryExport}
+              exportingInventoryCsv={endpointStore.exportingInventoryCsv}
               onDeleteClick={() => {
                 this.handleDeleteEndpointClick();
               }}
diff --git a/src/stores/EndpointStore.ts b/src/stores/EndpointStore.ts
index c8e6f73f..612c9731 100644
--- a/src/stores/EndpointStore.ts
+++ b/src/stores/EndpointStore.ts
@@ -66,6 +66,8 @@ class EndpointStore {
 
   @observable multiValidation: MultiValidationItem[] = [];
 
+  @observable exportingInventoryCsv = false;
+
   @action async getEndpoints(options?: {
     showLoading?: boolean;
     skipLog?: boolean;
@@ -199,14 +201,21 @@ class EndpointStore {
   }
 
   @action async exportInventoryCsv(endpoint: Endpoint): Promise<void> {
-    const csvContent = await EndpointSource.exportInventoryCsv(endpoint.id);
-    const now = new Date();
-    const pad = (n: number) => String(n).padStart(2, "0");
-    const d =
-      `${now.getFullYear()}-${pad(now.getMonth() + 1)}` +
-      `-${pad(now.getDate())}_${pad(now.getHours())}` +
-      `:${pad(now.getMinutes())}`;
-    DomUtils.download(csvContent, `vm_inventory_${endpoint.name}_${d}.csv`);
+    this.exportingInventoryCsv = true;
+    try {
+      const csvContent = await EndpointSource.exportInventoryCsv(endpoint.id);
+      const now = new Date();
+      const pad = (n: number) => String(n).padStart(2, "0");
+      const d =
+        `${now.getFullYear()}-${pad(now.getMonth() + 1)}` +
+        `-${pad(now.getDate())}_${pad(now.getHours())}` +
+        `:${pad(now.getMinutes())}`;
+      DomUtils.download(csvContent, `vm_inventory_${endpoint.name}_${d}.csv`);
+    } finally {
+      runInAction(() => {
+        this.exportingInventoryCsv = false;
+      });
+    }
   }
 
   @action async exportToJson(endpoint: Endpoint): Promise<void> {