psopeeps-newserv/Shell.cc

#include "Shell.hh"

#include <readline/history.h>
#include <readline/readline.h>
#include <stdio.h>

#include <phosg/Strings.hh>

using namespace std;



class exit_shell : public runtime_error {
public:
  exit_shell() : runtime_error("shell exited") { }
  ~exit_shell() = default;
};



void execute_command(shared_ptr<ServerState> state, const string& command) {
  // find the entry in the command table and run the command
  size_t command_end = skip_non_whitespace(command, 0);
  size_t args_begin = skip_whitespace(command, command_end);
  string command_name = command.substr(0, command_end);
  string command_args = command.substr(args_begin);

  if (command_name == "exit") {
    throw exit_shell();

  } else if (command_name == "help") {
    fprintf(stderr, "\
commands:\n\
  help\n\
    you\'re reading it now\n\
  exit (or ctrl+d)\n\
    shut down the server\n\
  reload <item> ...\n\
    reload data. <item> can be licenses, battle-params, level-table, or quests.\n\
  add-license <parameters>\n\
    add a license to the server. <parameters> is some subset of the following:\n\
      username=<username> (bb username)\n\
      bb-password=<password> (bb password)\n\
      gc-password=<password> (gc password)\n\
      access-key=<access-key> (gc/pc access key)\n\
      serial=<serial-number> (gc/pc serial number; required for all licenses)\n\
      privileges=<privilege-mask> (can be normal, mod, admin, root, or numeric)\n\
  delete-license <serial-number>\n\
    delete a license from the server\n\
  list-licenses\n\
    list all licenses registered on the server\n\
");

  } else if (command_name == "reload") {
    auto types = split(command_args, ' ');
    for (const string& type : types) {
      if (type == "licenses") {
        shared_ptr<LicenseManager> lm(new LicenseManager("system/licenses.nsi"));
        state->license_manager = lm;
      } else if (type == "battle-params") {
        shared_ptr<BattleParamTable> bpt(new BattleParamTable("system/blueburst/BattleParamEntry"));
        state->battle_params = bpt;
      } else if (type == "level-table") {
        shared_ptr<LevelTable> lt(new LevelTable("system/blueburst/PlyLevelTbl.prs", true));
        state->level_table = lt;
      } else if (type == "quests") {
        shared_ptr<QuestIndex> qi(new QuestIndex("system/quests"));
        state->quest_index = qi;
      } else {
        throw invalid_argument("incorrect data type");
      }
    }

  } else if (command_name == "add-license") {
    shared_ptr<License> l(new License());
    memset(l.get(), 0, sizeof(License));

    for (const string& token : split(command_args, ' ')) {
      if (starts_with(token, "username=")) {
        if (token.size() >= 29) {
          throw invalid_argument("username too long");
        }
        strcpy(l->username, token.c_str() + 9);

      } else if (starts_with(token, "bb-password=")) {
        if (token.size() >= 32) {
          throw invalid_argument("bb-password too long");
        }
        strcpy(l->bb_password, token.c_str() + 12);

      } else if (starts_with(token, "gc-password=")) {
        if (token.size() > 20) {
          throw invalid_argument("gc-password too long");
        }
        strcpy(l->gc_password, token.c_str() + 12);

      } else if (starts_with(token, "access-key=")) {
        if (token.size() > 23) {
          throw invalid_argument("access-key is too long");
        }
        strcpy(l->access_key, token.c_str() + 11);

      } else if (starts_with(token, "serial=")) {
        l->serial_number = stoul(token.substr(7));

      } else if (starts_with(token, "privileges=")) {
        string mask = token.substr(11);
        if (mask == "normal") {
          l->privileges = 0;
        } else if (mask == "mod") {
          l->privileges = Privilege::Moderator;
        } else if (mask == "admin") {
          l->privileges = Privilege::Administrator;
        } else if (mask == "root") {
          l->privileges = Privilege::Root;
        } else {
          l->privileges = stoul(mask);
        }

      } else {
        throw invalid_argument("incorrect field");
      }
    }

    if (!l->serial_number) {
      throw invalid_argument("license does not contain serial number");
    }

    state->license_manager->add(l);
    fprintf(stderr, "license added\n");

  } else if (command_name == "delete-license") {
    uint32_t serial_number = stoul(command_args);
    state->license_manager->remove(serial_number);
    fprintf(stderr, "license deleted\n");

  } else if (command_name == "list-licenses") {
    for (const auto& l : state->license_manager->snapshot()) {
      string s = l.str();
      fprintf(stderr, "%s\n", s.c_str());
    }

  } else {
    throw invalid_argument("unknown command; try \'help\'");
  }
}

void run_shell(shared_ptr<ServerState> state) {
  // initialize history
  using_history();
  // string history_filename = get_user_home_directory() + "/.newserv_history";
  // read_history(history_filename.c_str());
  // stifle_history(HISTORY_FILE_LENGTH);

  // read and execute commands
  bool should_continue = true;
  while (should_continue) {

    // read the command
    char* command = readline("newserv> ");
    if (!command) {
      fprintf(stderr, " -- exit\n");
      break;
    }

    // if there's a command, add it to the history
    const char* command_to_execute = command + skip_whitespace(command, 0);
    if (command_to_execute && *command_to_execute) {
      add_history(command);
    }

    // dispatch the command
    try {
      execute_command(state, command_to_execute);
    } catch (const exit_shell&) {
      should_continue = false;
    } catch (const exception& e) {
      fprintf(stderr, "FAILED: %s\n", e.what());
    }
    free(command);
  }
}