diff options
Diffstat (limited to 'src/header.hpp')
| -rw-r--r-- | src/header.hpp | 266 |
1 files changed, 266 insertions, 0 deletions
diff --git a/src/header.hpp b/src/header.hpp new file mode 100644 index 00000000..8ce070b9 --- /dev/null +++ b/src/header.hpp @@ -0,0 +1,266 @@ +// +// VMime library (http://vmime.sourceforge.net) +// Copyright (C) 2002-2004 Vincent Richard <[email protected]> +// +// 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 2 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., 675 Mass Ave, Cambridge, MA 02139, USA. +// + +#ifndef VMIME_HEADER_HPP_INCLUDED +#define VMIME_HEADER_HPP_INCLUDED + + +#include "base.hpp" +#include "component.hpp" +#include "exception.hpp" + +#include "headerField.hpp" +#include "headerFieldFactory.hpp" + +#include "addressListField.hpp" +#include "mailboxListField.hpp" +#include "mailboxField.hpp" +#include "textField.hpp" +#include "dateField.hpp" +#include "contentTypeField.hpp" +#include "contentEncodingField.hpp" +#include "defaultField.hpp" +#include "contentDispositionField.hpp" +#include "messageIdField.hpp" + + +namespace vmime +{ + + +class bodyPart; + + +/** Header section of a MIME part. + */ + +class header : public component +{ + friend class bodyPart; + friend class body; + friend class message; + +protected: + + header(); + ~header(); + +public: + + // A sub-class for field manipulation + class fieldsContainer + { + friend class header; + + protected: + + fieldsContainer(); + ~fieldsContainer(); + + public: + + // Field access + mailboxField& From() { return (dynamic_cast<mailboxField&>(get(headerField::From))); } + mailboxField& Sender() { return (dynamic_cast<mailboxField&>(get(headerField::Sender))); } + mailboxField& ReplyTo() { return (dynamic_cast<mailboxField&>(get(headerField::ReplyTo))); } + mailboxField& DeliveredTo() { return (dynamic_cast<mailboxField&>(get(headerField::DeliveredTo))); } + addressListField& To() { return (dynamic_cast<addressListField&>(get(headerField::To))); } + addressListField& Cc() { return (dynamic_cast<addressListField&>(get(headerField::Cc))); } + addressListField& Bcc() { return (dynamic_cast<addressListField&>(get(headerField::Bcc))); } + dateField& Date() { return (dynamic_cast<dateField&>(get(headerField::Date))); } + textField& Subject() { return (dynamic_cast<textField&>(get(headerField::Subject))); } + textField& Organization() { return (dynamic_cast<textField&>(get(headerField::Organization))); } + textField& UserAgent() { return (dynamic_cast<textField&>(get(headerField::UserAgent))); } + contentTypeField& ContentType() { return (dynamic_cast<contentTypeField&>(get(headerField::ContentType))); } + textField& ContentDescription() { return (dynamic_cast<textField&>(get(headerField::ContentDescription))); } + contentEncodingField& ContentTransferEncoding() { return (dynamic_cast<contentEncodingField&>(get(headerField::ContentTransferEncoding))); } + defaultField& MimeVersion() { return (dynamic_cast<defaultField&>(get(headerField::MimeVersion))); } + contentDispositionField& ContentDisposition() { return (dynamic_cast<contentDispositionField&>(get(headerField::ContentDisposition))); } + messageIdField& ContentId() { return (dynamic_cast<messageIdField&>(get(headerField::ContentId))); } + messageIdField& MessageId() { return (dynamic_cast<messageIdField&>(get(headerField::MessageId))); } + defaultField& ContentLocation() { return (dynamic_cast<defaultField&>(get(headerField::ContentLocation))); } + + const mailboxField& From() const { return (dynamic_cast<mailboxField&>(find(headerField::From))); } + const mailboxField& Sender() const { return (dynamic_cast<mailboxField&>(find(headerField::Sender))); } + const mailboxField& ReplyTo() const { return (dynamic_cast<mailboxField&>(find(headerField::ReplyTo))); } + const mailboxField& DeliveredTo() const { return (dynamic_cast<mailboxField&>(find(headerField::DeliveredTo))); } + const addressListField& To() const { return (dynamic_cast<addressListField&>(find(headerField::To))); } + const addressListField& Cc() const { return (dynamic_cast<addressListField&>(find(headerField::Cc))); } + const addressListField& Bcc() const { return (dynamic_cast<addressListField&>(find(headerField::Bcc))); } + const dateField& Date() const { return (dynamic_cast<dateField&>(find(headerField::Date))); } + const textField& Subject() const { return (dynamic_cast<textField&>(find(headerField::Subject))); } + const textField& Organization() const { return (dynamic_cast<textField&>(find(headerField::Organization))); } + const textField& UserAgent() const { return (dynamic_cast<textField&>(find(headerField::UserAgent))); } + const contentTypeField& ContentType() const { return (dynamic_cast<contentTypeField&>(find(headerField::ContentType))); } + const textField& ContentDescription() const { return (dynamic_cast<textField&>(find(headerField::ContentDescription))); } + const contentEncodingField& ContentTransferEncoding() const { return (dynamic_cast<contentEncodingField&>(find(headerField::ContentTransferEncoding))); } + const defaultField& MimeVersion() const { return (dynamic_cast<defaultField&>(find(headerField::MimeVersion))); } + const contentDispositionField& ContentDisposition() const { return (dynamic_cast<contentDispositionField&>(find(headerField::ContentDisposition))); } + const messageIdField& ContentId() const { return (dynamic_cast<messageIdField&>(find(headerField::ContentId))); } + const messageIdField& MessageId() const { return (dynamic_cast<messageIdField&>(find(headerField::MessageId))); } + const defaultField& ContentLocation() const { return (dynamic_cast<defaultField&>(find(headerField::ContentLocation))); } + + // Checks whether (at least) one field with this type/name exists + const bool has(const headerField::Types fieldType) const; + const bool has(const string& fieldName) const; + + // Find the first field that matches the specified type/name. + // If no field is found, an exception is thrown. + headerField& find(const headerField::Types fieldType) const; + headerField& find(const string& fieldName) const; + + // Find the first field that matches the specified type/name. + // If no field is found, one will be created. + headerField& get(const headerField::Types fieldType); + headerField& get(const string& fieldName); + + // Field iterator + class const_iterator; + + class iterator + { + friend class header::fieldsContainer::const_iterator; + friend class header::fieldsContainer; + + public: + + typedef std::vector <headerField*>::iterator::difference_type difference_type; + + iterator(std::vector <headerField*>::iterator it) : m_iterator(it) { } + iterator(const iterator& it) : m_iterator(it.m_iterator) { } + + iterator& operator=(const iterator& it) { m_iterator = it.m_iterator; return (*this); } + + headerField& operator*() const { return (**m_iterator); } + headerField* operator->() const { return (*m_iterator); } + + iterator& operator++() { ++m_iterator; return (*this); } + iterator operator++(int) { iterator i(*this); ++m_iterator; return (i); } + + iterator& operator--() { --m_iterator; return (*this); } + iterator operator--(int) { iterator i(*this); --m_iterator; return (i); } + + iterator& operator+=(difference_type n) { m_iterator += n; return (*this); } + iterator& operator-=(difference_type n) { m_iterator -= n; return (*this); } + + iterator operator-(difference_type x) const { return iterator(m_iterator - x); } + + headerField& operator[](difference_type n) const { return *(m_iterator[n]); } + + const bool operator==(const iterator& it) const { return (it.m_iterator == m_iterator); } + const bool operator!=(const iterator& it) const { return (!(*this == it)); } + + protected: + + std::vector <headerField*>::iterator m_iterator; + }; + + class const_iterator + { + public: + + typedef std::vector <headerField*>::const_iterator::difference_type difference_type; + + const_iterator(std::vector <headerField*>::const_iterator it) : m_iterator(it) { } + const_iterator(const iterator& it) : m_iterator(it.m_iterator) { } + const_iterator(const const_iterator& it) : m_iterator(it.m_iterator) { } + + const_iterator& operator=(const const_iterator& it) { m_iterator = it.m_iterator; return (*this); } + const_iterator& operator=(const iterator& it) { m_iterator = it.m_iterator; return (*this); } + + const headerField& operator*() const { return (**m_iterator); } + const headerField* operator->() const { return (*m_iterator); } + + const_iterator& operator++() { ++m_iterator; return (*this); } + const_iterator operator++(int) { const_iterator i(*this); ++m_iterator; return (i); } + + const_iterator& operator--() { --m_iterator; return (*this); } + const_iterator operator--(int) { const_iterator i(*this); --m_iterator; return (i); } + + const_iterator& operator+=(difference_type n) { m_iterator += n; return (*this); } + const_iterator& operator-=(difference_type n) { m_iterator -= n; return (*this); } + + const_iterator operator-(difference_type x) const { return const_iterator(m_iterator - x); } + + const headerField& operator[](difference_type n) const { return *(m_iterator[n]); } + + const bool operator==(const const_iterator& it) const { return (it.m_iterator == m_iterator); } + const bool operator!=(const const_iterator& it) const { return (!(*this == it)); } + + protected: + + std::vector <headerField*>::const_iterator m_iterator; + }; + + public: + + iterator begin() { return (m_fields.begin()); } + iterator end() { return (m_fields.end()); } + + const_iterator begin() const { return (const_iterator(m_fields.begin())); } + const_iterator end() const { return (const_iterator(m_fields.end())); } + + // Field insertion + void append(const headerField& field); + void insert(const iterator it, const headerField& field); + + // Field removing + void remove(const iterator it); + void clear(); + + // Field count + const std::vector <headerField*>::size_type count() const { return (m_fields.size()); } + const std::vector <headerField*>::size_type size() const { return (m_fields.size()); } + const bool empty() const { return (m_fields.empty()); } + + headerField& front() { return (*m_fields.front()); } + const headerField& front() const { return (*m_fields.front()); } + headerField& back() { return (*m_fields.back()); } + const headerField& back() const { return (*m_fields.back()); } + + fieldsContainer& operator=(const fieldsContainer& c); + + protected: + + void insertSorted(headerField* field); + + std::vector <headerField*> m_fields; + + } fields; + + typedef fieldsContainer::iterator iterator; + typedef fieldsContainer::const_iterator const_iterator; + + header& operator=(const header& h); + +public: + + using component::parse; + using component::generate; + + // Component parsing & assembling + void parse(const string& buffer, const string::size_type position, const string::size_type end, string::size_type* newPosition = NULL); + void generate(utility::outputStream& os, const string::size_type maxLineLength = lineLengthLimits::infinite, const string::size_type curLinePos = 0, string::size_type* newLinePos = NULL) const; +}; + + +} // vmime + + +#endif // VMIME_HEADER_HPP_INCLUDED |