import os
import logging
import random
from dotenv import load_dotenv

load_dotenv()

IGNORED_MODEL_NAMES = ["gpt-4", "gpt-3.5", "websearch", "dall-e-3", "gpt-4o"]
IMAGE_MODEL_NAMES = ["dalle3", "dalle-3", "dall-e-3"]
AUTH_TOKEN = os.getenv("AUTHORIZATION")
# G_TOKEN = os.getenv("G_TOKEN")
G_TOKEN = None
HISTORY_MSG_LIMIT = os.getenv("HISTORY_MSG_LIMIT", 0)
HTTP_PROXIES = os.getenv("HTTP_PROXIES")
HTTPS_PROXIES = os.getenv("HTTPS_PROXIES")
SOCKS_PROXIES = os.getenv("SOCKS_PROXIES")
RECAPTCHA_SECRET = os.getenv("RECAPTCHA_SECRET")
POPAI_BASE_URL = "https://www.popai.pro/"
log_level = os.getenv('LOG_LEVEL', 'INFO').upper()


def configure_logging():
    extended_log_format = (
        '%(asctime)s | %(levelname)s | %(name)s | '
        '%(process)d | %(filename)s:%(lineno)d | %(funcName)s | %(message)s'
    )
    logging.basicConfig(level=log_level, format=extended_log_format)


def _get_proxies_from_env(env_var):
    proxies = os.getenv(env_var, '')
    return [proxy.strip() for proxy in proxies.split(',') if proxy.strip()]


class ProxyPool:
    def __init__(self):
        self.http_proxies = _get_proxies_from_env('HTTP_PROXIES')
        self.https_proxies = _get_proxies_from_env('HTTPS_PROXIES')
        self.socks_proxies = _get_proxies_from_env('SOCKS_PROXIES')

    def get_random_proxy(self):
        proxy = {}
        if self.http_proxies:
            proxy['http'] = random.choice(self.http_proxies)
        if self.https_proxies:
            proxy['https'] = random.choice(self.https_proxies)
        if self.socks_proxies:
            socks_proxy = random.choice(self.socks_proxies)
            proxy['http'] = socks_proxy
            proxy['https'] = socks_proxy

        logging.info("proxy URL %s", proxy)

        return proxy if proxy else None