|
1 | 1 | #include <rift/parser.hpp> |
2 | 2 |
|
3 | 3 | #include <stack> |
| 4 | +#include <iterator> |
4 | 5 |
|
5 | 6 | #include <rift/nodes/accessor.hpp> |
6 | 7 | #include <rift/nodes/assign.hpp> |
@@ -123,15 +124,31 @@ namespace rift { |
123 | 124 | auto resNode = std::move(res.unwrap()); |
124 | 125 | if (resNode->type() == Node::Type::Value) { |
125 | 126 | auto* valNode = static_cast<ValueNode*>(resNode.get()); |
126 | | - // if it's a string, add the value to previous segment node (if any) |
127 | | - if (valNode->value().type() == Value::Type::String || valNode->value().type() == Value::Type::Integer || valNode->value().type() == Value::Type::Boolean) { |
| 127 | + auto const& val = valNode->value(); |
| 128 | + if ( |
| 129 | + val.type() == Value::Type::String || |
| 130 | + val.type() == Value::Type::Integer || |
| 131 | + val.type() == Value::Type::Boolean |
| 132 | + ) { |
| 133 | + auto appendValue = [](std::string& dest, Value const& value) { |
| 134 | + if (value.type() == Value::Type::String) { |
| 135 | + dest.append(value.getString()); |
| 136 | + } else { |
| 137 | + fmt::format_to(std::back_inserter(dest), "{}", value); |
| 138 | + } |
| 139 | + }; |
| 140 | + |
128 | 141 | if (!nodes.empty() && nodes.back()->type() == Node::Type::Segment) { |
129 | | - // append to the last segment node |
130 | 142 | auto* segNode = static_cast<SegmentNode*>(nodes.back().get()); |
131 | | - segNode->value() += valNode->value().toString(); |
| 143 | + appendValue(segNode->value(), val); |
132 | 144 | } else { |
133 | | - // create a new segment node |
134 | | - nodes.push_back(std::make_unique<SegmentNode>(valNode->value().toString(), valNode->fromIndex(), valNode->toIndex())); |
| 145 | + std::string segmentValue; |
| 146 | + if (val.type() == Value::Type::String) { |
| 147 | + segmentValue = val.getString(); |
| 148 | + } else { |
| 149 | + segmentValue = fmt::format("{}", val); |
| 150 | + } |
| 151 | + nodes.push_back(std::make_unique<SegmentNode>(std::move(segmentValue), valNode->fromIndex(), valNode->toIndex())); |
135 | 152 | } |
136 | 153 |
|
137 | 154 | // consume the right brace |
|
0 commit comments