InboundWebsocketSSLConfiguration.java

/*
 * Copyright (c) 2015, WSO2 Inc. (http://www.wso2.org) All Rights Reserved.
 * WSO2 Inc. licenses this file to you 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 org.wso2.carbon.inbound.endpoint.protocol.websocket.ssl;

import java.io.File;

public class InboundWebsocketSSLConfiguration {
    private File keyStore;
    private String keyStorePass;
    private String certPass;
    private File trustStore;
    private String trustStorePass;

    public InboundWebsocketSSLConfiguration(File keyStore, String keyStorePass) {
        this.keyStore = keyStore;
        this.keyStorePass = keyStorePass;
    }

    public String getCertPass() {
        return certPass;
    }

    public InboundWebsocketSSLConfiguration setCertPass(String certPass) {
        this.certPass = certPass;
        return this;
    }

    public File getTrustStore() {
        return trustStore;
    }

    public InboundWebsocketSSLConfiguration setTrustStore(File trustStore) {
        this.trustStore = trustStore;
        return this;
    }

    public String getTrustStorePass() {
        return trustStorePass;
    }

    public InboundWebsocketSSLConfiguration setTrustStorePass(String trustStorePass) {
        this.trustStorePass = trustStorePass;
        return this;
    }

    public File getKeyStore() {
        return keyStore;
    }

    public String getKeyStorePass() {
        return keyStorePass;
    }

    public static class SSLConfigurationBuilder {

        private String keyStoreFile;
        private String keyStorePass;
        private String trustStoreFile;
        private String trustStorePass;
        private String certPass;

        public SSLConfigurationBuilder(String keyStoreFile,
                                       String keyStorePass,
                                       String trustStoreFile,
                                       String trustStorePass,
                                       String certPass) {
            this.keyStoreFile = keyStoreFile;
            this.keyStorePass = keyStorePass;
            this.trustStoreFile = trustStoreFile;
            this.trustStorePass = trustStorePass;
            this.certPass = certPass;

        }

        public InboundWebsocketSSLConfiguration build() {
            if (certPass == null) {
                certPass = keyStorePass;
            }
            if (keyStoreFile == null || keyStorePass == null) {
                throw new IllegalArgumentException("keyStoreFile or keyStorePass not defined ");
            }
            File keyStore = new File(keyStoreFile);
            if (!keyStore.exists()) {
                throw new IllegalArgumentException("KeyStore File " + keyStoreFile + " not found");
            }
            InboundWebsocketSSLConfiguration sslConfig =
                    new InboundWebsocketSSLConfiguration(keyStore, keyStorePass).setCertPass(certPass);
            if (trustStoreFile != null) {
                File trustStore = new File(trustStoreFile);
                if (!trustStore.exists()) {
                    throw new IllegalArgumentException("trustStore File " + trustStoreFile + " not found");
                }
                if (trustStorePass == null) {
                    throw new IllegalArgumentException("trustStorePass is not defined ");
                }
                sslConfig.setTrustStore(trustStore).setTrustStorePass(trustStorePass);
            }
            return sslConfig;
        }

    }

}