1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#include <string.h>
#include <mbgl/style/conversion_impl.hpp>
#include <mbgl/style/expression/in.hpp>
#include <mbgl/style/expression/type.hpp>
#include <mbgl/util/string.hpp>
namespace mbgl {
namespace style {
namespace expression {
namespace {
bool isComparableType(type::Type type) {
return type == type::Boolean || type == type::String || type == type::Number || type == type::Null ||
type == type::Value;
}
bool isComparableRuntimeType(type::Type type) {
return type == type::Boolean || type == type::String || type == type::Number || type == type::Null;
}
bool isSearchableRuntimeType(type::Type type) {
return type == type::String || type.is<type::Array>() || type == type::Null;
}
} // namespace
In::In(std::unique_ptr<Expression> needle_, std::unique_ptr<Expression> haystack_)
: Expression(Kind::In, type::Boolean), needle(std::move(needle_)), haystack(std::move(haystack_)) {
assert(isComparableType(needle->getType()));
assert(isSearchableRuntimeType(haystack->getType()) || haystack->getType() == type::Value);
}
EvaluationResult In::evaluate(const EvaluationContext& params) const {
const EvaluationResult evaluatedHaystack = haystack->evaluate(params);
if (!evaluatedHaystack) {
return evaluatedHaystack.error();
}
const EvaluationResult evaluatedNeedle = needle->evaluate(params);
if (!evaluatedNeedle) {
return evaluatedNeedle.error();
}
type::Type evaluatedNeedleType = typeOf(*evaluatedNeedle);
if (!isComparableRuntimeType(evaluatedNeedleType)) {
return EvaluationError{"Expected first argument to be of type boolean, string or number, but found " +
toString(evaluatedNeedleType) + " instead."};
}
type::Type evaluatedHaystackType = typeOf(*evaluatedHaystack);
if (!isSearchableRuntimeType(evaluatedHaystackType)) {
return EvaluationError{"Expected second argument to be of type array or string, but found " +
toString(evaluatedHaystackType) + " instead."};
}
if (evaluatedNeedleType == type::Null || evaluatedHaystackType == type::Null) {
return EvaluationResult(false);
}
if (evaluatedHaystackType == type::String) {
const auto haystackString = evaluatedHaystack->get<std::string>();
std::string needleValue = "";
if (evaluatedNeedleType == type::Boolean) {
needleValue = evaluatedNeedle->get<bool>() ? "true" : " false";
} else if (evaluatedNeedleType == type::String) {
needleValue = evaluatedNeedle->get<std::string>();
} else if (evaluatedNeedleType == type::Number) {
needleValue = std::to_string(evaluatedNeedle->get<double>());
needleValue.erase(needleValue.find_last_not_of('0') + 1, std::string::npos);
needleValue.erase(needleValue.find_last_not_of('.') + 1, std::string::npos);
}
return EvaluationResult(haystackString.find(needleValue) != std::string::npos);
} else {
const auto haystackArray = evaluatedHaystack->get<std::vector<Value>>();
return EvaluationResult(std::find(haystackArray.begin(), haystackArray.end(), *evaluatedNeedle) !=
haystackArray.end());
}
}
void In::eachChild(const std::function<void(const Expression&)>& visit) const {
visit(*needle);
visit(*haystack);
}
using namespace mbgl::style::conversion;
ParseResult In::parse(const Convertible& value, ParsingContext& ctx) {
assert(isArray(value));
std::size_t length = arrayLength(value);
if (length != 3) {
ctx.error("Expected 2 arguments, but found " + util::toString(length - 1) + " instead.");
return ParseResult();
}
ParseResult needle = ctx.parse(arrayMember(value, 1), 1, {type::Value});
ParseResult haystack = ctx.parse(arrayMember(value, 2), 2, {type::Value});
if (!needle || !haystack) return ParseResult();
type::Type needleType = (*needle)->getType();
type::Type haystackType = (*haystack)->getType();
if (!isComparableType(needleType)) {
ctx.error("Expected first argument to be of type boolean, string or number, but found " + toString(needleType) +
" instead.");
return ParseResult();
}
return ParseResult(std::make_unique<In>(std::move(*needle), std::move(*haystack)));
}
} // namespace expression
} // namespace style
} // namespace mbgl
|
