From 58cba87ddbce03a92172ab127d72bc88d40bc422 Mon Sep 17 00:00:00 2001
From: Krzysztof <krzysztof.kowalczyk@elastic.co>
Date: Fri, 18 Oct 2024 14:13:02 +0200
Subject: [PATCH] Add type to GeneratePdfResponse

---
 .../server/formats/pdf/pdf_maker/worker.ts    | 34 +++++++++++--------
 1 file changed, 20 insertions(+), 14 deletions(-)

diff --git a/x-pack/plugins/screenshotting/server/formats/pdf/pdf_maker/worker.ts b/x-pack/plugins/screenshotting/server/formats/pdf/pdf_maker/worker.ts
index 98c10598703ae..ba6793621c682 100644
--- a/x-pack/plugins/screenshotting/server/formats/pdf/pdf_maker/worker.ts
+++ b/x-pack/plugins/screenshotting/server/formats/pdf/pdf_maker/worker.ts
@@ -34,21 +34,23 @@ export interface GeneratePdfRequest {
   data: GenerateReportRequestData;
 }
 
-export type GeneratePdfResponse = SuccessResponse | ErrorResponse;
-
-export interface SuccessResponse {
-  error?: undefined;
-  data: {
-    buffer: Uint8Array;
-    metrics: {
-      pages: number;
-    };
+export interface GeneratePdfData {
+  buffer: Uint8Array;
+  metrics: {
+    pages: number;
   };
 }
 
-export interface ErrorResponse {
-  error: string;
-  data: null;
+export enum GeneratePdfResponseType {
+  Log,
+  Data,
+}
+
+export interface GeneratePdfResponse {
+  type: GeneratePdfResponseType;
+  data?: GeneratePdfData;
+  error?: string;
+  message?: string;
 }
 
 if (!isMainThread) {
@@ -133,7 +135,8 @@ async function execute({ data: { layout, logo, title, content } }: GeneratePdfRe
       pdfDoc.end();
     });
 
-    const successResponse: SuccessResponse = {
+    const successResponse: GeneratePdfResponse = {
+      type: GeneratePdfResponseType.Data,
       data: {
         buffer,
         metrics: {
@@ -143,7 +146,10 @@ async function execute({ data: { layout, logo, title, content } }: GeneratePdfRe
     };
     port.postMessage(successResponse, [buffer.buffer /* Transfer buffer instead of copying */]);
   } catch (error) {
-    const errorResponse: ErrorResponse = { error: error.message, data: null };
+    const errorResponse: GeneratePdfResponse = {
+      type: GeneratePdfResponseType.Data,
+      error: error.message,
+    };
     port.postMessage(errorResponse);
   } finally {
     process.nextTick(() => {