<?php
namespace JVBase\managers;

use Exception;
use JVBase\Instance;
use JVBase\integrations\services\Cloudflare;
use JVBase\meta\Form;
use JVBase\meta\Meta;
use JVBase\meta\Sanitizer;
use JVBase\rest\Rest;
use JVBase\rest\routes\ReferralRoutes;
use JVBase\ui\CRUDSkeleton;
use JVBase\ui\Tabs;
use JVBase\base\Site;
use WP_User;
use WP_Error;

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

/**
 * Referral Tracking Manager
 *
 * Handles user referral codes, tracking, rewards, and reporting.
 * Keeps referrals separate from invitations system.
 */
class Referrals
{
	use Instance;
	protected MagicLinks $magic_link;
	protected Cache $cache;
	protected Cache $requestCache;
	protected Cache $statsCache;
	protected string $referrals_table;
	protected ?int $referralPage = null;
	protected string $rewards_table;

	protected CustomTable $referrals;
	protected CustomTable $codes;
	protected CustomTable $janeClients;
	protected CustomTable $rewards;
	protected CustomTable $treatments;
	protected Rest $route;

	// Default reward settings
	protected array $default_settings = [
		'from_user_reward_applies_to' => 'per_user',  // 'per_user' or 'flat_total'
		'from_user_reward_amount' => 25.00,
		'from_user_reward_type'	=> 'fixed',
		'to_user_reward_type' => 'percentage',  // 'percentage' or 'fixed'
		'to_user_reward_amount' => 20,  // 20% or $20
		'to_user_reward_applies_to' => 'first_order',  // 'first_order' or 'all_orders'
	];

	protected string $role;

	protected array $settings;

	private function __construct()
	{
		$this->route = ReferralRoutes::getInstance();
		$this->defineTables();
		$this->role = Site::getReferralRole();
		$this->default_settings['referral_role'] = $this->role;

		$this->cache = Cache::for('referrals');

		$this->statsCache = Cache::for('referral_stats', null, WEEK_IN_SECONDS)->user();

		$this->referralPage = $this->getReferralPageId();
		$this->settings = $this->getRewardSettings();


		add_action('jvbUserRegistered', [$this, 'processRegistrationToken'], 10, 3);
		add_action('jvb_add_token_inputs', [$this, 'addLoginInputs'], 10, 1);
		add_action('user_register', [$this, 'processReferral'], 10, 1);

		// Add meta boxes for admin to manage referrals
		add_action('show_user_profile', [$this, 'displayUserReferralInfo']);
		add_action('edit_user_profile', [$this, 'displayUserReferralInfo']);

		// Save referral code changes
		add_action('personal_options_update', [$this, 'saveUserReferralCode']);
		add_action('edit_user_profile_update', [$this, 'saveUserReferralCode']);

		add_filter('jvbNewUserEmail', [$this, 'addReferralToWelcomeEmail'], 99, 2);


		add_filter('jvbAdditionalActions', [$this, 'outputShareWidget']);

		add_action('wp_enqueue_scripts', [$this, 'enqueueScripts']);
		// Schedule cron jobs for reports
		$this->registerCronJobs();

		// Add admin notice to referral page edit screen
		add_action('admin_notices', [$this, 'showReferralPageNotice']);


		add_action(BASE.'dashboard_page_referrals', [$this, 'renderDashPage'], 10);
//		add_filter('jvbDashboardPage', [$this, 'renderDashPage'], 10, 2);

		// Handle settings save
		add_action('admin_init', [$this, 'registerSettings']);
		// Handle admin page form submission
		add_filter('jvb_admin_page_submission', [$this, 'handleAdminSubmission'], 10, 3);

		add_filter(BASE.'handle_bulk_operation', [$this, 'processOperation'], 10, 3);
	}

	protected function defineTables():void
	{
		$this->defineReferralsTable();
		$this->defineCodeTable();
		$this->defineJaneClientsTable();
		$this->defineRewardsTable();
		$this->defineTreatmentsTable();
	}
		protected function defineReferralsTable():void
		{
			$table = CustomTable::for('referrals');

			$table->setColumns([
				'id'		=> 'bigint(20) unsigned NOT NULL AUTO_INCREMENT',
				'from_user'	=> "{$table->getUserIDType()} NOT NULL",		//The person doing the referring
				'to_user'	=> "{$table->getUserIDType()} DEFAULT NULL",	//The person being referred
				'to_name'	=> 'varchar(255) DEFAULT NULL',
				'to_email'	=> 'varchar(255) DEFAULT NULL',
				'to_phone'	=> 'varchar(50) DEFAULT NULL',
				'referral_code'=> 'varchar(50) NOT NULL',
				'status'	=> "ENUM('invited', 'pending', 'consulted', 'treated', 'cancelled') DEFAULT 'pending'",
				'created_at'=> 'datetime DEFAULT CURRENT_TIMESTAMP',
				'last_invited'=> 'datetime DEFAULT CURRENT_TIMESTAMP',
				'consulted_at'=> 'datetime DEFAULT NULL',
				'treated_at'=> 'datetime DEFAULT NULL',
				'treatment_count' => 'int DEFAULT 0',
				'notes'		=> 'text DEFAULT NULL',
			]);

			$table->setKeys([
				['key' => 'PRIMARY', 'value' => 'id'],
//				['key' => 'UNIQUE', 'value' => '(`to_user`)'],
				'from_user (`from_user`)',
				'to_user (`to_user`)',
				'status (`status`)',
				'code (`referral_code`)',
				'date (`created_at`)',
				'consult (`consulted_at`)'
			]);

			$base = BASE;
			$table->setConstraints([
				"CONSTRAINT `{$base}referral_from_user_fk` FOREIGN KEY (`from_user`)
				REFERENCES `{$table->getUserTable()}` (`ID`) ON DELETE CASCADE",
				"CONSTRAINT `{$base}referral_to_user_fk` FOREIGN KEY (`to_user`)
				REFERENCES `{$table->getUserTable()}` (`ID`) ON DELETE CASCADE"
			]);
			$table->defineTable();
			$this->referrals = $table;
		}

		protected function defineCodeTable():void
		{
			$table = CustomTable::for('referrals_codes');

			$table->setColumns([
				'id'		=> 'bigint(20) unsigned NOT NULL AUTO_INCREMENT',
				'user_id'	=> "{$table->getUserIDType()} NOT NULL",
				'code'		=> 'varchar(50) NOT NULL',
				'created_at'=> 'datetime DEFAULT CURRENT_TIMESTAMP',
			]);

			$table->setKeys([
				['key' => 'PRIMARY', 'value' => '(`id`)'],
				['key' => 'UNIQUE', 'value' => '(`code`)'],
				'user (`user_id`)',
			]);

			$base = BASE;
			$table->setConstraints([
				"CONSTRAINT `{$base}referral_code_user_fk` FOREIGN KEY (`user_id`)
					REFERENCES `{$table->getUserTable()}` (`ID`) ON DELETE CASCADE",
			]);
			$table->defineTable();
			$this->codes = $table;
		}
		protected function defineJaneClientsTable():void
		{
			$table = CustomTable::for('referrals_jane_clients');

			$table->setColumns([
				'id'			=> 'bigint(20) unsigned NOT NULL AUTO_INCREMENT',
				'patient_guid'	=> 'varchar(50) NOT NULL',
				'user_id'		=> "{$table->getUserIDType()} NOT NULL",
				'first_name'	=> 'varchar(100) NOT NULL',
				'last_name'		=> 'varchar(100) NOT NULL',
				'email'			=> 'varchar(255) NOT NULL',
				'imported_at'	=> 'datetime DEFAULT CURRENT_TIMESTAMP',
				'updated_at'	=> 'datetime DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP',
			]);

			$table->setKeys([
				['key' => 'PRIMARY', 'value' => '(`id`)'],
				['key' => 'UNIQUE', 'value' => '(`patient_guid`)'],
				'user (`user_id`)',
				'email (`email`)',
			]);

			$base = BASE;
			$table->setConstraints([
				"CONSTRAINT `{$base}jane_clients_user` FOREIGN KEY (`user_id`)
				REFERENCES `{$table->getUserTable()}` (`ID`) ON DELETE CASCADE",
			]);
			$table->defineTable();
			$this->janeClients = $table;
		}
		protected function defineRewardsTable():void
		{
			$table = CustomTable::for('referrals_rewards');

			$table->setColumns([
				'id'				=> 'bigint(20) unsigned NOT NULL AUTO_INCREMENT',
				'referral_id'		=> 'bigint(20) unsigned NOT NULL',
				'user_id'			=> "{$table->getUserIDType()} NOT NULL",
				'reward_type'		=> "ENUM('from_user', 'to_user') NOT NULL",
				'amount'			=> 'decimal(10,2) NOT NULL',
				'reward_calculation'=> "ENUM('percentage', 'fixed')",
				'status'			=> "ENUM('available', 'redeemed', 'expired', 'cancelled') DEFAULT 'available'",
				'created_at'		=> 'datetime NOT NULL DEFAULT CURRENT_TIMESTAMP',
				'updated_at'		=> 'datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP',
				'redeemed_at'		=> 'datetime DEFAULT NULL',
				'expires_at'		=> 'datetime DEFAULT NULL',
				'notes'				=> 'text DEFAULT NULL',
			]);

			$table->setKeys([
				['key' => 'PRIMARY', 'value' => '(`id`)'],
				'referral (`referral_id`)',
				'user (`user_id`)',
				'status (`status`)',
				'type (`reward_type`)'
			]);

			$base = BASE;
			$table->setConstraints([
				"CONSTRAINT `{$base}reward_referral` FOREIGN KEY (`referral_id`)
				REFERENCES `{$this->referrals->getFullTableName()}` (`id`) ON DELETE CASCADE",
				"CONSTRAINT `{$base}reward_user` FOREIGN KEY (`user_id`)
				REFERENCES `{$table->getUserTable()}` (`ID`) ON DELETE CASCADE"
			]);
			$table->defineTable();
			$this->rewards = $table;
		}
		protected function defineTreatmentsTable():void
		{
			$table = CustomTable::for('referral_treatments');

			$table->setColumns([
				'id'			=> 'bigint(20) unsigned NOT NULL AUTO_INCREMENT',
				'referral_id'	=> 'bigint(20) unsigned NOT NULL',
				'user_id'		=> "{$table->getUserIDType()} NOT NULL",
				'treatment_type'=> 'varchar(100) NOT NULL',		//Tier 1-6, Brows, etc
				'treatment_date'=> 'datetime NOT NULL',
				'invoice_number'=> 'varchar(50) DEFAULT NULL',
				'amount'		=> 'decimal(10,2) DEFAULT NULL',
				'status'		=> "ENUM('completed', 'no_show', 'cancelled') DEFAULT 'completed'",
				'imported_at'	=> 'datetime DEFAULT CURRENT_TIMESTAMP',
			]);

			$table->setKeys([
				['key' => 'PRIMARY', 'value' => '(`id`)'],
				'referral (`referral_id`)',
				'user (`user_id`)',
				'date (`treatment_date`)',
				'type (`treatment_type`)',
			]);

			$base = BASE;
			$table->setConstraints([
				"CONSTRAINT `{$base}treatment_referral` FOREIGN KEY (`referral_id`)
				REFERENCES `{$this->referrals->getFullTableName()}` (`id`) ON DELETE CASCADE",
				"CONSTRAINT `{$base}treatment_user` FOREIGN KEY (`user_id`)
				REFERENCES `{$table->getUserTable()}` (`ID`) ON DELETE CASCADE"
			]);
			$table->defineTable();
			$this->treatments = $table;
		}

	public function getSettings():array
	{
		return $this->settings;
	}
	public function getRole():string
	{
		return $this->role;
	}
	public function addLoginInputs(string $action):void
	{
		if (array_key_exists('ref', $_GET)) {
			echo '<input type="hidden" name="referral_code" value="'.$_GET['ref'].'">';
		}
	}

	public function modifyLoginText(array $defaults):array
	{
		if (!array_key_exists('ref', $_GET)){
			return $defaults;
		}

		$code = $_GET['ref'];
		$user = $this->getUserByReferralCode($code);

		$desc = ($user) ? strtok($user->display_name, ' ') . ' invited you ' : 'You\'ve been invited ';

		$defaults['title'] = 'Register your Account. Get Your Reward.';
		$defaults['description'] = [
			$desc.' to see the difference with us.',
			'Oh, and you\'ll get 20% off your first treatment!',
			'Finish this account creation (you\'ll still need to make an account on Jane App to book), or let us know when you come in for your appointment.'
		];
		$defaults['extra'] = [
			'Once you\'re in, you\'ll get your own code you can share.',
			'Get a $25 credit for <b>each</b> person you send our way who comes in for their first treatment!'
		];
		return $defaults;
	}

	public function enqueueScripts():void
	{
		$requirements = [
			'jvb-utility',
			'jvb-a11y',
			'jvb-popup',
			'jvb-tabs',
			'jvb-data-store',
		];

		if (Site::hasIntegration('cloudflare') && Cloudflare::getInstance()->isSetUp()) {
			Cloudflare::getInstance()->enqueueTurnstileScripts();
		}
		if (is_singular(BASE.'dash')) {
			$requirements[] = 'jvb-form';
			$requirements[] = 'jvb-view';

			wp_enqueue_script('jvb-referral-admin',
			JVB_URL.'assets/js/min/referralAdmin.min.js',
			['jvb-referral'],
			'1.0.0',
			true);
		}
		wp_enqueue_script(
			'jvb-referral',
			JVB_URL . 'assets/js/min/referral.min.js',
			$requirements,
			'1.0.0',
			true
		);
	}

	/**
	 * Register cron jobs for automated reporting
	 */
	protected function registerCronJobs(): void
	{
		add_action(BASE . 'referral_daily_report', [$this, 'sendDailyReport']);
		add_action(BASE . 'referral_weekly_report', [$this, 'sendWeeklyReport']);

		// Schedule if not already scheduled
		if (!wp_next_scheduled(BASE . 'referral_daily_report')) {
			wp_schedule_event(strtotime('tomorrow 9:00'), 'daily', BASE . 'referral_daily_report');
		}

		if (!wp_next_scheduled(BASE . 'referral_weekly_report')) {
			wp_schedule_event(strtotime('next Monday 9:00'), 'weekly', BASE . 'referral_weekly_report');
		}
	}

	public function createCode(int $user_id, string $code):string|false
	{
		$code = sanitize_title($code);
		$existing = $this->codes->get(['code' => $code]);
		if ($existing) {
			if ($existing['user_id'] !== $user_id) {
				return false;
			}
			return $code;
		}
		$success = $this->codes->insert([
			'user_id' 	=> $user_id,
			'code'		=> $code
		]);
		if ($success) {
			return $code;
		}
		return false;
	}

	/**
	 * Generate or get existing referral code for a user
	 *
	 * @param int $user_id
	 * @param string|null $custom_code Optional custom code
	 * @return string|WP_Error
	 */
	public function getUserReferralCode(int $user_id, ?string $custom_code = null):string|false
	{
		$user = get_userdata($user_id);
		if (!$user) {
			return false;
		}

		$existing = $this->codes->pluck('code', ['user_id' => $user_id],'created_at', 'DESC');
		if (!empty($existing) && !$custom_code) {
			return $existing[0];
		}
		if ($custom_code && !empty($existing) && !in_array($custom_code, $existing)) {
			$test = $this->createCode($user_id, $custom_code);
			if ($test) {
				return $this->codes->pluck('code', ['user_id' => $user_id], 'created_at', 'DESC')[0];
			} else {
				return $existing[0];
			}
		}

		// Generate new code if custom provided or none exists
		$code = $this->generateReferralCode($user);

		$success = $this->createCode($user_id, $code);
		if ($success) {
			return $success;
		}

		return false;
	}

	/**
	 * Generate a referral code based on user info
	 *
	 * @param WP_User $user
	 * @return string
	 */
	protected function generateReferralCode(WP_User $user): string
	{
		// Create code from first and last name
		$first = sanitize_title($user->first_name ?: 'user');
		$last = sanitize_title($user->last_name ?: wp_generate_password(4, false));

		$base_code = strtoupper(substr($first, 0, 4) . substr($last, 0, 4));

		// Ensure uniqueness
		$code = $base_code;
		$suffix = 1;

		while ($this->isCodeTaken($code)) {
			$code = $base_code . $suffix;
			$suffix++;
		}

		return $code;
	}

	/**
	 * Check if a referral code is already taken
	 *
	 * @param string $code
	 * @return bool
	 */
	protected function isCodeTaken(string $code): bool
	{
		return (bool) $this->codes->get(['code' => $code]);
	}

	public function processRegistrationToken(int $user_id, string $email, array $data): void
	{
		// Check for referral code in data
		$code = $data['referral_code'] ?? '';
		if (empty($code)) {
			return;
		}

		// Store in session/cookie for processReferral to pick up
		if (session_status() === PHP_SESSION_NONE) {
			session_start();
		}
		$_SESSION[BASE . 'referral_code'] = sanitize_text_field($code);
		setcookie(
			BASE . 'referral_code',
			sanitize_text_field($code),
			time() + (86400 * 30),
			'/'
		);
	}
	/**
	 * Track a new referral when user registers
	 *
	 * @param int $user_id
	 * @param array $userData
	 * @return void
	 */
	public function processReferral(int $user_id, array $userData):void
	{

		if (!array_key_exists('ref', $_POST) || empty($_POST['ref'])) {
			return;
		}
		$code = sanitize_text_field($_POST['ref']);
		$referredBy = $this->codes->pluck('user_id', ['code' => $code]);
		if (!empty($referredBy)) {
			$this->createReferralFromID($referredBy[0], $user_id, $code);
		}
	}

	/**
	 * Create a referral record in the database
	 *
	 * @param int $from_user_id The person sending the referral
	 * @param string $name  The person who was referred
	 * @param string $email The person who was referred's email
	 * @param ?string $phone The person who was referred's phone (optional)
	 * @param ?string $code The person sending the referral's custom code
	 * @return array
	 */
	public function createReferralFromFields(int $from_user_id, string $name, string $email, ?string $phone = null, ?string $code = null):array
	{
		if (!get_userdata($from_user_id)) {
			return [
				'success'	=> false,
				'message'	=> 'From user does not exist',
			];
		}

		if (filter_var($email, FILTER_VALIDATE_EMAIL)) {
			if(email_exists($email)) {
				return [
					'success'	=> false,
					'message'	=> 'Email already exists',
				];
			}
		}
		if ($phone && !empty($this->referrals->pluck('to_phone', ['to_phone' => Sanitizer::sanitizePhone($phone)]))) {
			return [
				'success'	=> false,
				'message'	=> 'Phone already used'
			];
		}
		$insert = [
			'from_user'		=> $from_user_id,
			'referral_code'	=> is_null($code)? $this->getUserReferralCode($from_user_id) : sanitize_text_field($code),
			'to_name'		=> $name,
			'to_email'		=> $email,
			'status'		=> 'invited',
		];
		if ($phone) {
			$insert['to_phone'] = $phone;
		}

		$created = $this->referrals->insert($insert);
		$this->statsCache->forget($from_user_id);
		return [
			'success'	=> (bool) $created,
			'message'	=> 'Referral created'
		];
	}
	/**
	 * Create a referral record in the database
	 *
	 * @param int $from_user_id The person sending the referral
	 * @param int $to_user_id The person who was referred
	 * @param ?string $code Custom code, if not default
	 * @return array
	 */
	public function createReferralFromID(int $from_user_id, int $to_user_id, ?string $code = null):array
	{
		$user = get_userdata($to_user_id);

		$code = !is_null($code) ? sanitize_text_field($code) : $this->getUserReferralCode($to_user_id);

		$where = [
			'to_email'	=> $user->user_email,
			'to_name'	=> $user->first_name ?: $user->display_name
		];
		$phone = false;
		if (array_key_exists('phone', $_POST) && !empty($_POST['phone'])) {
			$phone = Sanitizer::sanitizePhone($_POST['phone']);
			if (!empty($phone)) {
				$where = [
					'to_phone'	=> $phone,
				];
			}
		} elseif (!empty(Meta::forUser($to_user_id)->get('phone'))) {
			$phone = Meta::forUser($to_user_id)->get('phone');
		}
		$where = [
			'OR'	=> $where,
			'status' => 'invited'
		];

		$update = [
			'from_user'	=> $from_user_id,
			'to_user'	=> $to_user_id,
			'to_name'	=> $user->first_name ?: $user->display_name,
			'to_email'	=> $user->user_email,
			'status'	=> 'pending',
			'referral_code'	=> $code
		];
		if ($phone) {
			$update ['to_phone'] = $phone;
		}
		$existing = $this->referrals->get($where);
		if ($existing) {
			if ($existing->from_user != $from_user_id) {
				$this->referrals->update([
					'status'	=> 'cancelled',
				], $where);

			} else {
				$updated = $this->referrals->update($update, $where);
				$this->sendReferrerNotification($from_user_id, $update['to_name']);
				return [
					'success'	=> (bool) $updated,
					'message'	=> $updated ? 'Successfully created referral' : 'Referral could not be updated'
				];
			}
		}
		$created = $this->referrals->insert($update);

		$this->statsCache->forget($from_user_id);
		return [
			'success'	=> (bool) $created,
			'message'	=> $created ? 'Successfully created referral' : 'Referral could not be added'
		];
	}

	/**
	 * Get user by referral code
	 *
	 * @param string $code
	 * @return WP_User|null
	 */
	public function getUserByReferralCode(string $code): ?WP_User
	{
		return $this->cache->remember(
			$code,
			function () use ($code) {
				$users = get_users([
					'meta_key' => BASE . 'referral_code',
					'meta_value' => $code,
					'number' => 1
				]);
				return $users[0] ?? null;
			}
		);
	}

	/**
	 * Get referral record by to_user ID
	 *
	 * @param int $to_user_id
	 * @return object|null
	 */
	public function getReferralByReferee(int $to_user_id): ?object
	{
		$result = $this->referrals->get([
			'to_user'	=> $to_user_id
		]);

		return $result ?: null;
	}

	/**
	 * Mark a referral as treated/rewarded
	 *
	 * @param int $referral_id
	 * @param bool $treated
	 * @return bool
	 */
	public function markAsTreated(int $referral_id, bool $treated = true): bool
	{
		$status = $treated ? 'treated' : 'pending';

		$result = $this->referrals->update(
			[
				'status'	=> $status,
				'treated_at' => $treated ? current_time('mysql') : null
			],
			[
				'id'	=> $referral_id
			]
		);

		if ($result && $treated) {
			// Create reward records when marking as treated
			$this->createRewardRecords($referral_id);
		}

		return $result !== false;
	}

	/**
	 * Create reward records for both from_user and to_user
	 *
	 * @param int $referral_id
	 */
	protected function createRewardRecords(int $referral_id): void
	{
		$referral = $this->referrals->get(['id' => $referral_id]);

		if (!$referral) {
			return;
		}

		// Create from_user reward
		$fromUserReward = $this->rewards->insert([
			'referral_id'	=> $referral_id,
			'user_id'		=> $referral->from_user,
			'reward_type'	=> 'from_user',
			'amount'		=> $this->settings['from_user_reward_amount'],
			'reward_calculation' => $this->settings['from_user_reward_type']
		]);


		// Create to_user reward
		$to_user_amount = $this->settings['to_user_reward_type'] === 'percentage'
			? $this->settings['to_user_reward_amount']  // Store as percentage
			: $this->settings['to_user_reward_amount'];  // Store as fixed amount

		$toUserReward = $this->rewards->insert([
			'referral_id'	=> $referral_id,
			'user_id'		=> $referral->to_user,
			'reward_type'	=> 'to_user',
			'amount'		=> $to_user_amount,
			'reward_calculation' => $this->settings['to_user_reward_type']
		]);

	}

	/**
	 * Get referrals for a user
	 *
	 * @param int $user_id
	 * @param array $args
	 * @return array
	 */
	public function getUserReferrals(int $user_id, array $args = []): array
	{
		$defaults = [
			'status' => 'all',
			'per_page' => 100,
			'page' => 1,
			'orderby' => 'created_at',
			'order' => 'DESC'
		];

		$where = [
			'from_user'	=> $user_id,
		];

		foreach ($this->referrals->getColumnKeys() as $col) {
			if (array_key_exists($col, $args)) {
				if ($col === 'status' && $args[$col] !== 'all') {
					$where[$col] = $args[$col];
				} else {
					$where[$col] = $args[$col];
				}
			}
		}


		$args = wp_parse_args($args, $defaults);
		$args['status'] = strtolower($args['status']);
		$tableArgs = [
			'where' 	=> $where,
			'per_page'		=> $args['per_page'],
			'page'		=> $args['page'],
			'orderby'	=> $args['orderby'],
			'order'		=> $args['order']
		];

		if ($args['status'] !== 'all'){
			$tableArgs['status'] = $args['status'];
		}
		$referrals = $this->referrals->getMany($tableArgs, false);

		return array_map(function($referral) {
			$can_resend = $this->checkCanResendEmail($referral->id);
			$status = match($referral->status) {
				'consulted' => 'Awaiting Treatment',
				'treated'	=> 'Rewarded!',
				'invited'	=> 'Invited, but has not used',
				default => 'Pending',
			};
			return [
				'id'			=> $referral->id,
				'to_name'	=> $referral->to_name,
				'to_email'	=> $referral->to_email,
				'referred_at'	=> ReferralRoutes::getInstance()->formatTimestamp($referral->referred_at),
				'referral_status'=> $status,
				'can_resend'	=> $can_resend
			];
		}, $referrals);

	}

	protected function checkCanResendEmail(int|object $referral): bool
	{
		if (is_int($referral)) {
			$referral = $this->referrals->get(['id'=>$referral]);
			if (!$referral) {
				return false;
			}
		}
		return (time() - $referral->updated_at) > WEEK_IN_SECONDS;
	}

	/**
	 * Get referral statistics for a user
	 *
	 * @param int $user_id
	 * @return array
	 */
	public function getUserStats(int $user_id): array
	{
		return $this->statsCache->remember(
			$user_id,
			function() use ($user_id) {
				$stats = [
					'total'		=> $this->referrals->count(['user_id' => $user_id]),
					'invited'	=> $this->referrals->count(['user_id' => $user_id, 'status' => 'invited']),
					'pending' 	=> $this->referrals->count(['user_id' => $user_id, 'status' => 'pending']),
					'consulted'	=>$this->referrals->count(['user_id' => $user_id, 'status' => 'consulted']),
					'treated' 	=> $this->referrals->count(['user_id' => $user_id, 'status' => 'treated']),
					'total_rewards'	=> $this->rewards->sum('amount', ['user_id' => $user_id]),
					'available_rewards'	=> $this->rewards->sum('amount', ['user_id' => $user_id, 'redeemed_at' => ['NOT NULL', null]]),
				];
				$stats['user_id'] = $user_id;
				return $stats;
			}
		);
	}

	/**
	 * Get top from_users for a time period
	 *
	 * @param int $limit
	 * @param string $period 'day'|'week'|'month'|'all'
	 * @return array
	 */
	public function getTopReferrers(int $limit = 10, string $period = 'all'): array
	{
		return $this->statsCache->remember(
			['per_page' => $limit, 'period' => $period],
			function() use ($limit, $period) {
				$where = '';
				if ($period !== 'all') {
					$date_clause = match($period) {
						'day'   => "created_at >= DATE_SUB(NOW(), INTERVAL 1 DAY)",
						'week'  => "created_at >= DATE_SUB(NOW(), INTERVAL 1 WEEK)",
						'month' => "created_at >= DATE_SUB(NOW(), INTERVAL 1 MONTH)",
						default => "1=1"
					};
					$where = "WHERE {$date_clause}";
				}

				$results = $this->referrals->queryResults(
					"SELECT
                    from_user,
                    COUNT(*) as referral_count,
                    SUM(CASE WHEN status = 'treated' THEN 1 ELSE 0 END) as treated_count
                FROM {table}
                {$where}
                GROUP BY from_user
                ORDER BY referral_count DESC
                LIMIT {$limit}"
				);

				foreach ($results as &$result) {
					$user = get_user_by('ID', $result->from_user);
					$result->user_name  = $user ? $user->display_name : 'Unknown';
					$result->user_email = $user ? $user->user_email   : '';
				}

				return $results;
			}
		);
	}

	/**
	 * Send daily report if there are new referrals
	 */
	public function sendDailyReport(): void
	{
		$yesterday = date('Y-m-d', strtotime('-1 day'));

		$new_referrals = $this->referrals->queryResults(
			"SELECT {table}.*, u.display_name as from_user_name
        FROM {table}
        JOIN {$this->referrals->getUserTable()} u ON {table}.from_user = u.ID
        WHERE DATE({table}.created_at) = %s
        ORDER BY {table}.created_at DESC",
			[$yesterday]
		);

		if (empty($new_referrals)) {
			return;
		}

		$content = Emails::getInstance()->h1('Daily Referral Report');
		$content .= Emails::getInstance()->stat(
			count($new_referrals),
			count($new_referrals) === 1 ? 'New Referral' : 'New Referrals',
			'From ' . $yesterday
		);

		$content .= Emails::getInstance()->spacer();
		$content .= Emails::getInstance()->h2('New Referrals');

		// Build list of referrals
		foreach ($new_referrals as $ref) {
			$cardContent = sprintf(
				'<p><strong>%s</strong> (%s)</p>',
				esc_html($ref->to_name),
				esc_html($ref->to_email)
			);
			$cardContent .= sprintf(
				'<p style="font-size:13px;color:%s;">Referred by: %s | Code: %s</p>',
				Emails::getInstance()->colours['dark-200'],
				esc_html($ref->from_name),
				Emails::getInstance()->badge($ref->referral_code, 'info')
			);

			$content .= Emails::getInstance()->card($cardContent);
		}

		$to = get_option('admin_email');
		$subject = sprintf(
			'[%s] %d New Referral%s',
			get_bloginfo('name'),
			count($new_referrals),
			count($new_referrals) !== 1 ? 's' : ''
		);

		Emails::getInstance()->sendEmail($to, $subject, $content, 'DAILY REPORT');
	}

	/**
	 * Send weekly report with top from_users
	 */
	public function sendWeeklyReport(): void
	{
		$total_referrals = $this->referrals->count(['created_at' => ['>=', "DATE_SUB(NOW(), INTERVAL 1 WEEK)"]]);

		if ($total_referrals == 0) {
			return;
		}
		$top_from_users = $this->getTopReferrers(10, 'week');

		$content = Emails::getInstance()->h1('Weekly Referral Summary');
		$content .= Emails::getInstance()->stat(
			$total_referrals,
			'Total Referrals',
			'This week'
		);

		$content .= Emails::getInstance()->spacer(30);
		$content .= Emails::getInstance()->h2('Top 10 Referrers');

		// Leaderboard style
		$rank = 1;
		foreach ($top_from_users as $from_user) {
			$rankBadge = $rank <= 3
				? Emails::getInstance()->badge('#' . $rank, $rank === 1 ? 'success' : 'info')
				: '<span style="font-weight:600;color:' . Emails::getInstance()->colours['dark-200'] . ';">#' . $rank . '</span>';

			$cardContent = sprintf(
				'<p>%s <strong>%s</strong></p>',
				$rankBadge,
				esc_html($from_user->user_name)
			);

			$stats = [
				Emails::getInstance()->stat($from_user->referral_count, 'Total Referrals'),
				Emails::getInstance()->stat($from_user->treated_count, 'Treated')
			];
			$cardContent .= Emails::getInstance()->grid($stats);

			$content .= Emails::getInstance()->card($cardContent);
			$rank++;
		}

		$to = get_option('admin_email');
		$subject = '[' . get_bloginfo('name') . '] Weekly Referral Summary - ' . date('F j, Y');

		Emails::getInstance()->sendEmail($to, $subject, $content, 'WEEKLY SUMMARY');
	}

	/**
	 * Generate CSV content from referrals
	 *
	 * @param array $referrals
	 * @return string
	 */
	protected function generateCSV(array $referrals): string
	{
		$cache = Cache::for('referralCSV');
		return $cache->remember(
			'csv',
			function () use ($referrals) {
				$csv = "Referred By,Referee Name,Referee Email,Referee Phone,Referral Code,Status,Referred At,Treated At\n";

				foreach ($referrals as $referral) {
					$csv .= sprintf(
						'"%s","%s","%s","%s","%s","%s","%s","%s"' . "\n",
						$referral->from_name ?? 'Unknown',
						$referral->to_name,
						$referral->to_email,
						$referral->to_phone,
						$referral->referral_code,
						$referral->status,
						$referral->referred_at,
						$referral->treated_at ?? 'Not yet'
					);
				}

				return $csv;
			}
		);

	}

	/**
	 * Generate HTML email for weekly report
	 *
	 * @param array $top_from_users
	 * @param int $total_referrals
	 * @return string
	 */
	protected function generateWeeklyReportEmail(array $top_from_users, int $total_referrals): string
	{
		$content = sprintf(
			'<p>This week you had <strong>%d total referral%s</strong>.</p>',
			$total_referrals,
			$total_referrals !== 1 ? 's' : ''
		);

		$from_users = [];
		$rank = 1;
		foreach ($top_from_users as $from_user) {
			$from_users[] = [
				'label' => '#' . $rank++ . ' - ' . esc_html($from_user->user_name),
				'value' => sprintf(
					'<strong>Total Referrals:</strong> %d | <strong>Treated:</strong> %d',
					$from_user->referral_count,
					$from_user->treated_count
				)
			];
		}

		$content .= Emails::getInstance()->table($from_users, 'Top 10 Referrers This Week');

		return $content;
	}

	/**
	 * Get reward settings
	 *
	 * @return array
	 */
	public function getRewardSettings(): array
	{
		$saved = get_option(BASE . 'referral_settings', []);
		return wp_parse_args($saved, $this->default_settings);
	}

	/**
	 * Display referral info in user profile
	 *
	 * @param WP_User $user
	 */
	public function displayUserReferralInfo(WP_User $user): void
	{
		if (!current_user_can('edit_user', $user->ID)) {
			return;
		}

		$referral_code = get_user_meta($user->ID, BASE . 'referral_code', true);
		$stats = $this->getUserStats($user->ID);
		$referrals = $this->getUserReferrals($user->ID, ['per_page' => 10]);

		?>
		<h2>Referral Information</h2>
		<table class="form-table">
			<tr>
				<th><label for="referral_code">Referral Code</label></th>
				<td>
					<input type="text"
						   name="referral_code"
						   id="referral_code"
						   value="<?php echo esc_attr($referral_code); ?>"
						   class="regular-text" />
					<p class="description">
						Users can sign up with this code.
						<?php if ($referral_code): ?>
							Share link: <?php echo home_url('/?ref=' . $referral_code); ?>
						<?php endif; ?>
					</p>
				</td>
			</tr>
		</table>

		<h3>Referral Statistics</h3>
		<table class="form-table">
			<tr>
				<th>Total Referrals:</th>
				<td><?php echo $stats['total_referrals'] ?? 0; ?></td>
			</tr>
			<tr>
				<th>Treated:</th>
				<td><?php echo $stats['treated_count'] ?? 0; ?></td>
			</tr>
			<tr>
				<th>Pending:</th>
				<td><?php echo $stats['pending_count'] ?? 0; ?></td>
			</tr>
			<tr>
				<th>Available Rewards:</th>
				<td>$<?php echo number_format($stats['available_rewards'] ?? 0, 2); ?></td>
			</tr>
			<tr>
				<th>Redeemed Rewards:</th>
				<td>$<?php echo number_format($stats['redeemed_rewards'] ?? 0, 2); ?></td>
			</tr>
		</table>

		<?php if (!empty($referrals)): ?>
		<h3>Recent Referrals</h3>
		<table class="widefat">
			<thead>
			<tr>
				<th>Name</th>
				<th>Email</th>
				<th>Status</th>
				<th>Referred At</th>
				<th>Actions</th>
			</tr>
			</thead>
			<tbody>
			<?php foreach ($referrals as $referral): ?>
				<tr>
					<td><?php echo esc_html($referral->to_name); ?></td>
					<td><?php echo esc_html($referral->to_email); ?></td>
					<td><?php echo esc_html(ucfirst($referral->status)); ?></td>
					<td><?php echo esc_html($referral->referred_at); ?></td>
					<td>
						<?php if ($referral->status === 'pending'): ?>
							<button type="button"
									onclick="markReferralTreated(<?php echo $referral->id; ?>)">
								Mark as Treated
							</button>
						<?php endif; ?>
					</td>
				</tr>
			<?php endforeach; ?>
			</tbody>
		</table>
	<?php endif; ?>

		<?php /**
		<script>
			function markReferralTreated(referralId) {
				if (!confirm('Mark this referral as treated? This will create reward records.')) {
					return;
				}

				fetch('<?php echo rest_url(BASE . '/v1/referrals/' ); ?>' + referralId + '/treat', {
					method: 'POST',
					headers: {
						'X-WP-Nonce': '<?php echo wp_create_nonce('wp_rest'); ?>'
					}
				})
					.then(response => response.json())
					.then(data => {
						if (data.success) {
							alert('Referral marked as treated!');
							location.reload();
						} else {
							alert('Error: ' + (data.message || 'Unknown error'));
						}
					});
			}
		</script>
		<?php
	 */
	}

	/**
	 * Save custom referral code
	 *
	 * @param int $user_id
	 */
	public function saveUserReferralCode(int $user_id): void
	{
		if (!current_user_can('edit_user', $user_id)) {
			return;
		}

		if (!isset($_POST['referral_code'])) {
			return;
		}

		$code = sanitize_text_field($_POST['referral_code']);

		if (empty($code)) {
			delete_user_meta($user_id, BASE . 'referral_code');
			return;
		}

		// Validate code format (alphanumeric only)
		if (!preg_match('/^[A-Z0-9]+$/i', $code)) {
			add_action('user_profile_update_errors', function($errors) {
				$errors->add('invalid_referral_code',
					'Referral code can only contain letters and numbers.');
			});
			return;
		}

		// Check if code is unique
		if ($this->isCodeTaken($code)) {
			add_action('user_profile_update_errors', function($errors) {
				$errors->add('referral_code_taken',
					'This referral code is already in use.');
			});
			return;
		}

		update_user_meta($user_id, BASE . 'referral_code', strtoupper($code));
	}


	/**
	 * Display user's referral code and share options
	 * Use in templates or dashboard with: echo jvbReferralShareWidget();
	 *
	 * @return string HTML output
	 */
	public function outputShareWidget(array $actions):array
	{

		$user_id = get_current_user_id();
		$content = '<aside class="main referral right">';
		if (!$user_id) {
			$content .= $this->getUnloggedInReferral();
		} else {
			$content .= $this->getLoggedInReferral($user_id);
		}
		$content .= '</aside>';

		$actions[] =[
			'button' => '<button type="button" class="attn toggle-referral row" title="Your Referrals" data-action="toggle-referral" aria-label="Open Referral Sidebar" aria-controls="referral" aria-expanded="false">
					'.jvbIcon('hand-heart').'<span class="screen-reader-text"></span>
				</button>',
			'content'	=> $content
		];

		return $actions;
	}

	/**
	 * Display referral sidebar for non-logged-in users
	 */
	function getUnloggedInReferral(): string
	{
		ob_start();
		Cloudflare::getInstance()->renderTurnstile();
		$turnstile = ob_get_clean();

		$reward_text = $this->getRewardText();

		// Pre-fill code if from referral link
		$prefill_code = $_GET['ref'] ?? '';
		$from_user_name = '';
		if ($prefill_code) {
			$from_user = $this->getUserByReferralCode($prefill_code);
			$from_user_name = $from_user ? strtok($from_user->display_name, ' ') : '';
		}

		$header = sprintf(
			'<header><h2>%sGet %s.</h2></header><h3>Have a code?</h3>%s<p>Enter your referral code to get started!</p>',
			jvbIcon('confetti'),
			esc_html($reward_text),
			($from_user_name ? '<p>' . esc_html($from_user_name) . ' invited you to join us</p>' : '')
		);

		$codeForm = sprintf(
			'<form id="referral-code-form">
				%s
   	 			<input type="hidden" name="user_select" value="%s">
   	 			%s%s%s%s
   	 			<div class="row">
					<button type="button" class="button-secondary check-code-btn">
						%s Verify Code
					</button>
					<button type="submit">
						Get Started
					</button>
				</div>

				<div class="code-status" hidden></div>

				<p class="hint">
					We\'ll send you a link to complete your registration.
				</p>
			</form>
			<div class="success-content" hidden>
				<h3>Check Your Email!</h3>
				<p>We\'ve sent you a magic link to complete your registration. Click the link to activate your account and claim your reward!</p>
				<p class="hint">Can\'t find it? Check your spam folder.</p>
			</div>',
			jvbFormStatus(),
			esc_attr(Site::getReferralRole()),
			Form::render('referral_name', '', [
				'required'	=> true,
				'type'		=> 'text',
				'label'		=> 'Your Name',
				'placeholder'=> 'Mister Meeseeks',
				'autocomplete'=>'name'
			]),
			Form::render('referral_email', '', [
				'required'	=> true,
				'type'		=> 'email',
				'label'		=> 'Your Email',
				'placeholder'=> 'look@me.com',
				'autocomplete'=> 'email'
			]),
			Form::render('referral_code', $prefill_code, [
				'required'	=> true,
				'type'		=> 'text',
				'label'		=> 'Referral Code',
				'pattern'	=> '[A-Za-z0-9]+',
				'maxLength'	=> 20,
				'autocomplete'=>'off',
				'data-from-user' => $from_user_name
			]),
			$turnstile,
			jvbIcon('check-circle')
		);

		$loginHeader = sprintf(
			'<header><h2>%sLogin</h2></header><p>Already have an account?<br>Log in to see your rewards!</p>',
			jvbIcon('sign-in')
		);
		$loginForm = sprintf(
			'<form id="login-form">%s%s%s
			<button type="submit">%sLogin With Magic Link</button>
		</form>

		<div class="success-content" hidden>
			<h3>Check Your Email!</h3>
			<p>We\'ve sent you a magic link to log in - no password required! Click the link in your email to log in.</p>
			<p class="hint">Can\'t find it? Check your spam folder.</p>
		</div>',
			jvbFormStatus(),
		Form::render('login_email', null, [
			'required'	=> true,
			'type'		=> 'email',
			'label'		=> 'Your Email',
			'autocomplete'=>'email'
		]),
		$turnstile,
			jvbIcon('magic-wand')
		);


		$footer = sprintf(
			'<p class="hint">
			<a href="%s" class="text-link">Prefer to use a password?</a>
	</p>',
			wp_login_url()
		);
		$tabs = [
			'enterCode' => [
				'title'	=> 'Have a Code?',
				'description'	=> [
					'Enter your referral code to get started'
				],
				'header' => $header,
				'content'	=> $codeForm
			],
			'login'	=> [
				'header'	=> $loginHeader,
				'title'		=> 'Login',
				'description'	=> [
					'Already have an account? Log in to see your rewards'
				],
				'content'	=> $loginForm.$footer
			]
		];


		return jvbRenderTabs($tabs, true);
	}


	protected function getReferralSuccessMessage(string $code): string
	{
		$from_user = $this->getUserByReferralCode($code);

		if (!$from_user) {
			return '';
		}

		$reward_amount = $this->settings['to_user_reward_amount'] ?? 20;
		$reward_type = $this->settings['to_user_reward_type'] ?? 'percentage';

		$reward_text = $reward_type === 'percentage'
			? $reward_amount . '% off'
			: '$' . number_format($reward_amount, 2) . ' off';

		$booking_url = apply_filters('jvb_referral_booking_url', home_url('/contact'));

		ob_start();
		?>
		<div class="success-icon">
			✓
		</div>

		<div class="success-content">
			<h3>Success! Your Reward is Ready!</h3>

			<div class="reward-highlight">
				<p style="margin: 0 0 8px 0;">You'll receive:</p>
				<p style="margin: 0;"><strong><?php echo esc_html($reward_text); ?> your first treatment!</strong></p>
			</div>

			<p>Your referral code <strong><?php echo esc_html($code); ?></strong> has been applied. Book your free consultation now to claim your reward!</p>

			<a href="<?php echo esc_url($booking_url); ?>" class="cta-button">
				Book Your Free Consultation
			</a>

			<div class="referred-by">
				Referred by <strong><?php echo esc_html($from_user->display_name); ?></strong>
			</div>
		</div>
		<?php
		return ob_get_clean();
	}

	/**
	 * Display referral sidebar for logged-in users
	 */
	public function getLoggedInReferral(int $user_id): string
	{
		$referral_code = $this->getUserReferralCode($user_id);
		if (!$referral_code) {
			return '';
		}

		$share_url = $this->getShareURL($referral_code);
		ob_start();
		?>
		<div class="wrap">
		<header>
			<h3>Share the ♡</h3>
			<p>Invite friends. Earn rewards.</p>
		</header>

		<?php $this->getShareButtons($user_id); ?>

		<section class="copy">
			<h4>Your Referral Link</h4>
			<div class="copy-group row x-btw nowrap">
				<code id="referral-link" class="copy-target"><?= esc_url($share_url) ?></code>
				<button type="button" class="copy-btn" data-target="referral-link" title="Copy referral link">
					<?= jvbIcon('copy'); ?>
					<?= jvbIcon('check-circle'); ?>
				</button>
			</div>
			<p class="hint">Quickest and easiest: autofills your code.</p>


			<h4>Your Code</h4>
			<div class="copy-group row x-btw nowrap">
				<code id="referral-code" class="copy-target"><?= esc_html($referral_code) ?></code>
				<button type="button" class="copy-btn" data-target="referral-code" title="Copy referral code">
					<?= jvbIcon('copy'); ?>
					<?= jvbIcon('check-circle'); ?>
				</button>
			</div>
			<p class="hint">Manually copy and paste the code</p>

		</section>

		<section class="recent-referrals">
			<h4>Recent Referrals</h4>
			<div class="recent-referrals-list" data-user-id="<?= $user_id ?>">
				<div class="loading">Loading...</div>
			</div>
		</section>

		<section class="stats-summary">
			<div class="row x-btw">
				<span class="stat-label">Total Referrals</span>
				<span class="stat-value" data-stat="total">-</span>
			</div>
			<div class="row x-btw">
				<span class="stat-label">Successful</span>
				<span class="stat-value" data-stat="treated">-</span>
			</div>
			<div class="row x-btw">
				<span class="stat-label">Pending</span>
				<span class="stat-value" data-stat="pending">-</span>
			</div>
			<div class="row x-btw highlight">
				<span class="stat-label">Available Rewards</span>
				<span class="stat-value" data-stat="rewards">$0.00</span>
			</div>
		</section>

		<a href="<?= get_home_url(null, '/dash/referrals')?>" class="view-dashboard-btn">
			Dashboard <?= jvbIcon('arrow-right', ['size' => 16]); ?>
		</a>
		<p class="hint">Bulk-invite your friends via email - the link will pre-fill their name, email, and code!</p>
		</div>
		<?php
		return ob_get_clean();
	}

	function addReferralToWelcomeEmail(string $content, WP_User $user): string
	{
		$referral = $this->getReferralByReferee($user->ID);

		if (!$referral) {
			return $content;
		}

		$reward_amount = $this->settings['to_user_reward_amount'] ?? 20;
		$reward_type = $this->settings['to_user_reward_type'] ?? 'percentage';

		$reward_text = $reward_type === 'percentage'
			? $reward_amount . '% off'
			: '$' . number_format($reward_amount, 2) . ' off';

		$bonus_content = sprintf(
			'
			<h3>Thanks for the ♡</h3>
			<p>Since you were referred by a friend, you\'ve earned <strong>%s</strong> your first booking!</p>
			<p>Your reward will be automatically applied when you book.</p>',
			$reward_text,
		);


		$code = $this->getUserReferralCode($user->ID);
		$yourCode = '';
		if ($code) {
			$share_url = $this->getShareURL($code);
			$yourCode = sprintf(
				'<div class="callout">
			<h3>Share the ♡ with Friends</h3>
			<p>If you find you love what we can do for you, you can share your own code!</p>
			<p>Your Referral Code: <strong>%s</strong></p>
			<p>Or click the button below:</p>
			%s
			</div>',
				Emails::getInstance()->link($code),
				Emails::getInstance()->button($share_url, 'Share Your Code')
			);
		}


		return $content . $bonus_content . $yourCode;
	}

	public function getShareURL(string $code):string
	{
		return add_query_arg(
			[
				'ref' => $code
			],
			get_home_url()
		);
	}

	/**
	 * Send referral invitation via email with magic link
	 *
	 * @param int $user_id Referrer's user ID
	 * @param string $invitee_email Email of person to invite
	 * @param string $invitee_name Name of person to invite
	 * @param string $subject
	 * @param string $message
	 * @return array|WP_Error Result with success/error
	 */
	public function sendReferralInvitation(int $user_id, string $invitee_email, string $invitee_name, string $subject, string $message):array|WP_Error
	{
		// Check email rate limit (15/hour)
		$rate_check = $this->checkEmailRateLimit($user_id);
		if ($rate_check !== true) {
			return new WP_Error('rate_limit', 'You can only send 15 invitations per hour. Please try again later.');
		}

		// Validate email
		$invitee_email = sanitize_email($invitee_email);
		if (!is_email($invitee_email)) {
			return new WP_Error('invalid_email', 'Invalid email address');
		}


		// Check if already registered
		if (email_exists($invitee_email)) {
			return new WP_Error('user_exists', 'This person already has an account');
		}

		// Get from_user info
		$from_user = get_userdata($user_id);
		$referral_code = $this->getUserReferralCode($user_id);

		// Record the invitation attempt (for rate limiting only)
		if ($this->emailAlreadyTreated($invitee_email)) {
			return [
				'success'	=> false,
				'message'	=> 'User already treated'
			];
		}

		$record = $this->referrals->findOrCreate(
			[
				'from_user'	=> $user_id,
				'to_email'	=> $invitee_email,
				'referral_code' => $referral_code,
			],
			[
				'to_name'	=> $invitee_name,
				'status'	=> 'invited',
				'last_invited'	=> current_time('mysql')
			]
		);

		// Encode the token
		$registration_url = add_query_arg([
			'ref' => $referral_code,
			'rname' => sanitize_text_field($invitee_name),
			'remail'=> rawurlencode($invitee_email),
		], home_url('/'));

		// Get reward text for email
		$reward_text = $this->settings['to_user_reward_type'] === 'percentage'
			? "{$this->settings['to_user_reward_amount']}% off"
			: "\${$this->settings['to_user_reward_amount']} off";

		// Build email content
		$email_content =
			sprintf(
				'<h2>%s invited you to %s!</h2>
			<p>%s</p>
			<div class="callout">
				<h3>Get %s your first treatment!</h3>
			</div>
			<p>Click the button below to register and claim your reward:</p>
			%s
			<p><small>This invitation expires in 30 days.</small></p>',
				esc_html($from_user->display_name),
				esc_html(get_bloginfo('name')),
				nl2br(esc_html($message)),
				esc_html($reward_text),
				Emails::getInstance()->button($registration_url, 'Register & Get Your Reward')
			);

		// Send email
		$sent = Emails::getInstance()->sendEmail(
			$invitee_email,
			$subject,
			$email_content
		);

		if (!$sent) {
			return new WP_Error('email_failed', 'Failed to send invitation email');
		}

		return [
			'success' => true,
			'message' => 'Invitation sent successfully',
			'email' => $invitee_email,
			'name' => $invitee_name
		];
	}

	/**
	 * Send multiple referral invitations
	 *
	 * @param int $user_id Referrer's user ID
	 * @param array $invitations Array of ['email' => '', 'name' => '']
	 * @return array Results with success/failed arrays
	 */
	public function sendBatchReferralInvitations(int $user_id, array $invitations, string $subject, string $message): array
	{
		$results = [
			'success' => [],
			'failed' => []
		];

		$invitations = array_filter($invitations, function ($invite) {
			if (!array_key_exists('email', $invite) || !filter_var($invite['email'], FILTER_VALIDATE_EMAIL)) {
				return false;
			}
			if (!array_key_exists('name', $invite) || empty ($invite['name'])) {
				return false;
			}
			return true;
		});

		foreach ($invitations as $invite) {
			$email = $invite['email'];
			$name = $invite['name'];

			$result = $this->sendReferralInvitation($user_id, $email, $name, $subject, $message);

			if (!$result['success']) {
				$results['failed'][] = [
					'email' => $email,
					'name' => $name,
					'reason' => $result['message']
				];
			} else {
				$results['success'][] = [
					'email' => $email,
					'name' => $name
				];
			}

			// Small delay between sends to be respectful
			usleep(100000); // 0.1 seconds
		}

		return [
			'success' => !empty($results['success']),
			'result' => $results,
			'summary' => sprintf(
				'Sent %d invitations, %d failed',
				count($results['success']),
				count($results['failed'])
			)
		];
	}

	/**
	 * Check email invitation rate limit (15 per hour)
	 *
	 * @param int $user_id
	 * @return true|string True if allowed, error message if limited
	 */
	protected function checkEmailRateLimit(int $user_id):bool|string
	{
		$key = BASE.'referral_email_limit';
		$stored = get_user_meta($user_id, $key, true);
		$count = empty($stored) ? 0 : count($stored);

		if ($count >= 15) {
			return false;
		}

		$stored = empty($stored) ? [] : $stored;
		$stored[] = time();
		$stored = array_filter($stored, function($time) {
			return time() - $time < HOUR_IN_SECONDS;
		});

		update_user_meta($user_id, $key, $stored);
		return true;
	}

	/**
	 * Export referrals for Jane App cross-reference
	 *
	 * @param string $start_date Y-m-d format
	 * @param string $end_date Y-m-d format
	 * @return string CSV content
	 */
	public function exportReferrals(string $start_date, string $end_date): string
	{
		$referrals = $this->referrals->queryResults(
			"SELECT
            {table}.id,
            {table}.to_name,
            {table}.to_email,
            {table}.to_phone,
            {table}.referral_code,
            {table}.created_at,
            {table}.status,
            {table}.treated_at,
            u.display_name as from_name,
            u.user_email as from_email
        FROM {table}
        JOIN {$this->referrals->getUserTable()} u ON {table}.from_user = u.ID
        WHERE DATE({table}.created_at) BETWEEN %s AND %s
        ORDER BY {table}.created_at DESC",
			[$start_date, $end_date]
		);

		// Build CSV
		$csv_lines = [];

		// Headers
		$csv_lines[] = [
			'Referral ID',
			'Referee Name',
			'Referee Email',
			'Referee Phone',
			'Referral Code',
			'Referred Date',
			'Status',
			'Treated Date',
			'Referrer Name',
			'Referrer Email'
		];

		// Data rows
		foreach ($referrals as $ref) {
			$csv_lines[] = [
				$ref->id,
				$ref->to_name,
				$ref->to_email,
				$ref->to_phone ?: 'N/A',
				$ref->referral_code,
				$ref->referred_at,
				ucfirst($ref->status),
				$ref->treated_at ?: 'N/A',
				$ref->from_name,
				$ref->from_email
			];
		}

		// Convert to CSV string
		$output = fopen('php://temp', 'r+');
		foreach ($csv_lines as $line) {
			fputcsv($output, $line);
		}
		rewind($output);
		$csv_content = stream_get_contents($output);
		fclose($output);

		return $csv_content;
	}

	/**
	 * Add referral settings subpage to admin menu
	 * Add referral settings subpage to admin menu
	 *
	 * @param array $subpages
	 * @return array
	 */
	public static function addSubpage():void
	{
		$subpage = [
			'page_title' => 'Referral System',
			'menu_title' => 'Referrals',
			'capability' => 'manage_options',
			'menu_slug' => BASE . 'referral-admin',
			'callback' => [self::class, 'renderAdminPage']
		];
		AdminPages::addSubPage(BASE.'referral-admin', $subpage);
	}

	/**
	 * Register settings
	 */
	public function registerSettings(): void
	{
		register_setting(
			BASE . 'referral_settings',
			BASE . 'referral_page_id',
			[
				'type' => 'integer',
				'sanitize_callback' => 'absint',
				'default' => 0
			]
		);

		register_setting(
			BASE . 'referral_settings',
			BASE . 'referral_reward_settings',
			[
				'type' => 'array',
				'sanitize_callback' => [$this, 'sanitizeRewardSettings'],
				'default' => $this->default_settings
			]
		);
	}

	/**
	 * Sanitize reward settings
	 */
	public function sanitizeRewardSettings(array $settings): array
	{
		return [
			'from_user_reward_applies_to' => in_array($settings['from_user_reward_applies_to'] ?? '', ['per_user', 'flat_total'])
				? $settings['from_user_reward_applies_to']
				: 'per_user',
			'from_user_reward_amount' => floatval($settings['from_user_reward_amount'] ?? 25.00),
			'from_user_reward_type' => in_array($settings['from_user_reward_type'] ?? '', ['fixed', 'percentage'])
				? $settings['from_user_reward_type']
				: 'fixed',
			'to_user_reward_type' => in_array($settings['to_user_reward_type'] ?? '', ['percentage', 'fixed'])
				? $settings['to_user_reward_type']
				: 'percentage',
			'to_user_reward_amount' => floatval($settings['to_user_reward_amount'] ?? 20),
			'to_user_reward_applies_to' => in_array($settings['to_user_reward_applies_to'] ?? '', ['first_order', 'all_orders'])
				? $settings['to_user_reward_applies_to']
				: 'first_order',
		];
	}

	/**
	 * Render the admin settings page
	 */
	public function renderAdminPage(): void
	{
		?>
		<div class="wrap jvb-admin-wrap">
			<h1>Referral System Management</h1>

			<!-- CSV Upload Section -->
			<div class="card">
				<h2>Import Data from Jane App</h2>
				<p>Upload your exported CSV files from Jane App to sync client and sales data.</p>

				<div class="jvb-upload-section" style="display: grid; grid-template-columns: 1fr 1fr; gap: 20px; margin-top: 20px;">
					<!-- Client List Upload -->
					<div class="jvb-upload-box">
						<h3>Client List</h3>
						<form id="client-upload-form" enctype="multipart/form-data">
							<input type="file"
								   name="client_file"
								   id="client_file"
								   accept=".csv"
								   required />
							<button type="submit" class="button button-primary" style="margin-top: 10px;">
								Upload Clients
							</button>
						</form>
						<div id="client-upload-status" style="margin-top: 10px;"></div>
					</div>

					<!-- Sales Export Upload -->
					<div class="jvb-upload-box">
						<h3>Sales Export</h3>
						<form id="sales-upload-form" enctype="multipart/form-data">
							<input type="file"
								   name="sales_file"
								   id="sales_file"
								   accept=".csv"
								   required />
							<button type="submit" class="button button-primary" style="margin-top: 10px;">
								Upload Sales
							</button>
						</form>
						<div id="sales-upload-status" style="margin-top: 10px;"></div>
					</div>
				</div>
			</div>

			<!-- Referrals Table -->
			<div class="card" style="margin-top: 20px;">
				<h2>Referrals Management</h2>

				<div class="jvb-table-controls" style="margin-bottom: 15px; display: flex; gap: 10px; align-items: center;">
					<label>
						Filter by Status:
						<select id="referral-status-filter">
							<option value="">All Statuses</option>
							<option value="pending">Pending</option>
							<option value="consulted">Consulted</option>
							<option value="treated">Treated</option>
							<option value="cancelled">Cancelled</option>
						</select>
					</label>
					<input type="text"
						   id="referral-search"
						   placeholder="Search by name or email..."
						   style="min-width: 250px;" />
					<button type="button" class="button" id="refresh-table">Refresh</button>
				</div>

				<div id="referrals-table-container">
					<div class="jvb-loading">Loading referrals...</div>
				</div>
			</div>

			<!-- Settings Section -->
			<?= $this->renderAdminHTML() ?>
		</div>
<?php /**
		<style>
			.jvb-upload-box {
				padding: 20px;
				background: #f9f9f9;
				border: 1px solid #ddd;
				border-radius: 4px;
			}
			.jvb-upload-box h3 {
				margin-top: 0;
			}
			.referrals-table {
				width: 100%;
				border-collapse: collapse;
			}
			.referrals-table th,
			.referrals-table td {
				padding: 12px;
				text-align: left;
				border-bottom: 1px solid #ddd;
			}
			.referrals-table th {
				background: #f5f5f5;
				font-weight: 600;
			}
			.referrals-table tr:hover {
				background: #f9f9f9;
			}
			.referral-status {
				padding: 4px 8px;
				border-radius: 3px;
				font-size: 12px;
				font-weight: 500;
			}
			.referral-status.pending {
				background: #fff3cd;
				color: #856404;
			}
			.referral-status.consulted {
				background: #d1ecf1;
				color: #0c5460;
			}
			.referral-status.treated {
				background: #d4edda;
				color: #155724;
			}
			.referral-actions {
				display: flex;
				gap: 5px;
			}
			.notice.notice-success,
			.notice.notice-error {
				margin: 10px 0;
			}
		</style>
*/
		if (is_admin()) {
?>
		<script>
			jQuery(document).ready(function($) {
				// Client upload
				$('#client-upload-form').on('submit', function(e) {
					e.preventDefault();
					const formData = new FormData(this);
					formData.append('file', $('#client_file')[0].files[0]);

					$('#client-upload-status').html('<span class="spinner is-active"></span> Uploading...');

					$.ajax({
						url: '<?= rest_url('jvb/v1/referrals/upload-clients') ?>',
						method: 'POST',
						data: formData,
						processData: false,
						contentType: false,
						beforeSend: function(xhr) {
							xhr.setRequestHeader('X-WP-Nonce', '<?= wp_create_nonce('wp_rest') ?>');
						},
						success: function(response) {
							if (response.success) {
								let message = '<div class="notice notice-success"><p>' + response.message;
								message += '<br>Created: ' + (response.stats.created || 0);
								message += ', Updated: ' + (response.stats.updated || 0);
								message += ', Skipped: ' + (response.stats.skipped || 0) + '</p>';

								// Show skipped details if any
								if (response.stats.skipped_details && response.stats.skipped_details.length > 0) {
									message += '<details style="margin-top: 10px;"><summary>View skipped records (' + response.stats.skipped_details.length + ')</summary>';
									message += '<table class="widefat" style="margin-top: 10px;"><thead><tr>';
									message += '<th>Line</th><th>Name</th><th>Email</th><th>GUID</th><th>Reason</th>';
									message += '</tr></thead><tbody>';
									response.stats.skipped_details.forEach(function(item) {
										message += '<tr>';
										message += '<td>' + (item.line || '-') + '</td>';
										message += '<td>' + (item.name || '-') + '</td>';
										message += '<td>' + (item.email || '-') + '</td>';
										message += '<td>' + (item.guid || '-') + '</td>';
										message += '<td>' + item.reason + '</td>';
										message += '</tr>';
									});
									message += '</tbody></table></details>';
								}

								message += '</div>';

								$('#client-upload-status').html(message);
								$('#client-upload-form')[0].reset();
								loadReferralsTable();
							} else {
								$('#client-upload-status').html(
									'<div class="notice notice-error"><p>' + response.message + '</p></div>'
								);
							}
						},
						error: function(xhr) {
							$('#client-upload-status').html(
								'<div class="notice notice-error"><p>Upload failed</p></div>'
							);
						}
					});
				});

				// Sales upload
				$('#sales-upload-form').on('submit', function(e) {
					e.preventDefault();
					const formData = new FormData(this);
					formData.append('file', $('#sales_file')[0].files[0]);

					$('#sales-upload-status').html('<span class="spinner is-active"></span> Uploading...');

					$.ajax({
						url: '<?= rest_url('jvb/v1/referrals/upload-sales') ?>',
						method: 'POST',
						data: formData,
						processData: false,
						contentType: false,
						beforeSend: function(xhr) {
							xhr.setRequestHeader('X-WP-Nonce', '<?= wp_create_nonce('wp_rest') ?>');
						},
						success: function(response) {
							if (response.success) {
								$('#sales-upload-status').html(
									'<div class="notice notice-success"><p>' + response.message +
									'<br>Consultations: ' + response.stats.consultations +
									', Treatments: ' + response.stats.treatments +
									', Skipped: ' + response.stats.skipped + '</p></div>'
								);
								$('#sales-upload-form')[0].reset();
								loadReferralsTable();
							} else {
								$('#sales-upload-status').html(
									'<div class="notice notice-error"><p>' + response.message + '</p></div>'
								);
							}
						},
						error: function(xhr) {
							$('#sales-upload-status').html(
								'<div class="notice notice-error"><p>Upload failed</p></div>'
							);
						}
					});
				});

				// Load referrals table
				function loadReferralsTable(page = 1) {
					const status = $('#referral-status-filter').val();
					const search = $('#referral-search').val();

					$.ajax({
						url: '<?= rest_url('jvb/v1/referrals') ?>',
						method: 'GET',
						data: {
							offset: page -1,
							limit: 20,
							status: status === '' ? 'all' : status,
							search: search
						},
						beforeSend: function(xhr) {
							xhr.setRequestHeader('X-WP-Nonce', '<?= wp_create_nonce('wp_rest') ?>');
							$('#referrals-table-container').html('<div class="jvb-loading">Loading...</div>');
						},
						success: function(response) {
							if (response.success) {
								renderReferralsTable(response);
							}
						}
					});
				}

				// Render table
				function renderReferralsTable(data) {
					let html = '<table class="referrals-table widefat">';
					html += '<thead><tr>';
					html += '<th>Referrer</th>';
					html += '<th>Referee</th>';
					html += '<th>Email</th>';
					html += '<th>Status</th>';
					html += '<th>Referred Date</th>';
					html += '<th>Total Referrals</th>';
					html += '<th>Actions</th>';
					html += '</tr></thead><tbody>';

					if (data.items.length === 0) {
						html += '<tr><td colspan="7" style="text-align: center;">No referrals found</td></tr>';
					} else {
						data.items.forEach(function(ref) {
							html += '<tr>';
							html += '<td>' + (ref.from_name || 'Unknown') + '</td>';
							html += '<td>' + (ref.to_user_display_name || ref.to_name) + '</td>';
							html += '<td>' + (ref.to_user_display_email || ref.to_email) + '</td>';
							html += '<td><span class="referral-status ' + ref.status + '">' + ref.status + '</span></td>';
							html += '<td>' + new Date(ref.referred_at).toLocaleDateString() + '</td>';
							html += '<td>' + (ref.from_user_total_referrals || 0) + '</td>';
							html += '<td class="referral-actions">';

							if (ref.status === 'pending') {
								html += '<button class="button button-small mark-consulted" data-id="' + ref.id + '">Mark Consulted</button>';
							}
							if (ref.status !== 'treated') {
								html += '<button class="button button-small mark-treated" data-id="' + ref.id + '">Mark Treated</button>';
							}

							html += '</td>';
							html += '</tr>';
						});
					}

					html += '</tbody></table>';

					// Add pagination
					if (data.total_pages > 1) {
						html += '<div class="tablenav"><div class="tablenav-pages">';
						for (let i = 1; i <= data.total_pages; i++) {
							const active = i === data.page ? 'button-primary' : 'button';
							html += '<button class="button ' + active + ' page-link" data-page="' + i + '">' + i + '</button> ';
						}
						html += '</div></div>';
					}

					$('#referrals-table-container').html(html);
				}

				// Event handlers for actions
				$(document).on('click', '.mark-consulted', function() {
					const id = $(this).data('id');
					if (!confirm('Mark this referral as consulted? This will create the consultation reward.')) return;

					$.ajax({
						url: '<?= rest_url('jvb/v1/referrals') ?>', // Changed from /mark-consulted
						method: 'POST',
						data: JSON.stringify({
							action: 'consulted',  // Added action parameter
							referral_id: id
						}),
						contentType: 'application/json',
						beforeSend: function(xhr) {
							xhr.setRequestHeader('X-WP-Nonce', '<?= wp_create_nonce('wp_rest') ?>');
						},
						success: function(response) {
							if (response.success) {
								alert(response.message);
								loadReferralsTable();
							} else {
								alert('Error: ' + response.message);
							}
						}
					});
				});

				$(document).on('click', '.mark-treated', function() {
					const id = $(this).data('id');
					if (!confirm('Mark this referral as treated? This will create rewards for both parties.')) return;

					$.ajax({
						url: '<?= rest_url('jvb/v1/referrals') ?>', // Changed from /mark-treated
						method: 'POST',
						data: JSON.stringify({
							action: 'treated',  // Added action parameter
							referral_id: id
						}),
						contentType: 'application/json',
						beforeSend: function(xhr) {
							xhr.setRequestHeader('X-WP-Nonce', '<?= wp_create_nonce('wp_rest') ?>');
						},
						success: function(response) {
							if (response.success) {
								alert(response.message);
								loadReferralsTable();
							} else {
								alert('Error: ' + response.message);
							}
						}
					});
				});

				$(document).on('click', '.page-link', function() {
					loadReferralsTable($(this).data('page'));
				});

				$('#referral-status-filter, #refresh-table').on('change click', function() {
					loadReferralsTable();
				});

				// Search with debounce
				let searchTimeout;
				$('#referral-search').on('keyup', function() {
					clearTimeout(searchTimeout);
					searchTimeout = setTimeout(function() {
						loadReferralsTable();
					}, 500);
				});

				// Initial load
				loadReferralsTable();
			});
		</script>
		<?php
		}
	}

	protected function renderAdminHTML():string
	{
		ob_start();
		?>
		<div class="wrap">
			<h1>Referral Settings</h1>

			<form method="post" action="">
				<?php wp_nonce_field(BASE . 'admin_page_nonce'); ?>

				<div class="card">
					<h2>Referral Page</h2>
					<p>Select the page where users can access their referral dashboard.</p>

					<table class="form-table">
						<tr>
							<th scope="row">
								<label for="<?= BASE ?>referral_page_id">Referral Page</label>
							</th>
							<td>
								<?php
								if (!$this->referralPage) {
									$this->referralPage = $this->getReferralPageId();
								}
								wp_dropdown_pages([
									'name' => BASE . 'referral_page_id',
									'id' => BASE . 'referral_page_id',
									'selected' => $this->referralPage,
									'show_option_none' => __('— Select —', 'jvbase'),
									'option_none_value' => '0'
								]);
								?>
								<p class="description">
									This page will show "Referral Page" in the admin bar when editing.
								</p>
							</td>
						</tr>
					</table>
				</div>

				<div class="card">
					<h2>Reward Settings</h2>

					<table class="form-table">
						<tr>
							<th colspan="2"><h3>Referrer Rewards</h3></th>
						</tr>
						<tr>
							<th scope="row">
								<label for="from_user_reward_type">Reward Type</label>
							</th>
							<td>
								<select name="from_user_reward_type" id="from_user_reward_type">
									<option value="fixed" <?php selected($this->settings['from_user_reward_type'], 'fixed'); ?>>Fixed Amount</option>
									<option value="percentage" <?php selected($this->settings['from_user_reward_type'], 'percentage'); ?>>Percentage</option>
								</select>
							</td>
						</tr>
						<tr>
							<th scope="row">
								<label for="from_user_reward_amount">Reward Amount</label>
							</th>
							<td>
								<input type="number"
									   name="from_user_reward_amount"
									   id="from_user_reward_amount"
									   value="<?= esc_attr($this->settings['from_user_reward_amount']) ?>"
									   step="0.01"
									   min="0">
								<p class="description">Amount in dollars or percentage</p>
							</td>
						</tr>
						<tr>
							<th scope="row">
								<label for="from_user_reward_applies_to">Applies To</label>
							</th>
							<td>
								<select name="from_user_reward_applies_to" id="from_user_reward_applies_to">
									<option value="per_user" <?php selected($this->settings['from_user_reward_applies_to'], 'per_user'); ?>>Per User Referred</option>
									<option value="flat_total" <?php selected($this->settings['from_user_reward_applies_to'], 'flat_total'); ?>>Flat Total</option>
								</select>
							</td>
						</tr>

						<tr>
							<th colspan="2"><h3>Referee (New User) Rewards</h3></th>
						</tr>
						<tr>
							<th scope="row">
								<label for="to_user_reward_type">Reward Type</label>
							</th>
							<td>
								<select name="to_user_reward_type" id="to_user_reward_type">
									<option value="percentage" <?php selected($this->settings['to_user_reward_type'], 'percentage'); ?>>Percentage</option>
									<option value="fixed" <?php selected($this->settings['to_user_reward_type'], 'fixed'); ?>>Fixed Amount</option>
								</select>
							</td>
						</tr>
						<tr>
							<th scope="row">
								<label for="to_user_reward_amount">Reward Amount</label>
							</th>
							<td>
								<input type="number"
									   name="to_user_reward_amount"
									   id="to_user_reward_amount"
									   value="<?= esc_attr($this->settings['to_user_reward_amount']) ?>"
									   step="0.01"
									   min="0">
								<p class="description">Amount in dollars or percentage</p>
							</td>
						</tr>
						<tr>
							<th scope="row">
								<label for="to_user_reward_applies_to">Applies To</label>
							</th>
							<td>
								<select name="to_user_reward_applies_to" id="to_user_reward_applies_to">
									<option value="first_order" <?php selected($this->settings['to_user_reward_applies_to'], 'first_order'); ?>>First Order Only</option>
									<option value="all_orders" <?php selected($this->settings['to_user_reward_applies_to'], 'all_orders'); ?>>All Orders</option>
								</select>
							</td>
						</tr>
						<tr>
							<th scope="row">
								<label for="<?= BASE ?>referral_role">Client Import Role</label>
							</th>
							<td>
								<?php
								$selected_role = get_option(BASE . 'referral_role', '');
								$roles = wp_roles()->get_names();
								?>
								<select name="<?= BASE ?>referral_role" id="<?= BASE ?>referral_role">
									<?php foreach ($roles as $role_value => $role_name): ?>
										<option value="<?= esc_attr($role_value) ?>" <?php selected($selected_role, $role_value); ?>>
											<?= esc_html($role_name) ?>
										</option>
									<?php endforeach; ?>
								</select>
								<p class="description">
									Role assigned to users imported from Jane App client list.
								</p>
							</td>
						</tr>
					</table>
				</div>

				<p class="submit">
					<button type="submit" name="submit" class="button button-primary">Save Settings</button>
				</p>
			</form>

			<?= $this->renderReferralStats(true) ?>
		</div>
		<?php
		return ob_get_clean();
	}

	/**
	 * Render referral statistics
	 */
	protected function renderReferralStats(bool $wrapCard = false): string
	{
		$isAdmin = current_user_can('manage_options');
		$current = get_current_user_id();
		$types = ['total' => 'Total Referrals', 'invited' => 'Invited', 'pending' => 'Pending', 'treated' => 'Treated'];

		$out = '<table class="widefat">';
		foreach ($types as $type => $label) {
			$status = $type === 'total' ? [] : ['status' => $type];

			$count = $isAdmin ?
				$this->referrals->count($status) :
				$this->referrals->count(array_merge(
					['from_user' => $current],
					$status
				));
			$out .= sprintf(
			'<tr>
					<th>%s</th>
					<td>%s</td>
				</tr>',
				$label,
				esc_html($count)
			);
		}
		$out .= '</table>';

		if ($wrapCard) {
			$out = '<div class="card">
            <h2>Referral Statistics</h2>
            ' . $out . '
        </div>';
		}

		return $out;
	}


	/**
	 * Get the referral page ID
	 *
	 * @return int|null
	 */
	public function getReferralPageId(): ?int
	{
		$page_id = get_option(BASE . 'referral_page_id');
		return $page_id ? (int) $page_id : null;
	}

	/**
	 * Show admin notice on referral page edit screen
	 */
	public function showReferralPageNotice(): void
	{
		global $pagenow, $post;

		if ('post.php' !== $pagenow || !$post) {
			return;
		}
		if (!$this->referralPage) {
			$this->referralPage = $this->getReferralPageId();
		}
		if ($post->ID === $this->referralPage) {
			echo '<div class="notice notice-info">';
			echo '<p>' . __('This page is designated as the <strong>Referral Page</strong>.', 'jvb') . '</p>';
			echo '</div>';
		}
	}

	public function renderDashPage():void
	{
		// Regular users get their referral dashboard
		$user_id = get_current_user_id();

		$referral_code = $this->getUserReferralCode($user_id);
		if (!$referral_code) {
			$user = get_userdata($user_id);
			$referral_code = $this->generateReferralCode($user);
		}

		$referrals = $this->getUserReferrals($user_id, ['per_page' => 20]);


		$tabs = new Tabs();
		$tabs->addTab('share')
			->title('Share')
			->icon('share-fat')
			->description('Share your code and earn rewards when your referrals complete their first treatment!')
			->content($this->shareDashboard($user_id, $referral_code));
		$tabs->addTab('referrals')
			->title('Your Referrals')
			->icon('hand-heart')
			->content($this->referralCRUD($user_id));

		?>
		<div class="referral-dashboard">
			<?= $tabs->render();?>
		</div>

		<?php
	}

	protected function shareDashboard(int $user_id, string $referral_code):string
	{
		ob_start();
		?>
		<?php $this->getShareButtons($user_id); ?>

		<!-- Referral Code Card -->
		<details open>
			<summary>Your Code</summary>
			<h3>Share Link</h3>
			<div class="row x-btw nowrap">
				<code id="referral-link" class="copy-target"><?= home_url('/?ref=' . $referral_code) ?></code>
				<button type="button" class="copy-btn" data-target="referral-link" title="Copy referral link">
					<?= jvbIcon('copy'); ?>
					<?= jvbIcon('check-circle'); ?>
				</button>
			</div>
			<h3>Share Code</h3>
			<div class="row x-btw nowrap">
				<code id="referral-code" class="copy-target"><?= esc_html($referral_code) ?></code>
				<button type="button" class="copy-btn" data-target="referral-code" title="Copy referral code">
					<?= jvbIcon('copy'); ?>
					<?= jvbIcon('check-circle'); ?>
				</button>
			</div>
		</details>
		<form class="invite">
			<h2>Invite your Friends</h2>
			<p>Or, if you prefer, enter your friends name(s) and email(s), and we'll send off some emails.</p>
			<p><small>(No data is stored. Your friends will get an email from our email.)</small></p>
			<?php
			$invite = [
				'type' => 'taglist',
				'label' => 'Invite Your Friends',
				'hint' => 'Add friends to send them a referral link',
				'add_label' => 'Add Invite',
				'tag_format' => '{{name}} ({{email}})', // or 'first_field', 'all_fields', 'email', etc.
				'fields' => [
					'name' => [
						'type' => 'text',
						'label' => 'Name',
						'placeholder' => 'Full Name',
						'required' => true
					],
					'email' => [
						'type' => 'email',
						'label' => 'Email',
						'placeholder' => 'email@example.com',
						'required' => true
					]
				]
			];
			$fields = [
				'subject'	=> [
					'type'	=> 'text',
					'label'	=> 'Email Subject',
					'value'	=> 'Try Legacy for Tattoo Removal',
				],
				'message'	=> [
					'type'		=> 'textarea',
					'label'		=> 'Customize message',
					'value'	=> 'I had a great experience at Legacy Tattoo Removal!

If you click the link below, you can get 20% off your first treatment with them.',
					'hint'		=> 'We\'ll add your code and a link automatically.'
				]
			];
			echo Form::render('invite', '', $invite);
			?>
			<details>
				<summary class="icon icon-caret-down">Customize Message</summary>
				<?php
				foreach ($fields as $fieldName => $field) {
					$value = (array_key_exists('value', $field)) ? $field['value'] : [];
					echo Form::render($fieldName, $value, $field);
				}
				?>
			</details>

			<button type="submit"><?=jvbIcon('envelope')?>Send Invites</button>
		</form>
		<?php
		return ob_get_clean();
	}

	protected function referralCRUD(int $user_id):string
	{
		$stats = $this->getUserStats($user_id);
		ob_start();
		?>
		<!-- Stats Grid with Updated Labels -->
		<div class="item-grid stats">
			<div class="card">
				<h4>Code Used</h4>
				<span class="stat-number" data-stat="code_used"><?= esc_html($stats['code_used'] ?? 0) ?></span>
				<p class="hint">People who used your code</p>
			</div>
			<div class="card">
				<h4>Treatments</h4>
				<span class="stat-number" data-stat="treatments"><?= esc_html($stats['treatments'] ?? 0) ?></span>
				<p class="hint">Completed first treatment</p>
			</div>
			<div class="card highlight">
				<h4>Total Rewards</h4>
				<span class="stat-number" data-stat="total_rewards">$<?= number_format($stats['total_rewards'] ?? 0, 2) ?></span>
				<p class="hint">Earned from referrals</p>
			</div>
		</div>

		<?php
		// Configure CRUDSkeleton for referrals
		$crud = new CRUDSkeleton();
		$crud->title('Your Referrals', 'Track friends you\'ve invited and rewards earned')
			->content('referral', 'Referral', 'Referrals')
//			->initMeta('custom', 'referral')
			->setFields([
				'to_name' => [
					'label' => 'Name',
					'type' => 'text',
				],
				'to_email' => [
					'label' => 'Email',
					'type' => 'text',
				],
				'referred_at' => [
					'label' => 'Code Used',
					'type' => 'date',
				],
				'referral_status' => [
					'label' => 'Status',
					'type' => 'text',
				]
			])
			->setStatuses(['all', 'unused', 'registered', 'consulted', 'completed'])
			->addViews(['table', 'list'])
			->defaultView('table')
			->addCapabilities(['view'])
			->addDateFilter('referred_at')
			->showBulkControls(false)
			->showFilters(false)
			->useCRUDjs(false); // We'll use our custom Referral.js with DataStore

		// Add custom template for actions column
		$crud->addItemActions(['resend', 'trash']);
		$crud->defineItemAction('resend', [
			'title'	=> 'Resend Invitation',
			'icon'	=> 'paper-plane-tilt'
		]);
		$crud->defineItemAction('trash', [
			'title'	=> 'Remove from List'
		]);

		// Custom empty state
		$crud->addTemplate('empty', '
		<template class="emptyState">
			<div class="empty-state">
				<h3>' . jvbDashIcon('hand-heart') . 'Nothing Yet' . jvbDashIcon('hand-heart') . '</h3>
				<p>Start sharing your referral code to earn rewards!</p>
				<p><small><i>Share your code using the "Share" tab below.</i></small></p>
			</div>
		</template>
	');

		$crud->render();

		return ob_get_clean();
	}

	/**
	 * Handle admin page form submission
	 *
	 * @param mixed $result Previous result
	 * @param string $page_slug Current page slug
	 * @param array $post_data POST data
	 * @return array|null Result array or null if not our page
	 */
	public function handleAdminSubmission(mixed $result, string $page_slug, array $post_data): ?array
	{
		// Only handle our page
		if ($page_slug !== BASE . 'referral-admin') {
			return $result;
		}

		try {
			// Save referral page
			$page_id = isset($post_data[BASE . 'referral_page_id']) ? absint($post_data[BASE . 'referral_page_id']) : 0;
			update_option(BASE . 'referral_page_id', $page_id);

			// Save client import role
			$import_role = sanitize_text_field($post_data[BASE . 'referral_role'] ?? Site::getDefaultReferralRole());
			update_option(BASE . 'referral_role', $import_role);

			// Save reward settings
			$settings = [
				'from_user_reward_type' => sanitize_text_field($post_data['from_user_reward_type'] ?? 'fixed'),
				'from_user_reward_amount' => floatval($post_data['from_user_reward_amount'] ?? 25.00),
				'from_user_reward_applies_to' => sanitize_text_field($post_data['from_user_reward_applies_to'] ?? 'per_user'),
				'to_user_reward_type' => sanitize_text_field($post_data['to_user_reward_type'] ?? 'percentage'),
				'to_user_reward_amount' => floatval($post_data['to_user_reward_amount'] ?? 20),
				'to_user_reward_applies_to' => sanitize_text_field($post_data['to_user_reward_applies_to'] ?? 'first_order')
			];

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

			return [
				'success' => true,
				'message' => 'Referral settings saved successfully!'
			];

		} catch (Exception $e) {
			return [
				'success' => false,
				'message' => 'Failed to save settings: ' . $e->getMessage()
			];
		}
	}

	/**
	 * Get formatted reward text for to_user
	 *
	 * @param bool $full Include "off your first treatment" text
	 * @return string
	 */
	public function getRewardText(bool $full = true): string
	{
		$reward_amount = $this->settings['to_user_reward_amount'] ?? 20;
		$reward_type = $this->settings['to_user_reward_type'] ?? 'percentage';

		$reward_text = $reward_type === 'percentage'
			? $reward_amount . '% off'
			: '$' . number_format($reward_amount, 2) . ' off';

		if ($full) {
			$reward_text .= ' your first treatment';
		}

		return $reward_text;
	}

	public function getShareButtons(int $user_id):void
	{
		$referral_code = $this->getUserReferralCode($user_id);
		if (!$referral_code) {
			return;
		}

		$share_url = $this->getShareURL($referral_code);
		$referral_page_id = $this->getReferralPageId();

		// SMS share text
		$sms_text = urlencode("Check out " . get_bloginfo('name') . "! " . $share_url);

		// Share message
		$share_message = urlencode("I love " . get_bloginfo('name') . "! Thought you might want to check them out.");

		?>
		<nav class="share">
			<h4>Quick Share</h4>
			<ul>
				<a href="mailto:?subject=<?php echo urlencode('Check out ' . get_bloginfo('name')); ?>&body=<?php echo urlencode($share_message . ' ' . $share_url); ?>"
				   class="button" title="Email">
					<?php echo jvbIcon('envelope'); ?>
				</a>
				<a href="sms:?&body=<?php echo $sms_text; ?>"
				   class="button" title="Text">
					<?php echo jvbIcon('chat'); ?>
				</a>
				<a href="https://www.facebook.com/sharer/sharer.php?u=<?php echo urlencode($share_url); ?>"
				   target="_blank"
				   rel="noopener noreferrer"
				   class="button" title="Facebook">
					<?php echo jvbIcon('facebook-logo'); ?>
				</a>
				<a href="https://twitter.com/intent/tweet?url=<?php echo urlencode($share_url); ?>&text=<?php echo urlencode($share_message); ?>"
				   target="_blank"
				   rel="noopener noreferrer"
				   class="button" title="Twitter">
					<?php echo jvbIcon('twitter-logo'); ?>
				</a>
				<a href="https://wa.me/?text=<?php echo $sms_text; ?>"
				   target="_blank"
				   rel="noopener noreferrer"
				   class="button" title="WhatsApp">
					<?php echo jvbIcon('whatsapp-logo'); ?>
				</a>
			</ul>
		</nav>
	<?php
	}

	/**
	 * Send notification to from_user when someone registers
	 *
	 * @param int $from_user_id
	 * @param string $to_name
	 */
	protected function sendReferrerNotification(int $from_user_id, string $to_name): void
	{
		$from_user = get_userdata($from_user_id);
		if (!$from_user) {
			return;
		}

		$subject = sprintf('%s signed up with your referral code!', $to_name);
		$message = sprintf(
			"Great news! %s just signed up using your referral code.\n\n" .
			"View your referrals: %s",
			$to_name,
			home_url('/dash/referrals')
		);

		Emails::getInstance()->sendEmail(
			$from_user->user_email,
			$subject,
			$message
		);
	}

	/**
	 * Get welcome message for newly referred user
	 *
	 * @param int $user_id
	 * @return string HTML content for welcome message
	 */
	public function getReferralWelcomeMessage(int $user_id): string
	{
		// Check if user was referred
		$referral = $this->getReferralByReferee($user_id);

		if (!$referral || $referral->status !== 'pending') {
			return '';
		}

		// Only show for recent registrations (within 7 days)
		$registered_time = strtotime($referral->referred_at);
		if ((time() - $registered_time) > (7 * DAY_IN_SECONDS)) {
			return '';
		}

		// Get from_user name
		$from_user = get_userdata($referral->from_user);
		$from_user_first_name = $from_user ? strtok($from_user->display_name, ' ') : 'Your friend';

		// Get reward text
		$reward_text = $this->getRewardText(); // Just "20% off" or "$25 off"

		$booking_url = apply_filters('jvb_referral_booking_url', home_url('/contact'));
		$estimate_url = apply_filters('jvb_referral_estimate_url', home_url('/estimate'));

		ob_start();
		?>
		<div class="welcome-banner referral-welcome">
			<div class="banner-content">
				<h3><?= jvbIcon('confetti') ?>Welcome! <small><b><?= esc_html($from_user_first_name) ?></b> invited you to save <b><?= esc_html($reward_text) ?></b>!</small></h3>
				<p>But we're not done yet! Here's what happens next:</p>
				<div class="callout">
					<ol>
						<li>Book your <b>free consultation</b></li>
						<li>Come in and we'll assess your tattoo</li>
						<li>Get <?= esc_html($reward_text) ?> your first treatment!</li>
					</ol>
				</div>
				<p class="hint">
					<strong>Important:</strong> If you book with a different email than
					<strong><?= esc_html(wp_get_current_user()->user_email) ?></strong>,
					please let us know so we can apply your reward!
				</p>
				<ul class="buttons">
					<li><a href="<?= esc_url($estimate_url) ?>" class="button-secondary">
						<?= jvbIcon('calculator') ?> Get an Estimate First
					</a></li>
					<li><a href="<?= esc_url($booking_url) ?>" class="button-primary">
						<?= jvbIcon('calendar') ?> Book Free Consult
					</a></li>
				</ul>


			</div>
		</div>
		<?php
		return ob_get_clean();
	}

	public function updateStatus(int $referral_id, string $status):array
	{
		$referral = $this->referrals->get(['id' => $referral_id]);
		if (!$referral) {
			return [
				'success'	=> false,
				'message'	=> 'Referral not found'
			];
		}

		$data = ['status' => $status, "{$status}_at" => current_time('mysql')];
		if ($status === 'treated') {
			$data['treatment_count'] = ($referral->treatment_count ?? 0) + 1;
		}

		$result = $this->referrals->update($data, ['id' => $referral_id]);
		if (!$result) {
			return [
				'success'	=> false,
				'message'	=> 'Failed to update referral status'
			];
		}

		if ($status === 'treated') {
			$this->createRewardRecords($referral_id);
		}

		$this->cache->forget($referral_id);
		return [
			'success'	=> true,
			'message'	=> 'Successfully set status to '.$status
		];
	}

	public function removeReferral(int $referral_id, int $user_id):array
	{
		$referral = $this->referrals->get(['id' => $referral_id]);
		if (!$referral) {
			return [
				'success'	=> false,
				'message'	=> 'Referral not found'
			];
		}

		if ($referral->from_user != $user_id && !current_user_can('manage_options')) {
			return [
				'success'	=> false,
				'message'	=> 'Unauthorized'
			];
		}

		if (!in_array($referral->status, ['invited', 'pending'])) {
			return [
				'success'	=> false,
				'message'	=> 'Cannot remove a completed or cancelled referral'
			];
		}

		$this->referrals->delete(['id' => $referral_id]);
		$this->cache->forget($referral_id);
		return [
			'success' => true,
			'message'	=> 'Successfully removed referral'
		];
	}

	public function resendInvitation(int $referral_id, int $user_id):array
	{
		$referral = $this->referrals->where(['id' => $referral_id, 'from_user' => $user_id])->first();
		if (!$referral) {
			return [
				'success'	=> false,
				'message'	=> 'Referral not found'
			];
		}

		if (!$this->checkCanResendEmail($referral)) {
			$days = (time() - $referral->last_invited) / DAY_IN_SECONDS;
			$days = match (true) {
				$days <= 1 => 'until tomorrow',
				$days <= 3 => 'a couple days',
				default => absint($days).' days'
			};

			return [
				'success'	=> false,
				'message'	=> 'Can\'t resend email. Wait '.$days.'.'
			];
		}

		$result = $this->sendReferralInvitation(
			$user_id,
			$referral->to_email,
			$referral->to_name,
			sprintf('Reminder: Join %s', get_bloginfo('name')),
			'Just a friendly reminder about my invitation!'
		);

		return [
			'success'	=> (bool) $result,
			'message'	=> $result ? 'Sent email' : 'Could not send email'
		];
	}

	public function getReferrals(array $args = []): array
	{
		$conditions = ['1=1'];
		$values = [];

		if (!empty($args['status']) && $args['status'] !== 'all') {
			$conditions[] = '{table}.status = %s';
			$values[] = $args['status'];
		}

		if (!empty($args['date_start'])) {
			$conditions[] = 'DATE({table}.created_at) >= %s';
			$values[] = $args['date_start'];
		}

		if (!empty($args['date_end'])) {
			$conditions[] = 'DATE({table}.created_at) <= %s';
			$values[] = $args['date_end'];
		}

		if (!empty($args['search'])) {
			global $wpdb;
			$like = '%' . $wpdb->esc_like($args['search']) . '%';
			$conditions[] = '({table}.to_name LIKE %s OR {table}.to_email LIKE %s OR {table}.referral_code LIKE %s OR u.display_name LIKE %s)';
			array_push($values, $like, $like, $like, $like);
		}

		array_push($values, absint($args['per_page'] ?? 50), absint($args['offset'] ?? 0));

		return $this->referrals->queryResults(
			"SELECT {table}.*, u.display_name as from_name
        FROM {table}
        LEFT JOIN {$this->referrals->getUserTable()} u ON {table}.from_user = u.ID
        WHERE " . implode(' AND ', $conditions) . "
        ORDER BY {table}.created_at DESC
        LIMIT %d OFFSET %d",
			$values
		);
	}

	public function emailAlreadyTreated(string $email):bool
	{
		if (!filter_var($email, FILTER_VALIDATE_EMAIL)) {
			return false;
		}
		if (email_exists($email)) {
			return false;
		}
		$referral = $this->referrals->get([
			'to_email'	=> $email,
			'status'	=> [
				'IN',
				['pending', 'consulted', 'treated', 'cancelled']
			]
		]);
		return !empty($referral);
	}


	/**
	 * Process queued referral operations
	 */
	public function processOperation(WP_Error|array $result, object $operation, array $data): array|WP_Error
	{
		if ($operation->type !== 'referral_invite') {
			return $result;
		}

		$result = $this->sendBatchReferralInvitations(
			$operation->user_id,
			$data['invitations'],
			$data['subject'],
			$data['message']
		);

		if ($result['success']) {
			$this->cache->flush();
		}

		return [
			'success' => true,
			'message' => sprintf(
				'Sent invitations. Success: %d. Failed: %d.',
				count($result['result']['success']),
				count($result['result']['failed'])
			),
			'details' => [
				'successful' => $result['result']['success'],
				'failed' => $result['result']['failed'],
				'total' => count($data['invitations'])
			]
		];
	}

}