MailClientIntegrationTest.java

package com.google.sitebricks.mail;

import com.google.common.base.Preconditions;
import com.google.common.util.concurrent.ListenableFuture;
import com.google.inject.Guice;
import com.google.sitebricks.mail.Mail.Auth;
import com.google.sitebricks.mail.MailClient.WireError;
import com.google.sitebricks.mail.imap.Folder;
import com.google.sitebricks.mail.imap.Message;
import com.google.sitebricks.mail.imap.MessageStatus;

import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * @author dhanji@gmail.com (Dhanji R. Prasanna)
 */
public class MailClientIntegrationTest {

  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);

    final String username = System.getProperty("sitebricks-mail.username");
    final String password = System.getProperty("sitebricks-mail.password");
    Preconditions.checkArgument(username != null && password != null);

    MailClientHandler.addUserForVerboseLogging(username, true);
    NettyImapClient.addUserForVerboseOutput(username, true);

    final MailClient client = mail.clientOf("imap.gmail.com", 993).prepare(Auth.SSL, username, password);
    try {
      client.connect();
    } catch (Exception e) {
      e.printStackTrace();
      System.out.println(">>>>>>>>>>>>>>>>>>>>>>>>>>>");
      WireError lastError = client.lastError();
      System.out.println(lastError.expected());
      System.out.println(lastError.message());
      System.out.println(lastError.trace());
      System.out.println(">>>>>>>>>>>>>>>>>>>>>>>>>>>");
    }

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

    System.out.println("FOLDERS: " + client.listFolders().get());
//    try {
//      Folder f = client.open("Thumping through the brush.", false).get();
//      System.out.println("Expected failure attempting to open invalid folder.");
//    } catch (ExecutionException ee) {
//      // expected.
//    }

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



    final ExecutorService executor = Executors.newCachedThreadPool();
    future.addListener(new Runnable() {
      @Override
      public void run() {
//        final ListenableFuture<List<Integer>> messageStatuses =
//            client.searchUid(allMail, "is:read", new Date(System.currentTimeMillis() - (500L * 24L * 60L * 60L * 1000L)));
        final ListenableFuture<List<MessageStatus>> messageStatuses =
            client.list(allMail, allMail.getCount() -5 , -1);

        try {
//          System.out.println(messageStatuses.get());
          for (MessageStatus messageStatus : messageStatuses.get()) {
            System.out.println(messageStatus);
          }

          fetchUidAndDump(client, allMail, 491, executor).await();
//          fetchUidAndDump(client, allMail, 33288, executor).await();
          client.disconnect();
        } catch (InterruptedException e) {
          e.printStackTrace();
        } catch (ExecutionException e) {
          e.printStackTrace();
        }
      }
    }, executor);
  }

  private static CountDownLatch fetchUidAndDump(final MailClient client, Folder allMail, int uid,
                                                ExecutorService executor) {
    final ListenableFuture<Message> msgFuture = client.fetchUid(allMail, uid);

    final CountDownLatch countDownLatch = new CountDownLatch(1);

    msgFuture.addListener(new Runnable() {
      @Override
      public void run() {
        try {
          Message message = msgFuture.get();
          //            System.out.println(ToStringBuilder.reflectionToString(message));
          for (Message.BodyPart bodyPart : message.getBodyParts()) {
            //              System.out.println(ToStringBuilder.reflectionToString(bodyPart));
          }

          System.out.println(">>>>>>>>>>>>>>>>>>>>>>>>>");
          System.out.println(message.getImapUid());
          System.out.println(message.getHeaders().get("Message-ID"));
          System.out.println(message.getHeaders());
          System.out.println(">>>>>>>>>>>>>>>>>>>>>>>>>\n\n\n");

//                System.out.println("Gmail flags set: " +
//                    client.addFlags(allMail, message.getImapUid(),
//                        ImmutableSet.of(Flag.SEEN)).get());

//                System.out
//                    .println("Matched UID: " + (message.getImapUid() == messageStatuses.get()
//                        .iterator()
//                        .next()
//                        .getImapUid()));
//                System.out.println("Fetched: " + message);
          dumpBodyParts(message.getBodyParts(), "");

          countDownLatch.countDown();
        } catch (InterruptedException e) {
          e.printStackTrace();
        } catch (ExecutionException e) {
          e.printStackTrace();
        }
      }
    }, executor);
    return countDownLatch;
  }

  private static void dumpBodyParts(List<Message.BodyPart> parts, String indent) {
    if (parts != null) {
      for (Message.BodyPart part : parts) {
        if (part != null) {
          System.out.println(indent + "Body part:");
          System.out.println(indent + part.getHeaders().entries());
          System.out.println(indent + part.getBody());
          dumpBodyParts(part.getBodyParts(), indent + ">>>>");
        }
      }
    }
  }
}