Chatto/chatto/src/main/java/org/ros/chatto/service/ChatServiceImpl.java

package org.ros.chatto.service;

import java.util.Date;
import java.util.List;
import java.util.stream.Collectors;

import org.ros.chatto.dto.ChatMessageDTO;
import org.ros.chatto.dto.MessageCipherDTO;
import org.ros.chatto.dto.ReencryptionDTO;
import org.ros.chatto.model.ChatMessage;
import org.ros.chatto.model.ChatUser;
import org.ros.chatto.model.MessageCipher;
import org.ros.chatto.repository.ChatMessageRepository;
import org.ros.chatto.repository.MessageCipherRepository;
import org.ros.chatto.repository.UserRepository;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.domain.PageRequest;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

@Service
@Transactional
public class ChatServiceImpl implements ChatService {

//	@Autowired
//	UserService userService;
	@Autowired
	UserRepository userRepository;
	@Autowired
	MessageCipherRepository messageCipherRepository;
	@Autowired
	ChatMessageRepository chatMessageRepository;
	@Autowired
	MyConversionService myConversionService;

	public ChatMessageDTO saveNewMessage(String fromUserName, String toUserName, MessageCipherDTO messageCipherDTO) {

		MessageCipher messageCipher = myConversionService.convertToMessageCipher(messageCipherDTO);
		ChatUser fromUser = userRepository.findByUserName(fromUserName);
		ChatUser toUser = userRepository.findByUserName(toUserName);
		
//		if(fromUser ==null || toUser == null)
//		{
//			System.out.println("User is null");
//			throw new SQLException();
//		}
		ChatMessage chatMessage = new ChatMessage();
		messageCipher = messageCipherRepository.save(messageCipher);
		chatMessage.setMessageCipher(messageCipher);
		chatMessage.setFromUser(fromUser);
		chatMessage.setToUser(toUser);

		chatMessage = chatMessageRepository.save(chatMessage);
		return myConversionService.convertToChatMessageDTO(chatMessage);

	}

	@Override
	@Transactional(readOnly = true)
	public List<ChatMessageDTO> getAllMessages(String fromUser, String toUser) {
		List<ChatMessage> chatMessages = chatMessageRepository.getAllMessages(fromUser, toUser);
		List<ChatMessageDTO> chatMessageDTOs = myConversionService.convertToChatMessageDTOs(chatMessages);
 		return chatMessageDTOs;
	}
	
	@Override
	@Transactional(readOnly = true)
	public List<ReencryptionDTO> getAllMessagesForReencryption(String fromUser, String toUser) {
		return myConversionService.convertToReencryptionDTOs(chatMessageRepository.getAllMessages(fromUser, toUser));
	}

	@Override
	@Transactional(readOnly = true)
	public List<ChatMessageDTO> getMessagePage(String fromUser, String toUser, int page, int size) {
//		Sort sort = Sort
//		Page<ChatMessage> chatMessages = chatMessageRepository.getAllMessages(fromUser, toUser,PageRequest.of(page, size));
//		List<ChatMessageDTO> chatMessageDTOs = myConversionService.convertToChatMessageDTOs(chatMessages);
//		return chatMessageDTOs;
		return myConversionService.convertToChatMessageDTOs(chatMessageRepository.getAllMessages(fromUser, toUser,PageRequest.of(page, size)));
	}

	@Override
	@Transactional(readOnly = true)
	public List<ChatMessageDTO> getNewMessages(String fromUser, String toUser, Date lastMessageTime) {
		List<ChatMessage> chatMessages = chatMessageRepository.getNewMessages(fromUser, toUser, lastMessageTime);
//		List<ChatMessageDTO> chatMessageDTOs 
		return myConversionService.convertToChatMessageDTOs(chatMessages);
	}


	@Override
	public void reencryptMessages(List<ReencryptionDTO> reencryptionDTOs) {
		List<MessageCipher> messageCiphers = reencryptionDTOs.stream().map(reencryptionDTO -> reencryptionDTO.getMessageCipher()).collect(Collectors.toList());
		messageCipherRepository.saveAll(messageCiphers);
	}

	@Override
	public List<ChatMessageDTO> getAllMessages(String name, String userName, PageRequest pageRequest) {
		// TODO Auto-generated method stub
		return null;
	}
}