Java tutorial
/** * Copyright (C) 2015 Frank Steiler <frank@steilerdev.de> * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package de.steilerdev.myVerein.server.controller.user; import de.steilerdev.myVerein.server.model.*; import de.steilerdev.myVerein.server.security.CurrentUser; import org.slf4j.Logger; import org.slf4j.LoggerFactory; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.http.HttpStatus; import org.springframework.http.ResponseEntity; import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RequestMethod; import org.springframework.web.bind.annotation.RequestParam; import org.springframework.web.bind.annotation.RestController; import java.time.LocalDateTime; import java.util.List; /** * This class handles all messages send and received by a user. */ @RestController @RequestMapping("/api/user/message") public class MessageController { private static Logger logger = LoggerFactory.getLogger(MessageController.class); @Autowired private MessageRepository messageRepository; @Autowired private DivisionRepository divisionRepository; @RequestMapping(produces = "application/json", method = RequestMethod.GET, params = "id") public ResponseEntity<Message> getMessage(@RequestParam(value = "id") String messageId, @CurrentUser User currentUser) { logger.trace("[{}] Getting message with id {}", currentUser, messageId); Message message; if (messageId.isEmpty()) { logger.warn("[{}] Required parameter id missing", currentUser); return new ResponseEntity<>(HttpStatus.BAD_REQUEST); } else if ((message = messageRepository.findOne(messageId)) == null) { logger.warn("[{}] Unable to find message with id {}", currentUser, messageId); return new ResponseEntity<>(HttpStatus.BAD_REQUEST); } else if (!message.getReceiver().containsKey(currentUser.getId())) { logger.warn("[{}] User is not allowed to read message {}, because he is not a receiver", currentUser, message); return new ResponseEntity<>(HttpStatus.FORBIDDEN); } else { logger.info("[{}] Delivering message {}", currentUser, message); return new ResponseEntity<>(message.getSendingObjectInternalSync(), HttpStatus.OK); } } /** * This function retrieves all unread messages of a user. The function is invoked bu GETting the URI /api/user/message. * @param currentUser The currently logged in user. * @return An HTTP response with a status code, together with the JSON list-object of all unread messages, or only an error code if an error occurred. */ @RequestMapping(produces = "application/json", method = RequestMethod.GET) public ResponseEntity<List<Message>> getMessages(@CurrentUser User currentUser, @RequestParam(required = false) String all) { logger.trace("[{}] Getting unread messages", currentUser); List<Message> messages = messageRepository.findAllByPrefixedReceiverIDAndMessageStatus( Message.receiverIDForUser(currentUser), Message.MessageStatus.PENDING); if (messages == null) { logger.debug("[{}] Unable to find any undelivered messages", currentUser); return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR); } else { messages.parallelStream().forEach(message -> message.setDelivered(currentUser)); if (all != null && !all.isEmpty()) { logger.debug("[{}] Retrieving all messages", currentUser); messages.addAll(messageRepository.findAllByPrefixedReceiverIDAndMessageStatus( Message.receiverIDForUser(currentUser), Message.MessageStatus.DELIVERED)); messages.addAll(messageRepository.findAllByPrefixedReceiverIDAndMessageStatus( Message.receiverIDForUser(currentUser), Message.MessageStatus.READ)); } try { messageRepository.save(messages); messages.replaceAll(Message::getSendingObjectOnlyId); logger.info("[{}] Returning messages", currentUser); return new ResponseEntity<>(messages, HttpStatus.OK); } catch (IllegalArgumentException e) { logger.warn("[" + currentUser + "] Unable to save messages for " + currentUser.getEmail() + ": " + e.getMessage()); return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR); } } } /** * This function sends a message to a specific group chat. The function is invoked by POSTing to the URI /api/user/message. * @param currentUser The currently logged in user. * @param division The name of the division where the message is send to. * @param content The content of the message. * @return An HTTP response with a status code. If no error occurred the system wide used id and timestamp are returned. The timestamp is set to the time or receiving on the server, to ensure the same view on the total order of messages for everyone. */ @RequestMapping(produces = "application/json", method = RequestMethod.POST) public ResponseEntity<Message> sendMessage(@CurrentUser User currentUser, @RequestParam String division, @RequestParam String content, @RequestParam(required = false) String timestamp) { logger.trace("[{}] Sending message to {}", currentUser, division); Division receivingDivision; if (division.isEmpty() || content.isEmpty()) { logger.warn("[{}] Required parameters for sending message missing", currentUser); return new ResponseEntity<>(HttpStatus.BAD_REQUEST); } else if ((receivingDivision = divisionRepository.findById(division)) == null) { logger.warn("[{}] Unable to find receiving division {}", currentUser, division); return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR); } else if (!currentUser.getDivisions().contains(receivingDivision)) { logger.warn("[{}] Trying to send a message to a division the user is not part of: {}", currentUser, division); return new ResponseEntity<>(HttpStatus.BAD_REQUEST); } else { if (receivingDivision.getMemberList().isEmpty()) { logger.warn("[{}] Empty receiver list", currentUser); return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR); } else { Message message = new Message(content, currentUser, receivingDivision); messageRepository.save(message); logger.info("[{}] Successfully saved and send message", currentUser); return new ResponseEntity<>(message.getSendingObjectOnlyIdAndTimestamp(), HttpStatus.OK); } } } }