sentry/static/app/views/performance/transactionSummary/transactionThresholdButton.tsx

import {Component} from 'react';

import {addErrorMessage} from 'sentry/actionCreators/indicator';
import {openModal} from 'sentry/actionCreators/modal';
import type {Client} from 'sentry/api';
import {Button} from 'sentry/components/button';
import {IconSettings} from 'sentry/icons';
import {t} from 'sentry/locale';
import type {Organization} from 'sentry/types/organization';
import type {Project} from 'sentry/types/project';
import {defined} from 'sentry/utils';
import type EventView from 'sentry/utils/discover/eventView';
import withApi from 'sentry/utils/withApi';
import withProjects from 'sentry/utils/withProjects';

import type {TransactionThresholdMetric} from './transactionThresholdModal';
import TransactionThresholdModal, {modalCss} from './transactionThresholdModal';

type Props = {
  api: Client;
  eventView: EventView;
  organization: Organization;
  projects: Project[];
  transactionName: string;
  onChangeThreshold?: (threshold: number, metric: TransactionThresholdMetric) => void;
};

type State = {
  loadingThreshold: boolean;
  transactionThreshold: number | undefined;
  transactionThresholdMetric: TransactionThresholdMetric | undefined;
};

class TransactionThresholdButton extends Component<Props, State> {
  state: State = {
    transactionThreshold: undefined,
    transactionThresholdMetric: undefined,
    loadingThreshold: false,
  };

  componentDidMount() {
    this.fetchTransactionThreshold();
  }

  getProject() {
    const {projects, eventView} = this.props;
    if (!defined(eventView)) {
      return undefined;
    }

    const projectId = String(eventView.project[0]);
    const project = projects.find(proj => proj.id === projectId);

    return project;
  }

  fetchTransactionThreshold = () => {
    const {api, organization, transactionName} = this.props;

    const project = this.getProject();
    if (!defined(project)) {
      return;
    }
    const transactionThresholdUrl = `/organizations/${organization.slug}/project-transaction-threshold-override/`;

    this.setState({loadingThreshold: true});

    api
      .requestPromise(transactionThresholdUrl, {
        method: 'GET',
        includeAllArgs: true,
        query: {
          project: project.id,
          transaction: transactionName,
        },
      })
      .then(([data]) => {
        this.setState({
          loadingThreshold: false,
          transactionThreshold: data.threshold,
          transactionThresholdMetric: data.metric,
        });
      })
      .catch(() => {
        const projectThresholdUrl = `/projects/${organization.slug}/${project.slug}/transaction-threshold/configure/`;
        this.props.api
          .requestPromise(projectThresholdUrl, {
            method: 'GET',
            includeAllArgs: true,
            query: {
              project: project.id,
            },
          })
          .then(([data]) => {
            this.setState({
              loadingThreshold: false,
              transactionThreshold: data.threshold,
              transactionThresholdMetric: data.metric,
            });
          })
          .catch(err => {
            this.setState({loadingThreshold: false});
            const errorMessage = err.responseJSON?.threshold ?? null;
            addErrorMessage(errorMessage);
          });
      });
  };

  onChangeThreshold(threshold: number, metric: TransactionThresholdMetric) {
    const {onChangeThreshold} = this.props;
    this.setState({
      transactionThreshold: threshold,
      transactionThresholdMetric: metric,
    });

    if (defined(onChangeThreshold)) {
      onChangeThreshold(threshold, metric);
    }
  }

  openModal() {
    const {organization, transactionName, eventView} = this.props;

    const {transactionThreshold, transactionThresholdMetric} = this.state;

    openModal(
      modalProps => (
        <TransactionThresholdModal
          {...modalProps}
          organization={organization}
          transactionName={transactionName}
          eventView={eventView}
          transactionThreshold={transactionThreshold}
          transactionThresholdMetric={transactionThresholdMetric}
          onApply={(threshold, metric) => this.onChangeThreshold(threshold, metric)}
        />
      ),
      {modalCss, closeEvents: 'escape-key'}
    );
  }

  render() {
    const {loadingThreshold} = this.state;
    return (
      <Button
        size="sm"
        onClick={() => this.openModal()}
        icon={<IconSettings />}
        disabled={loadingThreshold}
        aria-label={t('Settings')}
        data-test-id="set-transaction-threshold"
      />
    );
  }
}

export default withApi(withProjects(TransactionThresholdButton));