Function main

tools/shell/shell_runner.cpp:115–277

}

int main(int argc, char* argv[]) {
    args::ArgumentParser parser("Lbug shell");
    args::Positional<std::string> inputDirFlag(parser, "databasePath",
        "Path to the database. If not given or set to \":memory:\", the database will be opened "
        "under in-memory mode.");
    args::HelpFlag help(parser, "help", "Display this help menu", {'h', "help"});
    args::ValueFlag<uint64_t> bpSizeInMBFlag(parser, "",
        "Size of buffer pool for default and large page sizes in megabytes",
        {'d', "default_bp_size", "defaultbpsize"}, -1u);
    args::ValueFlag<uint64_t> maxDBSize(parser, "max_db_size",
        "Maximum size of the database in bytes", {"max_db_size", "maxdbsize"}, -1u);
    args::Flag disableCompression(parser, "no_compression", "Disable compression",
        {"no_compression", "nocompression"});
    args::Flag readOnlyMode(parser, "read_only", "Open database at read-only mode.",
        {'r', "read_only", "readonly"});
    args::ValueFlag<std::string> historyPathFlag(parser, "", "Path to directory for shell history",
        {'p', "path_history"});
    args::Flag version(parser, "version", "Display current database version", {'v', "version"});
    args::ValueFlag<std::string> mode(parser, "", "Set the output mode of the shell",
        {'m', "mode"});
    args::Flag stats(parser, "no_stats", "Disable query stats", {'s', "no_stats", "nostats"});
    args::Flag progress_bar(parser, "no_progress_bar", "Disable query progress bar",
        {'b', "no_progress_bar", "noprogressbar"});
    args::Flag ignore_wal_replay_errors(parser, "ignore_wal_replay_errors",
        "By default something goes wrong when replaying the WAL an exception will be thrown. With "
        "this flag enabled we will instead ignore these errors and stop replaying the WAL.",
        {'w', "ignore_wal_replay_errors"});
    args::ValueFlag<std::string> init(parser, "", "Path to file with script to run on startup",
        {'i', "init"});

    std::vector<std::string> lCaseArgsStrings;
    for (auto i = 0; i < argc; ++i) {
        std::string arg = argv[i];
        if (arg.size() > 1 && arg[0] == '-') {
            std::string lArg = arg;
            std::transform(lArg.begin(), lArg.end(), lArg.begin(),
                [](unsigned char c) { return std::tolower(c); });
            lCaseArgsStrings.push_back(lArg);
        } else {
            lCaseArgsStrings.push_back(argv[i]);
        }
    }
    std::vector<char*> lCaseArgs;
    for (auto& arg : lCaseArgsStrings) {
        lCaseArgs.push_back(const_cast<char*>(arg.c_str()));
    }

    try {
        parser.ParseCLI(lCaseArgs.size(), lCaseArgs.data());
    } catch (const args::Help&) {
        std::cout << parser;
        return 0;
    } catch (const args::ParseError& e) {
        std::cerr << e.what() << '\n';
        std::cerr << parser;
        return 1;
    }