Search in sources :

Example 1 with MailModelWrapper

use of org.dataportabilityproject.dataModels.mail.MailModelWrapper in project data-transfer-project by google.

the class ImapMailHelper method getMessages.

/**
 * Get all messages in an account.
 */
private MailModelWrapper getMessages(String host, String account, String password, Folder parentFolder, boolean fetchMessages, PaginationInformation paginationInformation) throws MessagingException, IOException {
    int foldersSize = 0;
    // Find containers to and be imported
    ImmutableCollection.Builder<MailContainerModel> folders = ImmutableList.builder();
    ImmutableCollection.Builder<Resource> folderIds = ImmutableList.builder();
    log("Calling list for folder: %s", parentFolder.getName());
    Folder[] subFolders = parentFolder.list();
    log("Folder: %s, subFolders: %d", parentFolder.getName(), subFolders.length);
    for (Folder folder : subFolders) {
        // This will tell the framework to create this folder on import
        folders.add(new MailContainerModel(folder.getName(), folder.getFullName()));
        // Content for these resources will be 'fetched' by the framework
        folderIds.add(new IdOnlyResource(folder.getName()));
        foldersSize++;
    }
    log("foldersSize: %d", foldersSize);
    // Get messages in the folder
    ImmutableCollection.Builder<MailMessageModel> resources = ImmutableList.builder();
    log("fetchMessages: %b", fetchMessages);
    PaginationInformation nextPaginationInformation = null;
    if (fetchMessages) {
        parentFolder.open(Folder.READ_ONLY);
        int start = getStart(paginationInformation);
        int end = getEnd(start, parentFolder.getMessageCount());
        if (end < parentFolder.getMessageCount()) {
            // Indicates page to be fetched on next request
            nextPaginationInformation = new IntPaginationToken(end + 1);
        }
        log("Fetching messages for foder: %s, start: %d, end: %d", parentFolder.getFullName(), start, end);
        Message[] messages = parentFolder.getMessages(start, end);
        log("Fetched message for folder: %s, messages: %s", parentFolder.getFullName(), messages.length);
        for (Message message : messages) {
            log("Message, contentType: %s ,size: %s", message.getContentType(), message.getSize());
            ImmutableList<String> folderId = ImmutableList.of(parentFolder.getName());
            resources.add(new MailMessageModel(createRawMessage(message), folderId));
        }
        parentFolder.close(false);
    }
    // TODO: add pagination below
    return new MailModelWrapper(folders.build(), resources.build(), new ContinuationInformation(folderIds.build(), nextPaginationInformation));
}
Also used : IntPaginationToken(org.dataportabilityproject.shared.IntPaginationToken) ImmutableCollection(com.google.common.collect.ImmutableCollection) Message(javax.mail.Message) Resource(org.dataportabilityproject.dataModels.Resource) IdOnlyResource(org.dataportabilityproject.shared.IdOnlyResource) IdOnlyResource(org.dataportabilityproject.shared.IdOnlyResource) Folder(javax.mail.Folder) ContinuationInformation(org.dataportabilityproject.dataModels.ContinuationInformation) MailContainerModel(org.dataportabilityproject.dataModels.mail.MailContainerModel) MailModelWrapper(org.dataportabilityproject.dataModels.mail.MailModelWrapper) MailMessageModel(org.dataportabilityproject.dataModels.mail.MailMessageModel) PaginationInformation(org.dataportabilityproject.dataModels.PaginationInformation)

Example 2 with MailModelWrapper

use of org.dataportabilityproject.dataModels.mail.MailModelWrapper in project data-transfer-project by google.

the class GoogleMailService method export.

// This currently exports each message with associated labels
@Override
public MailModelWrapper export(ExportInformation exportInformation) throws IOException {
    Messages.List request = gmail.users().messages().list(USER).setMaxResults(MAX_RESULTS_PER_REQUEST);
    if (exportInformation.getPaginationInformation().isPresent()) {
        request.setPageToken(((StringPaginationToken) exportInformation.getPaginationInformation().get()).getId());
    }
    ListMessagesResponse response = request.execute();
    List<MailMessageModel> results = new ArrayList<>(response.getMessages().size());
    // as we can't store all the mail messagess in memory at once.
    for (Message listMessage : response.getMessages()) {
        Message getResponse = gmail.users().messages().get(USER, listMessage.getId()).setFormat("raw").execute();
        // TODO: note this doesn't transfer things like labels
        results.add(new MailMessageModel(getResponse.getRaw(), getResponse.getLabelIds()));
    }
    PaginationInformation pageInfo = null;
    if (response.getNextPageToken() != null) {
        pageInfo = new StringPaginationToken(response.getNextPageToken());
    }
    // TODO: export by label or by message?
    return new MailModelWrapper(null, results, new ContinuationInformation(null, pageInfo));
}
Also used : ListMessagesResponse(com.google.api.services.gmail.model.ListMessagesResponse) ContinuationInformation(org.dataportabilityproject.dataModels.ContinuationInformation) Messages(com.google.api.services.gmail.Gmail.Users.Messages) Message(com.google.api.services.gmail.model.Message) ArrayList(java.util.ArrayList) MailModelWrapper(org.dataportabilityproject.dataModels.mail.MailModelWrapper) MailMessageModel(org.dataportabilityproject.dataModels.mail.MailMessageModel) PaginationInformation(org.dataportabilityproject.dataModels.PaginationInformation) StringPaginationToken(org.dataportabilityproject.shared.StringPaginationToken)

Aggregations

ContinuationInformation (org.dataportabilityproject.dataModels.ContinuationInformation)2 PaginationInformation (org.dataportabilityproject.dataModels.PaginationInformation)2 MailMessageModel (org.dataportabilityproject.dataModels.mail.MailMessageModel)2 MailModelWrapper (org.dataportabilityproject.dataModels.mail.MailModelWrapper)2 Messages (com.google.api.services.gmail.Gmail.Users.Messages)1 ListMessagesResponse (com.google.api.services.gmail.model.ListMessagesResponse)1 Message (com.google.api.services.gmail.model.Message)1 ImmutableCollection (com.google.common.collect.ImmutableCollection)1 ArrayList (java.util.ArrayList)1 Folder (javax.mail.Folder)1 Message (javax.mail.Message)1 Resource (org.dataportabilityproject.dataModels.Resource)1 MailContainerModel (org.dataportabilityproject.dataModels.mail.MailContainerModel)1 IdOnlyResource (org.dataportabilityproject.shared.IdOnlyResource)1 IntPaginationToken (org.dataportabilityproject.shared.IntPaginationToken)1 StringPaginationToken (org.dataportabilityproject.shared.StringPaginationToken)1