package com.pollex.pam.service.mapper;

import static java.util.stream.Collectors.toList;

import java.util.List;
import java.util.Objects;

import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.pollex.pam.domain.Consultant;
import com.pollex.pam.domain.Customer;
import com.pollex.pam.domain.Satisfaction;
import com.pollex.pam.repository.ConsultantRepository;
import com.pollex.pam.repository.CustomerRepository;
import com.pollex.pam.service.dto.SatisfactionDTO;

@Service
public class SatisfactionMapper {

	@Autowired
	ConsultantRepository consultantRepository;

	@Autowired
	CustomerRepository customerRepository;

	public SatisfactionDTO toDTO(Satisfaction source) {
		SatisfactionDTO target = new SatisfactionDTO();
		BeanUtils.copyProperties(source, target);
		target.setScore(source.getScore());
		Consultant consultant= consultantRepository.findOneByAgentNo(source.getAgentNo()).orElse(null);
		target.setAgentName(Objects.isNull(consultant) ? null : consultant.getName());
		Customer customer = customerRepository.findById(source.getCustomerId()).get();
		target.setCustomerName(customer.getName());
		return target;
	}

	public List<SatisfactionDTO> toDTO(List<Satisfaction> satisfactionList) {
		if(satisfactionList==null) return null;
		return satisfactionList.stream()
				.map(s -> toDTO(s))
				.collect(toList());
	}
}