org.jclouds.googlecomputeengine.config.GoogleComputeEngineRestClientModule.java Source code

Java tutorial

Introduction

Here is the source code for org.jclouds.googlecomputeengine.config.GoogleComputeEngineRestClientModule.java

Source

/**
 * Licensed to jclouds, Inc. (jclouds) under one or more
 * contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  jclouds 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 org.jclouds.googlecomputeengine.config;

import com.google.common.base.Function;
import com.google.common.base.Predicate;
import com.google.common.base.Splitter;
import com.google.common.base.Supplier;
import com.google.common.collect.ImmutableMap;
import com.google.common.collect.Iterables;
import com.google.inject.Provides;
import com.google.inject.TypeLiteral;
import org.jclouds.domain.Credentials;
import org.jclouds.googlecomputeengine.GoogleComputeEngineApi;
import org.jclouds.googlecomputeengine.GoogleComputeEngineAsyncApi;
import org.jclouds.googlecomputeengine.domain.Operation;
import org.jclouds.googlecomputeengine.features.DiskApi;
import org.jclouds.googlecomputeengine.features.DiskAsyncApi;
import org.jclouds.googlecomputeengine.features.FirewallApi;
import org.jclouds.googlecomputeengine.features.FirewallAsyncApi;
import org.jclouds.googlecomputeengine.features.ImageApi;
import org.jclouds.googlecomputeengine.features.ImageAsyncApi;
import org.jclouds.googlecomputeengine.features.InstanceApi;
import org.jclouds.googlecomputeengine.features.InstanceAsyncApi;
import org.jclouds.googlecomputeengine.features.KernelApi;
import org.jclouds.googlecomputeengine.features.KernelAsyncApi;
import org.jclouds.googlecomputeengine.features.MachineTypeApi;
import org.jclouds.googlecomputeengine.features.MachineTypeAsyncApi;
import org.jclouds.googlecomputeengine.features.NetworkApi;
import org.jclouds.googlecomputeengine.features.NetworkAsyncApi;
import org.jclouds.googlecomputeengine.features.OperationApi;
import org.jclouds.googlecomputeengine.features.OperationAsyncApi;
import org.jclouds.googlecomputeengine.features.ProjectApi;
import org.jclouds.googlecomputeengine.features.ProjectAsyncApi;
import org.jclouds.googlecomputeengine.features.ZoneApi;
import org.jclouds.googlecomputeengine.features.ZoneAsyncApi;
import org.jclouds.googlecomputeengine.handlers.GoogleComputeEngineErrorHandler;
import org.jclouds.googlecomputeengine.predicates.OperationDonePredicate;
import org.jclouds.http.HttpErrorHandler;
import org.jclouds.http.Uris;
import org.jclouds.http.annotation.ClientError;
import org.jclouds.http.annotation.Redirection;
import org.jclouds.http.annotation.ServerError;
import org.jclouds.json.config.GsonModule.DateAdapter;
import org.jclouds.json.config.GsonModule.Iso8601DateAdapter;
import org.jclouds.location.Provider;
import org.jclouds.rest.ConfiguresRestClient;
import org.jclouds.rest.config.RestClientModule;

import javax.inject.Named;
import javax.inject.Singleton;
import java.net.URI;
import java.util.Map;
import java.util.concurrent.atomic.AtomicReference;

import static com.google.common.base.Preconditions.checkState;
import static com.google.common.base.Suppliers.compose;

/**
 * Configures the GoogleCompute connection.
 *
 * @author David Alves
 */
@ConfiguresRestClient
public class GoogleComputeEngineRestClientModule
        extends RestClientModule<GoogleComputeEngineApi, GoogleComputeEngineAsyncApi> {
    public static final Map<Class<?>, Class<?>> DELEGATE_MAP = ImmutableMap.<Class<?>, Class<?>>builder()
            .put(DiskApi.class, DiskAsyncApi.class).put(FirewallApi.class, FirewallAsyncApi.class)
            .put(ImageApi.class, ImageAsyncApi.class).put(InstanceApi.class, InstanceAsyncApi.class)
            .put(KernelApi.class, KernelAsyncApi.class).put(MachineTypeApi.class, MachineTypeAsyncApi.class)
            .put(NetworkApi.class, NetworkAsyncApi.class).put(OperationApi.class, OperationAsyncApi.class)
            .put(ProjectApi.class, ProjectAsyncApi.class).put(ZoneApi.class, ZoneAsyncApi.class).build();

    public GoogleComputeEngineRestClientModule() {
        super(DELEGATE_MAP);
    }

    @Override
    protected void configure() {
        bind(DateAdapter.class).to(Iso8601DateAdapter.class);
        bind(new TypeLiteral<Predicate<AtomicReference<Operation>>>() {
        }).to(OperationDonePredicate.class);
        super.configure();
    }

    @Override
    protected void bindErrorHandlers() {
        bind(HttpErrorHandler.class).annotatedWith(Redirection.class).to(GoogleComputeEngineErrorHandler.class);
        bind(HttpErrorHandler.class).annotatedWith(ClientError.class).to(GoogleComputeEngineErrorHandler.class);
        bind(HttpErrorHandler.class).annotatedWith(ServerError.class).to(GoogleComputeEngineErrorHandler.class);
    }

    @Provides
    @Singleton
    @UserProject
    public Supplier<String> supplyProject(@org.jclouds.location.Provider final Supplier<Credentials> creds) {
        return compose(new Function<Credentials, String>() {
            public String apply(Credentials in) {
                checkState(in.identity.indexOf("@") != 1,
                        "identity should be in project_id@developer.gserviceaccount.com" + " format");
                return Iterables.get(Splitter.on("@").split(in.identity), 0);
            }
        }, creds);
    }

    @Provides
    @Singleton
    @Named("machineTypes")
    public Function<String, URI> provideMachineTypeNameToURIFunction(final @Provider Supplier<URI> endpoint,
            final @UserProject Supplier<String> userProject) {
        return new Function<String, URI>() {
            @Override
            public URI apply(String input) {
                return Uris.uriBuilder(endpoint.get()).appendPath("/projects/").appendPath(userProject.get())
                        .appendPath("/machineTypes/").appendPath(input).build();
            }
        };
    }

    @Provides
    @Singleton
    @Named("networks")
    public Function<String, URI> provideNetworkNameToURIFunction(final @Provider Supplier<URI> endpoint,
            final @UserProject Supplier<String> userProject) {
        return new Function<String, URI>() {
            @Override
            public URI apply(String input) {
                return Uris.uriBuilder(endpoint.get()).appendPath("/projects/").appendPath(userProject.get())
                        .appendPath("/networks/").appendPath(input).build();
            }
        };
    }

    @Provides
    @Singleton
    @Named("zones")
    public Function<String, URI> provideZoneNameToURIFunction(final @Provider Supplier<URI> endpoint,
            final @UserProject Supplier<String> userProject) {
        return new Function<String, URI>() {
            @Override
            public URI apply(String input) {
                return Uris.uriBuilder(endpoint.get()).appendPath("/projects/").appendPath(userProject.get())
                        .appendPath("/zones/").appendPath(input).build();
            }
        };
    }

    //   @Override
    //   protected void installLocations() {
    //      install(new LocationModule());
    //   }
}