GitHubSshKeyUploader.java

/*******************************************************************************
 * Copyright (c) 2012-2016 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.ide.ext.github.client;

import com.google.gwt.user.client.Window;
import com.google.gwt.user.client.rpc.AsyncCallback;
import com.google.inject.Inject;
import com.google.inject.Singleton;

import org.eclipse.che.ide.api.ProductInfoDataProvider;
import org.eclipse.che.ide.api.app.AppContext;
import org.eclipse.che.ide.api.notification.NotificationManager;
import org.eclipse.che.ide.api.notification.StatusNotification;
import org.eclipse.che.ide.commons.exception.UnauthorizedException;
import org.eclipse.che.ide.rest.AsyncRequestCallback;
import org.eclipse.che.ide.rest.RestContext;
import org.eclipse.che.ide.ui.dialogs.CancelCallback;
import org.eclipse.che.ide.ui.dialogs.ConfirmCallback;
import org.eclipse.che.ide.ui.dialogs.DialogFactory;
import org.eclipse.che.security.oauth.JsOAuthWindow;
import org.eclipse.che.security.oauth.OAuthCallback;
import org.eclipse.che.security.oauth.OAuthStatus;
import org.eclipse.che.ide.ext.git.ssh.client.SshKeyUploader;

import static org.eclipse.che.security.oauth.OAuthStatus.LOGGED_IN;

/**
 * Uploads SSH keys for github.com.
 *
 * @author Ann Shumilova
 */
@Singleton
public class GitHubSshKeyUploader implements SshKeyUploader, OAuthCallback {

    private final GitHubClientService        gitHubService;
    private final String                     baseUrl;
    private final GitHubLocalizationConstant constant;
    private final NotificationManager        notificationManager;
    private final ProductInfoDataProvider    productInfoDataProvider;
    private final DialogFactory              dialogFactory;
    private final AppContext                 appContext;

    private AsyncCallback<Void> callback;
    private String              userId;

    @Inject
    public GitHubSshKeyUploader(GitHubClientService gitHubService,
                                @RestContext String baseUrl,
                                GitHubLocalizationConstant constant,
                                NotificationManager notificationManager,
                                ProductInfoDataProvider productInfoDataProvider,
                                DialogFactory dialogFactory,
                                AppContext appContext) {
        this.gitHubService = gitHubService;
        this.baseUrl = baseUrl;
        this.constant = constant;
        this.notificationManager = notificationManager;
        this.productInfoDataProvider = productInfoDataProvider;
        this.dialogFactory = dialogFactory;
        this.appContext = appContext;
    }

    /** {@inheritDoc} */
    @Override
    public void uploadKey(final String userId, final AsyncCallback<Void> callback) {
        this.callback = callback;
        this.userId = userId;

        gitHubService.updatePublicKey(new AsyncRequestCallback<Void>() {
            @Override
            protected void onSuccess(Void o) {
                callback.onSuccess(o);
            }

            @Override
            protected void onFailure(Throwable e) {
                if (e instanceof UnauthorizedException) {
                    oAuthLoginStart();
                    return;
                }

                callback.onFailure(e);
            }
        });
    }

    /** Log in github */
    private void oAuthLoginStart() {
        dialogFactory.createConfirmDialog(constant.authorizationDialogTitle(),
                                          constant.authorizationDialogText(productInfoDataProvider.getName()),
                                          new ConfirmCallback() {
                                              @Override
                                              public void accepted() {
                                                  showPopUp();
                                              }
                                          },
                                          new CancelCallback() {
                                              @Override
                                              public void cancelled() {
                                                  callback.onFailure(new Exception(constant.authorizationRequestRejected()));
                                              }
                                          }).show();
    }

    private void showPopUp() {
        String authUrl = baseUrl + "/oauth/authenticate?oauth_provider=github"
                         + "&scope=user,repo,write:public_key&userId=" + userId + "&redirect_after_login=" +
                         Window.Location.getProtocol() + "//" + Window.Location.getHost() + "/ws/" + appContext.getWorkspace().getName();
        JsOAuthWindow authWindow = new JsOAuthWindow(authUrl, "error.url", 500, 980, this);
        authWindow.loginWithOAuth();
    }

    /** {@inheritDoc} */
    @Override
    public void onAuthenticated(OAuthStatus authStatus) {
        if (LOGGED_IN.equals(authStatus)) {
            uploadKey(userId, callback);
        } else {
            notificationManager.notify(constant.authorizationFailed(), StatusNotification.Status.FAIL, true);
            callback.onFailure(new Exception(constant.authorizationFailed()));
        }
    }
}