package nl.connectedit.swiss.mapper;

import lombok.RequiredArgsConstructor;
import nl.connectedit.swiss.domain.entity.Registration;
import nl.connectedit.swiss.dto.RegistrationDto;
import org.springframework.stereotype.Component;

@Component
@RequiredArgsConstructor
public class RegistrationMapper implements DtoMapper<Registration, RegistrationDto>, EntityMapper<RegistrationDto, Registration> {

    private final PlayerMapper playerMapper;

    @Override
    public Registration toEntity(RegistrationDto registrationDto) {
        Registration registration = new Registration();
        registration.setId(registrationDto.getId());
        registration.setPlayer(playerMapper.toEntity(registrationDto.getPlayer()));
        if (registrationDto.getPartner() != null) {
            registration.setPartner(playerMapper.toEntity(registrationDto.getPartner()));
        }
        return registration;
    }

    @Override
    public RegistrationDto toDto(Registration registration) {
        RegistrationDto registrationDto = new RegistrationDto();
        registrationDto.setId(registration.getId());
//        registrationDto.setTournament(registration.getEvent().getTournament().getId());
        registrationDto.setPlayer(playerMapper.toDto(registration.getPlayer()));
        if (registration.getPartner() != null) {
            registrationDto.setPartner(playerMapper.toDto(registration.getPartner()));
        }
        return registrationDto;
    }
}