package nl.connectedit.swiss.mapper;

import lombok.RequiredArgsConstructor;
import nl.connectedit.swiss.domain.entity.Team;
import nl.connectedit.swiss.dto.TeamDto;
import org.springframework.stereotype.Component;

@Component
@RequiredArgsConstructor
public class TeamMapper implements DtoMapper<Team, TeamDto> {

  private final PlayerMapper playerMapper;

  @Override
  public TeamDto toDto(Team team) {
    var teamDto = new TeamDto();
    teamDto.setId(team.getId());
    teamDto.setPlayer1(playerMapper.toDto(team.getPlayer1()));

    if (team.getPlayer2() != null) {
      teamDto.setPlayer2(playerMapper.toDto(team.getPlayer2()));
    }

    return teamDto;
  }
}