107 lines
2.4 KiB
C++
107 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include "lexer.h"
|
|
#include <stdexcept>
|
|
#include <string>
|
|
|
|
#include <vector>
|
|
|
|
|
|
enum class ColumnType {
|
|
integer_type,
|
|
float_type,
|
|
varchar_type
|
|
};
|
|
|
|
enum class NodeType {
|
|
create_table,
|
|
insert_into,
|
|
select_from,
|
|
column_name,
|
|
column_value,
|
|
column_def,
|
|
not_implemented_yet,
|
|
error
|
|
};
|
|
|
|
struct Node {
|
|
NodeType node_type;
|
|
|
|
Node(const NodeType type) : node_type(type) {}
|
|
};
|
|
|
|
struct ColNameNode : Node {
|
|
std::string name;
|
|
|
|
ColNameNode(const std::string col_name) :
|
|
Node(NodeType::column_name), name(col_name) {}
|
|
};
|
|
|
|
struct ColValueNode : Node {
|
|
std::string value;
|
|
|
|
ColValueNode(const std::string col_value) :
|
|
Node(NodeType::column_value), value(col_value) {}
|
|
};
|
|
|
|
// TODO add order in row
|
|
struct ColDefNode : Node {
|
|
std::string name;
|
|
ColumnType type;
|
|
int order;
|
|
int length;
|
|
bool null;
|
|
|
|
ColDefNode(const std::string col_name, const ColumnType col_type, int col_order, int col_len, bool nullable) :
|
|
Node(NodeType::column_def), name(col_name), type(col_type), order(col_order), length(col_len), null(nullable) {}
|
|
};
|
|
|
|
struct CreateTableNode : Node {
|
|
std::string table_name;
|
|
std::vector<ColDefNode> cols_defs;
|
|
|
|
CreateTableNode(const std::string name, std::vector<ColDefNode> defs) :
|
|
Node(NodeType::create_table), table_name(name), cols_defs(defs) {}
|
|
};
|
|
|
|
struct InsertIntoTableNode : Node {
|
|
std::string table_name;
|
|
std::vector<ColNameNode> cols_names;
|
|
std::vector<ColValueNode> cols_values;
|
|
|
|
InsertIntoTableNode(const std::string name, std::vector<ColNameNode> names, std::vector<ColValueNode> values) :
|
|
Node(NodeType::insert_into), table_name(name), cols_names(names), cols_values(values) {}
|
|
};
|
|
|
|
struct SelectFromTableNode : Node {
|
|
std::string table_name;
|
|
std::vector<ColNameNode> cols_names;
|
|
std::vector<Node> where;
|
|
|
|
SelectFromTableNode(const std::string name, std::vector<ColNameNode> names, std::vector<Node> where_clause) :
|
|
Node(NodeType::select_from), table_name(name), cols_names(names), where(where_clause) {}
|
|
};
|
|
|
|
struct UpdateTableNode : Node { };
|
|
struct DeleteFromTableNode : Node { };
|
|
|
|
|
|
|
|
|
|
class Parser {
|
|
private:
|
|
|
|
public:
|
|
Parser();
|
|
|
|
std::unique_ptr<Node> parse(const std::string &code);
|
|
|
|
private:
|
|
std::unique_ptr<Node> parse_create_table();
|
|
std::unique_ptr<Node> parse_insert_into_table();
|
|
std::unique_ptr<Node> parse_select_from_table();
|
|
|
|
private:
|
|
Lexer lexer;
|
|
};
|