main.cpp

#include "main.h"

void printHelp(std::string name);
void setNcursesUp();

int main(int argc, char *argv[]) {
    std::string sessionArg = "--session";
    std::string deleteArg = "--delete-session";
    std::string listArg = "--list-sessions";
    std::string helpArg = "--help";

    if (argc >= 2 && argv[1] == sessionArg) {
        // run timer with selected session
        if (argc >= 3) {
            setNcursesUp();
            // this is very vulnerable to SQL injection
            UI timer(DEFAULT_FILENAME, argv[2]);
            timer.mainloop();
        } else {
            printHelp(argv[0]);
        }
    } else if (argc >= 2 && argv[1] == deleteArg) {
        // delete selected session
        if (argc >= 3) {
            dbConnection conn;
            try {
                conn.deleteSession(argv[2]);
                std::cout << "Session " << argv[2] << " successfully deleted."
                          << std::endl;
            } catch (const std::invalid_argument &e) {
                std::cout << "ERROR: no session called " << argv[2] << " found."
                          << std::endl;
            }
        } else {
            printHelp(argv[0]);
        }
    } else if (argc >= 2 && argv[1] == listArg) {
        // list all sessions
        dbConnection conn;
        std::deque<std::string> sessions;
        conn.getAllSessions(sessions);
        for (unsigned i = 0; i < sessions.size(); i++) {
            std::cout << sessions.at(i) << std::endl;
        }
    } else if (argc >= 2 && argv[1] == helpArg) {
        // print help message
        printHelp(argv[0]);
    } else {
        // run timer normally
        setNcursesUp();
        UI timer;
        timer.mainloop();
    }

    return 0;
}

void printHelp(std::string name) {
    std::cout << "USAGE:\n\n";
    std::cout << name << ": open timer in default session\n";
    std::cout
        << name
        << " --delete-session sessionName: delete session called sessionName\n";
    std::cout << name << " --help: output this help message and quit\n";
    std::cout << name << " --list-sessions: list all sessions\n";
    std::cout << name
              << " --session sessionName: open timer in session called "
                 "sessionName, creating it if needed"
              << std::endl;
}

void setNcursesUp() {
    // functions that need to run before UI constructor to set ncurses up
    srand(time(nullptr));

    initscr();
    // don't print keystrokes to terminal
    noecho();
    // detect arrow keys, etc.
    keypad(stdscr, TRUE);
    // hide cursor
    curs_set(0);
    start_color();
    refresh();
}