001 /*
002 * Copyright 2010 the original author or authors.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016 package com.hs.mail.imap.message.response;
017
018 import java.util.Set;
019
020 import org.apache.log4j.Logger;
021
022 import com.hs.mail.imap.dao.DaoFactory;
023 import com.hs.mail.imap.dao.MailboxDao;
024 import com.hs.mail.imap.dao.MessageDao;
025 import com.hs.mail.imap.event.EventTracker;
026 import com.hs.mail.imap.mailbox.SelectedMailbox;
027 import com.hs.mail.imap.mailbox.UidToMsnMapper;
028
029 /**
030 *
031 * @author Won Chul Doh
032 * @since Aug 1, 2010
033 *
034 */
035 public class UnsolicitedResponseBuilder {
036
037 static Logger logger = Logger.getLogger(UnsolicitedResponseBuilder.class);
038
039 public UnsolicitedResponse build(SelectedMailbox selected,
040 EventTracker tracker) {
041 if (!tracker.isSizeChanged() && !tracker.hasExpungedUids()
042 && !tracker.hasUpdatedFlags()) {
043 // No change to the mailbox
044 return null;
045 }
046 UnsolicitedResponse response = new UnsolicitedResponse(tracker
047 .isSizeChanged());
048 if (tracker.isSizeChanged()) {
049 setExistsResponses(selected, response);
050 } else if (tracker.hasExpungedUids() || tracker.hasUpdatedFlags()) {
051 UidToMsnMapper map = new UidToMsnMapper(selected, false);
052 if (tracker.hasExpungedUids()) {
053 setExpungedResponses(response, tracker, map);
054 }
055 if (tracker.hasUpdatedFlags()) {
056 setFlagsResponses(response, tracker, map);
057 }
058 }
059 return response;
060 }
061
062 private void setExistsResponses(SelectedMailbox selected,
063 UnsolicitedResponse response) {
064 MailboxDao dao = DaoFactory.getMailboxDao();
065 response.setMessageCount(dao.getMessageCount(selected.getMailboxID()));
066 if (response.getMessageCount() > 0) {
067 response.setRecentMessageCount(dao.getRecentMessageCount(selected
068 .getMailboxID()));
069 }
070 }
071
072 private void setExpungedResponses(UnsolicitedResponse response,
073 EventTracker tracker, UidToMsnMapper map) {
074 Set<Long> expungedUids = tracker.getExpungedUids();
075 for (Long messageID : expungedUids) {
076 int msgnum = map.getMessageNumber(messageID);
077 if (msgnum != -1) {
078 response.addExpungedMsn(msgnum);
079 } else {
080 // This case is impossible.
081 logger.error("Failed to convert UID " + messageID
082 + " to message number.");
083 }
084 }
085 }
086
087 private void setFlagsResponses(UnsolicitedResponse response,
088 EventTracker tracker, UidToMsnMapper map) {
089 MessageDao dao = DaoFactory.getMessageDao();
090 Set<Long> flagUpdatedUids = tracker.getFlagUpdatedUids();
091 for (Long messageID : flagUpdatedUids) {
092 int msgnum = map.getMessageNumber(messageID);
093 if (msgnum != -1) {
094 response.addFlagsResponse(msgnum, dao.getFlags(messageID));
095 } else {
096 // This case is impossible.
097 logger.error("Failed to convert UID " + messageID
098 + " to message number.");
099 }
100 }
101 }
102
103 }