book.cpp

/* Main file for BookTracker CLI application.
 * Handles command-line argument parsing and command execution.
 *
 * Author: Harry Bevins
 */
#include "src/book.h"
#include "src/commands.h"
#include "src/database.h"
#include "src/plot.h"
#include "src/stats.h"
#include "src/version.h"
#include "src/help.h"
#include <algorithm>
#include <cstdlib>
#include <iostream>
#include <vector>

using namespace std;

int main(int argc, char *argv[]) {
  /* Main function for BookTracker CLI application.
   * Parses command-line arguments and executes corresponding commands.
   * Params:
   *   argc - number of command-line arguments
   *   argv - array of command-line argument strings
   */

  const vector<string> allowed_commands{"add",    "ls",      "list",   "del",
                                        "delete", "mod",     "modify", "help",
                                        "show",   "version", "plot",   "stats"};

  string command = argv[1];

  if (find(allowed_commands.begin(), allowed_commands.end(), command) ==
      allowed_commands.end()) {
    cout << "Unknown command: " << command << endl;
    return 1;
  }

  string homeDir = getenv("HOME");

  vector<Book> books = loadBooks(homeDir);
  vector<int> ids;
  for (Book b : books) {
    ids.push_back(b.id);
  }

  int maxId = ids.size() > 0 ? *max_element(ids.begin(), ids.end()) : 0;

  if (command == "add") {
    addBook(argv[2], maxId);
  } else if (command == "delete" || command == "del") {
    int id = stoi(argv[2]);
    deleteBook(id, books);
  } else if (command == "list" || command == "ls") {
    list(books);
  } else if (command == "show") {
    int id = stoi(argv[2]);
    showBook(id, books);
  } else if (command == "mod" || command == "modify") {
    int id = stoi(argv[2]);
    modifyBook(id, argv[3], argv[4], books);
  } else if (command == "help") {
    displayHelp();
  } else if (command == "version") {
    cout << "BookTracker version " << BOOKTRACKER_VERSION << endl;
  } else if (command == "plot") {
    plot(books);
  } else if (command == "stats") {
    stats(books);
  }
  return 0;
}