4c9d3042aeda3af0e834513b0ee2d00192f5a216
[oom/platform/cert-service.git] /
1 /*
2  * ============LICENSE_START=======================================================
3  * PROJECT
4  * ================================================================================
5  * Copyright (C) 2020 Nokia. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  *
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  *
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.aaf.certservice.certification;
22
23 import com.google.gson.Gson;
24 import org.onap.aaf.certservice.certification.exception.CsrDecryptionException;
25 import org.onap.aaf.certservice.certification.exception.ErrorResponseModel;
26 import org.onap.aaf.certservice.certification.exception.KeyDecryptionException;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
29 import org.springframework.http.HttpStatus;
30 import org.springframework.http.ResponseEntity;
31 import org.springframework.web.bind.annotation.ControllerAdvice;
32 import org.springframework.web.bind.annotation.ExceptionHandler;
33
34 @ControllerAdvice
35 public class CertificationExceptionController {
36
37     private static final Logger LOGGER = LoggerFactory.getLogger(CertificationExceptionController.class);
38
39     @ExceptionHandler(value = CsrDecryptionException.class)
40     public ResponseEntity<String> handle(CsrDecryptionException exception) {
41         LOGGER.error("Exception occurred during decoding certificate sign request:", exception);
42         return getErrorResponseEntity("Wrong certificate signing request (CSR) format");
43     }
44
45     @ExceptionHandler(value = KeyDecryptionException.class)
46     public ResponseEntity<String> handle(KeyDecryptionException exception) {
47         LOGGER.error("Exception occurred during decoding key:", exception);
48         return getErrorResponseEntity("Wrong key (PK) format");
49     }
50
51     private ResponseEntity<String> getErrorResponseEntity(String errorMessage) {
52         ErrorResponseModel errorResponse = new ErrorResponseModel(errorMessage);
53         return new ResponseEntity<>(
54                 new Gson().toJson(errorResponse),
55                 HttpStatus.BAD_REQUEST
56         );
57     }
58 }