mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-22 02:49:29 +03:00
ecd1862859
No functional changes.
177 lines
5.5 KiB
C++
177 lines
5.5 KiB
C++
/*
|
|
* Copyright (c) 2020-2022, the SerenityOS developers.
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/DeprecatedString.h>
|
|
#include <AK/GenericLexer.h>
|
|
#include <AK/OwnPtr.h>
|
|
#include <AK/Stream.h>
|
|
#include <AK/StringView.h>
|
|
#include <AK/Types.h>
|
|
#include <AK/Vector.h>
|
|
|
|
namespace Writer {
|
|
|
|
enum class WriterBehavior : u32 {
|
|
None = 0,
|
|
WriteHeaders = 1,
|
|
AllowNewlinesInFields = WriteHeaders << 1,
|
|
QuoteOnlyInFieldStart = WriteHeaders << 2,
|
|
QuoteAll = WriteHeaders << 3,
|
|
};
|
|
AK_ENUM_BITWISE_OPERATORS(WriterBehavior);
|
|
|
|
struct WriterTraits {
|
|
DeprecatedString separator;
|
|
DeprecatedString quote { "\"" };
|
|
enum QuoteEscape {
|
|
Repeat,
|
|
Backslash,
|
|
} quote_escape { Repeat };
|
|
};
|
|
|
|
constexpr WriterBehavior default_behaviors()
|
|
{
|
|
return WriterBehavior::None;
|
|
}
|
|
|
|
template<typename ContainerType, typename HeaderType = Vector<StringView>>
|
|
class XSV {
|
|
public:
|
|
static ErrorOr<void> generate(Stream& output, ContainerType const& data, WriterTraits traits, HeaderType headers = {}, WriterBehavior behaviors = default_behaviors())
|
|
{
|
|
auto writer = XSV(output, data, traits, headers, behaviors);
|
|
auto with_headers = has_flag(writer.m_behaviors, WriterBehavior::WriteHeaders);
|
|
if (with_headers) {
|
|
TRY(writer.write_row(writer.m_names));
|
|
TRY(writer.m_output.write_until_depleted({ "\n", 1 }));
|
|
}
|
|
|
|
for (auto&& row : writer.m_data) {
|
|
if (with_headers) {
|
|
if (row.size() != writer.m_names.size())
|
|
return Error::from_string_literal("Header count does not match given column count");
|
|
}
|
|
|
|
TRY(writer.write_row(row));
|
|
TRY(writer.m_output.write_until_depleted({ "\n", 1 }));
|
|
}
|
|
return {};
|
|
}
|
|
|
|
static ErrorOr<void> generate_preview(Stream& output, ContainerType const& data, WriterTraits traits, HeaderType headers = {}, WriterBehavior behaviors = default_behaviors())
|
|
{
|
|
auto writer = XSV(output, data, traits, headers, behaviors);
|
|
auto lines_written = 0;
|
|
constexpr auto max_preview_lines = 8;
|
|
|
|
auto with_headers = has_flag(writer.m_behaviors, WriterBehavior::WriteHeaders);
|
|
if (with_headers) {
|
|
TRY(writer.write_row(writer.m_names));
|
|
TRY(writer.m_output.write_until_depleted({ "\n", 1 }));
|
|
++lines_written;
|
|
}
|
|
|
|
for (auto&& row : writer.m_data) {
|
|
if (with_headers) {
|
|
if (row.size() != writer.m_names.size())
|
|
return Error::from_string_literal("Header count does not match given column count");
|
|
}
|
|
|
|
TRY(writer.write_row(row));
|
|
TRY(writer.m_output.write_until_depleted({ "\n", 1 }));
|
|
++lines_written;
|
|
|
|
if (lines_written >= max_preview_lines)
|
|
break;
|
|
}
|
|
return {};
|
|
}
|
|
|
|
private:
|
|
XSV(Stream& output, ContainerType const& data, WriterTraits traits, HeaderType headers = {}, WriterBehavior behaviors = default_behaviors())
|
|
: m_data(data)
|
|
, m_traits(move(traits))
|
|
, m_behaviors(behaviors)
|
|
, m_names(headers)
|
|
, m_output(output)
|
|
{
|
|
if (!headers.is_empty())
|
|
m_behaviors = m_behaviors | WriterBehavior::WriteHeaders;
|
|
}
|
|
|
|
template<typename T>
|
|
ErrorOr<void> write_row(T&& row)
|
|
{
|
|
bool first = true;
|
|
for (auto&& entry : row) {
|
|
if (!first) {
|
|
TRY(m_output.write_until_depleted(m_traits.separator.bytes()));
|
|
}
|
|
first = false;
|
|
TRY(write_entry(entry));
|
|
}
|
|
return {};
|
|
}
|
|
|
|
template<typename T>
|
|
ErrorOr<void> write_entry(T&& entry)
|
|
{
|
|
auto string = DeprecatedString::formatted("{}", FormatIfSupported(entry));
|
|
|
|
auto safe_to_write_normally = !has_flag(m_behaviors, WriterBehavior::QuoteAll)
|
|
&& !string.contains('\n')
|
|
&& !string.contains(m_traits.separator);
|
|
|
|
if (safe_to_write_normally) {
|
|
if (has_flag(m_behaviors, WriterBehavior::QuoteOnlyInFieldStart))
|
|
safe_to_write_normally = !string.starts_with(m_traits.quote);
|
|
else
|
|
safe_to_write_normally = !string.contains(m_traits.quote);
|
|
}
|
|
|
|
if (safe_to_write_normally) {
|
|
if (!string.is_empty())
|
|
TRY(m_output.write_until_depleted(string.bytes()));
|
|
return {};
|
|
}
|
|
|
|
TRY(m_output.write_until_depleted(m_traits.quote.bytes()));
|
|
|
|
GenericLexer lexer(string);
|
|
while (!lexer.is_eof()) {
|
|
if (lexer.consume_specific(m_traits.quote)) {
|
|
switch (m_traits.quote_escape) {
|
|
case WriterTraits::Repeat:
|
|
TRY(m_output.write_until_depleted(m_traits.quote.bytes()));
|
|
TRY(m_output.write_until_depleted(m_traits.quote.bytes()));
|
|
break;
|
|
case WriterTraits::Backslash:
|
|
TRY(m_output.write_until_depleted({ "\\", 1 }));
|
|
TRY(m_output.write_until_depleted(m_traits.quote.bytes()));
|
|
break;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
auto ch = lexer.consume();
|
|
TRY(m_output.write_until_depleted({ &ch, 1 }));
|
|
}
|
|
|
|
TRY(m_output.write_until_depleted(m_traits.quote.bytes()));
|
|
return {};
|
|
}
|
|
|
|
ContainerType const& m_data;
|
|
WriterTraits m_traits;
|
|
WriterBehavior m_behaviors;
|
|
HeaderType m_names;
|
|
Stream& m_output;
|
|
};
|
|
|
|
}
|