weseek__growi/apps/app/src/client/components/Admin/ElasticsearchManagement/RebuildIndexControls.jsx

import React from 'react';
import { useTranslation } from 'next-i18next';
import PropTypes from 'prop-types';

import { useAdminSocket } from '~/features/admin/states/socket-io';
import { SocketEventName } from '~/interfaces/websocket';

import LabeledProgressBar from '../Common/LabeledProgressBar';

class RebuildIndexControls extends React.Component {
  constructor(props) {
    super(props);

    this.state = {
      total: 0,
      current: 0,
    };
  }

  componentDidMount() {
    this.initWebSockets();
  }

  initWebSockets() {
    const { socket } = this.props;

    if (socket != null) {
      socket.on(SocketEventName.AddPageProgress, (data) => {
        this.setState({
          total: data.totalCount,
          current: data.count,
        });
      });

      socket.on(SocketEventName.FinishAddPage, (data) => {
        this.setState({
          total: data.totalCount,
          current: data.count,
        });
      });
    }
  }

  renderProgressBar() {
    const { isRebuildingProcessing, isRebuildingCompleted } = this.props;
    const { total, current } = this.state;
    const showProgressBar = isRebuildingProcessing || isRebuildingCompleted;

    if (!showProgressBar) {
      return null;
    }

    const header = isRebuildingCompleted ? 'Completed' : 'Processing..';

    return (
      <div className="mb-3">
        <LabeledProgressBar
          header={header}
          currentCount={current}
          totalCount={total}
        />
      </div>
    );
  }

  render() {
    const { t, isNormalized, isRebuildingProcessing } = this.props;

    const isEnabled = isNormalized && !isRebuildingProcessing;

    return (
      <>
        {this.renderProgressBar()}

        <button
          type="submit"
          className="btn btn-primary"
          onClick={() => {
            this.props.onRebuildingRequested();
          }}
          disabled={!isEnabled}
        >
          {t('full_text_search_management.rebuild_button')}
        </button>

        <p className="form-text text-muted">
          {t('full_text_search_management.rebuild_description_1')}
          <br />
          {t('full_text_search_management.rebuild_description_2')}
          <br />
        </p>
      </>
    );
  }
}

const RebuildIndexControlsFC = (props) => {
  const { t } = useTranslation('admin');
  const socket = useAdminSocket();
  return <RebuildIndexControls t={t} socket={socket} {...props} />;
};

RebuildIndexControls.propTypes = {
  t: PropTypes.func.isRequired, // i18next

  isRebuildingProcessing: PropTypes.bool.isRequired,
  isRebuildingCompleted: PropTypes.bool.isRequired,

  isNormalized: PropTypes.bool,
  onRebuildingRequested: PropTypes.func.isRequired,
  socket: PropTypes.object,
};

export default RebuildIndexControlsFC;