# ai_ads.py
import random
from datetime import datetime
from pymongo import MongoClient
import os

# MongoDB setup
MONGO_URI = os.getenv("MONGO_URI", "")
mongo_client = MongoClient(MONGO_URI)
ad_db = mongo_client["anime_ai_db"]
ad_logs = ad_db["ad_logs"]

# Example ad messages (can be replaced or extended from DB)
AD_MESSAGES = [
    {
        "text": "🗄️ Download Android ID to view your android system information such as Ram, Battery life, CPU, and etc, click this ad to view!",
        "url": "https://android-id.en.uptodown.com/android",
        "id": "android_id_001"
    },
    {
        "text": "🖼️ With DeepImagix you can create amazing portrays with different models, a text to image AI, now for free!",
        "url": "https://deepimagix-ai.en.uptodown.com/android",
        "id": "deepimagix_002"
    },
    {
        "text": "⛓️ Protect your privacy online by connecting to ACE INJECTOR VPN, good for internet speed and privacy!",
        "url": "https://ace-injector-ssh-v2r-dns-proxy-vpn.en.uptodown.com/android",
        "id": "ace_injector_003"
    },
    {
        "text": "🤖 Interested in coding.?Download Android IDE to creat your first android application!",
        "url": "https://androidide-coder.en.uptodown.com/android",
        "id": "androidide_005"
    },
    {
        "text": "💎 Subscribe to the premium feature to disable advertisements",
        "url": "",
        "id": "pro_upgrade_006"
    },
    {
        "text": "🎉 Want to advertise your business here.?email us at alysium.corporation.studios@gmail.com to buy ad, can be seen from over 10 000 users!",
        "url": "",
        "id": "ad_revenue_007"
    },
    {
        "text": "For best responsoes give the ai more information on the task and intructions",
        "url": "",
        "id": "no_ad_008"
    },
    {
        "text": "🎮 Optimise your gaming experience with GTA LAG fixer, Reduces lagging and increases fps for Grand theft auto san andreas!",
        "url": "https://gta-lag-fixer-increase-fps.en.uptodown.com/android",
        "id": "gta_lag_009"
    }
]

def pick_ad() -> dict:
    """Randomly select an ad from the pool."""
    return random.choice(AD_MESSAGES)

def inject_ad(response_text: str, user_id: str) -> str:
    """Add an ad to the AI response with a probability."""
    if random.random() < 0.25:  # 25% chance to inject ad
        ad = pick_ad()
        log_ad_display(user_id, ad["id"])
        ad_text = f"\n\n📢 **Sponsored AD**: [{ad['text']}]({ad['url']})"
        return response_text + ad_text
    return response_text

def log_ad_display(user_id: str, ad_id: str):
    """Log the ad display to MongoDB."""
    ad_logs.insert_one({
        "user_id": user_id,
        "ad_id": ad_id,
        "event": "impression",
        "timestamp": datetime.utcnow()
    })

def log_ad_click(user_id: str, ad_id: str):
    """Log when a user clicks on an ad."""
    ad_logs.insert_one({
        "user_id": user_id,
        "ad_id": ad_id,
        "event": "click",
        "timestamp": datetime.utcnow()
    })