org.lorislab.armonitor.client.MonitorClient.java Source code

Introduction

Here is the source code for org.lorislab.armonitor.client.MonitorClient.java

Source

/*
 * Copyright 2013 lorislab.org.
 *
 * 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 org.lorislab.armonitor.client;

import java.net.URL;
import org.lorislab.armonitor.agent.rs.model.Version;
import org.apache.http.auth.AuthScope;
import org.apache.http.auth.Credentials;
import org.apache.http.auth.UsernamePasswordCredentials;
import org.apache.http.impl.client.BasicCredentialsProvider;
import org.apache.http.impl.client.DefaultHttpClient;
import org.jboss.resteasy.client.ProxyFactory;
import org.jboss.resteasy.client.core.executors.ApacheHttpClient4Executor;
import org.jboss.resteasy.plugins.providers.RegisterBuiltin;
import org.jboss.resteasy.spi.ResteasyProviderFactory;
import org.lorislab.armonitor.rs.model.Request;
import org.lorislab.armonitor.rs.model.Result;
import org.lorislab.armonitor.rs.model.Status;
import org.lorislab.armonitor.rs.service.MonitorService;

/**
 * The ARMONITOR client.
 *
 * @author Andrej Petras
 */
public final class MonitorClient {

    /**
     * The agent monitor URL.
     */
    private static final String APP_URL = "armonitor-rs/rs";

    /**
     * The default constructor.
     */
    private MonitorClient() {
        // empty constructor
    }

    /**
     * Creates the monitor service.
     *
     * @param url the URL.
     * @param username the user name.
     * @param password the password.
     * @return the monitor service.
     */
    private static MonitorService createService(URL url, String username, String password, boolean auth) {
        ResteasyProviderFactory rpf = ResteasyProviderFactory.getInstance();
        RegisterBuiltin.register(rpf);

        String tmp = url.toString();
        if (!tmp.endsWith("/")) {
            tmp = tmp + "/";
        }
        tmp = tmp + APP_URL;

        if (auth) {
            Credentials credentials = new UsernamePasswordCredentials(username, password);
            DefaultHttpClient httpClient = new DefaultHttpClient();
            BasicCredentialsProvider provider = new BasicCredentialsProvider();
            provider.setCredentials(AuthScope.ANY, credentials);
            httpClient.setCredentialsProvider(provider);
            return ProxyFactory.create(MonitorService.class, tmp, new ApacheHttpClient4Executor(httpClient));
        }
        return ProxyFactory.create(MonitorService.class, tmp);
    }

    /**
     * Sends the version for the system <code>key</code> to the ARMONITOR
     * server.
     *
     * @param url the URL.
     * @param user the user name.
     * @param password the password.
     * @param auth the authentication flag.
     * @param key the system key.
     * @param version the version.
     * @throws java.lang.Exception if the method fails.
     */
    public static void install(URL url, String user, String password, boolean auth, String key, Version version)
            throws Exception {
        MonitorService service = createService(url, user, password, auth);
        Request request = createRequest(key, version);
        Result result = service.install(request);
        if (result != null) {
            if (result.status == null || result.status.equals(Status.ERROR)) {
                throw new Exception("Could not install the build for the application: " + result.message);
            }
        }
    }

    /**
     * Sends the version for the system <code>key</code> to the ARMONITOR
     * server.
     *
     * @param url the URL.
     * @param user the user name.
     * @param password the password.
     * @param auth the authentication flag.
     * @param key the system key.
     * @param version the version.
     * @throws java.lang.Exception if the method fails.
     */
    public static void deploy(URL url, String user, String password, boolean auth, String key, Version version)
            throws Exception {
        MonitorService service = createService(url, user, password, auth);
        Request request = createRequest(key, version);
        Result result = service.deploy(request);
        if (result != null) {
            if (result.status == null || result.status.equals(Status.ERROR)) {
                throw new Exception("Could not deploy the build for the system: " + result.message);
            }
        }
    }

    /**
     * Creates the request.
     *
     * @param key the key.
     * @param version the version.
     * @return the corresponding request.
     */
    private static Request createRequest(String key, Version version) {
        Request request = new Request();
        request.version = version;
        request.key = key;
        return request;
    }
}