org.eclipse.remote.internal.jsch.ui.JSchUserAuthenticator.java Source code

Introduction

Here is the source code for org.eclipse.remote.internal.jsch.ui.JSchUserAuthenticator.java

Source

/*******************************************************************************
 * Copyright (c) 2014, 2015 IBM Corporation and others.
 * 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:
 * IBM Corporation - Initial API and implementation
 * Bernd Hufmann - Implement IRemoteConnection.Service.Factory
 *******************************************************************************/
package org.eclipse.remote.internal.jsch.ui;

import java.net.PasswordAuthentication;

import org.eclipse.jface.dialogs.IDialogConstants;
import org.eclipse.jface.dialogs.MessageDialog;
import org.eclipse.jsch.ui.UserInfoPrompter;
import org.eclipse.remote.core.IRemoteConnection;
import org.eclipse.remote.core.IUserAuthenticatorService;
import org.eclipse.remote.internal.jsch.core.JSchConnection;
import org.eclipse.swt.widgets.Display;

import com.jcraft.jsch.JSch;
import com.jcraft.jsch.JSchException;

public class JSchUserAuthenticator implements IUserAuthenticatorService {

    private final IRemoteConnection remoteConnection;
    private UserInfoPrompter prompter;

    public JSchUserAuthenticator(IRemoteConnection conn) {
        this.remoteConnection = conn;
        try {
            String username = conn.getAttribute(JSchConnection.USERNAME_ATTR);
            String address = conn.getAttribute(JSchConnection.ADDRESS_ATTR);
            prompter = new UserInfoPrompter(new JSch().getSession(username, address));
        } catch (JSchException e) {
            // Not allowed
        }
    }

    @Override
    public IRemoteConnection getRemoteConnection() {
        return remoteConnection;
    }

    @Override
    public PasswordAuthentication prompt(String username, String message) {
        if (prompter.promptPassword(message)) {
            String sessionUserName = prompter.getSession().getUserName();
            if (sessionUserName != null) {
                username = sessionUserName;
            }
            PasswordAuthentication auth = new PasswordAuthentication(username,
                    prompter.getPassword().toCharArray());
            return auth;
        }
        return null;
    }

    @Override
    public String[] prompt(String destination, String name, String message, String[] prompt, boolean[] echo) {
        return prompter.promptKeyboardInteractive(destination, name, message, prompt, echo);
    }

    @Override
    public int prompt(final int promptType, final String title, final String message, final int[] promptResponses,
            final int defaultResponseIndex) {
        final Display display = getDisplay();
        final int[] retval = new int[1];
        final String[] buttons = new String[promptResponses.length];
        for (int i = 0; i < promptResponses.length; i++) {
            int prompt = promptResponses[i];
            switch (prompt) {
            case IDialogConstants.OK_ID:
                buttons[i] = IDialogConstants.OK_LABEL;
                break;
            case IDialogConstants.CANCEL_ID:
                buttons[i] = IDialogConstants.CANCEL_LABEL;
                break;
            case IDialogConstants.NO_ID:
                buttons[i] = IDialogConstants.NO_LABEL;
                break;
            case IDialogConstants.YES_ID:
                buttons[i] = IDialogConstants.YES_LABEL;
                break;
            }
        }

        display.syncExec(new Runnable() {
            @Override
            public void run() {
                final MessageDialog dialog = new MessageDialog(display.getActiveShell(), title,
                        null /* title image */, message, promptType, buttons, defaultResponseIndex);
                retval[0] = dialog.open();
            }
        });
        return promptResponses[retval[0]];
    }

    private Display getDisplay() {
        Display display = Display.getCurrent();
        if (display == null) {
            display = Display.getDefault();
        }
        return display;
    }

    public static class Factory implements IRemoteConnection.Service.Factory {
        @Override
        @SuppressWarnings("unchecked")
        public <T extends IRemoteConnection.Service> T getService(IRemoteConnection connection, Class<T> service) {
            if (IUserAuthenticatorService.class.equals(service)) {
                return (T) new JSchUserAuthenticator(connection);
            } else {
                return null;
            }
        }
    }
}