weseek__growi/apps/app/src/client/components/Admin/ExportArchiveDataPage.tsx

import React, { type JSX, useCallback, useEffect, useState } from 'react';
import { useTranslation } from 'react-i18next';

import { apiDelete } from '~/client/util/apiv1-client';
import { apiv3Get } from '~/client/util/apiv3-client';
import { toastError, toastSuccess } from '~/client/util/toastr';
import { useAdminSocket } from '~/features/admin/states/socket-io';

import LabeledProgressBar from './Common/LabeledProgressBar';
import ArchiveFilesTable from './ExportArchiveData/ArchiveFilesTable';
import SelectCollectionsModal from './ExportArchiveData/SelectCollectionsModal';

const IGNORED_COLLECTION_NAMES = [
  'sessions',
  'rlflx',
  'yjs-writings',
  'transferkeys',
];

const ExportArchiveDataPage = (): JSX.Element => {
  const socket = useAdminSocket();
  const { t } = useTranslation('admin');

  const [collections, setCollections] = useState<any[]>([]);
  const [zipFileStats, setZipFileStats] = useState<any[]>([]);
  const [progressList, setProgressList] = useState<any[]>([]);
  const [isExportModalOpen, setExportModalOpen] = useState(false);
  const [isExporting, setExporting] = useState(false);
  const [isZipping, setZipping] = useState(false);
  const [isExported, setExported] = useState(false);

  const fetchData = useCallback(async () => {
    const [{ data: collectionsData }, { data: statusData }] = await Promise.all(
      [
        apiv3Get<{ collections: any[] }>('/mongo/collections', {}),
        apiv3Get<{
          status: {
            zipFileStats: any[];
            isExporting: boolean;
            progressList: any[];
          };
        }>('/export/status', {}),
      ],
    );

    // filter only not ignored collection names
    const filteredCollections = collectionsData.collections.filter(
      (collectionName) => {
        return !IGNORED_COLLECTION_NAMES.includes(collectionName);
      },
    );

    const { zipFileStats, isExporting, progressList } = statusData.status;
    setCollections(filteredCollections);
    setZipFileStats(zipFileStats);
    setExporting(isExporting);
    setProgressList(progressList);
  }, []);

  const setupWebsocketEventHandler = useCallback(() => {
    if (socket == null) {
      return () => {};
    }

    const onProgress = ({ currentCount, totalCount, progressList }) => {
      setExporting(true);
      setProgressList(progressList);
    };

    const onStartZipping = () => {
      setZipping(true);
    };

    const onTerminateForExport = ({ addedZipFileStat }) => {
      setExporting(false);
      setZipping(false);
      setExported(true);
      setZipFileStats((prev) => prev.concat([addedZipFileStat]));

      toastSuccess(`New Archive Data '${addedZipFileStat.fileName}' is added`);
    };

    // Add listeners
    socket.on('admin:onProgressForExport', onProgress);
    socket.on('admin:onStartZippingForExport', onStartZipping);
    socket.on('admin:onTerminateForExport', onTerminateForExport);

    // Cleanup listeners
    return () => {
      socket.off('admin:onProgressForExport', onProgress);
      socket.off('admin:onStartZippingForExport', onStartZipping);
      socket.off('admin:onTerminateForExport', onTerminateForExport);
    };
  }, [socket]);

  const onZipFileStatRemove = useCallback(async (fileName) => {
    try {
      await apiDelete(`/v3/export/${fileName}`, {});

      setZipFileStats((prev) =>
        prev.filter((stat) => stat.fileName !== fileName),
      );

      toastSuccess(`Deleted ${fileName}`);
    } catch (err) {
      toastError(err);
    }
  }, []);

  const exportingRequestedHandler = useCallback(() => {}, []);

  const renderProgressBarsForCollections = useCallback(() => {
    const cols = progressList.map((progressData) => {
      const { collectionName, currentCount, totalCount } = progressData;
      return (
        <div className="col-md-6" key={collectionName}>
          <LabeledProgressBar
            header={collectionName}
            currentCount={currentCount}
            totalCount={totalCount}
          />
        </div>
      );
    });

    return <div className="row px-3">{cols}</div>;
  }, [progressList]);

  const renderProgressBarForZipping = useCallback(() => {
    const showZippingBar = isZipping || isExported;

    if (!showZippingBar) {
      return <></>;
    }

    return (
      <div className="row px-3">
        <div className="col-md-12" key="progressBarForZipping">
          <LabeledProgressBar
            header="Zip Files"
            currentCount={1}
            totalCount={1}
            isInProgress={isZipping}
          />
        </div>
      </div>
    );
  }, [isExported, isZipping]);

  useEffect(() => {
    fetchData();
    const cleanupWebsocket = setupWebsocketEventHandler();

    return () => {
      if (cleanupWebsocket) cleanupWebsocket();
    };
  }, [fetchData, setupWebsocketEventHandler]);

  const showExportingData = (isExported || isExporting) && progressList != null;

  return (
    <div data-testid="admin-export-archive-data">
      <h2>{t('export_management.export_archive_data')}</h2>

      <button
        type="button"
        className="btn btn-outline-secondary"
        disabled={isExporting}
        onClick={() => setExportModalOpen(true)}
      >
        {t('export_management.create_new_archive_data')}
      </button>

      {showExportingData && (
        <div className="mt-5">
          <h3>{t('export_management.exporting_collection_list')}</h3>
          {renderProgressBarsForCollections()}
          {renderProgressBarForZipping()}
        </div>
      )}

      <div className="mt-5">
        <h3 className="mb-3">{t('export_management.exported_data_list')}</h3>
        <ArchiveFilesTable
          zipFileStats={zipFileStats}
          onZipFileStatRemove={onZipFileStatRemove}
        />
      </div>

      <SelectCollectionsModal
        isOpen={isExportModalOpen}
        onExportingRequested={exportingRequestedHandler}
        onClose={() => setExportModalOpen(false)}
        collections={collections}
      />
    </div>
  );
};

export default ExportArchiveDataPage;