R/entities.R

In itscalledsoccer: American Soccer Analysis API Client

#' @importFrom rlang .data
.get_entity <- function(self, type, verbose) {
    plural_type <- if (type == "stadium") "stadia" else glue::glue("{type}s")
    if (verbose) cat(glue::glue("  Gathering {plural_type}"))

    entity_all <- list()

    for (league in self$LEAGUES) {
        url <- glue::glue("{self$base_url}/{league}/{plural_type}")

        response <- .execute_query(self, url)
        response <- response %>% dplyr::mutate(competition = league)

        entity_all <- append(entity_all, list(response))

        if (verbose) cat(".")
    }

    # can give player_name, etc.
    col_arrange <- paste0(type, "_name")
    entity_all <- data.table::rbindlist(entity_all, fill = TRUE) %>%
        dplyr::group_by(dplyr::pick(c(-dplyr::matches("competition"), -dplyr::starts_with("season"), -dplyr::ends_with("position")))) %>%
        dplyr::summarize(competitions = list(.data$competition), .groups = "drop") %>%
        dplyr::arrange(.data[[col_arrange]])

    if (verbose) cli::cat_bullet(bullet = "tick", bullet_col = "green")

    return(entity_all)
}

.initialize_entities <- function(self, verbose = FALSE) {
    self$players <- .get_entity(self, "player", verbose = verbose)
    self$teams <- .get_entity(self, "team", verbose = verbose)
    self$stadia <- .get_entity(self, "stadium", verbose = verbose)
    self$managers <- .get_entity(self, "manager", verbose = verbose)
    self$referees <- .get_entity(self, "referee", verbose = verbose)
}

#' @importFrom rlang .data
.filter_entity <- function(self, entity, leagues, ids, names) {
    .check_leagues(self, leagues)
    .check_ids_names(ids, names)
    .check_clear_cache(self)

    entity_filtered <- self[[entity]] %>%
        tidyr::unnest("competitions")

    if (!missing(leagues)) {
        entity_filtered <- entity_filtered %>%
            dplyr::filter(.data$competitions %in% leagues)
    }

    if (!missing(names)) {
        ids <- .convert_names_to_ids(entity_filtered, names)
    }

    if (!missing(names) | !missing(ids)) {
        entity_filtered <- entity_filtered %>%
            dplyr::filter(dplyr::if_any(dplyr::ends_with("_id"), ~ . %in% ids))
    }

    entity_filtered <- entity_filtered %>%
        dplyr::select(-"competitions") %>%
        dplyr::distinct()

    return(entity_filtered)
}

#' @importFrom rlang .data
.get_games <- function(self, leagues, game_ids, team_ids, team_names, seasons, stages) {
    .check_leagues(self, leagues)
    .check_ids_names(team_ids, team_names)
    .check_clear_cache(self)

    if (missing(leagues)) leagues <- self$LEAGUES

    query <- list()
    if (!missing(game_ids)) query$game_id <- game_ids
    if (!missing(team_ids)) query$team_id <- team_ids
    if (!missing(team_names)) query$team_id <- .convert_names_to_ids(self$teams, team_names)
    if (!missing(seasons)) query$season_name <- seasons
    if (!missing(stages)) query$stage_name <- stages

    games <- list()

    for (league in unique(leagues)) {
        url <- glue::glue("{self$base_url}/{league}/games")
        response <- .execute_query(self, url, query)
        games <- append(games, list(response))
    }

    games <- data.table::rbindlist(games, fill = TRUE) %>% dplyr::arrange(.data$date_time_utc)
    return(games)
}