added contest tabs to ProblemSetGenerator.

Author: Eagle-Konbu

atcoder-problems-frontend/src/components/ProblemSetGenerator.tsx

@@ -1,5 +1,6 @@
 import {
   Button,
+  ButtonGroup,
   Col,
   DropdownItem,
   DropdownMenu,
@@ -90,6 +91,15 @@ export const ProblemSetGenerator: React.FC<Props> = (props) => {
   );
   const [excludeExperimental, setExcludeExperimental] = useState(false);
   const [excludeOption, setExcludeOption] = useState<ExcludeOption>("Exclude");
+  const [contestTypeOption, setContestTypeOption] = useState({
+    ABC: true,
+    ARC: true,
+    AGC: true,
+    ABC_Like: true,
+    ARC_Like: true,
+    AGC_Like: true,
+    Other_Sponsored: true,
+  });
   const [selectedPreset, setSelectedPreset] = useState(ABC_PRESET);
   const problems = useProblems() ?? [];
   const problemModels = useProblemModelMap();
@@ -98,6 +108,16 @@ export const ProblemSetGenerator: React.FC<Props> = (props) => {
   const alreadySolvedProblemIds = new Set(submissions.map((s) => s.problem_id));
   const lastSolvedTimeMap = getLastSolvedTimeMap(submissions);
 
+  const contestTypeKeyToDisplayName = (key: string) => {
+    if (key.includes("Like")) {
+      return key.replace("_", "-");
+    } else if (key === "Other_Sponsored") {
+      return "Other Sponsored";
+    } else {
+      return key;
+    }
+  };
+
   return (
     <Form className={"w-100"}>
       <FormGroup row>
@@ -142,6 +162,31 @@ export const ProblemSetGenerator: React.FC<Props> = (props) => {
           </InputGroup>
         </Col>
       </FormGroup>
+      <FormGroup row>
+        <Col sm={6}>
+          <Label>Include / Exclude Contest Types</Label>
+          <InputGroup>
+            <ButtonGroup>
+              {Object.keys(contestTypeOption).map((contestType) => {
+                return (
+                  <Button
+                    key={contestType}
+                    active={contestTypeOption[contestType] as boolean}
+                    onClick={(): void => {
+                      setContestTypeOption({
+                        ...contestTypeOption,
+                        [contestType]: !contestTypeOption[contestType],
+                      });
+                    }}
+                  >
+                    {contestTypeKeyToDisplayName(contestType)}
+                  </Button>
+                );
+              })}
+            </ButtonGroup>
+          </InputGroup>
+        </Col>
+      </FormGroup>
       <FormGroup row>
         <Col sm={6}>
           <Label>Difficulty Adjustment Preset</Label>