<?php
/**
 * @package Restrict User Access
 * @author Joachim Jensen <joachim@dev.institute>
 * @license GPLv3
 * @copyright 2024 by Joachim Jensen
 */

/**
 * @since  1.1
 * @param  WP_User|int|null  $user null or omit for current user
 * @return RUA_User_Interface
 */
function rua_get_user($user = null)
{
    if (is_null($user) && is_user_logged_in()) {
        $user = wp_get_current_user();
    }

    if (!($user instanceof WP_User)) {
        $user = new WP_User($user);
    }

    return new RUA_User($user);
}

/**
 * @since 2.1
 * @param WP_Post|int $post
 * @return RUA_Level_Interface
 * @throws Exception
 */
function rua_get_level($post)
{
    if (is_numeric($post)) {
        $post = WP_Post::get_instance($post);
    }

    if (!($post instanceof WP_Post)) {
        throw new Exception();
    }

    return new RUA_Level($post);
}

/**
 * @since 2.1
 * @param RUA_Level_Interface|WP_Post|int $level
 * @param RUA_User_Interface|WP_User|int|null $user null or omit for current user
 * @return RUA_User_Level_Interface
 * @throws Exception
 */
function rua_get_user_level($level, $user = null)
{
    _deprecated_function(__FUNCTION__, '2.5', 'rua_get_user_levels()');

    if (!($level instanceof RUA_Level_Interface)) {
        $level = rua_get_level($level);
    }
    if (!($user instanceof RUA_User_Interface)) {
        $user = rua_get_user($user);
    }
    return $user->level_memberships()->get($level->get_id());
}

/**
 * @param RUA_User_Interface|WP_User|int|null $user null or omit for current user
 * @return RUA_Collection
 */
function rua_get_user_levels($user)
{
    if (!($user instanceof RUA_User_Interface)) {
        $user = rua_get_user($user);
    }

    $user_levels = new RUA_Collection();
    if($user->get_id() === 0) {
        return $user_levels;
    }

    $entities = _rua_get_comments([
        'user_id' => $user->get_id()
    ]);

    foreach ($entities as $entity) {
        $user_level = new RUA_User_Level($entity);
        $user_level->refresh();
        $user_levels->put($user_level->get_level_id(), $user_level);
    }
    return $user_levels;
}

/**
 * @param RUA_Level_Interface|WP_Post|int $level
 * @param array $query
 * @return RUA_Collection
 * @throws Exception
 */
function rua_get_level_members($level, $query = [])
{
    if (!($level instanceof RUA_Level_Interface)) {
        $level = rua_get_level($level);
    }

    $query['post_id'] = $level->get_id();
    $entities = _rua_get_comments($query);

    $user_levels = new RUA_Collection();
    foreach ($entities as $entity) {
        $user_level = new RUA_User_Level($entity);
        $user_level->refresh();
        $user_levels->put($user_level->get_user_id(), $user_level);
    }
    return $user_levels;
}

/**
 * @since  0.9
 * @param  string  $name
 * @return WP_Post|bool
 */
function rua_get_level_by_name($name)
{
    return RUA_App::instance()->level_manager->get_level_by_name($name);
}

/**
 * @since  0.13
 * @param  int   $level_id
 * @param  bool  $hierarchical
 * @return array
 */
function rua_get_level_caps($level_id, $hierarchical = false)
{
    $levels = [$level_id];
    if ($hierarchical) {
        $levels = array_merge($levels, get_post_ancestors((int) $level_id));
        $levels = array_reverse($levels);
    }
    $caps = RUA_App::instance()->level_manager->get_levels_caps($levels);
    return $caps;
}

/**
 * @internal
 * @param $args
 * @return int|int[]|WP_Comment[]
 */
function _rua_get_comments($args)
{
    $args = array_merge([
        'status'  => [RUA_User_Level::STATUS_ACTIVE, RUA_User_Level::STATUS_EXPIRED],
        'orderby' => false,
    ], $args);
    $args['type'] = RUA_User_Level::ENTITY_TYPE;

    $query = new RUA_Member_Query();
    return $query->query($args);
}