cpp-peglib/lint/peglint.cc

//
//  peglint.cc
//
//  Copyright (c) 2015 Yuji Hirose. All rights reserved.
//  MIT License
//

#include <peglib.h>
#include <fstream>
#include <sstream>

using namespace std;

bool read_file(const char* path, vector<char>& buff)
{
    ifstream ifs(path, ios::in | ios::binary);
    if (ifs.fail()) {
        return false;
    }

    buff.resize(static_cast<unsigned int>(ifs.seekg(0, ios::end).tellg()));
    if (!buff.empty()) {
        ifs.seekg(0, ios::beg).read(&buff[0], static_cast<streamsize>(buff.size()));
    }
    return true;
}

int main(int argc, const char** argv)
{
    auto opt_ast = false;
    auto opt_optimize_ast_nodes = false;
    auto opt_help = false;
    auto opt_source = false;
    vector<char> source;
    auto opt_trace = false;
    vector<const char*> path_list;

    auto argi = 1;
    while (argi < argc) {
        auto arg = argv[argi++];
        if (string("--help") == arg) {
            opt_help = true;
        } else if (string("--ast") == arg) {
            opt_ast = true;
        } else if (string("--optimize_ast_nodes") == arg || string("--opt") == arg) {
            opt_optimize_ast_nodes = true;
        } else if (string("--source") == arg) {
            opt_source = true;
            if (argi < argc) {
                std::string text = argv[argi++];
                source.assign(text.begin(), text.end());
            }
        } else if (string("--trace") == arg) {
            opt_trace = true;
        } else {
            path_list.push_back(arg);
        }
    }

    if (path_list.empty() || opt_help) {
        cerr << "usage: peglint [--ast] [--optimize_ast_nodes|--opt] [--source text] [--trace] [grammar file path] [source file path]" << endl;
        return 1;
    }

    // Check PEG grammar
    auto syntax_path = path_list[0];

    vector<char> syntax;
    if (!read_file(syntax_path, syntax)) {
        cerr << "can't open the grammar file." << endl;
        return -1;
    }

    peg::parser parser;

    parser.log = [&](size_t ln, size_t col, const string& msg) {
        cerr << syntax_path << ":" << ln << ":" << col << ": " << msg << endl;
    };

    if (!parser.load_grammar(syntax.data(), syntax.size())) {
        return -1;
    }

    if (path_list.size() < 2 && !opt_source) {
        return 0;
    }

    // Check source
    std::string source_path = "[commandline]";
    if (path_list.size() >= 2) {
        if (!read_file(path_list[1], source)) {
            cerr << "can't open the code file." << endl;
            return -1;
        }
        source_path = path_list[1];
    }

    parser.log = [&](size_t ln, size_t col, const string& msg) {
        cerr << source_path << ":" << ln << ":" << col << ": " << msg << endl;
    };

    if (opt_trace) {
        size_t prev_pos = 0;
        parser.enable_trace(
            [&](const char* name,
                const char* s,
                size_t /*n*/,
                const peg::SemanticValues& /*sv*/,
                const peg::Context& c,
                const peg::any& /*dt*/) {
                auto pos = static_cast<size_t>(s - c.s);
                auto backtrack = (pos < prev_pos ? "*" : "");
                string indent;
                auto level = c.trace_ids.size() - 1;
                while (level--) { indent += "│"; }
                std::cout
                    << "E " << pos << backtrack << "\t"
                    << indent << "┌" << name
                    << " #" << c.trace_ids.back()
                    << std::endl;
                prev_pos = static_cast<size_t>(pos);
            },
            [&](const char* name,
                const char* s,
                size_t /*n*/,
                const peg::SemanticValues& sv,
                const peg::Context& c,
                const peg::any& /*dt*/,
                size_t len) {
                auto pos = static_cast<size_t>(s - c.s);
                if (len != -1) {
                    pos += len;
                }
                string indent;
                auto level = c.trace_ids.size() - 1;
                while (level--) { indent += "│"; }
                auto ret = len != -1 ? "└o " : "└x ";
                std::stringstream choice;
                if (sv.choice_count() > 0) {
                    choice << " " << sv.choice() << "/" << sv.choice_count();
                }
                std::string token;
                if (!sv.tokens.empty()) {
                    const auto& tok = sv.tokens[0];
                    token += " '" + std::string(tok.first, tok.second) + "'";
                }
                std::cout
                    << "L " << pos << "\t"
                    << indent << ret << name
                    << " #" << c.trace_ids.back()
                    << choice.str()
                    << token
                    << std::endl;
            }
        );
    }

    if (opt_ast) {
	    parser.enable_ast();

	    std::shared_ptr<peg::Ast> ast;
	    if (!parser.parse_n(source.data(), source.size(), ast)) {
	        return -1;
	    }

        ast = peg::AstOptimizer(opt_optimize_ast_nodes).optimize(ast);
        std::cout << peg::ast_to_s(ast);

    } else {
	    if (!parser.parse_n(source.data(), source.size())) {
	        return -1;
	    }
    }

    return 0;
}

// vim: et ts=4 sw=4 cin cino={1s ff=unix