<?php

namespace LearnPress\Services;

use Exception;
use LearnPress\Helpers\Config;
use LearnPress\Helpers\Singleton;
use LearnPress\Models\UserModel;
use Throwable;
use WP_Error;
use WP_Role;
use WP_User;


/**
 * Class UserService
 *
 * Handle logic business for user.
 *
 * @since 4.3.4
 * @version 1.0.0
 */
class UserService {
	use Singleton;

	public function init() {}

	/**
	 * Check if pretty slug exists of another user.
	 *
	 * @param string $slug
	 *
	 * @return false|UserModel
	 * @throws Exception
	 * @version 1.0.1
	 * @since 4.3.4
	 */
	/*public function get_user_by_pretty_slug( string $slug ) {
		if ( '' === $slug ) {
			return false;
		}

		$lp_user_db          = UserDB::getInstance();
		$filter              = new UserFilter();
		$filter->only_fields = [ 'u.ID' ];
		$lp_user_db->get_query_single_row( $filter );
		$filter->join[]  = "INNER JOIN {$lp_user_db->wpdb->usermeta} AS um ON um.user_id = u.ID";
		$filter->where[] = $lp_user_db->wpdb->prepare( 'AND um.meta_key = %s', UserModel::META_KEY_USER_SLUG );
		$filter->where[] = $lp_user_db->wpdb->prepare( 'AND um.meta_value = %s', $slug );
		$query           = $lp_user_db->get_users( $filter );

		$user_id = (int) $lp_user_db->wpdb->get_var( $query );

		return UserModel::find( $user_id, true );
	}*/

	/**
	 * Generate pretty slug for all users who don't have it yet.
	 *
	 * @return array [ 'processed' => int, 'generated' => int, 'skipped' => int, 'failed' => int ]
	 * @since 4.3.4
	 * @version 1.0.0
	 */
	public function generate_users_pretty_slug(): array {
		$user_ids = get_users(
			[
				'fields' => 'ids',
				'number' => - 1,
			]
		);

		$result = [
			'processed' => 0,
			'generated' => 0,
			'skipped'   => 0,
			'failed'    => 0,
		];

		foreach ( $user_ids as $user_id ) {
			$user_id = (int) $user_id;
			++ $result['processed'];

			$userModel = UserModel::find( $user_id, true );
			if ( ! $userModel instanceof UserModel ) {
				++ $result['failed'];
				continue;
			}

			if ( '' !== $userModel->get_slug_link() ) {
				++ $result['skipped'];
				continue;
			}

			$generated = $this->generate_pretty_slug( $userModel );
			if ( is_wp_error( $generated ) ) {
				++ $result['failed'];
			} else {
				++ $result['generated'];
			}
		}

		return $result;
	}

	/**
	 * Create a unique pretty slug for user.
	 *
	 * If the user already has a pretty slug, it will return the existing one without generating a new one.
	 * The slug is generated based on the user's first name and last name.
	 * If empty user's first name and last name, it will use the username with uniqid() to generate a slug.
	 *
	 * @return string|WP_Error
	 * @since 4.3.4
	 * @version 1.0.1
	 */
	public function generate_pretty_slug( UserModel $userModel ) {
		$user_slug_new = '';

		try {
			// Check if pretty slug already exists, if exists, return it without generating a new one.
			$existing_slug = $userModel->get_slug_link();
			if ( ! empty( $existing_slug ) ) {
				return $existing_slug;
			}

			// Generate pretty slug based on first name and last name.
			$first_name  = $userModel->get_meta_value_by_key( 'first_name', '' );
			$last_name   = $userModel->get_meta_value_by_key( 'last_name', '' );
			$base_source = trim( "{$first_name} {$last_name}" );
			$base_slug   = sanitize_title( $base_source );

			if ( empty( $base_slug ) ) {
				// Shuffle username with uniqid to make it more unique and less guessable, get first 10 characters to make slug shorter.
				$base_slug = substr( str_shuffle( sanitize_title( $userModel->user_login . uniqid() ) ), 0, 10 );
			} else {
				$base_slug = $base_slug . substr( str_shuffle( uniqid() ), 0, 3 );
			}

			// Check slug exists.
			$userModelFind = UserService::instance()->get_user_by_slug_link( $base_slug );
			if ( ! $userModelFind ) {
				$userModel->user_nicename = $base_slug;
				$userModel->save();
			} else {
				// Regenerate slug by adding random string at the end of base slug until it is unique.
				$user_slug_new = $this->generate_pretty_slug( $userModel );
			}
		} catch ( Throwable $e ) {
			return new WP_Error( 'lp_user_slug_generation_failed', $e->getMessage() );
		}

		return $user_slug_new;
	}

	/**
	 * Detected user by slug link.
	 *
	 * @param string $slug_link
	 *
	 * @return false|UserModel
	 * @since 4.3.6
	 * @version 1.0.0
	 */
	public function get_user_by_slug_link( string $slug_link ) {
		// Get from column `user_nicename` in table `wp_users`.
		$wp_user = get_user_by( 'slug', $slug_link );
		if ( ! $wp_user instanceof WP_User ) {
			return false;
		}

		return new UserModel( $wp_user->data );
	}

	/**
	 * Set capabilities for roles.
	 *
	 * @param string $item_type Ex: "lp_course", "lp_lesson", "lp_assignment", "lp_h5p"
	 * @return void
	 * @since 4.4.2
	 * @version 1.0.0
	 */
	public function add_capabilities_for_roles( string $item_type ) {
		$role_capabilities = Config::instance()->get( 'user-roles-caps' );
		foreach ( $role_capabilities as $role_key => $role_data ) {
			$role_label  = $role_data['label'] ?? '';
			$prefix_caps = $role_data['prefix_capabilities'] ?? [];
			$role        = get_role( $role_key );
			if ( $role_key !== UserModel::ROLE_ADMINISTRATOR
				&& ! $role instanceof WP_Role ) {
				add_role( $role_key, $role_label );
				$role = get_role( $role_key );
			}

			if ( $role instanceof WP_Role ) {
				foreach ( $prefix_caps as $prefix_cap ) {
					// Example: "publish_lp_courses", "edit_lp_courses",...
					$cap = "{$prefix_cap}_{$item_type}s";
					if ( ! $role->has_cap( $cap ) ) {
						$role->add_cap( $cap );
					}
				}
			}
		}
	}

	/**
	 * Create or get CSRF token to verify
	 *
	 * @return mixed|string|null
	 * @since 4.4.6
	 * @version 1.0.0
	 */
	public static function csrf_token() {
		static $token = null;
		$cookie_name  = 'lp_csrf_token';

		if ( null !== $token ) {
			return $token;
		}

		if ( ! empty( $_COOKIE[ $cookie_name ] ) ) {
			$token = sanitize_text_field( wp_unslash( $_COOKIE[ $cookie_name ] ) );
			return $token;
		}

		$token = wp_generate_password( 20, false );
		if ( ! headers_sent() ) {
			learn_press_setcookie( $cookie_name, $token, time() + DAY_IN_SECONDS );
		}

		return $token;
	}
}