123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232 |
- /*
- * Copyright (c) 2021, Tim Flynn <trflynn89@pm.me>
- * All rights reserved.
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are met:
- *
- * 1. Redistributions of source code must retain the above copyright notice, this
- * list of conditions and the following disclaimer.
- *
- * 2. Redistributions in binary form must reproduce the above copyright notice,
- * this list of conditions and the following disclaimer in the documentation
- * and/or other materials provided with the distribution.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
- * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
- * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
- * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
- * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
- * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
- * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
- * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
- * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- */
- #include "Parser.h"
- namespace SQL {
- Parser::Parser(Lexer lexer)
- : m_parser_state(move(lexer))
- {
- }
- NonnullRefPtr<Statement> Parser::next_statement()
- {
- switch (m_parser_state.m_token.type()) {
- case TokenType::Create:
- return parse_create_table_statement();
- case TokenType::Drop:
- return parse_drop_table_statement();
- default:
- expected("CREATE or DROP");
- return create_ast_node<ErrorStatement>();
- }
- }
- NonnullRefPtr<CreateTable> Parser::parse_create_table_statement()
- {
- // https://sqlite.org/lang_createtable.html
- consume(TokenType::Create);
- bool is_temporary = false;
- if (consume_if(TokenType::Temp) || consume_if(TokenType::Temporary))
- is_temporary = true;
- consume(TokenType::Table);
- bool is_error_if_table_exists = true;
- if (consume_if(TokenType::If)) {
- consume(TokenType::Not);
- consume(TokenType::Exists);
- is_error_if_table_exists = false;
- }
- String schema_or_table_name = consume(TokenType::Identifier).value();
- String schema_name;
- String table_name;
- if (consume_if(TokenType::Period)) {
- schema_name = move(schema_or_table_name);
- table_name = consume(TokenType::Identifier).value();
- } else {
- table_name = move(schema_or_table_name);
- }
- // FIXME: Parse "AS select-stmt".
- NonnullRefPtrVector<ColumnDefinition> column_definitions;
- consume(TokenType::ParenOpen);
- do {
- column_definitions.append(parse_column_definition());
- if (match(TokenType::ParenClose))
- break;
- consume(TokenType::Comma);
- } while (!match(TokenType::Eof));
- // FIXME: Parse "table-constraint".
- consume(TokenType::ParenClose);
- consume(TokenType::SemiColon);
- return create_ast_node<CreateTable>(move(schema_name), move(table_name), move(column_definitions), is_temporary, is_error_if_table_exists);
- }
- NonnullRefPtr<DropTable> Parser::parse_drop_table_statement()
- {
- // https://sqlite.org/lang_droptable.html
- consume(TokenType::Drop);
- consume(TokenType::Table);
- bool is_error_if_table_does_not_exist = true;
- if (consume_if(TokenType::If)) {
- consume(TokenType::Exists);
- is_error_if_table_does_not_exist = false;
- }
- String schema_or_table_name = consume(TokenType::Identifier).value();
- String schema_name;
- String table_name;
- if (consume_if(TokenType::Period)) {
- schema_name = move(schema_or_table_name);
- table_name = consume(TokenType::Identifier).value();
- } else {
- table_name = move(schema_or_table_name);
- }
- consume(TokenType::SemiColon);
- return create_ast_node<DropTable>(move(schema_name), move(table_name), is_error_if_table_does_not_exist);
- }
- NonnullRefPtr<ColumnDefinition> Parser::parse_column_definition()
- {
- // https://sqlite.org/syntax/column-def.html
- auto name = consume(TokenType::Identifier).value();
- auto type_name = match(TokenType::Identifier)
- ? parse_type_name()
- // https://www.sqlite.org/datatype3.html: If no type is specified then the column has affinity BLOB.
- : create_ast_node<TypeName>("BLOB", NonnullRefPtrVector<SignedNumber> {});
- // FIXME: Parse "column-constraint".
- return create_ast_node<ColumnDefinition>(move(name), move(type_name));
- }
- NonnullRefPtr<TypeName> Parser::parse_type_name()
- {
- // https: //sqlite.org/syntax/type-name.html
- auto name = consume(TokenType::Identifier).value();
- NonnullRefPtrVector<SignedNumber> signed_numbers;
- if (consume_if(TokenType::ParenOpen)) {
- signed_numbers.append(parse_signed_number());
- if (consume_if(TokenType::Comma))
- signed_numbers.append(parse_signed_number());
- consume(TokenType::ParenClose);
- }
- return create_ast_node<TypeName>(move(name), move(signed_numbers));
- }
- NonnullRefPtr<SignedNumber> Parser::parse_signed_number()
- {
- // https://sqlite.org/syntax/signed-number.html
- bool is_positive = true;
- if (consume_if(TokenType::Plus))
- is_positive = true;
- else if (consume_if(TokenType::Minus))
- is_positive = false;
- if (match(TokenType::NumericLiteral)) {
- auto number = consume(TokenType::NumericLiteral).double_value();
- return create_ast_node<SignedNumber>(is_positive ? number : (number * -1));
- }
- expected("NumericLiteral");
- return create_ast_node<SignedNumber>(0);
- }
- Token Parser::consume()
- {
- auto old_token = m_parser_state.m_token;
- m_parser_state.m_token = m_parser_state.m_lexer.next();
- return old_token;
- }
- Token Parser::consume(TokenType expected_type)
- {
- if (!match(expected_type)) {
- expected(Token::name(expected_type));
- }
- return consume();
- }
- bool Parser::consume_if(TokenType expected_type)
- {
- if (!match(expected_type))
- return false;
- consume();
- return true;
- }
- bool Parser::match(TokenType type) const
- {
- return m_parser_state.m_token.type() == type;
- }
- void Parser::expected(StringView what)
- {
- syntax_error(String::formatted("Unexpected token {}, expected {}", m_parser_state.m_token.name(), what));
- }
- void Parser::syntax_error(String message)
- {
- m_parser_state.m_errors.append({ move(message), position() });
- }
- Parser::Position Parser::position() const
- {
- return {
- m_parser_state.m_token.line_number(),
- m_parser_state.m_token.line_column()
- };
- }
- Parser::ParserState::ParserState(Lexer lexer)
- : m_lexer(move(lexer))
- , m_token(m_lexer.next())
- {
- }
- }
|