xml_file.h

/*****************************************************/
/* lean XML                     (c) Tobias Zirr 2011 */
/*****************************************************/

#ifndef LEAN_XML_XML_FILE
#define LEAN_XML_XML_FILE

#include "../lean.h"
#include "../strings/types.h"
#include "../meta/constexpr.h"
#include "../io/raw_file.h"
#include "../io/raw_file_inserter.h"
#include "../rapidxml/rapidxml.hpp"
#include "../rapidxml/rapidxml_print.hpp"

#ifndef LEAN_XML_FILE_SAVE_BATCH_SIZE


    #define LEAN_XML_FILE_SAVE_BATCH_SIZE 4096
#endif

namespace lean
{
namespace xml
{

namespace impl
{
    template <class Char>
    inline char* load_xml_source(const utf8_ntri &fileName, rapidxml::xml_document<Char> &document)
    {
        raw_file file(fileName, file::read);
        
        size_t fileSize = static_cast<size_t>(file.size());
        char *source = document.allocate_string(nullptr, fileSize);
        
        file.read(source, fileSize); // TODO: strict (exception on error)
        return source;
    }
}

template <int ParseFlags, class Char>
LEAN_INLINE void load_xml_file(const utf8_ntri &fileName, rapidxml::xml_document<Char> &document)
{
    try
    {
        document.parse<ParseFlags>(
            impl::load_xml_source(fileName, document) );
    }
    catch(rapidxml::parse_error &error)
    {
        throw std::runtime_error(error.what());
    }
}

template <int PrintFlags, class Char>
LEAN_INLINE void save_xml_file(const utf8_ntri &fileName, const rapidxml::xml_node<Char> &document)
{
    raw_file file(fileName, file::write, file::overwrite);
    print(raw_file_inserter<LEAN_XML_FILE_SAVE_BATCH_SIZE>(file).iter(), document, PrintFlags);
}

template <class Char = char>
class xml_file
{
private:
    rapidxml::xml_document<Char> m_document;

public:
    LEAN_INLINE xml_file() { }
    LEAN_INLINE explicit xml_file(const utf8_ntri &name)
    {
        load_xml_file<rapidxml::parse_trim_whitespace | rapidxml::parse_normalize_whitespace>(name, m_document);
    }
    template <int ParseFlags>
    LEAN_INLINE xml_file(const utf8_ntri &name, ce_int<ParseFlags>)
    {
        load_xml_file<ParseFlags>(name, m_document);
    }

    LEAN_INLINE void save(const utf8_ntri &name) const
    {
        save_xml_file<0>(name, m_document);
    }
    template <int ParseFlags>
    LEAN_INLINE void save(const utf8_ntri &name, ce_int<ParseFlags>) const
    {
        save_xml_file<PrintFlags>(name, m_document);
    }

    LEAN_INLINE rapidxml::xml_document<Char>& document() { return m_document; }
    LEAN_INLINE const rapidxml::xml_document<Char>& document() const { return m_document; }
};

} // namespace

using xml::xml_file;

} // namespace

#endif