Skip to content

Commit

Permalink
[issue1106] Support string options in parser.
Browse files Browse the repository at this point in the history
We now support string arguments in double quotes. Strings may use escape symbols '\"', '\\', and '\n' for double quotes, backslashes and newlines.
  • Loading branch information
FlorianPommerening authored Nov 21, 2023
1 parent c5ffc2d commit 47cc432
Show file tree
Hide file tree
Showing 8 changed files with 145 additions and 44 deletions.
4 changes: 4 additions & 0 deletions src/search/parser/abstract_syntax_tree.cc
Original file line number Diff line number Diff line change
Expand Up @@ -420,6 +420,8 @@ DecoratedASTNodePtr LiteralNode::decorate(DecorateContext &context) const {
switch (value.type) {
case TokenType::BOOLEAN:
return utils::make_unique_ptr<BoolLiteralNode>(value.content);
case TokenType::STRING:
return utils::make_unique_ptr<StringLiteralNode>(value.content);
case TokenType::INTEGER:
return utils::make_unique_ptr<IntLiteralNode>(value.content);
case TokenType::FLOAT:
Expand All @@ -441,6 +443,8 @@ const plugins::Type &LiteralNode::get_type(DecorateContext &context) const {
switch (value.type) {
case TokenType::BOOLEAN:
return plugins::TypeRegistry::instance()->get_type<bool>();
case TokenType::STRING:
return plugins::TypeRegistry::instance()->get_type<string>();
case TokenType::INTEGER:
return plugins::TypeRegistry::instance()->get_type<int>();
case TokenType::FLOAT:
Expand Down
52 changes: 52 additions & 0 deletions src/search/parser/decorated_abstract_syntax_tree.cc
Original file line number Diff line number Diff line change
Expand Up @@ -218,6 +218,46 @@ void BoolLiteralNode::dump(string indent) const {
cout << indent << "BOOL: " << value << endl;
}

StringLiteralNode::StringLiteralNode(const string &value)
: value(value) {
}

plugins::Any StringLiteralNode::construct(ConstructContext &context) const {
utils::TraceBlock block(context, "Constructing string value from '" + value + "'");
if (!(value.starts_with('"') && value.ends_with('"'))) {
ABORT("String literal value is not enclosed in quotation marks"
" (this should have been caught before constructing this node).");
}
/*
We are not doing any further syntax checking. Escaped symbols other than
\n will just ignore the escaping \ (e.g., \t is treated as t, not as a
tab). Strings ending in \ will not produce an error but should be excluded
by the previous steps.
*/
string result;
result.reserve(value.length() - 2);
bool escaped = false;
for (char c : value.substr(1, value.size() - 2)) {
if (escaped) {
escaped = false;
if (c == 'n') {
result += '\n';
} else {
result += c;
}
} else if (c == '\\') {
escaped = true;
} else {
result += c;
}
}
return result;
}

void StringLiteralNode::dump(string indent) const {
cout << indent << "STRING: " << value << endl;
}

IntLiteralNode::IntLiteralNode(const string &value)
: value(value) {
}
Expand Down Expand Up @@ -473,6 +513,18 @@ shared_ptr<DecoratedASTNode> BoolLiteralNode::clone_shared() const {
return make_shared<BoolLiteralNode>(*this);
}

StringLiteralNode::StringLiteralNode(const StringLiteralNode &other)
: value(other.value) {
}

unique_ptr<DecoratedASTNode> StringLiteralNode::clone() const {
return utils::make_unique_ptr<StringLiteralNode>(*this);
}

shared_ptr<DecoratedASTNode> StringLiteralNode::clone_shared() const {
return make_shared<StringLiteralNode>(*this);
}

IntLiteralNode::IntLiteralNode(const IntLiteralNode &other)
: value(other.value) {
}
Expand Down
14 changes: 14 additions & 0 deletions src/search/parser/decorated_abstract_syntax_tree.h
Original file line number Diff line number Diff line change
Expand Up @@ -157,6 +157,20 @@ class BoolLiteralNode : public DecoratedASTNode {
BoolLiteralNode(const BoolLiteralNode &other);
};

class StringLiteralNode : public DecoratedASTNode {
std::string value;
public:
StringLiteralNode(const std::string &value);

plugins::Any construct(ConstructContext &context) const override;
void dump(std::string indent) const override;

// TODO: once we get rid of lazy construction, this should no longer be necessary.
virtual std::unique_ptr<DecoratedASTNode> clone() const override;
virtual std::shared_ptr<DecoratedASTNode> clone_shared() const override;
StringLiteralNode(const StringLiteralNode &other);
};

class IntLiteralNode : public DecoratedASTNode {
std::string value;
public:
Expand Down
56 changes: 35 additions & 21 deletions src/search/parser/lexical_analyzer.cc
Original file line number Diff line number Diff line change
Expand Up @@ -24,12 +24,22 @@ static vector<pair<TokenType, regex>> construct_token_type_expressions() {
{TokenType::CLOSING_BRACKET, R"(\])"},
{TokenType::COMMA, R"(,)"},
{TokenType::EQUALS, R"(=)"},
{TokenType::LET, R"(let)"},
{TokenType::BOOLEAN, R"(true|false)"},
{TokenType::STRING, R"("(\\\\|\\"|\\n|[^"\\])*")"},
/*
Floats have to be parsed before integers, so tokens like '1.2' are
parsed as one float token rather than an integer token '1' followed
by a float token '.2'.
*/
{TokenType::FLOAT,
R"([+-]?(((\d*\.\d+|\d+\.)(e[+-]?\d+|[kmg]\b)?)|\d+e[+-]?\d+))"},
{TokenType::INTEGER,
R"([+-]?(infinity|\d+([kmg]\b)?))"},
{TokenType::BOOLEAN, R"(true|false)"},
{TokenType::LET, R"(let)"},
{TokenType::INTEGER, R"([+-]?(infinity|\d+([kmg]\b)?))"},
/*
Identifiers have to be parsed last to prevent reserved words (
'infinity', 'true', 'false', and 'let') from being recognized as
identifiers.
*/
{TokenType::IDENTIFIER, R"([a-zA-Z_]\w*)"}
};
vector<pair<TokenType, regex>> token_type_expression;
Expand All @@ -42,6 +52,24 @@ static vector<pair<TokenType, regex>> construct_token_type_expressions() {
static const vector<pair<TokenType, regex>> token_type_expressions =
construct_token_type_expressions();

static string highlight_position(const string &text, string::const_iterator pos) {
ostringstream message_stream;
int distance_to_highlight = pos - text.begin();
for (const string &line : utils::split(text, "\n")) {
int line_length = line.size();
bool highlight_in_line =
distance_to_highlight < line_length && distance_to_highlight >= 0;
message_stream << (highlight_in_line ? "> " : " ") << line << endl;
if (highlight_in_line) {
message_stream << string(distance_to_highlight + 2, ' ') << "^"
<< endl;
}
distance_to_highlight -= line.size() + 1;
}
string message = message_stream.str();
utils::rstrip(message);
return message;
}

TokenStream split_tokens(const string &text) {
utils::Context context;
Expand All @@ -59,29 +87,15 @@ TokenStream split_tokens(const string &text) {
TokenType token_type = type_and_expression.first;
const regex &expression = type_and_expression.second;
if (regex_search(start, end, match, expression, regex_constants::match_continuous)) {
tokens.push_back({utils::tolower(match[1]), token_type});
tokens.push_back({match[1], token_type});
start += match[0].length();
has_match = true;
break;
}
}
if (!has_match) {
ostringstream error;
error << "Unable to recognize next token:" << endl;
int distance_to_error = start - text.begin();
for (const string &line : utils::split(text, "\n")) {
int line_length = line.size();
bool error_in_line =
distance_to_error < line_length && distance_to_error >= 0;
error << (error_in_line ? "> " : " ") << line << endl;
if (error_in_line)
error << string(distance_to_error + 2, ' ') << "^" << endl;

distance_to_error -= line.size() + 1;
}
string message = error.str();
utils::rstrip(message);
context.error(message);
context.error("Unable to recognize next token:\n" +
highlight_position(text, start));
}
}
return TokenStream(move(tokens));
Expand Down
33 changes: 18 additions & 15 deletions src/search/parser/syntax_analyzer.cc
Original file line number Diff line number Diff line change
Expand Up @@ -159,9 +159,10 @@ static ASTNodePtr parse_function(TokenStream &tokens,
}

static unordered_set<TokenType> literal_tokens {
TokenType::FLOAT,
TokenType::INTEGER,
TokenType::BOOLEAN,
TokenType::STRING,
TokenType::INTEGER,
TokenType::FLOAT,
TokenType::IDENTIFIER
};

Expand Down Expand Up @@ -191,40 +192,42 @@ static ASTNodePtr parse_list(TokenStream &tokens, SyntaxAnalyzerContext &context
return utils::make_unique_ptr<ListNode>(move(elements));
}

static vector<TokenType> PARSE_NODE_TOKEN_TYPES = {
TokenType::LET, TokenType::IDENTIFIER, TokenType::BOOLEAN,
TokenType::INTEGER, TokenType::FLOAT, TokenType::OPENING_BRACKET};
static vector<TokenType> parse_node_token_types = {
TokenType::OPENING_BRACKET, TokenType::LET, TokenType::BOOLEAN,
TokenType::STRING, TokenType::INTEGER, TokenType::FLOAT,
TokenType::IDENTIFIER};

static ASTNodePtr parse_node(TokenStream &tokens,
SyntaxAnalyzerContext &context) {
utils::TraceBlock block(context, "Identify node type");
Token token = tokens.peek(context);
if (find(PARSE_NODE_TOKEN_TYPES.begin(),
PARSE_NODE_TOKEN_TYPES.end(),
token.type) == PARSE_NODE_TOKEN_TYPES.end()) {
if (find(parse_node_token_types.begin(),
parse_node_token_types.end(),
token.type) == parse_node_token_types.end()) {
ostringstream message;
message << "Unexpected token '" << token
<< "'. Expected any of the following token types: "
<< utils::join(PARSE_NODE_TOKEN_TYPES, ", ");
<< utils::join(parse_node_token_types, ", ");
context.error(message.str());
}

switch (token.type) {
case TokenType::OPENING_BRACKET:
return parse_list(tokens, context);
case TokenType::LET:
return parse_let(tokens, context);
case TokenType::BOOLEAN:
case TokenType::STRING:
case TokenType::INTEGER:
case TokenType::FLOAT:
return parse_literal(tokens, context);
case TokenType::IDENTIFIER:
if (tokens.has_tokens(2)
&& tokens.peek(context, 1).type == TokenType::OPENING_PARENTHESIS) {
return parse_function(tokens, context);
} else {
return parse_literal(tokens, context);
}
case TokenType::BOOLEAN:
case TokenType::INTEGER:
case TokenType::FLOAT:
return parse_literal(tokens, context);
case TokenType::OPENING_BRACKET:
return parse_list(tokens, context);
default:
ABORT("Unknown token type '" + token_type_name(token.type) + "'.");
}
Expand Down
22 changes: 17 additions & 5 deletions src/search/parser/token_stream.cc
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,18 @@
using namespace std;

namespace parser {
static string case_insensitive_to_lower(const string &content, TokenType type) {
if (type == TokenType::BOOLEAN ||
type == TokenType::INTEGER ||
type == TokenType::FLOAT ||
type == TokenType::IDENTIFIER) {
return utils::tolower(content);
} else {
return content;
}
}
Token::Token(const string &content, TokenType type)
: content(content), type(type) {
: content(case_insensitive_to_lower(content, type)), type(type) {
}

TokenStream::TokenStream(vector<Token> &&tokens)
Expand Down Expand Up @@ -90,16 +100,18 @@ string token_type_name(TokenType token_type) {
return ",";
case TokenType::EQUALS:
return "=";
case TokenType::LET:
return "Let";
case TokenType::BOOLEAN:
return "Boolean";
case TokenType::STRING:
return "String";
case TokenType::INTEGER:
return "Integer";
case TokenType::FLOAT:
return "Float";
case TokenType::BOOLEAN:
return "Boolean";
case TokenType::IDENTIFIER:
return "Identifier";
case TokenType::LET:
return "Let";
default:
ABORT("Unknown token type.");
}
Expand Down
7 changes: 4 additions & 3 deletions src/search/parser/token_stream.h
Original file line number Diff line number Diff line change
Expand Up @@ -16,11 +16,12 @@ enum class TokenType {
CLOSING_BRACKET,
COMMA,
EQUALS,
LET,
BOOLEAN,
STRING,
INTEGER,
FLOAT,
BOOLEAN,
IDENTIFIER,
LET
IDENTIFIER
};

struct Token {
Expand Down
1 change: 1 addition & 0 deletions src/search/plugins/types.cc
Original file line number Diff line number Diff line change
Expand Up @@ -292,6 +292,7 @@ BasicType TypeRegistry::NO_TYPE = BasicType(typeid(void), "<no type>");

TypeRegistry::TypeRegistry() {
insert_basic_type<bool>();
insert_basic_type<string>();
insert_basic_type<int>();
insert_basic_type<double>();
}
Expand Down

0 comments on commit 47cc432

Please sign in to comment.