Chatto/chatto/src/main/java/org/ros/chatto/controller/AdminRESTController.java

package org.ros.chatto.controller;

import java.security.Principal;
import java.time.OffsetDateTime;
import java.time.format.DateTimeFormatter;
import java.time.format.DateTimeFormatterBuilder;
import java.util.Date;
import java.util.List;

import javax.validation.Valid;

import org.ros.chatto.dto.ChatMessageDTO;
import org.ros.chatto.dto.ReencryptionDTO;
import org.ros.chatto.service.ChatService;
import org.ros.chatto.service.UserService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

@RestController
@RequestMapping("/api/admin")
public class AdminRESTController {
	@Autowired
	private ChatService chatService;

	@Autowired
	private UserService userService;

	@PostMapping(value = "/post/re-encrypt", consumes = { "application/json" })
	public ResponseEntity<ReencryptionDTO> reencryptMessages(
			@RequestBody @Valid List<ReencryptionDTO> reencryptionDTOs,
			Principal principal) {
		if (reencryptionDTOs.size() > 0) {
			chatService.reencryptMessages(reencryptionDTOs);
		}
		return new ResponseEntity<ReencryptionDTO>(HttpStatus.OK);
	}

	@GetMapping(value = "/get/messages/{userName}")
	public List<ReencryptionDTO> sendAllMessages(@PathVariable String userName,
			Principal principal) {
		List<ReencryptionDTO> reencryptionDTOs = chatService
				.getAllMessagesForReencryption(principal.getName(), userName);
		return reencryptionDTOs;
	}

	@GetMapping(value = "/get/messages/{userName}/{lastMessageTime}")
	public List<ChatMessageDTO> sendNewMessages(@PathVariable String userName,
			@PathVariable String lastMessageTime, Principal principal) {
		System.out.println("Last message time = " + lastMessageTime);
		DateTimeFormatter formatter = new DateTimeFormatterBuilder()
				// date/time
				.append(DateTimeFormatter.ISO_LOCAL_DATE_TIME)
				// offset (hh:mm - "+00:00" when it's zero)
				.optionalStart().appendOffset("+HH:MM", "+00:00").optionalEnd()
				// offset (hhmm - "+0000" when it's zero)
				.optionalStart().appendOffset("+HHMM", "+0000").optionalEnd()
				// offset (hh - "Z" when it's zero)
				.optionalStart().appendOffset("+HH", "Z").optionalEnd()
				// create formatter
				.toFormatter();
		Date date = Date.from(
				OffsetDateTime.parse(lastMessageTime, formatter).toInstant());

		List<ChatMessageDTO> chatMessageDTOs = chatService
				.getNewMessages(principal.getName(), userName, date);
		return chatMessageDTOs;
	}

	@GetMapping("/get/users")
	public List<String> getAllOtherUsers(Principal principal) {
		return userService.findAllOtherUsers(principal.getName());
	}
}

// public ResponseEntity<List<ChatMessage>> getMessages(@PathVariable String
// userName, Principal principal) {
//// List<ChatMessage> chatMessages =
// chatMessageRepository.getAllMessages(principal.getName(), userName);
//
//// return posts.stream()
//// .map(post -> convertToDto(post))
//// .collect(Collectors.toList());
// return new ResponseEntity<List<ChatMessage>>(chatMessages, HttpStatus.OK);
// }