UserDtoConverter.java

package com.sungardas.enhancedsnapshots.dto.converter;

import com.sungardas.enhancedsnapshots.aws.dynamodb.Roles;
import com.sungardas.enhancedsnapshots.aws.dynamodb.model.User;
import com.sungardas.enhancedsnapshots.dto.UserDto;
import org.springframework.beans.BeanUtils;

import java.util.ArrayList;
import java.util.List;

public final class UserDtoConverter {
	private UserDtoConverter() {
	}

	public static UserDto convert(User user) {
		UserDto userDto = new UserDto();

		userDto.setAdmin("admin".equals(user.getRole()));
		userDto.setEmail(user.getEmail());
		userDto.setFirstName(user.getFirstName());
		userDto.setLastName(user.getLastName());
		userDto.setRole(user.getRole());

		return userDto;
	}

	public static List<UserDto> convert(Iterable<User> users) {
		List<UserDto> dtos = new ArrayList<>();
		for (User user : users) {
			dtos.add(convert(user));
		}
		return dtos;
	}

	public static User convert(UserDto userDto) {
		User user = new User();
		BeanUtils.copyProperties(userDto, user);
		if (userDto.isAdmin()) {
			user.setRole(Roles.ADMIN.getName());
		} else {
			user.setRole(Roles.USER.getName());
		}
		return user;
	}
}