com.dowdandassociates.gentoo.bootstrap.DefaultTestSessionInformationProvider.java Source code

Introduction

Here is the source code for com.dowdandassociates.gentoo.bootstrap.DefaultTestSessionInformationProvider.java

Source

/*
 *
 * DefaultTestSessionInformationProvider.java
 *
 *-----------------------------------------------------------------------------
 * Copyright 2013 Dowd and Associates
 *
 * Licensed 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.dowdandassociates.gentoo.bootstrap;

import com.amazonaws.services.ec2.model.Instance;

import com.google.common.base.Optional;
import com.google.common.base.Strings;
import com.google.common.base.Supplier;
import com.google.common.base.Suppliers;

import com.google.inject.Inject;
import com.google.inject.Provider;
import com.google.inject.name.Named;

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

import com.netflix.governator.annotations.Configuration;

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

public class DefaultTestSessionInformationProvider implements Provider<TestSessionInformation> {
    private static Logger log = LoggerFactory.getLogger(DefaultTestSessionInformationProvider.class);

    @Configuration("com.dowdandassociates.gentoo.bootstrap.TestSession.user")
    private Supplier<String> user = Suppliers.ofInstance("ec2-user");

    @Configuration("com.dowdandassociates.gentoo.bootstrap.TestSession.port")
    private Supplier<Integer> port = Suppliers.ofInstance(22);

    private Optional<JSch> jsch;
    private UserInfo userInfo;
    private TestInstanceInformation instanceInfo;

    @Inject
    public DefaultTestSessionInformationProvider(Optional<JSch> jsch, UserInfo userInfo,
            TestInstanceInformation instanceInfo) {
        this.jsch = jsch;
        this.userInfo = userInfo;
        this.instanceInfo = instanceInfo;
    }

    @Override
    public TestSessionInformation get() {
        return new TestSessionInformation().withInstanceInfo(instanceInfo).withSession(getSession());
    }

    private Optional<Session> getSession() {
        try {
            if (null == jsch || !jsch.isPresent()) {
                return Optional.absent();
            }

            Optional<String> host = getHost();
            if (!host.isPresent()) {
                return Optional.absent();
            }

            Session session = jsch.get().getSession(user.get(), host.get(), port.get());
            session.setUserInfo(userInfo);
            session.connect();
            return Optional.of(session);
        } catch (JSchException jse) {
            log.error(jse.getMessage(), jse);
            return Optional.absent();
        }
    }

    private Optional<String> getHost() {
        if (null == instanceInfo) {
            return Optional.absent();
        }

        Optional<Instance> instance = instanceInfo.getInstance();

        if (instance.isPresent()) {
            return Optional.fromNullable(Strings.emptyToNull(instance.get().getPublicDnsName()));
        } else {
            return Optional.absent();
        }
    }
}