StsAddressProviderImpl.java

/**
 * Copyright (c) Codice Foundation
 * <p>
 * This is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser
 * General Public License as published by the Free Software Foundation, either version 3 of the
 * License, or any later version.
 * <p>
 * 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
 * Lesser General Public License for more details. A copy of the GNU Lesser General Public License
 * is distributed along with this program and can be found at
 * <http://www.gnu.org/licenses/lgpl.html>.
 */
package ddf.security.realm.sts;

import java.net.URI;
import java.net.URISyntaxException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import ddf.security.sts.client.configuration.STSClientConfiguration;
import ddf.security.sts.client.configuration.StsAddressProvider;

public class StsAddressProviderImpl implements StsAddressProvider {
    private static final Logger LOGGER = LoggerFactory.getLogger(StsAddressProviderImpl.class);

    private boolean useWss = false;

    private final STSClientConfiguration internalSts;

    private final STSClientConfiguration wssSts;

    public StsAddressProviderImpl(STSClientConfiguration internalSts,
            STSClientConfiguration wssSts) {
        this.internalSts = internalSts;
        this.wssSts = wssSts;
    }

    public boolean isUseWss() {
        return useWss;
    }

    public void setUseWss(boolean useWss) {
        this.useWss = useWss;
    }

    @Override
    public String getWsdlAddress() {
        String currentStsAddress;
        if (useWss) {
            currentStsAddress = wssSts.getAddress();
        } else {
            currentStsAddress = internalSts.getAddress();
        }
        return currentStsAddress;
    }

    @Override
    public String getProtocol() {
        try {
            URI uri = new URI(getWsdlAddress());
            return uri.getScheme();
        } catch (URISyntaxException e) {
            LOGGER.info("Unable to parse STS url", e);
            return "";
        }
    }

    @Override
    public String getHost() {
        try {
            URI uri = new URI(getWsdlAddress());
            return uri.getHost();
        } catch (URISyntaxException e) {
            LOGGER.info("Unable to parse STS url", e);
            return "";
        }
    }

    @Override
    public String getPort() {
        try {
            URI uri = new URI(getWsdlAddress());
            return Integer.toString(uri.getPort());
        } catch (URISyntaxException e) {
            LOGGER.info("Unable to parse STS url", e);
            return "";
        }
    }
}