GoogleCredentials.java

/*
 * Copyright (c) 2015 Google, Inc.
 *
 * 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.cloudera.director.google.internal;

import com.cloudera.director.google.util.Names;
import com.google.api.client.googleapis.auth.oauth2.GoogleCredential;
import com.google.api.client.googleapis.javanet.GoogleNetHttpTransport;
import com.google.api.client.http.HttpTransport;
import com.google.api.client.json.JsonFactory;
import com.google.api.client.json.jackson2.JacksonFactory;
import com.google.api.services.compute.Compute;
import com.google.api.services.compute.ComputeScopes;
import com.google.api.services.sqladmin.SQLAdmin;
import com.google.api.services.sqladmin.SQLAdminScopes;
import com.typesafe.config.Config;

import java.io.ByteArrayInputStream;
import java.util.Collection;
import java.util.Collections;

public class GoogleCredentials {

  private final Config applicationProperties;
  private final String projectId;
  private final String jsonKey;
  private final Compute compute;
  private final SQLAdmin sqlAdmin;

  public GoogleCredentials(Config applicationProperties, String projectId, String jsonKey) {
    this.applicationProperties = applicationProperties;
    this.projectId = projectId;
    this.jsonKey = jsonKey;
    this.compute = buildCompute();
    this.sqlAdmin = buildSQLAdmin();
  }

  private Compute buildCompute() {
    try {
      JsonFactory JSON_FACTORY = JacksonFactory.getDefaultInstance();
      HttpTransport httpTransport = GoogleNetHttpTransport.newTrustedTransport();
      GoogleCredential credential;

      if (jsonKey != null) {
        credential = GoogleCredential.fromStream(
            new ByteArrayInputStream(jsonKey.getBytes()), httpTransport, JSON_FACTORY)
            .createScoped(Collections.singleton(ComputeScopes.COMPUTE));
      } else {
        Collection COMPUTE_SCOPES = Collections.singletonList(ComputeScopes.COMPUTE);

        credential = GoogleCredential
            .getApplicationDefault(httpTransport, JSON_FACTORY)
            .createScoped(COMPUTE_SCOPES);
      }

      return new Compute.Builder(httpTransport,
          JSON_FACTORY,
          null)
          .setApplicationName(Names.buildApplicationNameVersionTag(applicationProperties))
          .setHttpRequestInitializer(credential)
          .build();

    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  private SQLAdmin buildSQLAdmin() {
    try {
      JsonFactory JSON_FACTORY = JacksonFactory.getDefaultInstance();
      HttpTransport httpTransport = GoogleNetHttpTransport.newTrustedTransport();
      GoogleCredential credential;

      if (jsonKey != null) {
        credential = GoogleCredential.fromStream(
            new ByteArrayInputStream(jsonKey.getBytes()), httpTransport, JSON_FACTORY)
            .createScoped(Collections.singleton(SQLAdminScopes.SQLSERVICE_ADMIN));
      } else {
        Collection SQLSERVICE_ADMIN_SCOPES = Collections.singletonList(SQLAdminScopes.SQLSERVICE_ADMIN);

        credential = GoogleCredential
            .getApplicationDefault(httpTransport, JSON_FACTORY)
            .createScoped(SQLSERVICE_ADMIN_SCOPES);
      }

      return new SQLAdmin.Builder(httpTransport,
          JSON_FACTORY,
          null)
          .setApplicationName(Names.buildApplicationNameVersionTag(applicationProperties))
          .setHttpRequestInitializer(credential)
          .build();

    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  public String getProjectId() {
    return projectId;
  }

  public String getJsonKey() {
    return jsonKey;
  }

  public Compute getCompute() {
    return compute;
  }

  public SQLAdmin getSQLAdmin() {
    return sqlAdmin;
  }

  public boolean match(String projectId, String jsonKey) {
    return projectId.equals(this.projectId) &&
        jsonKey.equals(this.jsonKey);
  }
}