com.liferay.mail.vaadin.MessageView.java Source code

Introduction

Here is the source code for com.liferay.mail.vaadin.MessageView.java

Source

/**
 * Copyright (c) 2000-2011 Liferay, Inc. All rights reserved.
 *
 * This library is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Lesser General Public License as published by the Free
 * Software Foundation; either version 2.1 of the License, or (at your option)
 * any later version.
 *
 * This library is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
 * details.
 */

package com.liferay.mail.vaadin;

import com.liferay.mail.model.Attachment;
import com.liferay.mail.model.Message;
import com.liferay.mail.service.AttachmentLocalServiceUtil;
import com.liferay.mail.util.AttachmentHandler;
import com.liferay.portal.kernel.exception.PortalException;
import com.liferay.portal.kernel.exception.SystemException;
import com.liferay.portal.kernel.log.Log;
import com.liferay.portal.kernel.log.LogFactoryUtil;
import com.liferay.portal.kernel.util.MimeTypesUtil;

import com.vaadin.terminal.StreamResource;
import com.vaadin.ui.Button.ClickEvent;
import com.vaadin.ui.Button.ClickListener;
import com.vaadin.ui.Button;
import com.vaadin.ui.FormLayout;
import com.vaadin.ui.Label;
import com.vaadin.ui.VerticalLayout;
import com.vaadin.ui.themes.BaseTheme;

import java.io.IOException;
import java.io.InputStream;

import java.text.DateFormat;

import java.util.Date;
import java.util.List;

/**
 * @author Henri Sara
 */
public class MessageView extends VerticalLayout implements ClickListener {

    private static Log _log = LogFactoryUtil.getLog(MessageView.class);

    private Label messageLabel;
    private VerticalLayout headersAndAttachmentLayout = new VerticalLayout();
    private FormLayout headersLayout;

    public MessageView() {

        setMargin(false, true, true, true);

        messageLabel = new Label("", Label.CONTENT_XHTML);
        messageLabel.setSizeFull();

        headersLayout = new FormLayout();

        headersAndAttachmentLayout.setStyleName("message-header");
        headersAndAttachmentLayout.addComponent(headersLayout);
        headersAndAttachmentLayout.setVisible(false);

        addComponent(headersAndAttachmentLayout);
        addComponent(new Label("<hr />", Label.CONTENT_XHTML));
        addComponent(messageLabel);

        setExpandRatio(messageLabel, 1);
    }

    public void showMessage(Message msg) {

        if (msg == null) {
            messageLabel.setVisible(false);
            headersAndAttachmentLayout.setVisible(false);
            return;
        } else {
            messageLabel.setVisible(true);
            headersAndAttachmentLayout.setVisible(true);
        }
        // Body
        String text = "";
        if (msg != null) {
            text = msg.getBody();
        }
        messageLabel.setValue(text);

        // Headers
        headersLayout = new FormLayout();
        headersLayout.setSpacing(false);
        headersLayout.setMargin(false);
        if (msg != null) {
            String to = msg.getTo();
            String cc = msg.getCc();
            // String replyTo = msg.get();

            Label subject = new Label(msg.getSubject());
            subject.setCaption(Lang.get("subject"));
            headersLayout.addComponent(subject);

            Label from = new Label(msg.getSender());
            from.setCaption(Lang.get("from"));
            headersLayout.addComponent(from);

            if (to != null && !to.equals("")) {
                Label toLabel = new Label(to);
                toLabel.setCaption(Lang.get("to"));
                headersLayout.addComponent(toLabel);
            }
            if (cc != null && !cc.equals("")) {
                Label ccLabel = new Label(cc);
                ccLabel.setCaption(Lang.get("cc"));
                headersLayout.addComponent(ccLabel);
            }

            Label date = new Label(formatDate(msg.getSentDate()));
            date.setCaption(Lang.get("date"));
            headersLayout.addComponent(date);

            if (MessageUtil.isImportant(msg)) {
                Label flag = new Label(Lang.get("important"));
                flag.setStyleName(MessageList.STYLE_IMPORTANT);
                flag.setCaption(Lang.get("flag"));
                headersLayout.addComponent(flag);

            }
        }

        // Attachments
        try {
            headersAndAttachmentLayout.removeAllComponents();
            headersAndAttachmentLayout.addComponent(headersLayout);

            Controller controller = Controller.get();
            List<Attachment> attachments = AttachmentLocalServiceUtil.getAttachments(msg.getMessageId());
            if (attachments != null && !attachments.isEmpty()) {
                for (Attachment attachment : attachments) {
                    Button attachmentDownload = new Button();
                    attachmentDownload.setStyleName(BaseTheme.BUTTON_LINK);

                    attachmentDownload.setCaption(attachment.getFileName() + " "
                            + MessageUtil.formatSize(attachment.getSize(), controller.getUserLocale()));
                    attachmentDownload.setData(attachment);
                    attachmentDownload.addListener(this);

                    headersAndAttachmentLayout.addComponent(attachmentDownload);
                }
            }
        } catch (SystemException e) {
            _log.debug(e);
        }
    }

    private String formatDate(Date date) {

        if (date == null) {
            return "";
        }

        return DateFormat.getDateTimeInstance().format(date);
    }

    public void buttonClick(ClickEvent event) {

        Button b = event.getButton();
        Object data = b.getData();
        if (data != null && data instanceof Attachment) {
            final Attachment attachment = (Attachment) data;
            try {
                final AttachmentHandler handler = Controller.get().getMailManager()
                        .getAttachment(attachment.getAttachmentId());
                StreamResource r = new StreamResource(new StreamResource.StreamSource() {
                    public InputStream getStream() {

                        return handler.getInputStream();
                    }

                }, attachment.getFileName(), Controller.get().getApplication());

                String contentType = MimeTypesUtil.getContentType(attachment.getFileName());
                r.setMIMEType(contentType);

                Controller.get().getApplication().getMainWindow().open(r);
            } catch (PortalException e) {
                Controller.get().showError(Lang.get("unable-to-fetch-attachment"), e);
            } catch (SystemException e) {
                Controller.get().showError(Lang.get("unable-to-fetch-attachment"), e);
            } catch (IOException e) {
                Controller.get().showError(Lang.get("unable-to-fetch-attachment"), e);
            }
        }
    }

}