<?php
namespace JVBase\rest\routes;

use JVBase\rest\RestRouteManager;
use WP_REST_Request;
use WP_REST_Response;
use WP_Error;

if (!defined('ABSPATH')) {
	exit;
}

/**
 * REST API routes for referral system
 */
class ReferralRoutes extends RestRouteManager
{

	public function __construct()
	{
		$this->route = 'referrals';
		parent::__construct();
	}

	public function registerRoutes(): void
	{
		// Get user's referrals
		register_rest_route($this->namespace, "/{$this->route}", [
			'methods' => 'GET',
			'callback' => [$this, 'getUserReferrals'],
			'permission_callback' => [$this, 'checkPermission']
		]);

		register_rest_route($this->namespace, '/referrals/register', [
			'methods' => 'POST',
			'callback' => [$this, 'registerWithReferral'],
			'permission_callback' => '__return_true',
			'args' => [
				'name' => [
					'required' => true,
					'type' => 'string',
					'sanitize_callback' => 'sanitize_text_field'
				],
				'email' => [
					'required' => true,
					'type' => 'string',
					'format' => 'email',
					'validate_callback' => function($param) {
						return is_email($param);
					}
				],
				'code' => [
					'required' => true,
					'type' => 'string',
					'sanitize_callback' => function($code) {
						return strtoupper(sanitize_text_field($code));
					}
				]
			]
		]);

		register_rest_route($this->namespace, '/referrals/check-code', [
			'methods' => 'POST',
			'callback' => [$this, 'checkReferralCode'],
			'permission_callback' => '__return_true',
			'args' => [
				'code' => [
					'required' => true,
					'type' => 'string',
					'sanitize_callback' => function($code) {
						return strtoupper(sanitize_text_field($code));
					}
				]
			]
		]);

		// Get or create referral code
		register_rest_route($this->namespace, "/{$this->route}/code", [
			'methods' => 'GET',
			'callback' => [$this, 'getReferralCode'],
			'permission_callback' => [$this, 'checkPermission']
		]);

		// Track referral click (public endpoint)
		register_rest_route($this->namespace, "/{$this->route}/track", [
			'methods' => 'POST',
			'callback' => [$this, 'trackReferralClick'],
			'permission_callback' => '__return_true',
			'args' => [
				'code' => [
					'required' => true,
					'type' => 'string'
				]
			]
		]);

		// Mark referral as treated
		register_rest_route($this->namespace, "/{$this->route}/(?P<id>\d+)/treat", [
			'methods' => 'POST',
			'callback' => [$this, 'markAsTreated'],
			'permission_callback' => function() {
				return current_user_can('manage_options');
			},
			'args' => [
				'id' => [
					'required' => true,
					'validate_callback' => function($param) {
						return is_numeric($param);
					}
				]
			]
		]);

		// Send referral invitation
		register_rest_route($this->namespace, '/'.$this->route.'/invite', [
			'methods' => 'POST',
			'callback' => [$this, 'sendInvitation'],
			'permission_callback' => [$this, 'checkPermission'],
			'args' => [
				'email' => [
					'required' => true,
					'type' => 'string',
					'format' => 'email',
					'validate_callback' => function($param) {
						return is_email($param);
					}
				],
				'name' => [
					'required' => true,
					'type' => 'string',
					'sanitize_callback' => 'sanitize_text_field'
				]
			]
		]);

		// Send batch invitations
		register_rest_route($this->namespace, '/'.$this->route.'/invite/batch', [
			'methods' => 'POST',
			'callback' => [$this, 'sendBatchInvitations'],
			'permission_callback' => [$this, 'checkPermission'],
			'args' => [
				'invitations' => [
					'required' => true,
					'type' => 'array',
					'validate_callback' => function($param) {
						return is_array($param) && !empty($param);
					}
				]
			]
		]);

		// Get invitation stats for current user
		register_rest_route($this->namespace, '/'.$this->route.'/invite/stats', [
			'methods' => 'GET',
			'callback' => [$this, 'getInvitationStats'],
			'permission_callback' => [$this, 'checkPermission']
		]);

		// Export referrals for Jane App
		register_rest_route($this->namespace, '/'.$this->route.'/export', [
			'methods' => 'POST',
			'callback' => [$this, 'exportReferrals'],
			'permission_callback' => function() {
				return current_user_can('manage_options');
			},
			'args' => [
				'start_date' => [
					'required' => true,
					'type' => 'string',
					'validate_callback' => function($param) {
						return (bool) strtotime($param);
					}
				],
				'end_date' => [
					'required' => true,
					'type' => 'string',
					'validate_callback' => function($param) {
						return (bool) strtotime($param);
					}
				]
			]
		]);

		// Get top referrers (admin only)
		register_rest_route($this->namespace, "/{$this->route}/leaderboard", [
			'methods' => 'GET',
			'callback' => [$this, 'getTopReferrers'],
			'permission_callback' => function() {
				return current_user_can('manage_options');
			},
			'args' => [
				'period' => [
					'default' => 'week',
					'enum' => ['day', 'week', 'month', 'all']
				],
				'limit' => [
					'default' => 10,
					'type' => 'integer'
				]
			]
		]);

		// Get/Update referral settings (admin only)
		register_rest_route($this->namespace, "/{$this->route}/settings", [
			[
				'methods' => 'GET',
				'callback' => [$this, 'getSettings'],
				'permission_callback' => function() {
					return current_user_can('manage_options');
				}
			],
			[
				'methods' => 'POST',
				'callback' => [$this, 'updateSettings'],
				'permission_callback' => function() {
					return current_user_can('manage_options');
				}
			]
		]);
	}

	public function checkPermission(WP_REST_Request $request): bool
	{
		return is_user_logged_in();
	}

	/**
	 * Get user's referrals
	 */
	public function getUserReferrals(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();

		$args = [
			'status' => $request->get_param('status') ?? 'all',
			'limit' => $request->get_param('limit') ?? 50,
			'offset' => $request->get_param('offset') ?? 0
		];

		$referrals = JVB()->referrals()->getUserReferrals($user_id, $args);

		return new WP_REST_Response([
			'success' => true,
			'referrals' => $referrals
		]);
	}

	/**
	 * Get user's referral code
	 */
	public function getReferralCode(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();
		$code = JVB()->referrals()->getUserReferralCode($user_id);

		if (is_wp_error($code)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => $code->get_error_message()
			], 400);
		}

		return new WP_REST_Response([
			'success' => true,
			'code' => $code,
			'share_url' => home_url('/?ref=' . $code)
		]);
	}

	/**
	 * Update user's referral code
	 */
	public function updateReferralCode(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();
		$new_code = strtoupper(sanitize_text_field($request->get_param('code')));

		$result = JVB()->referrals()->getUserReferralCode($user_id, $new_code);

		if (is_wp_error($result)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => $result->get_error_message()
			], 400);
		}

		return new WP_REST_Response([
			'success' => true,
			'code' => $result,
			'message' => 'Referral code updated successfully'
		]);
	}

	/**
	 * Track referral click and store in session
	 */
	public function trackReferralClick(WP_REST_Request $request): WP_REST_Response
	{
		$code = strtoupper(sanitize_text_field($request->get_param('code')));

		// Start session if not already started
		if (session_status() === PHP_SESSION_NONE) {
			session_start();
		}

		// Store referral code in both session and cookie (30 day expiry)
		$_SESSION[BASE . 'referral_code'] = $code;
		setcookie(BASE . 'referral_code', $code, time() + (30 * DAY_IN_SECONDS), '/');

		return new WP_REST_Response([
			'success' => true,
			'message' => 'Referral tracked'
		]);
	}

	/**
	 * Mark referral as treated
	 */
	public function markAsTreated(WP_REST_Request $request): WP_REST_Response
	{
		$referral_id = intval($request->get_param('id'));

		$result = JVB()->referrals()->markAsTreated($referral_id, true);

		if (!$result) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'Failed to update referral'
			], 400);
		}

		return new WP_REST_Response([
			'success' => true,
			'message' => 'Referral marked as treated and rewards created'
		]);
	}

	/**
	 * Get user stats
	 */
	public function getUserStats(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();
		$stats = JVB()->referrals()->getUserStats($user_id);

		return new WP_REST_Response([
			'success' => true,
			'stats' => $stats
		]);
	}

	/**
	 * Get top referrers
	 */
	public function getTopReferrers(WP_REST_Request $request): WP_REST_Response
	{
		$period = $request->get_param('period') ?? 'week';
		$limit = $request->get_param('limit') ?? 10;

		$top_referrers = JVB()->referrals()->getTopReferrers($limit, $period);

		return new WP_REST_Response([
			'success' => true,
			'period' => $period,
			'referrers' => $top_referrers
		]);
	}

	/**
	 * Get referral settings
	 */
	public function getSettings(WP_REST_Request $request): WP_REST_Response
	{
		$settings = get_option(BASE . 'referral_settings', []);

		return new WP_REST_Response([
			'success' => true,
			'settings' => $settings
		]);
	}

	/**
	 * Update referral settings
	 */
	public function updateSettings(WP_REST_Request $request): WP_REST_Response
	{
		$settings = [
			'referrer_reward_type' => $request->get_param('referrer_reward_type') ?? 'per_user',
			'referrer_reward_amount' => floatval($request->get_param('referrer_reward_amount') ?? 25),
			'referee_reward_type' => $request->get_param('referee_reward_type') ?? 'percentage',
			'referee_reward_amount' => floatval($request->get_param('referee_reward_amount') ?? 20),
			'referee_reward_applies_to' => $request->get_param('referee_reward_applies_to') ?? 'first_order'
		];

		update_option(BASE . 'referral_settings', $settings);

		return new WP_REST_Response([
			'success' => true,
			'message' => 'Settings updated successfully',
			'settings' => $settings
		]);
	}

	/**
	 * Send a single referral invitation
	 *
	 * @param WP_REST_Request $request
	 * @return WP_REST_Response
	 */
	public function sendInvitation(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();
		$email = sanitize_email($request->get_param('email'));
		$name = sanitize_text_field($request->get_param('name'));

		// Send invitation via ReferralManager
		$referral_manager = JVB()->referrals();
		$result = $referral_manager->sendReferralInvitation($user_id, $email, $name);

		if (is_wp_error($result)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => $result->get_error_message(),
				'code' => $result->get_error_code()
			], 400);
		}

		return new WP_REST_Response($result, 200);
	}

	/**
	 * Send batch referral invitations
	 *
	 * @param WP_REST_Request $request
	 * @return WP_REST_Response
	 */
	public function sendBatchInvitations(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();
		$invitations = $request->get_param('invitations');

		// Validate invitation format
		foreach ($invitations as $invite) {
			if (empty($invite['email']) || empty($invite['name'])) {
				return new WP_REST_Response([
					'success' => false,
					'message' => 'Each invitation must have email and name'
				], 400);
			}
		}

		// Send batch via ReferralManager
		$referral_manager = JVB()->referrals();
		$result = $referral_manager->sendBatchReferralInvitations($user_id, $invitations);

		return new WP_REST_Response($result, 200);
	}

	/**
	 * Get invitation stats for current user
	 *
	 * @param WP_REST_Request $request
	 * @return WP_REST_Response
	 */
	public function getInvitationStats(WP_REST_Request $request): WP_REST_Response
	{
		$user_id = get_current_user_id();

		$referral_manager = JVB()->referrals();
		$stats = $referral_manager->getUserInvitationStats($user_id);

		return new WP_REST_Response([
			'success' => true,
			'stats' => $stats
		], 200);
	}

	/**
	 * Export referrals for Jane App cross-reference
	 * Admin only
	 *
	 * @param WP_REST_Request $request
	 * @return WP_REST_Response
	 */
	public function exportReferrals(WP_REST_Request $request): WP_REST_Response
	{
		$start_date = sanitize_text_field($request->get_param('start_date'));
		$end_date = sanitize_text_field($request->get_param('end_date'));

		$referral_manager = JVB()->referrals();
		$csv_content = $referral_manager->exportReferrals($start_date, $end_date);

		// Return CSV for download
		return new WP_REST_Response([
			'success' => true,
			'csv' => $csv_content,
			'filename' => sprintf('referrals_%s_to_%s.csv', $start_date, $end_date)
		], 200);
	}

	public function registerWithReferral(WP_REST_Request $request): WP_REST_Response
	{
		$name = sanitize_text_field($request->get_param('name'));
		$email = sanitize_email($request->get_param('email'));
		$code = strtoupper(sanitize_text_field($request->get_param('code')));

		// Validate email
		if (!is_email($email)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'Invalid email address'
			], 400);
		}

		// Check if user exists
		if (email_exists($email)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'An account with this email already exists'
			], 400);
		}

		// Validate referral code
		$referral_manager = JVB()->referrals();
		$referrer = $referral_manager->getUserByReferralCode($code);

		if (!$referrer) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'Invalid referral code'
			], 404);
		}

		// Get reward text
		$settings = $referral_manager->getRewardSettings();
		$reward_amount = $settings['referee_reward_amount'] ?? 20;
		$reward_type = $settings['referee_reward_type'] ?? 'percentage';
		$reward_text = $reward_type === 'percentage'
			? "{$reward_amount}% off your first treatment!"
			: "\${$reward_amount} off your first treatment!";

		// Send magic link with referral context via MagicLinkManager
		$magic_link_manager = new \JVBase\managers\MagicLinkManager();

		$result = $magic_link_manager->sendMagicLink(
			$email,
			\JVBase\managers\MagicLinkManager::TYPE_REFERRAL,
			[
				'name' => $name,
				'referral_code' => $code,
				'referrer_id' => $referrer->ID,
				'referrer_name' => $referrer->display_name,
				'reward_text' => $reward_text
			]
		);

		if (is_wp_error($result)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'Failed to send registration link. Please try again.'
			], 500);
		}

		return new WP_REST_Response([
			'success' => true,
			'message' => 'Check your email! We sent you a link to complete your registration.',
			'email' => $email
		], 200);
	}

	public function checkReferralCode(WP_REST_Request $request): WP_REST_Response
	{
		$code = strtoupper(sanitize_text_field($request->get_param('code')));

		if (empty($code)) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'Code is required'
			], 400);
		}

		$referral_manager = JVB()->referrals();
		$referrer = $referral_manager->getUserByReferralCode($code);

		if (!$referrer) {
			return new WP_REST_Response([
				'success' => false,
				'message' => 'Invalid referral code'
			], 404);
		}

		// Return basic referrer info (no sensitive data)
		return new WP_REST_Response([
			'success' => true,
			'code' => $code,
			'referrer_name' => $referrer->display_name,
		], 200);
	}
}