1 files changed, 163 insertions, 0 deletions
diff --git a/vmime-master/src/vmime/misc/importanceHelper.cpp b/vmime-master/src/vmime/misc/importanceHelper.cpp
new file mode 100644
index 0000000..20c8e13
--- /dev/null
+++ b/vmime-master/src/vmime/misc/importanceHelper.cpp
@@ -0,0 +1,163 @@
+//
+// VMime library (http://www.vmime.org)
+// Copyright (C) 2002 Vincent Richard <vincent@vmime.org>
+//
+// This program is free software; you can redistribute it and/or
+// modify it under the terms of the GNU General Public License as
+// published by the Free Software Foundation; either version 3 of
+// the License, or (at your option) any later version.
+//
+// This program 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
+// General Public License for more details.
+//
+// You should have received a copy of the GNU General Public License along
+// with this program; if not, write to the Free Software Foundation, Inc.,
+// 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+//
+// Linking this library statically or dynamically with other modules is making
+// a combined work based on this library.  Thus, the terms and conditions of
+// the GNU General Public License cover the whole combination.
+//
+
+#include "vmime/misc/importanceHelper.hpp"
+#include "vmime/exception.hpp"
+
+#include "vmime/text.hpp"
+
+
+namespace vmime {
+namespace misc {
+
+
+void importanceHelper::resetImportance(const shared_ptr <message>& msg) {
+
+	resetImportanceHeader(msg->getHeader());
+}
+
+
+void importanceHelper::resetImportanceHeader(const shared_ptr <header>& hdr) {
+
+	shared_ptr <headerField> fld;
+
+	if ((fld = hdr->findField("X-Priority")))
+		hdr->removeField(fld);
+
+	if ((fld = hdr->findField("Importance")))
+		hdr->removeField(fld);
+}
+
+
+importanceHelper::Importance importanceHelper::getImportance(const shared_ptr <const message>& msg) {
+
+	return getImportanceHeader(msg->getHeader());
+}
+
+
+importanceHelper::Importance importanceHelper::getImportanceHeader(const shared_ptr <const header>& hdr) {
+
+	// Try "X-Priority" field
+	shared_ptr <const headerField> fld = hdr->findField("X-Priority");
+
+	if (fld) {
+
+		const string value = fld->getValue <text>()->getWholeBuffer();
+
+		int n = IMPORTANCE_NORMAL;
+
+		std::istringstream iss(value);
+		iss.imbue(std::locale::classic());
+
+		iss >> n;
+
+		Importance i = IMPORTANCE_NORMAL;
+
+		switch (n) {
+			case 1: i = IMPORTANCE_HIGHEST; break;
+			case 2: i = IMPORTANCE_HIGH; break;
+			case 3: i = IMPORTANCE_NORMAL; break;
+			case 4: i = IMPORTANCE_LOW; break;
+			case 5: i = IMPORTANCE_LOWEST; break;
+		}
+
+		return i;
+
+	} else {
+
+		// Try "Importance" field
+		fld = hdr->findField("Importance");
+
+		if (fld) {
+
+			const string value = utility::stringUtils::toLower(utility::stringUtils::trim
+				(fld->getValue <text>()->getWholeBuffer()));
+
+			if (value == "low") {
+				return IMPORTANCE_LOWEST;
+			} else if (value == "high") {
+				return IMPORTANCE_HIGHEST;
+			} else {
+				return IMPORTANCE_NORMAL;
+			}
+
+		} else {
+
+			// Default
+			return IMPORTANCE_NORMAL;
+		}
+	}
+
+	// Should not go here...
+	return IMPORTANCE_NORMAL;
+}
+
+
+void importanceHelper::setImportance(const shared_ptr <message>& msg, const Importance i) {
+
+	setImportanceHeader(msg->getHeader(), i);
+}
+
+
+void importanceHelper::setImportanceHeader(const shared_ptr <header>& hdr, const Importance i) {
+
+	// "X-Priority:" Field
+	shared_ptr <headerField> fld = hdr->getField("X-Priority");
+
+	switch (i) {
+		case IMPORTANCE_HIGHEST: fld->setValue("1 (Highest)"); break;
+		case IMPORTANCE_HIGH:    fld->setValue("2 (High)"); break;
+		default:
+		case IMPORTANCE_NORMAL:  fld->setValue("3 (Normal)"); break;
+		case IMPORTANCE_LOW:     fld->setValue("4 (Low)"); break;
+		case IMPORTANCE_LOWEST:  fld->setValue("5 (Lowest)"); break;
+	}
+
+	// "Importance:" Field
+	fld = hdr->getField("Importance");
+
+	switch (i) {
+
+		case IMPORTANCE_HIGHEST:
+		case IMPORTANCE_HIGH:
+
+			fld->setValue("high");
+			break;
+
+		default:
+		case IMPORTANCE_NORMAL:
+
+			fld->setValue("normal");
+			break;
+
+		case IMPORTANCE_LOWEST:
+		case IMPORTANCE_LOW:
+
+			fld->setValue("low");
+			break;
+	}
+}
+
+
+} // misc
+} // vmime