Update application class to handle https
[aai/data-router.git] / src / main / java / org / onap / aai / datarouter / Application.java
1 /**
2  * ============LICENSE_START=======================================================
3  * org.onap.aai
4  * ================================================================================
5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017-2018 Amdocs
7  * ================================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *       http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END=========================================================
20  */
21 package org.onap.aai.datarouter;
22
23 import java.util.HashMap;
24 import javax.annotation.PostConstruct;
25 import org.apache.camel.component.servlet.CamelHttpTransportServlet;
26 import org.eclipse.jetty.util.security.Password;
27 import org.springframework.beans.factory.annotation.Autowired;
28 import org.springframework.boot.autoconfigure.SpringBootApplication;
29 import org.springframework.boot.builder.SpringApplicationBuilder;
30 import org.springframework.boot.web.servlet.ServletRegistrationBean;
31 import org.springframework.boot.web.support.SpringBootServletInitializer;
32 import org.springframework.context.annotation.Bean;
33 import org.springframework.core.env.Environment;
34
35
36 @SpringBootApplication
37 public class Application extends SpringBootServletInitializer{
38
39     private static final String CAMEL_URL_MAPPING = "/*";
40     private static final String CAMEL_SERVLET_NAME = "CamelServlet";
41
42     @Autowired
43     private Environment env;
44
45     public static void main(String[] args) {
46       String keyStorePassword = System.getenv("KEY_STORE_PASSWORD");
47       if(keyStorePassword==null || keyStorePassword.isEmpty()){
48         throw new RuntimeException("Env property KEY_STORE_PASSWORD not set");
49       }
50       HashMap<String, Object> props = new HashMap<>();
51       props.put("server.ssl.key-store-password", Password.deobfuscate(keyStorePassword));
52       new Application().configure(new SpringApplicationBuilder(Application.class).properties(props)).run(args);
53
54
55     }
56
57     @Bean
58     public ServletRegistrationBean getServletRegistrationBean() {
59         ServletRegistrationBean registration = new ServletRegistrationBean(new CamelHttpTransportServlet(), CAMEL_URL_MAPPING);
60         registration.setName(CAMEL_SERVLET_NAME);
61         return registration;
62     }
63
64     /**
65      * Set required trust store system properties using values from application.properties
66      */
67     @PostConstruct
68     public void setSystemProperties() {
69         String trustStorePath = env.getProperty("server.ssl.key-store");
70         if (trustStorePath != null) {
71             String trustStorePassword = env.getProperty("server.ssl.key-store-password");
72
73             if (trustStorePassword != null) {
74                 System.setProperty("javax.net.ssl.trustStore", trustStorePath);
75                 System.setProperty("javax.net.ssl.trustStorePassword", trustStorePassword);
76             } else {
77                 throw new IllegalArgumentException("Env property server.ssl.key-store-password not set");
78             }
79         }
80     }
81
82
83
84 }