root/msg/multipart.cc

/*  This file is part of Alue, the multiprotocol Internet discussion daemon

    Copyright © 2010 Antti-Juhani Kaijanaho

    Alue 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.

    Alue 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 Alue.  If not, see <http://www.gnu.org/licenses/>.

 */

#include "content_type.hh"
#include "multipart.hh"
#include "../tlate/data_model.hh"
#include "../tlate/list_value.hh"

namespace msg
{
        multipart::multipart(entity_header::ptr hdr, std::string str)
                : entity(hdr)
        {
                content_type ct = get_content_type();
                if (ct.get_type() != content_type::MULTIPART)
                        throw std::logic_error("not multipart");
                std::string boundary = ct.get_param("boundary");
                size_t bs = str.find("\r\n--"+boundary);
                preamble = str.substr(0,bs);
                std::string default_ct = ct.get_subtype() == "digest"
                        ? "message/rfc822"
                        : "text/plain; charset=\"US-ASCII\"";
                while (bs != std::string::npos)
                {
                        size_t be = str.find("\r\n", bs+2);
                        if (be == std::string::npos) be = str.length() - 2;
                        if (str.substr(bs + 4 + boundary.length(), 2) == "--")
                        {
                                trailer = be + 2 < str.length()
                                        ? str.substr(be+2)
                                        : "";
                                break;
                        }
                        size_t ns = str.find("\r\n--"+boundary, be);
                        std::string sub = str.substr(be+2, ns - be);
                        entity::ptr sen(entity::mk(sub, false, default_ct));
                        parts.push_back(sen);
                        bs = ns;
                }
        }

        boost::shared_ptr<tlate::value>
        multipart::get_tlate_value(bool hide_addresses) const
        {
                tlate::list_value::ptr li(new tlate::list_value);
                for (size_t i = 0; i < parts.size(); i++)
                {
                        entity::ptr en = parts[i];
                        li->push_back(en->get_tlate_value(hide_addresses));
                }
                tlate::data_model::ptr am(new tlate::data_model);
                am->insert("multipart", li);
                return am;
        }

        std::string multipart::get_body() const
        {
                content_type ct = get_content_type();
                std::string boundary = ct.get_param("boundary");
                std::ostringstream os;
                os << preamble;
                for (size_t i = 0; i < parts.size(); i++)
                {
                        os << "\r\n--" << boundary << "\r\n";
                        os << parts[i]->get_entity();
                }
                os << "\r\n--" << boundary << "--\r\n";
                os << trailer;
                return os.str();
        }
}