com.blackducksoftware.integration.hub.rest.oauth.TokenManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hub-common-rest Show documentation
null
There is a newer version: 12.0.4
/**
 * Hub Common Rest
 *
 * Copyright (C) 2017 Black Duck Software, Inc.
 * http://www.blackducksoftware.com/
 *
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements. See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership. The ASF 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 com.blackducksoftware.integration.hub.rest.oauth;

import java.net.MalformedURLException;
import java.net.URL;

import org.apache.commons.lang3.StringUtils;

import com.blackducksoftware.integration.exception.IntegrationException;
import com.blackducksoftware.integration.hub.api.oauth.OAuthConfiguration;
import com.blackducksoftware.integration.hub.api.oauth.Token;
import com.blackducksoftware.integration.hub.rest.RestConnection;
import com.blackducksoftware.integration.hub.rest.UnauthenticatedRestConnection;
import com.blackducksoftware.integration.hub.service.HubOAuthTokenService;
import com.blackducksoftware.integration.log.IntLogger;

public class TokenManager {
    public static final String WWW_AUTH_RESP = "Authorization";

    private final IntLogger logger;

    private final int timeout;

    private OAuthConfiguration configuration;

    private Token clientToken;

    public TokenManager(final IntLogger logger, final int timeout) {
        this.logger = logger;
        this.timeout = timeout;
    }

    public IntLogger getLogger() {
        return logger;
    }

    public int getTimeout() {
        return timeout;
    }

    public OAuthConfiguration getConfiguration() {
        return configuration;
    }

    public void setConfiguration(final OAuthConfiguration configuration) {
        this.configuration = configuration;
    }

    public String createTokenCredential(final String token) {
        return String.format("Bearer %s", token);
    }

    public Token exchangeForUserToken(final String authorizationCode) throws IntegrationException {
        Token result = null;
        try {
            final URL url = new URL(configuration.tokenUri);
            final RestConnection connection = new UnauthenticatedRestConnection(logger, url, timeout);

            final HubOAuthTokenService tokenService = new HubOAuthTokenService(connection);
            result = tokenService.requestUserToken(configuration.clientId, authorizationCode, configuration.callbackUrl);
        } catch (final IntegrationException | MalformedURLException ex) {
            throw new IntegrationException("Error refreshing client token", ex);
        }

        return result;
    }

    public Token refreshToken(final AccessType accessType) throws IntegrationException {
        Token result = null;
        if (AccessType.USER.equals(accessType)) {
            result = refreshUserAccessToken();
        } else if (AccessType.CLIENT.equals(accessType)) {
            result = refreshClientAccessToken();
        }

        return result;
    }

    public Token getToken(final AccessType accessType) throws IntegrationException {
        Token result = null;
        if (AccessType.USER.equals(accessType)) {
            result = refreshUserAccessToken();
        } else if (AccessType.CLIENT.equals(accessType)) {
            if (clientToken == null) {
                refreshClientAccessToken();
            }
            result = clientToken;
        }

        return result;
    }

    private Token refreshUserAccessToken() throws IntegrationException {
        Token result = null;
        if (StringUtils.isNotBlank(configuration.refreshToken)) {
            try {
                final URL url = new URL(configuration.tokenUri);
                final RestConnection connection = new UnauthenticatedRestConnection(logger, url, timeout);

                final HubOAuthTokenService tokenService = new HubOAuthTokenService(connection);
                result = tokenService.refreshUserToken(configuration.clientId,
                        configuration.refreshToken);
            } catch (final IntegrationException | MalformedURLException ex) {
                throw new IntegrationException("Error refreshing user token", ex);
            }
        } else {
            throw new IntegrationException("No token present to refresh");
        }

        return result;
    }

    private Token refreshClientAccessToken() throws IntegrationException {
        Token result = null;
        try {
            final URL url = new URL(configuration.tokenUri);
            final RestConnection connection = new UnauthenticatedRestConnection(logger, url, timeout);

            final HubOAuthTokenService tokenService = new HubOAuthTokenService(connection);
            result = tokenService.refreshClientToken(configuration.clientId);
            clientToken = result;
        } catch (final IntegrationException | MalformedURLException ex) {
            throw new IntegrationException("Error refreshing client token", ex);
        }
        return result;
    }
}