import aiohttp
import var
import requests
from beloved_logger import logger
from functions.roll_token import roll_token



async def check_ip_tracker():
    with open('data/tracked-users.txt', 'r') as file:
        reader = csv.reader(file)
        user_ids = [row[0] for row in reader]
    
        async with aiohttp.ClientSession() as session:
            tasks = [fetch_data(session, user_id) for user_id in user_ids]
            results = await asyncio.gather(*tasks)
    
        for user_id, log_times in results:
            all_results.append((user_id, log_times))
    
        print("Results stored in memory.")
        print(all_results)


async def get_user_ip(user_id):
    await roll_token()
    
        url = f"https://sketchersunited.org/admin/ip_tracker?profile_id={user_id}"
    
        cookies = {
            var.cookie_name: var.cookie_value,
            'sketchers_united_session': var.session,
            'XSRF-TOKEN': var.xsrf_token
        }
    
        headers = {
            'User-Agent': 'automod',
            'Accept': 'application/json, */*'
        }
    
        async with aiohttp.ClientSession() as session:
            try:
                async with session.get(url, cookies=cookies, headers=headers) as response:
                    if response.status != 200:
                        logger.info(f"Failed to fetch data for user {user_id}: {response.status}")
                        return None
    
                    data = await response.json()
                    logger.info(f"Data for user {user_id}: {data}")
                    return data
            except Exception as e:
                logger.error(f"Error fetching data for user {user_id}: {e}")
                return None