mirror of
https://github.com/RGBCube/serenity
synced 2025-07-27 02:27:43 +00:00
AK: Introduce SourceGenerator::fork().
Previously, I abused the copy constructor, this is a lot better.
This commit is contained in:
parent
bf1ed34236
commit
61e73b1a7b
3 changed files with 48 additions and 67 deletions
|
@ -34,54 +34,32 @@
|
|||
namespace AK {
|
||||
|
||||
class SourceGenerator {
|
||||
AK_MAKE_NONCOPYABLE(SourceGenerator);
|
||||
|
||||
public:
|
||||
using MappingType = HashMap<StringView, String>;
|
||||
|
||||
explicit SourceGenerator(SourceGenerator& parent, char opening = '@', char closing = '@')
|
||||
: m_parent(&parent)
|
||||
explicit SourceGenerator(StringBuilder& builder, char opening = '@', char closing = '@')
|
||||
: m_builder(builder)
|
||||
, m_opening(opening)
|
||||
, m_closing(closing)
|
||||
{
|
||||
}
|
||||
explicit SourceGenerator(char opening = '@', char closing = '@')
|
||||
: m_parent(nullptr)
|
||||
explicit SourceGenerator(StringBuilder& builder, const MappingType& mapping, char opening = '@', char closing = '@')
|
||||
: m_builder(builder)
|
||||
, m_mapping(mapping)
|
||||
, m_opening(opening)
|
||||
, m_closing(closing)
|
||||
{
|
||||
m_builder = new StringBuilder();
|
||||
}
|
||||
|
||||
SourceGenerator fork() { return SourceGenerator { m_builder, m_mapping, m_opening, m_closing }; }
|
||||
|
||||
void set(StringView key, String value) { m_mapping.set(key, value); }
|
||||
String get(StringView key) const { return m_mapping.get(key).value(); }
|
||||
|
||||
String lookup(StringView key) const
|
||||
{
|
||||
if (auto opt = m_mapping.get(key); opt.has_value())
|
||||
return opt.value();
|
||||
|
||||
if (m_parent == nullptr) {
|
||||
dbgln("Can't find key '{}'", key);
|
||||
ASSERT_NOT_REACHED();
|
||||
}
|
||||
|
||||
return m_parent->lookup(key);
|
||||
}
|
||||
|
||||
String generate() const { return builder().build(); }
|
||||
|
||||
StringBuilder& builder()
|
||||
{
|
||||
if (m_parent)
|
||||
return m_parent->builder();
|
||||
else
|
||||
return *m_builder;
|
||||
}
|
||||
const StringBuilder& builder() const
|
||||
{
|
||||
if (m_parent)
|
||||
return m_parent->builder();
|
||||
else
|
||||
return *m_builder;
|
||||
}
|
||||
StringView as_string_view() const { return m_builder.string_view(); }
|
||||
String as_string() const { return m_builder.build(); }
|
||||
|
||||
void append(StringView pattern)
|
||||
{
|
||||
|
@ -94,7 +72,7 @@ public:
|
|||
return lexer.consume_while([&](char ch) { return ch != stop; });
|
||||
};
|
||||
|
||||
builder().append(consume_until_without_consuming_stop_character(m_opening));
|
||||
m_builder.append(consume_until_without_consuming_stop_character(m_opening));
|
||||
|
||||
if (lexer.consume_specific(m_opening)) {
|
||||
const auto placeholder = consume_until_without_consuming_stop_character(m_closing);
|
||||
|
@ -102,7 +80,7 @@ public:
|
|||
if (!lexer.consume_specific(m_closing))
|
||||
ASSERT_NOT_REACHED();
|
||||
|
||||
builder().append(lookup(placeholder));
|
||||
m_builder.append(get(placeholder));
|
||||
} else {
|
||||
ASSERT(lexer.is_eof());
|
||||
}
|
||||
|
@ -110,9 +88,8 @@ public:
|
|||
}
|
||||
|
||||
private:
|
||||
SourceGenerator* m_parent;
|
||||
StringBuilder& m_builder;
|
||||
MappingType m_mapping;
|
||||
OwnPtr<StringBuilder> m_builder;
|
||||
char m_opening, m_closing;
|
||||
};
|
||||
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue