insomnia/packages/insomnia-app/app/ui/components/modals/ask-modal.tsx

import React, { PureComponent } from 'react';
import { autoBindMethodsForReact } from 'class-autobind-decorator';
import { AUTOBIND_CFG } from '../../../common/constants';
import Modal from '../base/modal';
import ModalBody from '../base/modal-body';
import ModalHeader from '../base/modal-header';
import ModalFooter from '../base/modal-footer';

interface State {
  title: string;
  message: string;
  yesText: string;
  noText: string;
  loading: boolean;
}

@autoBindMethodsForReact(AUTOBIND_CFG)
class AskModal extends PureComponent<{}, State> {
  state: State = {
    title: '',
    message: '',
    yesText: 'Yes',
    noText: 'No',
    loading: false,
  }

  modal: Modal | null = null;
  yesButton: HTMLButtonElement | null = null;
  _doneCallback: Function = () => {};
  _promiseCallback: Function = () => {};

  _setModalRef(m: Modal) {
    this.modal = m;
  }

  _setYesButtonRef(n: HTMLButtonElement) {
    this.yesButton = n;
  }

  async _handleYes() {
    this.setState({
      loading: true,
    });

    if (this._doneCallback) {
      // Wait for the callback to finish before closing
      await this._doneCallback(true);
    }

    this._promiseCallback(true);

    this.hide();
  }

  _handleNo() {
    this.hide();
    this?._doneCallback(false);

    this._promiseCallback(false);
  }

  hide() {
    this.modal?.hide();
  }

  show(options: any = {}) {
    const { title, message, onDone, yesText, noText } = options;
    this._doneCallback = onDone;
    this.setState({
      title: title || 'Confirm',
      message: message || 'No message provided',
      yesText: yesText || 'Yes',
      noText: noText || 'No',
      loading: false,
    });
    this.modal?.show();
    setTimeout(() => {
      this.yesButton && this.yesButton.focus();
    }, 100);
    return new Promise(resolve => {
      this._promiseCallback = resolve;
    });
  }

  render() {
    const { message, title, yesText, noText, loading } = this.state;
    return (
      <Modal noEscape ref={this._setModalRef} closeOnKeyCodes={[13]}>
        <ModalHeader>{title || 'Confirm?'}</ModalHeader>
        <ModalBody className="wide pad">{message}</ModalBody>
        <ModalFooter>
          <div>
            <button className="btn" onClick={this._handleNo}>
              {noText}
            </button>
            <button
              ref={this._setYesButtonRef}
              className="btn"
              onClick={this._handleYes}
              disabled={loading}
            >
              {loading && <i className="fa fa-refresh fa-spin" />} {yesText}
            </button>
          </div>
        </ModalFooter>
      </Modal>
    );
  }
}

export default AskModal;