import requests
import logging
import time
import os

# Configuration du logger
logging.basicConfig(
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] %(message)s",
)

logging.info("Démarrage du script")

# Récupération des variables d'environnement
try:
    DISCORD_WEBHOOK_URL = os.environ['DISCORD_WEBHOOK_URL']
    REFRESH_TIME = int(os.environ['REFRESH_TIME'])  # Convertir en entier
except KeyError as e:
    logging.error(f"Variable d'environnement manquante : {e}")
    exit(1)  # Quitter le script proprement en cas d'erreur
except ValueError:
    logging.error("REFRESH_TIME doit être un entier valide.")
    exit(1)

# Afficher les valeurs des variables d'environnement
print(f"url du webhook Discord: {DISCORD_WEBHOOK_URL}")
print(f"Temps d'actualisation (en secondes) : {REFRESH_TIME}")

# L’URL de l’API (exemple)
API_URL = "https://api.nvidia.partners/edge/product/search?page=1&limit=100&locale=fr-fr"

# GPUs à surveiller
GPU_TARGETS = ["RTX 5070 Ti", "RTX 5070", "RTX 5080", "RTX 5090"]

# Entêtes HTTP pour la requête
HEADERS = {
    "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) "
                  "AppleWebKit/537.36 (KHTML, like Gecko) "
                  "Chrome/131.0.0.0 Safari/537.36",
    "Accept": "text/html,application/xhtml+xml,application/xml;q=0.9,"
              "image/avif,image/webp,image/apng,*/*;q=0.8",
    "Accept-Encoding": "gzip, deflate, br, zstd",
    "Accept-Language": "en-US,en;q=0.5",
    "Cache-Control": "max-age=0",
    "Connection": "keep-alive",
    "Sec-Fetch-Dest": "document",
    "Sec-Fetch-Mode": "navigate",
    "Sec-Fetch-Site": "none",
    "Sec-Fetch-User": "?1",
    "Upgrade-Insecure-Requests": "1",
    "Sec-GPC": "1",
}

# Dictionnaire stockant l'état de stock pour chaque GPU.
stock_status = {gpu.upper(): False for gpu in GPU_TARGETS}

session = requests.Session()


def send_discord_notification(message: str):
    """Envoie un message sur un salon Discord via un webhook."""
    payload = {
        "content": message
    }
    try:
        response = requests.post(DISCORD_WEBHOOK_URL, json=payload)
        if response.status_code == 204:
            logging.info("Notification envoyée sur Discord.")
        else:
            logging.error(f"Échec de l'envoi du webhook : {response.status_code} - {response.text}")
    except Exception as e:
        logging.error(f"Erreur lors de l'appel au webhook Discord : {e}")


def check_rtx_50_founders():
    """Vérifie l'état de stock des GPU Founders Edition et notifie Discord si un GPU repasse en stock."""
    try:
        session.get("https://api.nvidia.partners/edge/product/search?page=1&limit=1",
                    headers=HEADERS,
                    timeout=10)

        response = session.get(API_URL, headers=HEADERS, timeout=10)
        logging.info(f"Réponse de l'API : {response.status_code}")
        response.raise_for_status()
        data = response.json()

    except requests.exceptions.ReadTimeout:
        logging.info("Time-out lors de l'appel API.")
        return
    except Exception as e:
        logging.info(f"Erreur lors de l'appel API : {e}")
        return

    products = data.get("searchedProducts", {}).get("productDetails", [])
    found_in_stock = set()

    for p in products:
        gpu_name = p.get("gpu", "").upper()
        is_founder = p.get("isFounderEdition") is True
        #is_nvidia = (p.get("manufacturer") == "NVIDIA")
        is_buy_now = (p.get("prdStatus") == "buy_now" or p.get("productAvailable") is True)

        if is_founder and is_buy_now:
            if any(target.upper() in gpu_name for target in GPU_TARGETS):
                found_in_stock.add(gpu_name)

    for gpu in GPU_TARGETS:
        gpu_upper = gpu.upper()
        currently_in_stock = (gpu_upper in found_in_stock)
        previously_in_stock = stock_status[gpu_upper]

        if currently_in_stock and not previously_in_stock:
            for product in products:
                product_name = product.get("gpu", "").upper()
                if product_name == gpu_upper:
                    real_gpu_name = product.get("gpu", "Inconnu")
                    product_link = product.get("internalLink", "(pas de lien trouvé)")
                    msg = (
                        f"@everyone **En stock !**\n"
                        f"Carte : `{real_gpu_name}`\n"
                        f"Lien : {product_link}"
                    )
                    send_discord_notification(msg)

            stock_status[gpu_upper] = True
            print(f"{gpu} est maintenant en stock!")

        elif (not currently_in_stock) and previously_in_stock:
            logging.info(f"{gpu} n'est plus en stock.")
            stock_status[gpu_upper] = False
            print(f"{gpu} est hors stock !")

        elif not currently_in_stock:
            print(f"{gpu} est actuellement hors stock.")


if __name__ == "__main__":
    while True:
        check_rtx_50_founders()
        time.sleep(REFRESH_TIME)