60 lines
1.5 KiB
C++
60 lines
1.5 KiB
C++
#include <cstdlib>
|
|
#include <vector>
|
|
#include <vector>
|
|
|
|
#include "compiler.hpp"
|
|
#include "debug/tree_print.hpp"
|
|
#include "codegen/cppgen.hpp"
|
|
#include "utils/command_line/arguments.hpp"
|
|
#include "utils/string/filereader.hpp"
|
|
|
|
#include "utils/terminate_handler.hpp"
|
|
|
|
using std::cout;
|
|
using std::endl;
|
|
|
|
// * INPUT ARGUMENTS *
|
|
// -q -> query
|
|
// -v -> visitor
|
|
// -f -> file
|
|
|
|
std::string extract_query(const vector_str& arguments)
|
|
{
|
|
if (contain_argument(arguments, "-q"))
|
|
return get_argument(arguments, "-q", "CREATE (n {a:1, b:2}) RETURN n");
|
|
auto default_file = "query/read/match/match-where.cypher";
|
|
auto file = get_argument(arguments, "-f", default_file);
|
|
// TODO: error handling
|
|
return read_file(file.c_str());
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
std::set_terminate(&terminate_handler);
|
|
|
|
// arguments parsing
|
|
auto arguments = all_arguments(argc, argv);
|
|
|
|
// query extraction
|
|
auto cypher_query = extract_query(arguments);
|
|
cout << "QUERY: " << cypher_query << endl;
|
|
|
|
// traversers
|
|
auto traverser = get_argument(arguments, "-t", "code");
|
|
auto print_traverser = Traverser::sptr(new PrintVisitor(cout));
|
|
auto cppgen_traverser = Traverser::sptr(new CppGen());
|
|
std::map<std::string, Traverser::sptr> traversers = {
|
|
{"print", print_traverser},
|
|
{"code", cppgen_traverser}
|
|
};
|
|
|
|
cypher::Compiler compiler;
|
|
auto tree = compiler.syntax_tree(cypher_query);
|
|
|
|
auto t = traversers[traverser];
|
|
tree.root->accept(*t);
|
|
|
|
return 0;
|
|
}
|
|
|