XoauthMailClientIntegrationTest.java

package com.google.sitebricks.mail;

import com.google.common.util.concurrent.ListenableFuture;
import com.google.inject.Guice;
import com.google.sitebricks.mail.imap.Folder;
import com.google.sitebricks.mail.imap.FolderStatus;
import com.google.sitebricks.mail.imap.Message;
import com.google.sitebricks.mail.oauth.OAuthConfig;
import org.scribe.builder.ServiceBuilder;
import org.scribe.builder.api.GoogleApi;
import org.scribe.model.Token;
import org.scribe.model.Verifier;
import org.scribe.oauth.OAuthService;

import java.util.List;
import java.util.Scanner;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.Executors;

/**
 * @author dhanji@gmail.com (Dhanji R. Prasanna)
 */
public class XoauthMailClientIntegrationTest {
  // Same as CONSUMER_KEY
  public static final String CLIET_ID_PROP = "xoauth.clientId";
  // Same as CONSUMER_SECRET
  public static final String CLIENT_SECRET_PROP = "xoauth.clientSecret";

  static {
    java.util.logging.ConsoleHandler fh = new java.util.logging.ConsoleHandler();
    java.util.logging.Logger.getLogger("").addHandler(fh);
    java.util.logging.Logger.getLogger("").setLevel(java.util.logging.Level.FINEST);
  }

  public static void main(String... args) throws InterruptedException, ExecutionException {
    Mail mail = Guice.createInjector().getInstance(Mail.class);
    String clientId = System.getProperty(CLIET_ID_PROP);
    String clientSecret = System.getProperty(CLIENT_SECRET_PROP);

    OAuthorize oauthorize = oauthorize(clientId, clientSecret);
    System.out.println("Requesting email via OAuth...");


    final MailClient client = mail.clientOf("imap.gmail.com", 993)
        .prepareOAuth("dhanji@gmail.com", new OAuthConfig(
            oauthorize.accessToken,
            oauthorize.code,
            clientId,
            clientSecret));

    client.connect();

    List<String> capabilities = client.capabilities();
    System.out.println("CAPS: " + capabilities);

    final ListenableFuture<FolderStatus> fStatus =
        client.statusOf("[Gmail]/All Mail");
    ListenableFuture<Folder> future = client.open("[Gmail]/All Mail");
    final Folder allMail = future.get();
    final FolderStatus folderStatus = fStatus.get();
    System.out
        .println("Folder opened: " + allMail.getName() + " with count " + folderStatus.getMessages());

    future.addListener(new Runnable() {
      @Override
      public void run() {
        ListenableFuture<List<Message>> messages = client.fetch(allMail, 80034, 80084);
        try {
          for (Message message : messages.get()) {
            for (Message.BodyPart bodyPart : message.getBodyParts()) {
            }
          }

          System.out.println("Fetched: " + messages.get().size());
        } catch (InterruptedException e) {
          e.printStackTrace();
        } catch (ExecutionException e) {
          e.printStackTrace();
        }
        client.disconnect();

        System.exit(0);
      }
    }, Executors.newCachedThreadPool());


  }

  private static OAuthorize oauthorize(String clientId, String clientSecret) {

    // Execute out-of-band OAuth steps to obtain an access token.
    OAuthService oAuthService = new ServiceBuilder()
        .provider(GoogleApi.class)
        .apiKey(clientId)
        .apiSecret(clientSecret)
        .scope("https://mail.google.com/")
        .build();

    Token requestToken = oAuthService.getRequestToken();
    System.out.println("Retrieved Request token. Please paste the verifier from this URL:");
    System.out.println("https://www.google.com/accounts/OAuthAuthorizeToken?oauth_token="
        + requestToken.getToken());

    System.out.print("\n> ");
    Scanner in = new Scanner(System.in);

    Verifier verifier = new Verifier(in.nextLine());
    Token accessToken = oAuthService.getAccessToken(requestToken, verifier);
    System.out.println("\nAccess token successfully retrieved: " + accessToken.getToken());

    return new OAuthorize(accessToken.getToken(), accessToken.getSecret());
  }

  private static class OAuthorize {
    private final String accessToken;
    private final String code;

    private OAuthorize(String accessToken, String code) {
      this.accessToken = accessToken;
      this.code = code;
    }
  }
}