1 files changed, 141 insertions, 0 deletions
diff --git a/libmail/smapfetch.C b/libmail/smapfetch.C
new file mode 100644
index 0000000..7f360b6
--- /dev/null
+++ b/libmail/smapfetch.C
@@ -0,0 +1,141 @@
+/*
+** Copyright 2003, Double Precision Inc.
+**
+** See COPYING for distribution information.
+*/
+
+#include "smapfetch.H"
+
+using namespace std;
+
+const char *mail::smapFETCH::getName()
+{
+	return "FETCHMESSAGE";
+}
+
+mail::smapFETCH::smapFETCH(const std::vector<size_t> &messages,
+			   bool peek,
+			   std::string mime_id,
+			   mail::readMode getType,
+			   const char *decoded,
+			   mail::callback::message &msgcallbackArg,
+			   mail::imap &imapAccount)
+	: uidSet(imapAccount, messages),
+	  msgcallback(msgcallbackArg),
+	  expectedCount(messages.size()),
+	  countProcessed(0)
+{
+	defaultCB= &msgcallbackArg;
+
+	string element;
+
+	switch (getType) {
+	case mail::readHeadersFolded:
+		element="HEADERS";
+		break;
+	case mail::readContents:
+		element="BODY";
+		break;
+	case mail::readBoth:
+		element="ALL";
+		break;
+	case mail::readHeaders:
+		element="RAWHEADERS";
+		break;
+	}
+
+	fetchingMessageNum=0;
+
+	if (decoded && *decoded)
+		element += decoded;
+
+	if (mime_id.size() > 0)
+		mime_id="[" + mime_id + "]";
+
+	fetchCmd= mail::imap::quoteSMAP((peek ? "CONTENTS.PEEK=":"CONTENTS=")
+					+ element + mime_id);
+}
+
+mail::smapFETCH::~smapFETCH()
+{
+}
+
+void mail::smapFETCH::installed(imap &imapAccount)
+{
+	msgRange.init(imapAccount, uidSet);
+	uidSet.clear();
+	if (go())
+		return;
+
+	ok("OK");
+	imapAccount.uninstallHandler(this);
+}
+
+bool mail::smapFETCH::ok(std::string msg)
+{
+	if (go())
+	{
+		doDestroy=false;
+		return true;
+	}
+
+	return smapHandler::ok(msg);
+}
+
+bool mail::smapFETCH::go()
+{
+	ostringstream msgList;
+
+	msgList << "FETCH";
+
+	if (!myimap->currentFolder ||
+	    myimap->currentFolder->closeInProgress ||
+	    !(msgRange >> msgList))
+		return false;
+
+	msgList << " " << fetchCmd << "\n";
+	myimap->imapcmd("", msgList.str());
+	return true;
+}
+
+void mail::smapFETCH::beginProcessData(imap &imapAccount,
+				       std::vector<const char *> &words,
+				       unsigned long estimatedSizeArg)
+{
+	estimatedSize=estimatedSizeArg;
+	sizeDone=0;
+
+	if (words.size() >= 2 && strcasecmp(words[0], "FETCH") == 0)
+	{
+		string n=words[1];
+		istringstream i(n);
+
+		fetchingMessageNum=0;
+
+		i >> fetchingMessageNum;
+	}
+}
+
+void mail::smapFETCH::processData(imap &imapAccount,
+				  std::string data)
+{
+	if (fetchingMessageNum > 0)
+	{
+		if ((sizeDone += data.size()) > estimatedSize)
+			estimatedSize=sizeDone;
+
+		msgcallback.reportProgress(sizeDone, estimatedSize,
+					   countProcessed, expectedCount);
+		msgcallback.messageTextCallback(fetchingMessageNum-1,
+						data);
+	}
+}
+
+void mail::smapFETCH::endData(imap &imapAccount)
+{
+	if (++countProcessed > expectedCount)
+		expectedCount=countProcessed;
+
+	msgcallback.reportProgress(sizeDone, sizeDone, countProcessed,
+				   expectedCount);
+}