add an option to save all messages as csv/json, Issue kafbat#688

Author: alexanderlz

frontend/src/components/Topics/Topic/Messages/MessagesTable.tsx

@@ -7,6 +7,9 @@ import { Button } from 'components/common/Button/Button';
 import * as S from 'components/common/NewTable/Table.styled';
 import { usePaginateTopics, useIsLiveMode } from 'lib/hooks/useMessagesFilters';
 import { useMessageFiltersStore } from 'lib/hooks/useMessageFiltersStore';
+import useDataSaver from 'lib/hooks/useDataSaver';
+import Select from 'components/common/Select/Select';
+import { SelectOption } from 'components/common/Select/Select';
 
 import PreviewModal from './PreviewModal';
 import Message, { PreviewFilter } from './Message';
@@ -16,6 +19,33 @@ export interface MessagesTableProps {
   isFetching: boolean;
 }
 
+interface MessageData {
+  Value: string | undefined;
+  Offset: number;
+  Key: string | undefined;
+  Partition: number;
+  Headers: { [key: string]: string | undefined; } | undefined;
+  Timestamp: Date;
+}
+
+type DownloadFormat = 'json' | 'csv';
+
+function padCurrentDateTimeString(): string {
+  const now: Date = new Date();
+
+  const year: string = now.getFullYear().toString();
+  const month: string = (now.getMonth() + 1).toString().padStart(2, '0');
+  const day: string = now.getDate().toString().padStart(2, '0');
+  const hours: string = now.getHours().toString().padStart(2, '0');
+  const minutes: string = now.getMinutes().toString().padStart(2, '0');
+  const seconds: string = now.getSeconds().toString().padStart(2, '0');
+
+  const dateTimeString: string = `${year}-${month}-${day}_${hours}-${minutes}-${seconds}`;
+
+  return `_${dateTimeString}`;
+}
+
+
 const MessagesTable: React.FC<MessagesTableProps> = ({
   messages,
   isFetching,
@@ -28,8 +58,80 @@ const MessagesTable: React.FC<MessagesTableProps> = ({
   const nextCursor = useMessageFiltersStore((state) => state.nextCursor);
   const isLive = useIsLiveMode();
 
+  const [selectedFormat, setSelectedFormat] = useState<DownloadFormat>('json');
+
+  const formatOptions: SelectOption<DownloadFormat>[] = [
+    { label: 'JSON', value: 'json' },
+    { label: 'CSV', value: 'csv' }
+  ];
+
+  const handleFormatSelect = (format: DownloadFormat) => {
+    setSelectedFormat(format);
+  };
+
+  const handleDownload = () => {
+
+  const savedMessagesJson: MessageData[] = messages.map(message => ({
+    Value: message.content,
+    Offset: message.offset,
+    Key: message.key,
+    Partition: message.partition,
+    Headers: message.headers,
+    Timestamp: message.timestamp,
+  }));
+
+  const convertToCSV = (messages: MessageData[]) => {
+    const headers = ['Value', 'Offset', 'Key', 'Partition', 'Headers', 'Timestamp'] as const;
+    type Header = typeof headers[number];
+
+    const rows = messages.map(msg =>
+      headers.map(header => {
+        const value = msg[header];
+        if (header === 'Headers') {
+          return JSON.stringify(value || {});
+        }
+        return String(value ?? '');
+      }).join(',')
+    );
+
+    return [headers.join(','), ...rows].join('\n');
+  };
+
+    const baseFileName = 'topic-messages'+padCurrentDateTimeString();
+    const jsonSaver = useDataSaver(baseFileName+'.json', JSON.stringify(savedMessagesJson, null, '\t'));
+    const csvSaver = useDataSaver(baseFileName+'.csv', convertToCSV(savedMessagesJson));
+
+    if (selectedFormat === 'json') {
+      jsonSaver.saveFile();
+    } else {
+      csvSaver.saveFile();
+    }
+  };
+
   return (
     <div style={{ position: 'relative' }}>
+      <div style={{ display: 'flex', gap: '8px', marginLeft: '1rem', marginBottom: '1rem' }}>
+        <Select<DownloadFormat>
+          id="download-format"
+          name="download-format"
+          onChange={handleFormatSelect}
+          options={formatOptions}
+          value={selectedFormat}
+	  minWidth="70px"
+	  selectSize="M"
+          placeholder="Select format to download"
+          disabled={isFetching || messages.length === 0}
+        />
+        <Button
+          disabled={isFetching || messages.length === 0}
+          buttonType="secondary"
+          buttonSize="M"
+          onClick={handleDownload}
+        >
+          Download All Messages
+        </Button>
+      </div>
+
       {previewFor !== null && (
         <PreviewModal
           values={previewFor === 'key' ? keyFilters : contentFilters}