Changeset 247 for trunk/src/org/expeditee/agents/mail/MailSession.java

Timestamp:

08/22/08 10:54:44 (16 years ago)

Author:

ra33

Message:

Added more import and mail stuff... including text importer

File:

• trunk/src/org/expeditee/agents/mail/MailSession.java (modified) (16 diffs)

trunk/src/org/expeditee/agents/mail/MailSession.java

@@ -1 +1 @@
 package org.expeditee.agents.mail;
 
+import java.awt.Color;
+import java.awt.Point;
 import java.io.BufferedReader;
 import java.io.InputStream;
 import java.io.InputStreamReader;
+import java.util.Collection;
 import java.util.Date;
+import java.util.LinkedList;
 import java.util.Properties;
 
@@ -18 +22 @@
 import javax.mail.Transport;
 import javax.mail.Flags.Flag;
+import javax.mail.event.MessageCountAdapter;
+import javax.mail.event.MessageCountEvent;
 import javax.mail.internet.InternetAddress;
 import javax.mail.internet.MimeMessage;
@@ -23 +29 @@
 import org.expeditee.gui.AttributeValuePair;
 import org.expeditee.gui.Frame;
+import org.expeditee.gui.FrameCreator;
 import org.expeditee.gui.MessageBay;
+import org.expeditee.importer.FrameDNDTransferHandler;
+import org.expeditee.items.Item;
 import org.expeditee.items.Text;
 
 public class MailSession {
-        public static boolean _autoConnect = true;
+        public static boolean _autoConnect = false;
 
         private static MailSession _theMailSession = null;
@@ -45 +54 @@
         private String _password;
 
-        private String _popServer;
+        private String _mailServer;
+
+        private String _serverType;
 
         private Boolean _bConnecting;
@@ -56 +67 @@
                 _username = null;
                 _password = "";
-                _popServer = null;
+                _mailServer = null;
+                _serverType = null;
 
                 // Set the settings
                 for (Text item : settingsFrame.getBodyTextItems(false)) {
+                        if (item.getText().toLowerCase().trim().equals("autoconnect")) {
+                                _autoConnect = true;
+                        }
                         AttributeValuePair avp = new AttributeValuePair(item.getText());
                         if (!avp.hasPair())
@@ -81 +96 @@
                                 props.setProperty("mail.smtp.auth", "true");
                         } else if (attribute.equals("popserver")) {
-                                _popServer = avp.getValue();
+                                _mailServer = avp.getValue();
+                                _serverType = "pop3s";
+                                props.setProperty("mail.pop3.host", _mailServer);
+                        } else if (attribute.equals("imapserver")) {
+                                _mailServer = avp.getValue();
+                                _serverType = "imaps";
+                                props.setProperty("mail.imap.host", _mailServer);
                         }
                 }
@@ -97 +118 @@
                 try {
                         // Set up the mail receiver
-                        _store = _session.getStore("pop3s");
+                        _store = _session.getStore(_serverType);
 
                         // Connect the mail sender
@@ -133 +154 @@
                                 _bConnecting = true;
                                 _transport.connect();
-                                MessageBay.overwriteMessage("SMTP server connected");
+                                MessageBay.overwriteMessage("SMTP server connected",
+                                                Color.green);
                         } else {
                                 MessageBay.warningMessage("SMTP server already connected");
@@ -141 +163 @@
                 }
 
-                if (_popServer != null && !_store.isConnected()) {
+                if (_mailServer != null && !_store.isConnected()) {
                         try {
-                                MessageBay.displayMessage("Connecting to POP3 server...");
-                                _store.connect(_popServer, _username, _password);
+                                MessageBay.displayMessage("Connecting to " + _mailServer
+                                                + "...");
+                                _store.connect(_mailServer, _username, _password);
 
                                 // -- Try to get hold of the default folder --
@@ -153 +176 @@
                                 _folder = _folder.getFolder("INBOX");
                                 if (_folder == null)
-                                        throw new Exception("No POP3 INBOX");
+                                        throw new Exception("No INBOX");
                                 // -- Open the folder for read only --
-                                _folder.open(Folder.READ_ONLY);
-                                MessageBay.overwriteMessage("POP3 server connected");
-                                MessageBay.warningMessage(getUnreadMessageCount() + " unread messages are in your inbox");
+                                _folder.open(Folder.READ_WRITE);
+                                _folder.addMessageCountListener(new MessageCountAdapter() {
+                                        @Override
+                                        public void messagesAdded(MessageCountEvent e) {
+                                                MessageBay.displayMessage("New mail message!",
+                                                                Color.green);
+                                                displayUnreadMailCount();
+                                        }
+                                });
+
+                                new Thread() {
+                                        public void run() {
+                                                for (;;) {
+                                                        try {
+                                                                Thread.sleep(1000); // sleep for freq
+                                                                // milliseconds
+                                                                // This is to force the IMAP server to send us
+                                                                // EXISTS notifications
+                                                                // TODO: Is synchronisation needed?
+                                                                _folder.getMessageCount();
+                                                        } catch (Exception e) {
+                                                        }
+                                                }
+                                        }
+                                }.start();
+
+                                MessageBay.overwriteMessage("Connection complete", Color.GREEN);
+
+                                displayUnreadMailCount();
+
                         } catch (Exception e) {
                                 // e.printStackTrace();
-                                MessageBay.errorMessage("Error connecting to POP3 server");
+                                MessageBay.errorMessage("Error connecting to " + _mailServer);
                         }
                 }
                 _bConnecting = false;
+        }
+
+        private void displayUnreadMailCount() {
+                Text text = new Text(-1, getUnreadCount() + " unread message"
+                                + (getUnreadCount() == 1 ? "" : "s"), Color.BLUE, null);
+                text.addAction("getUnreadMail");
+                MessageBay.displayMessage(text);
         }
 
@@ -287 +344 @@
         }
 
-        public int getUnreadMessageCount() {
-                // -- Get the message wrappers and process them --
-                Message[] msgs;
-                int count = 0;
-                try {
-                        msgs = _folder.getMessages();
-                        for (int msgNum = 0; msgNum < msgs.length; msgNum++) {
-                                if (msgs[msgNum].getFlags().contains(Flag.SEEN))
-                                        continue;
-                                count++;
-                        }
+        public int getUnreadCount() {
+                try {
+                        return _folder.getUnreadMessageCount();
                 } catch (MessagingException e) {
                         // TODO Auto-generated catch block
                         e.printStackTrace();
                 }
-                return count;
-        }
-
-        public String getMail() {
+                return 0;
+        }
+
+        /**
+         * Gets mail and puts a list of mail items with links to the messages
+         * content. TODO: Put reply and forward button on the frame...
+         * 
+         * @param flag
+         * @param isPresent
+         * @param frame
+         * @param point
+         * @return
+         */
+        public String getMailString(Flag flag, Boolean isPresent) {
                 StringBuffer sb = new StringBuffer();
                 // -- Get the message wrappers and process them --
@@ -312 +371 @@
                         msgs = _folder.getMessages();
                         for (int msgNum = 0; msgNum < msgs.length; msgNum++) {
-                                if (msgs[msgNum].getFlags().contains(Flag.SEEN))
-                                        continue;
-                                // Only get messages that have not been read
-                                sb.append(getTextMessage(msgs[msgNum]));
-                                sb.append('\n');
+
+                                if (flag == null
+                                                || msgs[msgNum].getFlags().contains(flag) == isPresent) {
+                                        if (sb.length() > 0) {
+                                                sb.append('\n').append('\n').append(
+                                                                "-----------------------------").append('\n')
+                                                                .append('\n');
+                                        }
+                                        // Only get messages that have not been read
+                                        sb.append(getTextMessage(msgs[msgNum]));
+                                }
                         }
                 } catch (MessagingException e) {
                         e.printStackTrace();
                 }
-                sb.deleteCharAt(sb.length() - 1);
                 return sb.toString();
+        }
+
+        /**
+         * Gets mail and puts a list of mail items with links to the messages
+         * content. TODO: Put reply and forward button on the frame...
+         * 
+         * @param flag
+         * @param isPresent
+         * @param frame
+         * @param point
+         * @return
+         */
+        public Collection<Item> getMail(Flag flag, Boolean isPresent, Frame frame,
+                        Point point) {
+                Collection<Item> mailItems = new LinkedList<Item>();
+                // -- Get the message wrappers and process them --
+                Message[] msgs;
+                try {
+                        msgs = _folder.getMessages();
+                        for (int msgNum = 0; msgNum < msgs.length; msgNum++) {
+                                if (flag == null
+                                                || msgs[msgNum].getFlags().contains(flag) == isPresent) {
+                                        Item newItem = readMessage(msgs[msgNum], msgNum, frame, point);
+                                        if (newItem != null) {
+                                                mailItems.add(newItem);
+                                                point.y += newItem.getBoundsHeight();
+                                        }
+                                }
+
+                        }
+                } catch (MessagingException e) {
+                        e.printStackTrace();
+                }
+                return mailItems;
+        }
+
+        private Item readMessage(final Message message, int messageNo, Frame frame, Point point) {
+                String subject;
+                try {
+                        subject = message.getSubject();
+                } catch (MessagingException e) {
+                        MessageBay.errorMessage("GetMail error: " + e.getMessage());
+                        return null;
+                }
+
+                final Text source = FrameDNDTransferHandler
+                                .importString(subject, point);
+                // Create a frameCreator
+                final FrameCreator frames = new FrameCreator(frame.getFramesetName(),
+                                frame.path, messageNo + ". " + subject, false);
+
+                new Thread() {
+                        public void run() {
+                                try {
+                                        // Get the header information
+                                        String from = ((InternetAddress) message.getFrom()[0])
+                                                        .getPersonal();
+                                        if( from != null){
+                                                frames.addText("@fromName: " + from, null, null, null, false);
+                                        }
+                                        from = ((InternetAddress) message.getFrom()[0])
+                                                                .getAddress();
+                                        frames.addText("@fromAddress: " + from, null, null, null, false);
+
+                                        // -- Get the message part (i.e. the message itself) --
+                                        Part messagePart = message;
+                                        Object content = messagePart.getContent();
+                                        // -- or its first body part if it is a multipart
+                                        // message --
+                                        if (content instanceof Multipart) {
+                                                messagePart = ((Multipart) content).getBodyPart(0);
+                                                // System.out.println("[ Multipart Message ]");
+                                        }
+                                        // -- Get the content type --
+                                        String contentType = messagePart.getContentType().toLowerCase();
+                                        // -- If the content is plain text, we can print it --
+                                        // System.out.println("CONTENT:" + contentType);
+                                        if (contentType.startsWith("text/plain")
+                                                        || contentType.startsWith("text/html")) {
+                                                InputStream is = messagePart.getInputStream();
+                                                BufferedReader reader = new BufferedReader(
+                                                                new InputStreamReader(is));
+                                                String thisLine = reader.readLine();
+                                                while (thisLine != null) {
+                                                        frames.addText(thisLine, null, null, null, false);
+                                                        thisLine = reader.readLine();
+                                                }
+                                        }
+                                        message.setFlag(Flag.SEEN, true);
+
+                                        frames.save();
+                                        source.setLink(frames.getName());
+                                } catch (Exception e) {
+                                        MessageBay.errorMessage("GetMail error: " + e.getMessage());
+                                }
+                        }
+                }.start();
+                return source;
         }
 
@@ -330 +492 @@
         public String getTextMessage(Message message) {
                 StringBuffer sb = new StringBuffer();
-                
+
                 try {
                         // Get the header information
@@ -337 +499 @@
                         if (from == null)
                                 from = ((InternetAddress) message.getFrom()[0]).getAddress();
-                        sb.append("FROM: " + from);
+                        sb.append("FROM: " + from).append('\n');
                         String subject = message.getSubject();
-                        sb.append("SUBJECT: " + subject);
+                        sb.append("SUBJECT: " + subject).append('\n').append('\n');
                         // -- Get the message part (i.e. the message itself) --
                         Part messagePart = message;
@@ -351 +513 @@
                         String contentType = messagePart.getContentType();
                         // -- If the content is plain text, we can print it --
-                        //System.out.println("CONTENT:" + contentType);
+                        // System.out.println("CONTENT:" + contentType);
                         if (contentType.startsWith("text/plain")
                                         || contentType.startsWith("text/html")) {
@@ -359 +521 @@
                                 String thisLine = reader.readLine();
                                 while (thisLine != null) {
-                                        sb.append(thisLine);
+                                        sb.append(thisLine).append('\n');
                                         thisLine = reader.readLine();
                                 }
                         }
-                        sb.append("-----------------------------");
+                        message.setFlag(Flag.SEEN, true);
                 } catch (Exception ex) {
                         ex.printStackTrace();
                 }
+                sb.deleteCharAt(sb.length() - 1);
                 return sb.toString();
         }
+
+        public Folder getFolder() {
+                return _folder;
+        }
 }