/************************************************************************
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at

     http://www.apache.org/licenses/LICENSE-2.0

 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
 ************************************************************************/
package core.plugin.spring;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
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;

import common.exception.http.BadRequestException;
import common.exception.http.ConflictException;
import common.exception.http.ExpectationFailedException;
import common.exception.http.ForbiddenException;
import common.exception.http.GoneException;
import common.exception.http.LengthRequiredException;
import common.exception.http.MethodNotAllowedException;
import common.exception.http.NotAcceptableException;
import common.exception.http.NotFoundException;
import common.exception.http.PaymentRequiredException;
import common.exception.http.PreconditionFailedException;
import common.exception.http.ProxyAuthenticationRequiredException;
import common.exception.http.RequestEntityTooLargeException;
import common.exception.http.RequestTimeoutException;
import common.exception.http.RequestURITooLongException;
import common.exception.http.RequestedRangeNotSatisfiableException;
import common.exception.http.UnauthorizedException;
import common.exception.http.UnsupportedMediaTypeException;

/**
 * <p>
 * 全局异常处理器
 * 
 * @author CSJ
 */
@ControllerAdvice
public class GlobalExceptionHandler extends ResponseEntityExceptionHandler {

    @Override
    protected ResponseEntity<Object> handleExceptionInternal(Exception ex, Object body, HttpHeaders headers,
            HttpStatus status, WebRequest request) {
        L.error(ex.getClass().getSimpleName(), ex);
        // no cache
        headers.add("Content-Type", "text/plain; charset=utf-8");
        headers.add("Cache-Control", "no-cache, no-store, must-revalidate");
        headers.add("Pragma", "no-cache");
        headers.add("Expires", "0");
        return super.handleExceptionInternal(ex, ex.getLocalizedMessage(), headers, status, request);
    }

    /**
     * 500 默认的异常处理
     */
    @ExceptionHandler(Exception.class)
    public ResponseEntity<?> handException(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(),
                HttpStatus.INTERNAL_SERVER_ERROR, webRequest);
    }

    /**
     * 400 Bad Request.
     */
    @ExceptionHandler(BadRequestException.class)
    public ResponseEntity<?> badRequest(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.BAD_REQUEST,
                webRequest);
    }

    /**
     * 401 Unauthorized.
     */
    @ExceptionHandler(UnauthorizedException.class)
    public ResponseEntity<?> unauthorized(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.UNAUTHORIZED,
                webRequest);
    }

    /**
     * 402 Payment Required.
     */
    @ExceptionHandler(PaymentRequiredException.class)
    public ResponseEntity<?> paymentRequired(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.PAYMENT_REQUIRED,
                webRequest);
    }

    /**
     * 403 Forbidden,
     */
    @ExceptionHandler(ForbiddenException.class)
    public ResponseEntity<?> forbidden(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.FORBIDDEN,
                webRequest);
    }

    /**
     * 404 Not Found.
     */
    @ExceptionHandler(NotFoundException.class)
    public ResponseEntity<?> notFound(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.NOT_FOUND,
                webRequest);
    }

    /**
     * 405 Method Not Allowed.
     */
    @ExceptionHandler(MethodNotAllowedException.class)
    public ResponseEntity<?> methodNotAllowed(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.METHOD_NOT_ALLOWED,
                webRequest);
    }

    /**
     * 406 Not Acceptable.
     */
    @ExceptionHandler(NotAcceptableException.class)
    public ResponseEntity<?> notAcceptable(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.NOT_ACCEPTABLE,
                webRequest);
    }

    /**
     * 407 Proxy Authentication Required.
     */
    @ExceptionHandler(ProxyAuthenticationRequiredException.class)
    public ResponseEntity<?> proxyAuthenticationRequired(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(),
                HttpStatus.PROXY_AUTHENTICATION_REQUIRED, webRequest);
    }

    /**
     * 408 Request Timeout.
     */
    @ExceptionHandler(RequestTimeoutException.class)
    public ResponseEntity<?> requestTimeout(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.REQUEST_TIMEOUT,
                webRequest);
    }

    /**
     * 409 Conflict.
     */
    @ExceptionHandler(ConflictException.class)
    public ResponseEntity<?> conflict(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.CONFLICT,
                webRequest);
    }

    /**
     * 410 Gone.
     */
    @ExceptionHandler(GoneException.class)
    public ResponseEntity<?> gone(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.GONE, webRequest);
    }

    /**
     * 411 Length Required.
     */
    @ExceptionHandler(LengthRequiredException.class)
    public ResponseEntity<?> lengthRequired(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.LENGTH_REQUIRED,
                webRequest);
    }

    /**
     * 412 Precondition Failed.
     */
    @ExceptionHandler(PreconditionFailedException.class)
    public ResponseEntity<?> preconditionFail(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.PRECONDITION_FAILED,
                webRequest);
    }

    /**
     * 413 Request Entity Too Large.
     */
    @ExceptionHandler(RequestEntityTooLargeException.class)
    public ResponseEntity<?> requestEntityTooLarge(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.PAYLOAD_TOO_LARGE,
                webRequest);
    }

    /**
     * 414 Request URI Too Long.
     */
    @ExceptionHandler(RequestURITooLongException.class)
    public ResponseEntity<?> requestURITooLong(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.URI_TOO_LONG,
                webRequest);
    }

    /**
     * 415 Unsupported Media Type.
     */
    @ExceptionHandler(UnsupportedMediaTypeException.class)
    public ResponseEntity<?> unsupportedMediaType(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(),
                HttpStatus.UNSUPPORTED_MEDIA_TYPE, webRequest);
    }

    /**
     * 416 Requested Range Not Satisfiable.
     */
    @ExceptionHandler(RequestedRangeNotSatisfiableException.class)
    public ResponseEntity<?> requestedRangeNotSatisfiable(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(),
                HttpStatus.REQUESTED_RANGE_NOT_SATISFIABLE, webRequest);
    }

    /**
     * 417 Expectation Failed.
     */
    @ExceptionHandler(ExpectationFailedException.class)
    public ResponseEntity<?> expectationFailed(Exception ex, WebRequest webRequest) {
        return handleExceptionInternal(ex, ex.getLocalizedMessage(), new HttpHeaders(), HttpStatus.EXPECTATION_FAILED,
                webRequest);
    }

    private static final Logger L = LoggerFactory.getLogger(GlobalExceptionHandler.class);
}