<?php

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

define('NR_SRC_PARAM', 'nr_src');
define('NR_COOKIE_DAYS', 30);

define('NR_COOKIE_FIRST', 'nr_src_first');
define('NR_COOKIE_LAST',  'nr_src_last');

function nr_src_clean($v){
  $v = sanitize_text_field(wp_unslash($v));
  $v = trim($v);
  if ($v === '') return '';
  $v = substr($v, 0, 140);
  $v = preg_replace('/[^a-zA-Z0-9_\-\.]+/', '_', $v);
  return $v;
}

function nr_root_domain(){
  // Si algún día cambias el dominio, aquí lo ajustas una vez.
  return '.neorejuvenation.com';
}

function nr_set_cookie_root($name, $value, $expire){
  setcookie($name, $value, [
    'expires'  => $expire,
    'path'     => '/',
    'domain'   => nr_root_domain(),
    'secure'   => is_ssl(),
    'httponly' => true,
    'samesite' => 'Lax',
  ]);
  $_COOKIE[$name] = $value;
}

add_action('init', function(){
  if (is_admin()) return;
  if (empty($_GET[NR_SRC_PARAM])) return;

  $src = nr_src_clean($_GET[NR_SRC_PARAM]);
  if ($src === '') return;

  $expire = time() + (NR_COOKIE_DAYS * DAY_IN_SECONDS);

  // LAST
  nr_set_cookie_root(NR_COOKIE_LAST, $src, $expire);

  // FIRST
  if (empty($_COOKIE[NR_COOKIE_FIRST])) {
    nr_set_cookie_root(NR_COOKIE_FIRST, $src, $expire);
  }
}, 20);