main.cpp

#include <iostream>
#include <print>
#include <string>

#include "SearchEngine.h"

enum class Command { UNKNOWN, EXIT, FIND };

Command getCommand(const std::string &input) {
    static const std::unordered_map<std::string, Command> commandMap = {
        {"find", Command::FIND}, {"exit", Command::EXIT}
    };

    auto it = commandMap.find(input);
    if (it != commandMap.end()) return it->second;
    return Command::UNKNOWN;
}

int main(int argc, char *argv[]) {
    if (argc < 2) {
        std::println("Missing data directory");
    }

    if (argc == 2) {
        SearchEngine search = SearchEngine();
        search.addDir(argv[1]);
        search.compile();

        std::string command = "";
        bool shouldExit = false;
        while (!shouldExit) {
            std::print("> ");
            std::cin >> command;
            switch (getCommand(command)) {
                case Command::FIND: {
                    std::string searchString = "";
                    std::getline(std::cin, searchString);

                    search.find(searchString);
                    break;
                }
                case Command::EXIT: {
                    shouldExit = true;
                    break;
                }
                case Command::UNKNOWN: {
                    std::println("Invalid command");
                    break;
                }
                default: {
                    shouldExit = true;
                    break;
                }
            }
        }
    }

    return 0;
}