From ef0a8b50135408fdb5070bec96644c1435111acf Mon Sep 17 00:00:00 2001
From: Bryan Frimin <bryan@getprobo.com>
Date: Thu, 17 Jul 2025 00:03:14 +0200
Subject: [PATCH] Fix download button removed

Signed-off-by: Bryan Frimin <bryan@getprobo.com>
---
 .../documents/DocumentDetailPage.tsx          | 68 ++++++++++++--
 ...mentDetailPageExportPDFMutation.graphql.ts | 92 +++++++++++++++++++
 2 files changed, 151 insertions(+), 9 deletions(-)
 create mode 100644 apps/console/src/pages/organizations/documents/__generated__/DocumentDetailPageExportPDFMutation.graphql.ts

diff --git a/apps/console/src/pages/organizations/documents/DocumentDetailPage.tsx b/apps/console/src/pages/organizations/documents/DocumentDetailPage.tsx
index 614f7e2e4..71f9482a0 100644
--- a/apps/console/src/pages/organizations/documents/DocumentDetailPage.tsx
+++ b/apps/console/src/pages/organizations/documents/DocumentDetailPage.tsx
@@ -17,6 +17,7 @@ import type {
   DocumentDetailPageDocumentFragment$data,
   DocumentDetailPageDocumentFragment$key,
 } from "./__generated__/DocumentDetailPageDocumentFragment.graphql";
+import type { DocumentDetailPageExportPDFMutation } from "./__generated__/DocumentDetailPageExportPDFMutation.graphql";
 import { useTranslate } from "@probo/i18n";
 import {
   ActionDropdown,
@@ -27,6 +28,7 @@ import {
   Drawer,
   Dropdown,
   DropdownItem,
+  IconArrowDown,
   IconCheckmark1,
   IconChevronDown,
   IconClock,
@@ -116,12 +118,22 @@ const publishDocumentVersionMutation = graphql`
   }
 `;
 
+const exportDocumentVersionPDFMutation = graphql`
+  mutation DocumentDetailPageExportPDFMutation(
+    $input: ExportDocumentVersionPDFInput!
+  ) {
+    exportDocumentVersionPDF(input: $input) {
+      data
+    }
+  }
+`;
+
 export default function DocumentDetailPage(props: Props) {
   const { versionId } = useParams<{ versionId?: string }>();
   const node = usePreloadedQuery(documentNodeQuery, props.queryRef).node;
-  const document = useFragment(
+  const document = useFragment<DocumentDetailPageDocumentFragment$key>(
     documentFragment,
-    node as DocumentDetailPageDocumentFragment$key,
+    node
   );
   const { __, dateFormat } = useTranslate();
   const organizationId = useOrganizationId();
@@ -138,9 +150,17 @@ export default function DocumentDetailPage(props: Props) {
     {
       successMessage: __("Document published successfully."),
       errorMessage: __("Failed to publish document. Please try again."),
-    },
+    }
   );
   const [deleteDocument, isDeleting] = useDeleteDocumentMutation();
+  const [exportDocumentVersionPDF, isExporting] =
+    useMutationWithToasts<DocumentDetailPageExportPDFMutation>(
+      exportDocumentVersionPDFMutation,
+      {
+        successMessage: __("PDF download started."),
+        errorMessage: __("Failed to generate PDF. Please try again."),
+      }
+    );
   const versionConnectionId = document.versions.__id;
 
   usePageTitle(document.title);
@@ -156,7 +176,7 @@ export default function DocumentDetailPage(props: Props) {
           props.queryRef.environment,
           documentNodeQuery,
           props.queryRef.variables,
-          { fetchPolicy: "network-only" },
+          { fetchPolicy: "network-only" }
         );
       },
     });
@@ -170,7 +190,7 @@ export default function DocumentDetailPage(props: Props) {
         new Promise<void>((resolve) => {
           const connectionId = ConnectionHandler.getConnectionID(
             organizationId,
-            DocumentsConnectionKey,
+            DocumentsConnectionKey
           );
           deleteDocument({
             variables: {
@@ -186,14 +206,32 @@ export default function DocumentDetailPage(props: Props) {
       {
         message: sprintf(
           __(
-            'This will permanently delete the document "%s". This action cannot be undone.',
+            'This will permanently delete the document "%s". This action cannot be undone.'
           ),
-          document.title,
+          document.title
         ),
-      },
+      }
     );
   };
 
+  const handleDownloadPdf = () => {
+    exportDocumentVersionPDF({
+      variables: {
+        input: { documentVersionId: currentVersion.id },
+      },
+      onCompleted: (data) => {
+        if (data.exportDocumentVersionPDF?.data) {
+          const link = window.document.createElement("a");
+          link.href = data.exportDocumentVersionPDF.data;
+          link.download = `${document.title}-v${currentVersion.version}.pdf`;
+          window.document.body.appendChild(link);
+          link.click();
+          window.document.body.removeChild(link);
+        }
+      },
+    });
+  };
+
   const updateDialogRef = useRef<{ open: () => void }>(null);
   const controlsCount = document.controlsInfo.totalCount;
   const urlPrefix = versionId
@@ -257,6 +295,13 @@ export default function DocumentDetailPage(props: Props) {
               >
                 {isDraft ? __("Edit draft document") : __("Create new draft")}
               </DropdownItem>
+              <DropdownItem
+                onClick={handleDownloadPdf}
+                icon={IconArrowDown}
+                disabled={isExporting}
+              >
+                {__("Download PDF")}
+              </DropdownItem>
               <DropdownItem
                 variant="danger"
                 icon={IconTrashCan}
@@ -357,7 +402,12 @@ function VersionItem({
       {...props}
     >
       <div className="flex gap-3 w-full overflow-hidden">
-                 <div className={clsx("flex-shrink-0 flex items-center justify-center size-10", active && "bg-active rounded")}>
+        <div
+          className={clsx(
+            "flex-shrink-0 flex items-center justify-center size-10",
+            active && "bg-active rounded"
+          )}
+        >
           <div className="text-base text-txt-primary whitespace-nowrap font-bold text-center">
             {version.version}
           </div>
diff --git a/apps/console/src/pages/organizations/documents/__generated__/DocumentDetailPageExportPDFMutation.graphql.ts b/apps/console/src/pages/organizations/documents/__generated__/DocumentDetailPageExportPDFMutation.graphql.ts
new file mode 100644
index 000000000..fc54352aa
--- /dev/null
+++ b/apps/console/src/pages/organizations/documents/__generated__/DocumentDetailPageExportPDFMutation.graphql.ts
@@ -0,0 +1,92 @@
+/**
+ * @generated SignedSource<<804ed692b3f8c67b2a73ca09260e5d25>>
+ * @lightSyntaxTransform
+ * @nogrep
+ */
+
+/* tslint:disable */
+/* eslint-disable */
+// @ts-nocheck
+
+import { ConcreteRequest } from 'relay-runtime';
+export type ExportDocumentVersionPDFInput = {
+  documentVersionId: string;
+};
+export type DocumentDetailPageExportPDFMutation$variables = {
+  input: ExportDocumentVersionPDFInput;
+};
+export type DocumentDetailPageExportPDFMutation$data = {
+  readonly exportDocumentVersionPDF: {
+    readonly data: string;
+  };
+};
+export type DocumentDetailPageExportPDFMutation = {
+  response: DocumentDetailPageExportPDFMutation$data;
+  variables: DocumentDetailPageExportPDFMutation$variables;
+};
+
+const node: ConcreteRequest = (function(){
+var v0 = [
+  {
+    "defaultValue": null,
+    "kind": "LocalArgument",
+    "name": "input"
+  }
+],
+v1 = [
+  {
+    "alias": null,
+    "args": [
+      {
+        "kind": "Variable",
+        "name": "input",
+        "variableName": "input"
+      }
+    ],
+    "concreteType": "ExportDocumentVersionPDFPayload",
+    "kind": "LinkedField",
+    "name": "exportDocumentVersionPDF",
+    "plural": false,
+    "selections": [
+      {
+        "alias": null,
+        "args": null,
+        "kind": "ScalarField",
+        "name": "data",
+        "storageKey": null
+      }
+    ],
+    "storageKey": null
+  }
+];
+return {
+  "fragment": {
+    "argumentDefinitions": (v0/*: any*/),
+    "kind": "Fragment",
+    "metadata": null,
+    "name": "DocumentDetailPageExportPDFMutation",
+    "selections": (v1/*: any*/),
+    "type": "Mutation",
+    "abstractKey": null
+  },
+  "kind": "Request",
+  "operation": {
+    "argumentDefinitions": (v0/*: any*/),
+    "kind": "Operation",
+    "name": "DocumentDetailPageExportPDFMutation",
+    "selections": (v1/*: any*/)
+  },
+  "params": {
+    "cacheID": "4c17d48a790ff5f1617c7db604766b83",
+    "id": null,
+    "metadata": {},
+    "name": "DocumentDetailPageExportPDFMutation",
+    "operationKind": "mutation",
+    "text": "mutation DocumentDetailPageExportPDFMutation(\n  $input: ExportDocumentVersionPDFInput!\n) {\n  exportDocumentVersionPDF(input: $input) {\n    data\n  }\n}\n"
+  }
+};
+})();
+
+(node as any).hash = "e8854650389c3ebd1020ec278d8be831";
+
+export default node;