Endgame-Analyzer/src/download.cpp

#include <boost/json/src.hpp>
#include <cpr/cpr.h>

#include "parse_game.h"

#include "download.h"


namespace Download {

    std::optional<boost::json::object> download_game_json(int game_id) {
        std::string request_str = "https://hanab.live/export/" + std::to_string(game_id);
        cpr::Response r = cpr::Get(cpr::Url(request_str));
        if (r.header["content-type"] != "application/json; charset=utf-8") {
            return std::nullopt;
        }
        return boost::json::parse(r.text).as_object();
    }

    std::optional<boost::json::object> open_game_json(const char *filename) {
        std::ifstream file(filename);
        if (!file.is_open()) {
            return std::nullopt;
        }
        std::string game_json((std::istreambuf_iterator<char>(file)), std::istreambuf_iterator<char>());
        return boost::json::parse(game_json).as_object();
    }

    std::unique_ptr<Hanabi::HanabiStateIF> get_base_state(
          std::size_t num_suits,
          Hanabi::player_t num_players,
          std::vector<Hanabi::Card> const & deck,
          std::optional<uint8_t> score_goal) {
        uint8_t actual_score_goal = score_goal.value_or(5 * num_suits);
        switch(num_players) {
            case 2:
                switch(num_suits) {
                    case 3:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<3,2,5>(deck, actual_score_goal));
                    case 4:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<4,2,5>(deck, actual_score_goal));
                    case 5:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<5,2,5>(deck, actual_score_goal));
                    case 6:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<6,2,5>(deck, actual_score_goal));
                    default:
                        throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
                }
            case 3:
                switch(num_suits) {
                    case 3:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<3,3,5>(deck, actual_score_goal));
                    case 4:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<4,3,5>(deck, actual_score_goal));
                    case 5:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<5,3,5>(deck, actual_score_goal));
                    case 6:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<6,3,5>(deck, actual_score_goal));
                    default:
                        throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
                }
            case 4:
                switch(num_suits) {
                    case 3:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<3,4,4>(deck, actual_score_goal));
                    case 4:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<4,4,4>(deck, actual_score_goal));
                    case 5:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<5,4,4>(deck, actual_score_goal));
                    case 6:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<6,4,4>(deck, actual_score_goal));
                    default:
                        throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
                }
            case 5:
                switch(num_suits) {
                    case 3:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<3,5,4>(deck, actual_score_goal));
                    case 4:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<4,5,4>(deck, actual_score_goal));
                    case 5:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<5,5,4>(deck, actual_score_goal));
                    case 6:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<6,5,4>(deck, actual_score_goal));
                    default:
                        throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
                }
            case 6:
                switch(num_suits) {
                    case 3:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<3,6,3>(deck, actual_score_goal));
                    case 4:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<4,6,3>(deck, actual_score_goal));
                    case 5:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<5,6,3>(deck, actual_score_goal));
                    case 6:
                        return std::unique_ptr<Hanabi::HanabiStateIF>(new Hanabi::HanabiState<6,6,3>(deck, actual_score_goal));
                    default:
                        throw std::runtime_error("Invalid number of suits: " + std::to_string(num_suits));
                }
            default:
                throw std::runtime_error("Invalid number of players: " + std::to_string(num_players));
        }
    }

  Hanabi::Game get_game(std::variant<int, std::string> game_spec, std::optional<uint8_t> score_goal){
    const std::optional<boost::json::object> game_json_opt = [&game_spec]() {
      if (game_spec.index() == 0) {
        return download_game_json(std::get<int>(game_spec));
      } else {
        return open_game_json(std::get<std::string>(game_spec).c_str());
      }
    }();

    if (!game_json_opt.has_value() or game_json_opt.value().empty()) {
      return {nullptr, {}, {}};
    }

    Parsing::GameInfo game_info = Parsing::parse_game(game_json_opt.value());
    return {get_base_state(game_info.num_suits, game_info.num_players, game_info.deck, score_goal), game_info.actions, game_info.deck};
  }

} // namespace Download