v3/packages/design-system/molecules/demo-picker/DemoPicker.tsx

import { Role } from '@roleypoly/design-system/atoms/role';
import { Role as RPCRole } from '@roleypoly/types';
import { demoData } from '@roleypoly/types/demoData';
import * as React from 'react';
import styled from 'styled-components';

const Container = styled.div`
  display: flex;
  flex-wrap: wrap;
  justify-content: center;
  align-items: center;
  align-content: center;
  height: 95px;
`;

const RoleWrap = styled.div`
  padding: 2.5px;
  display: inline-block;
`;

export const DemoPicker = () => {
  const [selectedStates, setSelectedStates] = React.useState<{
    [key in RPCRole['id']]: boolean;
  }>(demoData.reduce((acc, role) => ({ ...acc, [role.id]: false }), {}));

  return (
    <Container>
      {demoData.map((role) => (
        <RoleWrap key={`role${role.id}`}>
          <Role
            role={role}
            selected={selectedStates[role.id]}
            onClick={() => {
              setSelectedStates({
                ...selectedStates,
                [role.id]: !selectedStates[role.id],
              });
            }}
          />
        </RoleWrap>
      ))}
    </Container>
  );
};