subprojects/frontend/src/editor/GenerateButton.tsx


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118

/*
 * SPDX-FileCopyrightText: 2021-2023 The Refinery Authors <https://refinery.tools/>
 *
 * SPDX-License-Identifier: EPL-2.0
 */

import CancelIcon from '@mui/icons-material/Cancel';
import CloseIcon from '@mui/icons-material/Close';
import PlayArrowIcon from '@mui/icons-material/PlayArrow';
import { observer } from 'mobx-react-lite';

import AnimatedButton from './AnimatedButton';
import type EditorStore from './EditorStore';

const GENERATE_LABEL = 'Generate';

const GenerateButton = observer(function GenerateButton({
  editorStore,
  hideWarnings,
}: {
  editorStore: EditorStore | undefined;
  hideWarnings?: boolean | undefined;
}): JSX.Element {
  if (editorStore === undefined) {
    return (
      <AnimatedButton color="inherit" disabled>
        Loading&hellip;
      </AnimatedButton>
    );
  }

  const {
    delayedErrors: { analyzing, errorCount, warningCount, semanticsError },
    generating,
  } = editorStore;

  if (analyzing) {
    return (
      <AnimatedButton color="inherit" disabled>
        Analyzing&hellip;
      </AnimatedButton>
    );
  }

  if (generating) {
    return (
      <AnimatedButton
        color="inherit"
        onClick={() => editorStore.cancelModelGeneration()}
        startIcon={<CloseIcon />}
      >
        Cancel
      </AnimatedButton>
    );
  }

  if (semanticsError !== undefined && editorStore.opened) {
    return (
      <AnimatedButton
        color="error"
        disabled
        startIcon={<CancelIcon />}
        sx={(theme) => ({
          '&.Mui-disabled': {
            color: `${theme.palette.error.main} !important`,
          },
        })}
      >
        Analysis error
      </AnimatedButton>
    );
  }

  const diagnostics: string[] = [];
  if (errorCount > 0) {
    diagnostics.push(`${errorCount} error${errorCount === 1 ? '' : 's'}`);
  }
  if (!(hideWarnings ?? false) && warningCount > 0) {
    diagnostics.push(`${warningCount} warning${warningCount === 1 ? '' : 's'}`);
  }
  const summary = diagnostics.join(' and ');

  if (errorCount > 0) {
    return (
      <AnimatedButton
        aria-label={`Select next diagnostic out of ${summary}`}
        onClick={() => editorStore.nextDiagnostic()}
        color="error"
        startIcon={<CancelIcon />}
      >
        {summary}
      </AnimatedButton>
    );
  }

  return (
    <AnimatedButton
      disabled={!editorStore.opened}
      color={warningCount > 0 ? 'warning' : 'primary'}
      startIcon={<PlayArrowIcon />}
      onClick={(event) => {
        if (event.shiftKey) {
          editorStore.startModelGeneration(1);
        } else {
          editorStore.startModelGeneration();
        }
      }}
    >
      {summary === '' ? GENERATE_LABEL : `${GENERATE_LABEL} (${summary})`}
    </AnimatedButton>
  );
});

GenerateButton.defaultProps = {
  hideWarnings: false,
};

export default GenerateButton;