com.fredericvauchelles.ConnectMojo.java Source code

Introduction

Here is the source code for com.fredericvauchelles.ConnectMojo.java

Source

package com.fredericvauchelles;

/*
 * Copyright 2001-2005 The Apache Software Foundation.
 *
 * 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.
 */

import org.apache.maven.plugin.logging.Log;
import org.apache.maven.plugin.AbstractMojo;
import org.apache.maven.plugin.MojoExecutionException;

import com.google.api.client.googleapis.auth.oauth2.GoogleAuthorizationCodeFlow;
import com.google.api.client.googleapis.auth.oauth2.GoogleCredential;
import com.google.api.client.auth.oauth2.Credential;
import com.google.api.client.googleapis.auth.oauth2.GoogleTokenResponse;
import com.google.api.client.http.FileContent;
import com.google.api.client.http.HttpTransport;
import com.google.api.client.http.javanet.NetHttpTransport;
import com.google.api.client.json.JsonFactory;
import com.google.api.client.json.jackson.JacksonFactory;
import com.google.api.services.drive.Drive;
import com.google.api.services.drive.DriveScopes;
import com.google.api.services.drive.model.File;

import java.io.FileWriter;
import java.io.IOException;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.FileOutputStream;
import java.io.FileInputStream;
import java.util.Arrays;
import java.util.Properties;

/**
 * @goal connect
 */
public class ConnectMojo extends AbstractMojo {
    private static String REDIRECT_URI = "urn:ietf:wg:oauth:2.0:oob";

    /**
     * @parameter
     */
    public java.io.File googleClientProperties;

    /**
     * @parameter expression="${basedir}/src/main/resources"
     */
    private java.io.File googleDrivePropertiesDirectory;

    /**
     * @paramter expression="${googleDrive.authToken}"
     */
    public String authToken;

    @Override
    public void execute() throws MojoExecutionException {

        try {
            Properties clientProperties = new Properties();
            clientProperties.load(new FileInputStream(googleClientProperties));
            String clientId = clientProperties.getProperty("clientId");
            String clientSecret = clientProperties.getProperty("clientSecret");

            authToken = authToken == null ? clientProperties.getProperty("authToken") : authToken;

            HttpTransport httpTransport = new NetHttpTransport();
            JsonFactory jsonFactory = new JacksonFactory();

            if (clientId == null)
                throw new Exception("clientId is not defined in " + googleClientProperties.getAbsolutePath());

            if (clientSecret == null)
                throw new Exception("clientSecret is not defined in " + googleClientProperties.getAbsolutePath());

            MavenCredentialStore store = new MavenCredentialStore(googleDrivePropertiesDirectory, getLog());

            GoogleAuthorizationCodeFlow flow = new GoogleAuthorizationCodeFlow.Builder(httpTransport, jsonFactory,
                    clientId, clientSecret, Arrays.asList(DriveScopes.DRIVE)).setAccessType("offline")
                            .setApprovalPrompt("auto").setCredentialStore(store).build();

            Credential credential = new GoogleCredential.Builder().setTransport(httpTransport)
                    .setJsonFactory(jsonFactory).setClientSecrets(clientId, clientSecret).build();

            if (!store.load("service", credential)) {

                if (authToken != null) {
                    try {
                        GoogleTokenResponse response = flow.newTokenRequest(authToken).setRedirectUri(REDIRECT_URI)
                                .execute();
                        credential = flow.createAndStoreCredential(response, "service");
                        store.store("service", credential);
                    } catch (Exception e) {
                        String url = flow.newAuthorizationUrl().setRedirectUri(REDIRECT_URI).build();
                        throw new Exception("Authorization token expired, get a new one at " + url, e);
                    }
                } else {
                    String url = flow.newAuthorizationUrl().setRedirectUri(REDIRECT_URI).build();
                    throw new Exception("Application must be authorized at " + url);
                }
            }
        } catch (Exception e) {
            throw new MojoExecutionException(e.getMessage(), e);
        }

    }
}