package fr.ifremer.quadrige3.synchro.server.rest;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 Synchro server
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 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 Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  import fr.ifremer.quadrige3.core.dao.technical.gson.Gsons;
  import fr.ifremer.quadrige3.core.dao.technical.http.CustomHttpHeaders;
  import fr.ifremer.quadrige3.core.dao.technical.http.CustomHttpStatus;
  import fr.ifremer.quadrige3.core.exception.BadUpdateDtException;
  import fr.ifremer.quadrige3.core.exception.DataLockedException;
  import fr.ifremer.quadrige3.core.exception.DeleteForbiddenException;
  import fr.ifremer.quadrige3.core.exception.SaveForbiddenException;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.springframework.http.HttpHeaders;
  import org.springframework.http.HttpStatus;
  import org.springframework.http.ResponseEntity;
  import org.springframework.web.bind.annotation.ControllerAdvice;
  import org.springframework.web.bind.annotation.ExceptionHandler;
  import org.springframework.web.context.request.WebRequest;
  import org.springframework.web.servlet.mvc.method.annotation.ResponseEntityExceptionHandler;
  
  @ControllerAdvice
  public class RestResponseEntityExceptionHandler extends ResponseEntityExceptionHandler {
      /* Logger */
      private static final Log log = LogFactory.getLog(RestResponseEntityExceptionHandler.class);
      private static final String MESSAGE_FORMAT = "[%s] %s";
  
      public RestResponseEntityExceptionHandler() {
          super();
      }
  
      /**
       * Transform an exception on bad update_dt, into a HTTP error 500 + a body response with the exact error code.
       */
      @ExceptionHandler(value = {BadUpdateDtException.class})
      protected ResponseEntity<Object> handleBadUpdateDt(RuntimeException ex, WebRequest request) {
          String message = String.format(MESSAGE_FORMAT, CustomHttpStatus.SC_BAD_UPDATE_DT, ex.getMessage());
          if (log.isDebugEnabled()) {
              log.debug(message);
          }
          return handleExceptionInternal(ex, message, new HttpHeaders(), HttpStatus.INTERNAL_SERVER_ERROR, request);
      }
  
      /**
       * Transform an exception on locked data, into a HTTP error 500 + a body response with the exact error code.
       */
      @ExceptionHandler(value = {DataLockedException.class})
      protected ResponseEntity<Object> handleLock(RuntimeException ex, WebRequest request) {
          String message = String.format(MESSAGE_FORMAT, CustomHttpStatus.SC_DATA_LOCKED, ex.getMessage());
          if (log.isDebugEnabled()) {
              log.debug(message);
          }
          return handleExceptionInternal(ex, message, new HttpHeaders(), HttpStatus.INTERNAL_SERVER_ERROR, request);
      }
  
      /**
       * Transform an exception on delete data, into a HTTP error 500 + a body response with the exact error code.
       */
      @ExceptionHandler(value = {DeleteForbiddenException.class})
      protected ResponseEntity<Object> handleDeleteForbidden(RuntimeException ex, WebRequest request) {
          String message = String.format(MESSAGE_FORMAT, CustomHttpStatus.SC_DELETE_FORBIDDEN, ex.getMessage());
          if (log.isDebugEnabled()) {
              log.debug(message);
          }
          HttpHeaders httpHeaders = new HttpHeaders();
          // Add objects in specific header in json format
          httpHeaders.add(
                  CustomHttpHeaders.HH_DELETE_FORBIDDEN_OBJECT_IDS,
                  Gsons.newBuilder().create().toJson(((DeleteForbiddenException) ex).getObjectIds()));
          return handleExceptionInternal(ex, message, httpHeaders, HttpStatus.INTERNAL_SERVER_ERROR, request);
      }
  
      /**
       * Transform an exception on save data, into a HTTP error 500 + a body response with the exact error code.
       */
      @ExceptionHandler(value = {SaveForbiddenException.class})
      protected ResponseEntity<Object> handleSaveForbidden(RuntimeException ex, WebRequest request) {
          String message = String.format(MESSAGE_FORMAT, CustomHttpStatus.SC_SAVE_FORBIDDEN, ex.getMessage());
         if (log.isDebugEnabled()) {
             log.debug(message);
         }
         HttpHeaders httpHeaders = new HttpHeaders();
         // Add objects in specific header in json format
         httpHeaders.add(
                 CustomHttpHeaders.HH_SAVE_FORBIDDEN_TYPE,
                 ((SaveForbiddenException)ex).getType().name().toUpperCase()
         );
         httpHeaders.add(
                 CustomHttpHeaders.HH_SAVE_FORBIDDEN_OBJECT_IDS,
                 Gsons.newBuilder().create().toJson(((SaveForbiddenException) ex).getObjectIds())
         );
         return handleExceptionInternal(ex, message, httpHeaders, HttpStatus.INTERNAL_SERVER_ERROR, request);
     }
 
 }