htmlwriter.cpp

/*
   SPDX-FileCopyrightText: 2017 Volker Krause <[email protected]>
 
   SPDX-License-Identifier: LGPL-2.0-or-later
*/
 
#include "htmlwriter.h"
 
#include <QTextStream>
 
using namespace MessageViewer;
 
HtmlWriter::HtmlWriter() = default;
HtmlWriter::~HtmlWriter() = default;
 
void HtmlWriter::begin()
{
    if (!m_stream) {
        m_stream = std::make_unique<QTextStream>();
    }
    m_stream->setDevice(device());
#if QT_VERSION < QT_VERSION_CHECK(6, 0, 0)
    m_stream->setCodec("UTF-8");
#endif
}
 
void HtmlWriter::write(const QString &html)
{
    Q_ASSERT(m_stream);
    if (!m_stream) {
        return;
    }
    *stream() << html;
}
 
void HtmlWriter::end()
{
    if (!m_stream) {
        return;
    }
    m_stream->flush();
    m_stream->setDevice(nullptr);
}
 
void HtmlWriter::reset()
{
    if (!m_stream) {
        return;
    }
    m_stream->setDevice(nullptr);
}
 
QTextStream *HtmlWriter::stream() const
{
    return m_stream.get();
}
 
void HtmlWriter::setCodec(const QByteArray &codec)
{
    mCodec = codec;
}
 
QByteArray HtmlWriter::codec() const
{
    return mCodec;
}