refactor: Improve reuseability of export-config.controller

bluehalo · Aug 8, 2024 · 40c30c1 · 40c30c1
1 parent b241a12
commit 40c30c1
Showing 1 changed file with 20 additions and 19 deletions.
diff --git a/src/app/core/export/export-config.controller.ts b/src/app/core/export/export-config.controller.ts
@@ -49,24 +49,7 @@ export const exportCSV = (
 	data: Array<unknown> | Readable
 ) => {
 	if (null !== data) {
-		exportStream(
-			req,
-			res,
-			filename,
-			'text/csv',
-			buildExportStream(
-				data,
-				(stream) => () => {
-					if (Array.isArray(data)) {
-						data.forEach((row) => {
-							stream.push(row);
-						});
-					}
-					stream.push(null);
-				},
-				[csvStream.streamToCsv(columns)]
-			)
-		);
+		exportStream(req, res, filename, 'text/csv', buildCSVStream(data, columns));
 	}
 };
 
@@ -94,6 +77,24 @@ export const exportPlaintext = (req, res, filename: string, text: string) => {
 	}
 };
 
+export const buildCSVStream = (
+	data: Readable | unknown,
+	columns: ExportColumnDef[]
+) => {
+	return buildExportStream(
+		data,
+		(stream) => () => {
+			if (Array.isArray(data)) {
+				data.forEach((row) => {
+					stream.push(row);
+				});
+			}
+			stream.push(null);
+		},
+		[csvStream.streamToCsv(columns)]
+	);
+};
+
 /**
  * Build a readable stream from data and pipe through a chain of transform streams
  *
@@ -167,7 +168,7 @@ const buildExportStream = (
  * @param contentType
  * @param stream
  */
-const exportStream = (
+export const exportStream = (
 	req,
 	res,
 	fileName: string,